Update agents.py

agents.py

@@ -5,7 +5,7 @@ from openai import OpenAI
 MODEL = "gpt-4o-mini"
 
 def create_triage_agent(client):
-    assistant = client.beta.assistants.create(
+    return client.beta.assistants.create(
         name="Triage Agent",
         instructions=(
             "You are a customer service bot for ACME Inc. "
@@ -18,11 +18,9 @@ def create_triage_agent(client):
                {"type": "transfer_to_issues_repairs_agent"},
                {"type": "escalate_to_human"}],
     )
-    show_json("assistant", assistant)
-    return assistant
 
 def create_sales_agent(client):
-    assistant = client.beta.assistants.create(
+    return client.beta.assistants.create(
         name="Sales Agent",
         instructions=(
             "You are a sales agent for ACME Inc."
@@ -37,14 +35,12 @@ def create_sales_agent(client):
             ""
         ),
         model=MODEL,
-        tools=[{"type": "execute_order"},
-               {"type": "transfer_to_triage_agent"}],
+        tools=[{"type": "function", "function": execute_order},
+               {"type": "function", "function": transfer_to_triage_agent}],
     )
-    show_json("assistant", assistant)
-    return assistant
     
 def create_issues_repairs_agent(client):
-    assistant = client.beta.assistants.create(
+    return client.beta.assistants.create(
         name="Issues and Repairs Agent",
         instructions=(
             "You are a customer support agent for ACME Inc."
@@ -58,9 +54,7 @@ def create_issues_repairs_agent(client):
             ""
         ),
         model=MODEL,
-        tools=[{"type": "look_up_item"},
-               {"type": "execute_refund"},
-               {"type": "transfer_to_triage_agent"}],
-    )
-    show_json("assistant", assistant)
-    return assistant
+        tools=[{"type": "function", "function": look_up_item},
+               {"type": "function", "function": execute_refund},
+               {"type": "function", "function": transfer_to_triage_agent}],
+    )