Update app.py

app.py

@@ -16,6 +16,7 @@ def transfer_to_triage_agent():
     global triage_agent, triage_thread
     set_current_agent(triage_agent)
     set_current_thread(triage_thread)
+    return "transfer_to_triage_agent"
 
 def transfer_to_sales_agent():
     """Use for anything sales or buying related."""
@@ -23,6 +24,7 @@ def transfer_to_sales_agent():
     global sales_agent, sales_thread
     set_current_agent(sales_agent)
     set_current_thread(sales_thread)
+    return "transfer_to_sales_agent"
 
 def transfer_to_issues_repairs_agent():
     """Use for issues, repairs, or refunds."""
@@ -30,6 +32,7 @@ def transfer_to_issues_repairs_agent():
     global issues_repairs_agent, issues_repairs_thread
     set_current_agent(issues_repairs_agent)
     set_current_thread(issues_repairs_thread)
+    return "transfer_to_issues_repairs_agent"
 
 #
 
@@ -37,7 +40,8 @@ def escalate_to_human(summary):
     """Only call this if explicitly asked to."""
     print(f"=> escalate_to_human: summary: {summary}")
     #exit()
-
+    return "escalate_to_human"
+    
 #
 
 def execute_order(product, price: int):
@@ -148,9 +152,9 @@ def create_thread(client):
     
     return thread
 
-def create_message(client, thread, msg):
+def create_message(client, thread, msg, type = "user"):
     message = client.beta.threads.messages.create(
-        role="user",
+        role=type,
         thread_id=thread.id,
         content=msg,
     )
@@ -197,16 +201,18 @@ def execute_tool_call(tool_call):
     return tools[name](**args)
 
 def execute_tool_calls(steps):
+    result = ""
+    
     for step in steps.data:
         step_details = step.step_details
         show_json("step_details", step_details)
 
         if hasattr(step_details, "tool_calls"):
             for tool_call in step_details.tool_calls:
-                execute_tool_call(tool_call)
+                result = execute_tool_call(tool_call)
                 show_json("tool_call", tool_call)
 
-    return "TODO: execute_tool_calls return"
+    return result
                 
 def list_messages(client, thread):
     messages = client.beta.threads.messages.list(
@@ -281,8 +287,11 @@ def chat(message, history, openai_api_key):
 
     steps = list_steps(client, thread, run)
 
-    result = execute_tool_calls(steps)
-    print(result)
+    message = execute_tool_calls(steps)
+    print("\n\n\n[" + message + "]\n\n\n")
+
+    if message != "":
+        create_message(client, thread, message, "tool")
     
     messages = list_messages(client, thread)