optimate ui

app.py

@@ -7,7 +7,7 @@ from embedding import CustomEmbedding
 from memories import HumenFeedbackBufferMemory
 from langchain.memory import ConversationBufferMemory
 from promopts import FEEDBACK, FEEDBACK_PROMPT
-from code_generate import code_agent_executor
+from code_generate import code_agent_executor, tools
 
 # llm = ChatOpenAI(temperature=0.1)
 
@@ -17,8 +17,6 @@ baChain = HumanFeedBackChain(verbose=True, memory=baMemory)
 
 """读取document/business_context.py文件内容作为context"""
 context_path = "./documents/bussiness_context/business_context.md"
-textloader = TextLoader(context_path)
-CONTEXT = textloader.load()[0].page_content
 
 
 def sendMessage(chatbot, input):
@@ -33,12 +31,18 @@ def clearMemory(chatbot):
         baMemory.clear()
     return chatbot, ""
 
+def loadContext():
+    textloader = TextLoader(context_path)
+    return textloader.load()[0].page_content
+
+def saveContext(context):
+    with open(context_path, 'w') as f:
+        f.write(context)
 
 def feedBack(context, story, chatbot=[], input=""):
     if len(input) > 0:
         context += (f"\n\n {input}")
-        with open(context_path, 'w') as f:
-            f.write(context)
+        saveContext(context)
     response = baChain.run(
         input=(input if len(input) == 0 else input), context=context, story=story, stop="\nAnswer:")
     chatbot[-1][1] = response
@@ -61,7 +65,7 @@ def generateEmbeddings():
     customerEmbedding.calculateNotionEmbedding()
 
 
-def generateCode(input: str, chatbot=[]):
+def generateCode(input: str, chatbot=[], returnCode=False):
     if len(input) <=0:
         chatbot[-1][1] = None
         return chatbot, ""
@@ -70,46 +74,83 @@ def generateCode(input: str, chatbot=[]):
     chatbot[-1][1] = response
     return chatbot, ""
 
+def generateCodeByMultiPart(context: str, relateCode: str, toolName: str, chatbot=[]):
+    input = f"请根据如下信息{toolName}:\n{context}\n\n{relateCode}"
+    return generateCode(input, chatbot)
+
+def sendMessageByMultiPart(chatbot, context: str, relateCode: str, toolName: str):
+    input = f"请根据如下信息{toolName}:\n{context}\n\n{relateCode}"
+    chatbot.append((input, None))
+    return chatbot
+
+
+
 
+
+index = 0
+toolTextBox = []
 with gr.Blocks() as demo:
     with gr.Row():
-        with gr.Column():
-            chatbot = gr.Chatbot(elem_id="chatbot").style()
-
-            with gr.Row():
-                txt = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(
-                    container=False)
-            with gr.Row():
-                code = gr.Textbox(show_label=True, label="Code Generate", placeholder="Enter text and press enter").style(
-                    container=False)
-            with gr.Row():
-                faq = gr.Textbox(show_label=False, placeholder="Q&A from local context").style(
-                    container=False)
-        with gr.Column():
-            with gr.Row():
-                context = gr.Textbox(show_label=True, label="Context", value=CONTEXT, placeholder="Enter Context").style(
-                    container=False)
-
-            with gr.Row():
-                story = gr.Textbox(show_label=True, label="Story", placeholder="Enter Story").style(
-                    container=False)
-
-            with gr.Row():
-                button = gr.Button("Generate Scenarios")
-
-            with gr.Blocks():
+            
+            with gr.Tab("Business"):
+                with gr.Row():
+                    with gr.Column():
+                        chatbot = gr.Chatbot(elem_id="chatbot").style()
+                        with gr.Row():
+                            txt = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(
+                                container=False)
+                    with gr.Column():
+                        with gr.Row():
+                            context = gr.Textbox(show_label=True, value=loadContext(), label="Context", placeholder="Enter Context").style(
+                                container=False)
+                        with gr.Row():
+                            story = gr.Textbox(show_label=True, label="User Story", placeholder="Enter User Story").style(
+                                container=False)
+                        with gr.Row():
+                            buisinessButton = gr.Button("Generate Scenarios")
+                        with gr.Row():
+                            gr.Button("Save Context").click(saveContext, [context], [])
+                        # with gr.Row():
+                        #     generateEmbedding = gr.Button("Generate embedding")
+            with gr.Tab("Tech"):
+                with gr.Row():
+                    with gr.Column():
+                        code_chatbot = gr.Chatbot(elem_id="chatbot").style()
+                        with gr.Row():
+                            code = gr.Textbox(show_label=False, label="Code Generate", placeholder="Enter text and press enter").style(
+                            container=False)
+                    with gr.Column():
+                        with gr.Row():
+                            code_context = gr.Textbox(show_label=True, label="Context", placeholder="Enter Context").style(
+                                container=False)
+                        with gr.Row():
+                            relateCode = gr.Textbox(show_label=True, label="Relate Code", placeholder="Enter Relate Code").style(
+                                container=False)
+                        for tool in tools:
+                            with gr.Row():
+                                toolTextBox.append(gr.Textbox(show_label=False, visible=False, label=tool.name, value=tool.name).style())
+                                gr.Button(tool.name).click(
+                                    sendMessageByMultiPart, [code_chatbot, code_context, relateCode, toolTextBox[index]], [code_chatbot]).then(
+                                    generateCodeByMultiPart, [code_context, relateCode, toolTextBox[index], code_chatbot], [code_chatbot, code])
+                        index += 1
+            with gr.Tab("FAQ"):
+                faq_chatbot = gr.Chatbot(elem_id="chatbot").style()
                 with gr.Row():
-                    generateEmbedding = gr.Button("Generate embedding")
+                    faq = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(
+                        container=False)
 
-    button.click(clearMemory, [chatbot], [chatbot, txt]).then(sendMessage, [chatbot, txt], [chatbot]).then(
+    
+    buisinessButton.click(clearMemory, [chatbot], [chatbot, txt]).then(sendMessage, [chatbot, txt], [chatbot]).then(
         feedBack, [context, story, chatbot], [chatbot, txt])
+    
     txt.submit(sendMessage, [chatbot, txt], [chatbot]).then(
         feedBack, [context, story, chatbot, txt], [chatbot, txt, context])
     
-    code.submit(sendMessage, [chatbot, code], [chatbot]).then(
-        generateCode, [code, chatbot], [chatbot, code])
+    code.submit(sendMessage, [code_chatbot, code], [code_chatbot]).then(
+        generateCode, [code, code_chatbot], [code_chatbot, code])
+    
+    faq.submit(faqFromLocal, [faq, faq_chatbot], [faq_chatbot, faq])
+    # generateEmbedding.click(generateEmbeddings)
     
-    faq.submit(faqFromLocal, [faq, chatbot], [chatbot, faq])
-    generateEmbedding.click(generateEmbeddings)
 
 demo.launch()

code_generate.py

@@ -94,13 +94,13 @@ class CustomOutputParser(AgentOutputParser):
 
 # agent = initialize_agent(
 #     tools=tools, llm=llm_chain, template=AGENT_PROMPT, stop=["\nObservation:"], agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
+tools = [domainLayerCodeGenerator, persistentLayerCodeGenerator, apiLayerCodeGenerator]
 
 
 def code_agent_executor() -> AgentExecutor:
     
 
-    tools = [domainLayerCodeGenerator, persistentLayerCodeGenerator, apiLayerCodeGenerator]
-
+    
     output_parser = CustomOutputParser()
 
 

documents/bussiness_context/business_context.md

@@ -19,4 +19,12 @@ FeatureConfig新增后status为DRAFT、执行发布操作后变为PUBLISHED、
 
 新创建的FeatureConfig状态为DRAFT
 
- 一个FeatureFlag中可以包含多个FeatureConfig
+ 一个FeatureFlag中可以包含多个FeatureConfig
+
+ 客户端用户需要查到FeatureConfig中data、saData、更新时间、id
+
+ FeatureConfig 可以关联圈人条件，符合圈人条件的配置，可展示给客户端用户
+
+ 客户端用户仅能查看符合圈人条件的PUBLISHED状态的数据
+
+ 圈人条件包括上传用户白名单，按照比例灰度发布，地理位置，人群标签等

promopts.py

@@ -102,8 +102,20 @@ For the Association，do not write tests because it is has no impletation.
 Use the following format:
 request: the request that you need to fulfill
 
-the code just include Enity Association and tests for Entity that you write to fulfill the request, follow TechStack Architecture TestStrategy to fulfill request, you should cover more test case as posible 
+Entity:
+```
+the Entity code that you write to fulfill the request, follow TechStack and Architecture
+```
 
+Association:
+```
+the Association code that you write to fulfill the request, follow TechStack and Architecture
+```
+
+Test:
+```
+the test code that you write to fulfill the request, follow TechStack Architecture and TestStrategy
+```
 
 request: {input}"""
 
@@ -246,8 +258,25 @@ For Association Impletation，we writ unit test, and stub repository method with
 Use the following format:
 request: the request that you need to fulfill include Entity and Association of domain layer
 
-the code just include DbEnity、Repository、Association Implement and tests code that you write to fulfill the request, follow TechStack Architecture TestStrategy to fulfill request, you should cover more test case as posible 
+DBEntity:
+```
+the DBEntity code that you write to fulfill the request, follow TechStack and Architecture
+```
+
+Repository:
+```
+the Repository code that you write to fulfill the request, follow TechStack and Architecture
+```
+
+Association Impletation:
+```
+the Association Impletation code that you write to fulfill the request, follow TechStack and Architecture
+```
 
+Test:
+```
+the test code that you write to fulfill the request, follow TechStack Architecture and TestStrategy
+```
 
 request: {input}"""
 
@@ -317,7 +346,20 @@ For the Controller and DTO, we can write component test to test the actual imple
 Use the following format:
 request: the request that you need to fulfill include Entity and Association of domain layer
 
-the code just include Controller、DTO and tests code that you write to fulfill the request, follow TechStack Architecture TestStrategy to fulfill request, you should cover as more test case as posible
+DTO:
+```
+the DTO code that you write to fulfill the request, follow TechStack and Architecture
+```
+
+Controller:
+```
+the Controller code that you write to fulfill the request, follow TechStack and Architecture
+```
+
+Test:
+```
+the test code that you write to fulfill the request, follow TechStack Architecture and TestStrategy
+```
 
 request: {input}"""