Update app.py

app.py

@@ -21,6 +21,11 @@ def transfer_to_issues_repairs_agent():
     """Use for issues, repairs, or refunds."""
     set_current_agent(issues_repairs_agent)
 
+tools = {"transfer_to_triage_agent": transfer_to_triage_agent,
+         "transfer_to_sales_agent": transfer_to_sales_agent,
+         "transfer_to_issues_repairs_agent": transfer_to_issues_repairs_agent,
+         "escalate_to_human": escalate_to_human}
+
 #
 
 def escalate_to_human(summary):
@@ -159,12 +164,20 @@ def wait_on_run(client, thread, run):
     
     return run
 
-def execute_tool_call(tools, tool_call):
+def execute_tool_call(tool_call):
     name = tool_call.function.name
     args = json.loads(tool_call.function.arguments)
 
     print(f"{name}, {args}")
-        
+
+    global tools
+
+    print("===")
+    print(tools)
+    print("===")
+    print(tools[name])
+    print("===")
+    
     return tools[name](**args)
 
 def list_run_steps(client, thread, run):
@@ -176,31 +189,13 @@ def list_run_steps(client, thread, run):
 
     for step in run_steps.data:
         step_details = step.step_details
-        show_json("step_details", step_details)
+        #show_json("step_details", step_details)
 
         if not hasattr(step_details, "tool_calls"):
             break
 
-        print("XXXXX")
-        print(get_current_agent().tools)
-        print("XXXXX")
-        #print(tools)
-        #print("XXXXX")
-
-        ###
-        tools = {}
-        for tool in get_current_agent().tools:
-            print("===")
-            print(tool)
-            print("===")
-            print(tool.function)
-            print("===")
-        ###
-
-        tools = {tool.function.__name__: tool.function for tool in get_current_agent().tools}
-
         for tool_call in step_details.tool_calls:
-            result = execute_tool_call(tools, tool_call)
+            result = execute_tool_call(tool_call)
     
     return run_steps