Update agents.py

agents.py

@@ -1,8 +1,10 @@
 # Reference: https://github.com/openai/build-hours/blob/main/2-assistants/7_orchestration.py
 
 from openai import OpenAI
-from tools import transfer_to_sales_agent, transfer_to_issues_repairs_agent, escalate_to_human
-from tools import execute_order, execute_refund, look_up_item, transfer_to_triage_agent
+
+from tools import escalate_to_human_json
+from tools import transfer_to_sales_agent_json, transfer_to_issues_repairs_agent_json, transfer_to_triage_agent_json 
+from tools import execute_order_json, execute_refund_json, look_up_item_json
 
 MODEL = "gpt-4o-mini"
 
@@ -16,9 +18,9 @@ def create_triage_agent(client):
             "But make your questions subtle and natural."
         ),
         model=MODEL,
-        tools=[{"type": "function", "function": function_to_schema(transfer_to_sales_agent)},
-               {"type": "function", "function": function_to_schema(transfer_to_issues_repairs_agent)},
-               {"type": "function", "function": function_to_schema(escalate_to_human)}],
+        tools=[{"type": "function", "function": transfer_to_sales_agent_json},
+               {"type": "function", "function": transfer_to_issues_repairs_agent_json},
+               {"type": "function", "function": escalate_to_human_json}],
     )
 
 def create_sales_agent(client):
@@ -37,8 +39,8 @@ def create_sales_agent(client):
             ""
         ),
         model=MODEL,
-        tools=[{"type": "function", "function": function_to_schema(execute_order)},
-               {"type": "function", "function": function_to_schema(transfer_to_triage_agent)}],
+        tools=[{"type": "function", "function": execute_order_json},
+               {"type": "function", "function": transfer_to_triage_agent_json}],
     )
     
 def create_issues_repairs_agent(client):
@@ -56,7 +58,7 @@ def create_issues_repairs_agent(client):
             ""
         ),
         model=MODEL,
-        tools=[{"type": "function", "function": function_to_schema(look_up_item)},
-               {"type": "function", "function": function_to_schema(execute_refund)},
-               {"type": "function", "function": function_to_schema(transfer_to_triage_agent)}],
+        tools=[{"type": "function", "function": look_up_item_json},
+               {"type": "function", "function": execute_refund_json},
+               {"type": "function", "function": transfer_to_triage_agent_json}],
     )