Spaces:
Running
Running
| # Reference: https://github.com/openai/build-hours/blob/main/2-assistants/7_orchestration.py | |
| from openai import OpenAI | |
| from app import transfer_to_sales_agent, transfer_to_issues_repairs_agent, transfer_to_triage_agent | |
| from tools import escalate_to_human | |
| from tools import execute_order, execute_refund, look_up_item | |
| from utils import function_to_schema | |
| MODEL = "gpt-4o-mini" | |
| _assistant, _thread = None, None | |
| def set_current_agent(agent): | |
| _assistant = agent | |
| def set_current_thread(thread): | |
| _thread = thread | |
| 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)}], | |
| ) |