Update app.py

app.py

@@ -21,6 +21,8 @@ def transfer_to_issues_repairs_agent():
     """Use for issues, repairs, or refunds."""
     set_current_agent(issues_repairs_agent)
 
+#
+
 def escalate_to_human(summary):
     """Only call this if explicitly asked to."""
     print("Escalating to human agent...")
@@ -29,6 +31,8 @@ def escalate_to_human(summary):
     print("=========================\n")
     exit()
 
+#
+
 def execute_order(product, price: int):
     """Price should be in USD."""
     print("\n\n=== Order Summary ===")
@@ -58,16 +62,6 @@ def execute_refund(item_id, reason="not provided"):
     print("Refund execution successful!")
     return "Success"
 
-#
-
-def execute_tool_call(tool_call, tools, agent_name):
-    name = tool_call.function.name
-    args = json.loads(tool_call.function.arguments)
-
-    print(f"{agent_name}, {name}, {args}")
-
-    return tools[name](**args)  # call corresponding function with provided arguments
-
 # Agents
 
 MODEL = "gpt-4o-mini"
@@ -167,9 +161,12 @@ def list_run_steps(client, thread, run):
         run_id=run.id,
         order="asc",
     )
+
     for step in run_steps.data:
         step_details = step.step_details
         show_json("step_details", step_details)
+        # TODO: Call function
+    
     return run_steps
     
 def list_messages(client, thread):
@@ -178,23 +175,25 @@ def list_messages(client, thread):
     )
     #show_json("messages", messages)
     return messages
-    
+
 def extract_content_values(data):
     content_values = []
     for item in data.data:
-        print("###")
-        print(item)
-        print("###")
         for content in item.content:
-            print("###")
-            print(content)
-            print("###")
             if content.type == 'text':
                 content_values.append(content.text.value)
     return content_values
 
 #
 
+def execute_tool_call(tool_call, tools, agent_name):
+    name = tool_call.function.name
+    args = json.loads(tool_call.function.arguments)
+    print(f"{agent_name}, {name}, {args}")
+    return tools[name](**args)
+
+#
+
 current_agent, current_thread = None, None
 
 def set_current_agent(agent):
@@ -215,7 +214,7 @@ def get_current_thread():
 
 #
 
-_client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
 triage_agent = create_triage_agent(_client)
 sales_agent = create_sales_agent(_client)
@@ -230,34 +229,25 @@ issues_repairs_thread = create_thread(_client)
 set_current_thread(triage_thread)
 
 def chat(message, history, openai_api_key):
-    global _client
+    global client
+
+    assistant = get_current_agent()
+    show_json("Current Agent", assistant)
+    thread = get_current_thread()
+    show_json("Current Thread", thread)
 
-    _assistant = get_current_agent()
-    #show_json("_assistant", _assistant)
-    _thread = get_current_thread()
-    #show_json("_thread", _thread)
-            
-    create_message(_client, _thread, message)
+    create_message(client, thread, message)
 
     # async
-    run = create_run(_client, _assistant, _thread)
-    run = wait_on_run(_client, _thread, run)
+    run = create_run(client, assistant, thread)
+    run = wait_on_run(client, thread, run)
 
-    list_run_steps(_client, _thread, run)
+    list_run_steps(client, thread, run)
     
-    messages = list_messages(_client, _thread)
-
-    print("###")
-    tmp = extract_content_values(messages)
-    #print(messages[0])
-    print("###")
-    #print(messages[0].content.text.value)
-    print("###")
-
-    # Call function
+    messages = list_messages(client, thread)
     
-    return "TODO" #extract_content_values(messages)[0]
-        
+    return extract_content_values(messages)[0]
+
 gr.ChatInterface(
     chat,
     chatbot=gr.Chatbot(height=300),