Update app.py

app.py

@@ -4,11 +4,62 @@ import openai, os, time
 from openai import OpenAI
 from utils import function_to_schema, show_json
 
-###
-MODEL = "gpt-4o-mini"
+# Tools
 
-current_agent, current_thread = None, None
+def transfer_to_sales_agent():
+    """Use for anything sales or buying related."""
+    set_current_agent(sales_agent)
+
+def transfer_to_issues_repairs_agent():
+    """Use for issues, repairs, or refunds."""
+    set_current_agent(issues_repairs_agent)
+
+def transfer_to_triage_agent():
+    """Call this if the user brings up a topic outside of your purview,
+    including escalating to human."""
+    set_current_agent(triage_agent)
+
+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 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"
+
+# Agents
+
+MODEL = "gpt-4o-mini"
+
 def create_triage_agent(client):
     return client.beta.assistants.create(
         name="Triage Agent",
@@ -64,102 +115,8 @@ def create_issues_repairs_agent(client):
                {"type": "function", "function": function_to_schema(transfer_to_triage_agent)}],
     )
 
-def set_current_agent(agent):
-    current_agent = agent
-
-def set_current_thread(thread):
-    current_thread = thread
-
-def get_current_agent():
-    return current_agent
-
-def get_current_thread():
-    return current_thread
-
-_client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
-
-#_assistant, _thread = None, None
-
-###
-triage_agent = create_triage_agent(_client)
-sales_agent = create_sales_agent(_client)
-issues_repairs_agent = create_issues_repairs_agent(_client)
-
-set_current_agent(triage_agent)
-
-triage_thread = create_thread(_client)
-sales_thread = create_thread(_client)
-issues_repairs_thread = create_thread(_client)
-
-set_current_thread(triage_thread)
-
 #
 
-def transfer_to_sales_agent():
-    """Use for anything sales or buying related."""
-    set_current_agent(sales_agent)
-
-def transfer_to_issues_repairs_agent():
-    """Use for issues, repairs, or refunds."""
-    set_current_agent(issues_repairs_agent)
-
-def transfer_to_triage_agent():
-    """Call this if the user brings up a topic outside of your purview,
-    including escalating to human."""
-    set_current_agent(triage_agent)
-
-#
-
-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 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 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"
-    
-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 create_assistant(client):
-#    assistant = client.beta.assistants.create(
-#        name="Math Tutor",
-#        instructions="You are a personal math tutor. Answer questions briefly, in a sentence or less.",
-#        model="gpt-4-1106-preview",
-#        tools=[{"type": "code_interpreter"}],
-#    )
-#    show_json("assistant", assistant)
-#    return assistant
-    
 def create_thread(client):
     thread = client.beta.threads.create()
     #show_json("thread", thread)
@@ -218,6 +175,38 @@ def extract_content_values(data):
                 content_values.append(content.text.value)
     return content_values
 
+#
+
+current_agent, current_thread = None, None
+
+def set_current_agent(agent):
+    current_agent = agent
+
+def set_current_thread(thread):
+    current_thread = thread
+
+def get_current_agent():
+    return current_agent
+
+def get_current_thread():
+    return current_thread
+
+#
+
+_client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+
+triage_agent = create_triage_agent(_client)
+sales_agent = create_sales_agent(_client)
+issues_repairs_agent = create_issues_repairs_agent(_client)
+
+set_current_agent(triage_agent)
+
+triage_thread = create_thread(_client)
+sales_thread = create_thread(_client)
+issues_repairs_thread = create_thread(_client)
+
+set_current_thread(triage_thread)
+
 def chat(message, history, openai_api_key):
     global _client, _assistant, _thread