blocks experiment

app.py

@@ -21,7 +21,8 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-# What is the average each customer paid? Create a sql statement and invoke your sql_engine tool
+# What is the average each customer paid?
+#  Create a sql statement and invoke your sql_engine tool
 
 
 @spaces.GPU
@@ -30,7 +31,7 @@ def dummy():
 
 
 @tool
-def sql_engine_tool(query: str) -> str:
+def sql_engine_tool(query: str, engine: any) -> str:
     """
     Allows you to perform SQL queries on the table. Returns a string representation of the result.
     The table is named 'receipts'. Its description is as follows:
@@ -42,7 +43,7 @@ def sql_engine_tool(query: str) -> str:
 
     Args:
         query: The query to perform. This should be correct SQL.
-
+        engine: Use the given engine in the additional_args
     """
     output = ""
     print("debug sql_engine_tool")
@@ -119,9 +120,6 @@ if __name__ == "__main__":
 
     agent = CodeAgent(
         tools=[sql_engine_tool],
-        #         system_prompt="""
-        # You are a text to sql converter
-        # """,
         model=model,
         max_steps=1,
         verbosity_level=1,
@@ -129,20 +127,20 @@ if __name__ == "__main__":
     # agent.run("What is the average each customer paid?")
     # GradioUI(agent).launch()
 
-    def enter_message(message, chat_history):
+    def enter_message(message, chat_history, engine):
         print()
         print("enter_message debug")
         print(message)
         print(chat_history)
         chat_history.append({"role": "user", "content": message})
-        x = agent.run(message)
+        x = agent.run(message, additional_args=dict(engine=engine))
         print(type(x))
-        print(x)
+        print("\n\n\n", x, "\n\n\n")
         return "", x
 
     with gr.Blocks() as b:
         chatbot = gr.Chatbot(type="messages")
         input = gr.Textbox()
         button = gr.Button("reply")
-        button.click(enter_message, [input, chatbot], [input, chatbot])
+        button.click(enter_message, [input, chatbot, engine], [input, chatbot, engine])
     b.launch()