Update assistants.py

assistants.py

@@ -38,7 +38,7 @@ def create_sales_agent(client):
         ),
         model=MODEL,
         tools=[{"type": "execute_order"},
-               {"type": "transfer_back_to_triage"}],
+               {"type": "transfer_to_triage_agent"}],
     )
     show_json("assistant", assistant)
     return assistant
@@ -58,9 +58,59 @@ def create_issues_repairs_agent(client):
             ""
         ),
         model=MODEL,
-        tools=[{"type": "execute_refund"},
-               {"type": "look_up_item"},
-               {"type": "transfer_back_to_triage"}],
+        tools=[{"type": "look_up_item"},
+               {"type": "execute_refund"},
+               {"type": "transfer_to_triage_agent"}],
     )
     show_json("assistant", assistant)
-    return assistant
+    return assistant
+
+def escalate_to_human(summary):
+    """Only call this if explicitly asked to."""
+    print("Escalating to human agent...")
+    print("\n=== Escalation Report ===")
+    print(f"Summary: {summary}")
+    print("=========================\n")
+    exit()
+
+def transfer_to_sales_agent():
+    """Use for anything sales or buying related."""
+    return sales_agent
+
+def transfer_to_issues_repairs_agent():
+    """Use for issues, repairs, or refunds."""
+    return issues_and_repairs_agent
+
+def transfer_to_triage_agent():
+    """Call this if the user brings up a topic outside of your purview,
+    including escalating to human."""
+    return triage_agent
+
+def execute_order(product, price: int):
+    """Price should be in USD."""
+    print("\n\n=== Order Summary ===")
+    print(f"Product: {product}")
+    print(f"Price: ${price}")
+    print("=================\n")
+    confirm = input("Confirm order? y/n: ").strip().lower()
+    if confirm == "y":
+        print("Order execution successful!")
+        return "Success"
+    else:
+        print(color("Order cancelled!", "red"))
+        return "User cancelled order."
+
+def look_up_item(search_query):
+    """Use to find item ID.
+    Search query can be a description or keywords."""
+    item_id = "item_132612938"
+    print("Found item:", item_id)
+    return item_id
+
+def execute_refund(item_id, reason="not provided"):
+    print("\n\n=== Refund Summary ===")
+    print(f"Item ID: {item_id}")
+    print(f"Reason: {reason}")
+    print("=================\n")
+    print("Refund execution successful!")
+    return "Success"