Update app.py

app.py

@@ -1,15 +1,68 @@
 import gradio as gr
 import openai, os, time
 
-from agents import create_triage_agent, create_sales_agent, create_issues_repairs_agent
-from agents import get_current_agent, get_current_thread, set_current_agent, set_current_thread
 from openai import OpenAI
-from utils import show_json
+from utils import function_to_schema, show_json
 
 ###
+MODEL = "gpt-4o-mini"
+
 current_agent, current_thread = None, None
+    
+def create_triage_agent(client):
+    return client.beta.assistants.create(
+        name="Triage Agent",
+        instructions=(
+            "You are a customer service bot for ACME Inc. "
+            "Introduce yourself. Always be very brief. "
+            "Gather information to direct the customer to the right department. "
+            "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)}],
+    )
 
-###
+def create_sales_agent(client):
+    return client.beta.assistants.create(
+        name="Sales Agent",
+        instructions=(
+            "You are a sales agent for ACME Inc."
+            "Always answer in a sentence or less."
+            "Follow the following routine with the user:"
+            "1. Ask them about any problems in their life related to catching roadrunners.\n"
+            "2. Casually mention one of ACME's crazy made-up products can help.\n"
+            " - Don't mention price.\n"
+            "3. Once the user is bought in, drop a ridiculous price.\n"
+            "4. Only after everything, and if the user says yes, "
+            "tell them a crazy caveat and execute their order.\n"
+            ""
+        ),
+        model=MODEL,
+        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):
+    return client.beta.assistants.create(
+        name="Issues and Repairs Agent",
+        instructions=(
+            "You are a customer support agent for ACME Inc."
+            "Always answer in a sentence or less."
+            "Follow the following routine with the user:"
+            "1. First, ask probing questions and understand the user's problem deeper.\n"
+            " - unless the user has already provided a reason.\n"
+            "2. Propose a fix (make one up).\n"
+            "3. ONLY if not satesfied, offer a refund.\n"
+            "4. If accepted, search for the ID and then execute refund."
+            ""
+        ),
+        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)}],
+    )
 
 def set_current_agent(agent):
     current_agent = agent
@@ -23,22 +76,39 @@ def get_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."""
-    current_agent = sales_agent
+    set_current_agent(sales_agent)
 
 def transfer_to_issues_repairs_agent():
     """Use for issues, repairs, or refunds."""
-    current_agent = issues_repairs_agent
+    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."""
-    current_agent = triage_agent
+    set_current_agent(triage_agent)
 
-###
+#
 
 def escalate_to_human(summary):
     """Only call this if explicitly asked to."""
@@ -48,7 +118,7 @@ def escalate_to_human(summary):
     print("=========================\n")
     exit()
 
-###
+#
 
 def execute_order(product, price: int):
     """Price should be in USD."""
@@ -148,24 +218,6 @@ def extract_content_values(data):
                 content_values.append(content.text.value)
     return content_values
 
-_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 chat(message, history, openai_api_key):
     global _client, _assistant, _thread