Update agents.py

agents.py

@@ -10,19 +10,6 @@ from utils import function_to_schema
 
 MODEL = "gpt-4o-mini"
 
-escalate_to_human_json = function_to_schema(escalate_to_human)
-print("###")
-print(escalate_to_human_json)
-print("###")
-
-transfer_to_sales_agent_json = function_to_schema(transfer_to_sales_agent)
-transfer_to_issues_repairs_agent_json = function_to_schema(transfer_to_issues_repairs_agent)
-transfer_to_triage_agent_json = function_to_schema(transfer_to_triage_agent)
-
-execute_order_json = function_to_schema(execute_order)
-execute_refund_json = function_to_schema(execute_refund)
-look_up_item_json = function_to_schema(look_up_item)
-
 def create_triage_agent(client):
     return client.beta.assistants.create(
         name="Triage Agent",
@@ -33,9 +20,9 @@ def create_triage_agent(client):
             "But make your questions subtle and natural."
         ),
         model=MODEL,
-        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}],
+        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)}],
     )
 
 def create_sales_agent(client):
@@ -54,8 +41,8 @@ def create_sales_agent(client):
             ""
         ),
         model=MODEL,
-        tools=[{"type": "function", "function": execute_order_json},
-               {"type": "function", "function": transfer_to_triage_agent_json}],
+        tools=[{"type": "function", "function": function_to_schema(execute_order)},
+               {"type": "function", "function": function_to_schema(transfer_to_triage_agent)}],
     )
     
 def create_issues_repairs_agent(client):
@@ -73,7 +60,7 @@ def create_issues_repairs_agent(client):
             ""
         ),
         model=MODEL,
-        tools=[{"type": "function", "function": look_up_item_json},
-               {"type": "function", "function": execute_refund_json},
-               {"type": "function", "function": transfer_to_triage_agent_json}],
+        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)}],
     )