Update app.py

app.py

@@ -13,6 +13,8 @@
 # https://platform.openai.com/docs/assistants/tools
 
 import gradio as gr
+import yfinance as yf
+
 import json, openai, os, time
 
 from datetime import date
@@ -24,13 +26,22 @@ client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 assistant, thread = None, None
 
 def today_tool(text: str) -> str:
-    """Returns today's date. Use this for any questions related to knowing today's date. 
-       The input should always be an empty string, and this function will always return today's date. 
-       Any date mathematics should occur outside this function."""
+    """Returns today's date. 
+       Use this function for any questions related to knowing today's date. 
+       There is no input. 
+       This function always returns today's date."""
     return str(date.today())
 
+def yfinance_download_tool(tickers: str[], start: date, end: date) -> str:
+    """Returns stock data for given tickers from a start date to an end date. 
+       Use this function for any questions related to getting stock data. 
+       The input should be a string array of tickers, a start date, and an end date. 
+       This function always returns yf.download."""
+    return yf.download(tickers, start=start_date, end=end_date)
+
 tools = {
     "today_tool": today_tool,
+    "yf_download_tool": yf_download_tool,
 }
 
 def create_assistant(client):
@@ -39,12 +50,13 @@ def create_assistant(client):
         instructions=(
                          "You are a Python programming language expert that "
                          "generates Pylint-compliant code and explains it. "
-                         "Only execute code when explicitly asked to."
+                         "Execute code when explicitly asked to."
                      ),
         model="gpt-4o",
         tools=[
             {"type": "code_interpreter"},
             {"type": "function", "function": function_to_schema(today_tool)},
+            {"type": "function", "function": function_to_schema(yf_download_tool)},
         ],
     )
     
@@ -97,7 +109,7 @@ def wait_on_run(client, thread, run):
             run_id=run.id,
         )
     
-        time.sleep(0.25)
+        time.sleep(0.5)
     
     #show_json("run", run)
     
@@ -206,6 +218,26 @@ def chat(message, history):
         run_steps = get_run_steps(client, thread, run)
         tool_call_id, tool_call_result = execute_tool_calls(run_steps)
     ###
+    if tool_call_result:
+        print("### tool_call_id=" + tool_call_id)
+        print("### tool_call_result=" + tool_call_result)
+
+        # https://platform.openai.com/docs/api-reference/runs/submitToolOutputs
+        run = client.beta.threads.runs.submit_tool_outputs(
+            thread_id=thread.id,
+            run_id=run.id,
+            tool_outputs=[
+                {
+                  "tool_call_id": tool_call_id,
+                  "output": tool_call_result
+                }
+            ]
+        )
+    
+        run = wait_on_run(client, thread, run)
+        run_steps = get_run_steps(client, thread, run)
+        tool_call_id, tool_call_result = execute_tool_calls(run_steps)
+    ###
     
     messages = get_messages(client, thread)