tool_choice.py

"""When working with the tool_choice parameter, we have three possible options:

auto allows Claude to decide whether to call any provided tools or not.
any tells Claude that it must use one of the provided tools, but doesn't force a particular tool.
tool allows us to force Claude to always use a particular tool.
"""

import wikipedia
from anthropic import Anthropic
from dotenv import load_dotenv
from database import FakeDatabase
import json

load_dotenv()

client = Anthropic()
MODEL_NAME = "claude-3-5-sonnet-20240620"
system_prompt = """
You are a customer support chat bot for an online retailer called TechNova. 
Your job is to help users look up their account, orders, and cancel orders.
Be helpful and brief in your responses.
You have access to a set of tools, but only use them when needed.  
If you do not have enough information to use a tool correctly, ask a user follow up questions to get the required inputs.
Do not call any of the tools unless you have the required data from a user. 
"""

# ANSI color codes
BLUE = "\033[94m"
GREEN = "\033[92m"
YELLOW = "\033[93m"
CYAN = "\033[96m"
RESET = "\033[0m"



    
db = FakeDatabase()

tools = [
    {
        "name": "get_user",
        "description": "Looks up a user by email, phone, or username.",
        "input_schema": {
            "type": "object",
            "properties": {
                "key": {
                    "type": "string",
                    "enum": ["email", "phone", "username"],
                    "description": "The attribute to search for a user by (email, phone, or username)."
                },
                "value": {
                    "type": "string",
                    "description": "The value to match for the specified attribute."
                }
            },
            "required": ["key", "value"]
        }
    },
    {
        "name": "get_order_by_id",
        "description": "Retrieves the details of a specific order based on the order ID. Returns the order ID, product name, quantity, price, and order status.",
        "input_schema": {
            "type": "object",
            "properties": {
                "order_id": {
                    "type": "string",
                    "description": "The unique identifier for the order."
                }
            },
            "required": ["order_id"]
        }
    },
    {
        "name": "get_customer_orders",
        "description": "Retrieves the list of orders belonging to a user based on a user's customer id.",
        "input_schema": {
            "type": "object",
            "properties": {
                "customer_id": {
                    "type": "string",
                    "description": "The customer_id belonging to the user"
                }
            },
            "required": ["customer_id"]
        }
    },
    {
        "name": "cancel_order",
        "description": "Cancels an order based on a provided order_id.  Only orders that are 'processing' can be cancelled",
        "input_schema": {
            "type": "object",
            "properties": {
                "order_id": {
                    "type": "string",
                    "description": "The order_id pertaining to a particular order"
                }
            },
            "required": ["order_id"]
        }
    },
    {
        "name": "update_customer_info",
        "description": "Updates a customer's information based on a provided key, and value.",
        "input_schema": {
            "type": "object",
            # search_key, search_value, update_key, update_value
            "properties": {
                "search_key": {
                    "type": "string",
                    "enum": ["email", "phone", "username"],
                    "description": "The attribute to search for a user by (email, phone, username)."
                },
                "search_value": {
                    "type": "string",
                    "description": "The customer_id belonging to the user"
                },
                "update_key": {
                    "type": "string",
                    "enum": ["email", "phone", "username"],
                    "description": "The attribute to update a user by (email, phone, username)."
                },
                "update_value": {
                    "type": "string",
                    "description": "The value to update the user with"
                }
            },
            "required": ["search_key", "search_value", "update_key", "update_value"]
        }
    }
]

def process_tool_call(tool_name, tool_input):
    if tool_name == "get_user":
        return db.get_user(tool_input["key"], tool_input["value"])
    elif tool_name == "get_order_by_id":
        return db.get_order_by_id(tool_input["order_id"])
    elif tool_name == "get_customer_orders":
        return db.get_customer_orders(tool_input["customer_id"])
    elif tool_name == "cancel_order":
        return db.cancel_order(tool_input["order_id"])
    elif tool_name == "update_customer_info":
        return db.update_customer_info(tool_input["search_key"], tool_input["search_value"], tool_input["update_key"], tool_input["update_value"])




def simple_chat():
    user_message = input(f"\n{YELLOW}User: {RESET}")
    
    # Check for exit commands
    if user_message.lower() in ['quit', 'exit', 'bye']:
        print(f"\n{CYAN}Goodbye! Have a great day!{RESET}")
        return
        
    messages = [{"role": "user", "content": user_message}]
    while True:
        if messages[-1].get("role") == "assistant":
            user_message = input(f"\n{YELLOW}User: {RESET}")
            
            # Check for exit commands
            if user_message.lower() in ['quit', 'exit', 'bye']:
                print(f"\n{CYAN}Goodbye! Have a great day!{RESET}")
                return
                
            messages.append({"role": "user", "content": user_message})

        #Send a request to Claude
        response = client.messages.create(
            model=MODEL_NAME,
            system=system_prompt,
            max_tokens=4096,
            tools=tools,
            messages=messages
        )
        # Update messages to include Claude's response
        messages.append(
            {"role": "assistant", "content": response.content}
        )

        #If Claude stops because it wants to use a tool:
        if response.stop_reason == "tool_use":
            tool_use = response.content[-1] #Naive approach assumes only 1 tool is called at a time
            tool_name = tool_use.name
            tool_input = tool_use.input
            print(f"\n{BLUE}======Claude wants to use the {tool_name} tool======{RESET}")

            #Actually run the underlying tool functionality on our db
            tool_result = process_tool_call(tool_name, tool_input)

            #Add our tool_result message:
            messages.append(
                {
                    "role": "user",
                    "content": [
                        {
                            "type": "tool_result",
                            "tool_use_id": tool_use.id,
                            "content": str(tool_result),
                        }
                    ],
                },
            )
        else: 
            #If Claude does NOT want to use a tool, just print out the text reponse
            print(f"\n{GREEN}TechNova Support:{RESET} {response.content[0].text}")


simple_chat()

# r = db.update_customer_info("email", "kwameo@yahoo.com", "username", "tilly")
# print(json.dumps(r, indent=2))