add python code tool

agents/tools/python_code_tool.py

@@ -0,0 +1,76 @@
+import re
+from langchain import LLMChain, OpenAI, PromptTemplate
+from langchain.chat_models import ChatOpenAI
+from langchain.agents import tool
+from langchain.utilities import PythonREPL
+
+
+generate_python_code = """
+请用python写一段代码实现如下需求:
+
+---
+{input}
+---
+"""
+
+generate_python_code_promopt = PromptTemplate(input_variables=["input"], template=generate_python_code,)
+
+generate_code_chain = LLMChain(llm = OpenAI(temperature=0.1), prompt=generate_python_code_promopt, output_key="code")
+
+
+@tool("Generate and Excute Python Code ", return_direct=True)
+def generate_and_excute_python_code(input: str) -> str:
+    '''useful for when you need to generate python code and excute it'''
+    answer_code = generate_code_chain.run(input)
+    print(answer_code)
+    start = answer_code.find('python') + len('python')
+    end = answer_code.rfind('```')
+    code_content = answer_code[start:end].strip()
+
+    print(code_content)
+    python_repl = PythonREPL()
+    result = python_repl.run(code_content)
+    return f"""
+code:
+```
+{code_content}
+```
+
+execute result:
+---
+{result}
+---
+    """
+
+
+
+if __name__ == "__main__":
+    input = """
+我有一个json文件url为: https://artwork-assets-staging-sbux.starbucks.com.cn/accountavatars.json
+并按照如下Example进行格式转换
+文件格式为:
+```
+{
+'artworks': {
+        'file1.png': {
+            'middle@1x': '***', 
+            'middle@2x': '***', 
+            'middle@3x': '***'
+        },
+        'file2.png': {
+            'middle@1x': '***', 
+            'middle@2x': '***', 
+            'middle@3x': '***'
+        }
+    }
+}
+```
+输出格式:
+```
+curl https://active.stg.starbucks.com.cn/accountAvatar/file1.png
+curl https://active.stg.starbucks.com.cn/accountAvatar/file2.png
+```
+"""
+
+    result = generate_and_excute_python_code(input)
+    print(result)

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from langchain.chains import LLMChain
 from langchain.chat_models import ChatOpenAI
 from langchain.document_loaders import TextLoader
+from agents.tools.python_util_tool import generate_and_excute_python_code
 from chains import HumanFeedBackChain, contextRewriteChain
 from embedding import CustomEmbedding
 from memories import HumenFeedbackBufferMemory
@@ -91,7 +92,10 @@ def rewriteContext(input, chatbot):
     chatbot.append((input, response))
     return chatbot, response
 
-
+def generateCodeAndExcute(input, chatbot=[]):
+    result = generate_and_excute_python_code.run(input)
+    chatbot.append((input, result))
+    return chatbot
 
 toolTextBox = []
 with gr.Blocks() as demo:
@@ -99,7 +103,7 @@ with gr.Blocks() as demo:
             with gr.Tab("Business"):
                 with gr.Row():
                     with gr.Column():
-                        chatbot = gr.Chatbot(elem_id="chatbot").style()
+                        chatbot = gr.Chatbot().style()
                         with gr.Row():
                             txt = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(
                                 container=False)
@@ -123,12 +127,12 @@ with gr.Blocks() as demo:
 
             with gr.Tab("Tech"):
                 with gr.Row():
-                    with gr.Column(min_width="50%"):
-                        code_chatbot = gr.Chatbot(elem_id="chatbot").style()
+                    with gr.Column():
+                        code_chatbot = gr.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(min_width="50%"):
+                    with gr.Column():
                         with gr.Row():
                             code_context = gr.Textbox(show_label=True, label="Context", placeholder="Enter Context").style(
                                 container=False)
@@ -142,12 +146,21 @@ with gr.Blocks() as demo:
                                     sendMessageByMultiPart, [code_chatbot, code_context, relateCode, toolTextBox[index]], [code_chatbot]).then(
                                     generateCodeByMultiPart, [code_context, relateCode, toolTextBox[index], code_chatbot], [code_chatbot, code])
             with gr.Tab("FAQ"):
-                faq_chatbot = gr.Chatbot(elem_id="chatbot").style()
+                faq_chatbot = gr.Chatbot().style(max_width="50%")
                 with gr.Row():
                     faq = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(
                         container=False)
                 with gr.Row():
                     gr.Button("Regenerate embedding").click(generateEmbeddings,[faq_chatbot], [faq_chatbot])
+            with gr.Tab("TOOL"):
+                with gr.Row():
+                    with gr.Column():
+                        tool_request = gr.Textbox(show_label=False, placeholder="Enter your tool Request").style(
+                            container=False, show_copy_button=True)
+                        tool_button = gr.Button("Generate Code and Execute")
+                    with gr.Column():
+                        tool_chatbot = gr.Chatbot(elem_id="chatbot").style(container=False)
+                    tool_button.click(generateCodeAndExcute,[tool_request, tool_chatbot], [tool_chatbot])
 
         
     txt.submit(sendMessage, [chatbot, txt], [chatbot]).then(