# Reference: https://github.com/openai/build-hours/blob/main/2-assistants/7_orchestration.py

from openai import OpenAI

from tools import escalate_to_human
from tools import transfer_to_sales_agent, transfer_to_issues_repairs_agent, transfer_to_triage_agent 
from tools import execute_order, execute_refund, look_up_item

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",
        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=[{transfer_to_sales_agent_json},
               {transfer_to_issues_repairs_agent_json},
               {escalate_to_human_json}],
    )

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=[{execute_order_json},
               {transfer_to_triage_agent_json}],
    )
    
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=[{look_up_item_json},
               {execute_refund_json},
               {transfer_to_triage_agent_json}],
    )