tools.py

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

###

def transfer_to_sales_agent():
    """Use for anything sales or buying related."""
    current_agent = sales_agent

def transfer_to_issues_repairs_agent():
    """Use for issues, repairs, or refunds."""
    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

###

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"