from openai import OpenAI
import json

client = OpenAI()

# Define tools the LLM can call
tools = [
    {
        "type": "function",
        "function": {
            "name": "get_weather",
            "description": "Get current weather for a city",
            "parameters": {
                "type": "object",
                "properties": {
                    "city": {"type": "string", "description": "City name"},
                    "units": {"type": "string", "enum": ["celsius", "fahrenheit"]}
                },
                "required": ["city"]
            }
        }
    },
    {
        "type": "function",
        "function": {
            "name": "search_database",
            "description": "Search the product database",
            "parameters": {
                "type": "object",
                "properties": {
                    "query": {"type": "string"},
                    "category": {"type": "string"}
                },
                "required": ["query"]
            }
        }
    }
]

def run_tool(name: str, args: dict) -> str:
    """Execute the requested tool."""
    if name == "get_weather":
        # Call real weather API
        return f"Weather in {args['city']}: 22°C, sunny"
    elif name == "search_database":
        # Query your database
        return f"Found 3 products matching '{args['query']}'"
    return "Tool not found"

def agent_loop(user_message: str) -> str:
    """Simple agent that can use tools."""
    messages = [{"role": "user", "content": user_message}]

    while True:
        response = client.chat.completions.create(
            model="gpt-4o",
            messages=messages,
            tools=tools,
            tool_choice="auto"
        )

        msg = response.choices[0].message

        # If no tool calls, return the final answer
        if not msg.tool_calls:
            return msg.content

        # Process tool calls
        messages.append(msg)  # add assistant message with tool_calls
        for tool_call in msg.tool_calls:
            args = json.loads(tool_call.function.arguments)
            result = run_tool(tool_call.function.name, args)

            messages.append({
                "role": "tool",
                "tool_call_id": tool_call.id,
                "content": result
            })

# Usage
answer = agent_loop("What's the weather in Tel Aviv and do you have any umbrellas in stock?")
print(answer)
AI APIs — OpenAI, Anthropic, Integration Patterns

AI APIs — OpenAI, Anthropic, Integration Patterns

AI APIs — OpenAI, Anthropic, Integration Patterns
