Build Your First AI Agent Using the OpenAI Agents SDK

%pip install --quiet openai-agents httpx python-dotenv nest-asyncio

import asyncio
import os

import httpx
import nest_asyncio
from agents import Agent, Runner, function_tool
from dotenv import load_dotenv

if not globals().get("ENV_LOADED", False):
    load_dotenv()
    ENV_LOADED = True
    env_status = "Loaded environment variables from .env"
else:
    env_status = "Environment variables already loaded"

print("\n Exercise 1: Agent Hello World - Solution\n")
print(f"[OK] {env_status}")

@function_tool
async def get_weather_alerts(state: str) -> str:
    """
    Get current weather alerts for a US state from the National Weather Service.

    Args:
        state: Two-letter US state code (e.g., 'CA', 'NY', 'TX')

    Returns:
        String describing current weather alerts
    """
    print(f" Calling NWS API for state: {state}")

    try:
        # Call the National Weather Service API
        url = f"https://api.weather.gov/alerts/active/area/{state.upper()}"
        headers = {"User-Agent": "OpenAI-Agents-Workshop (educational)"}

        async with httpx.AsyncClient() as client:
            response = await client.get(url, headers=headers, timeout=10.0)
            response.raise_for_status()

            data = response.json()
            features = data.get("features", [])

            if not features:
                print("[OK] No active alerts")
                return f"No active weather alerts for {state.upper()}."

            # Extract alert information
            alerts = []
            for feature in features[:5]:  # Limit to first 5 alerts
                properties = feature.get("properties", {})
                event = properties.get("event", "Unknown")
                headline = properties.get("headline", "No headline")
                severity = properties.get("severity", "Unknown")
                alerts.append(f"- {event} ({severity}): {headline}")

            result = f"Active weather alerts for {state.upper()}:\n" + "\n".join(alerts)
            print(f"[OK] Found {len(features)} alert(s)")
            return result

    except httpx.HTTPError as e:
        error_msg = f"Failed to fetch weather alerts: {str(e)}"
        print(f"[ERROR] {error_msg}")
        return error_msg
    except Exception as e:
        error_msg = f"Unexpected error: {str(e)}"
        print(f"[ERROR] {error_msg}")
        return error_msg

print("[OK] Weather tool defined with @function_tool")
print("ℹ️  This tool makes real API calls to weather.gov")

agent = Agent(
    name="Weather Agent",
    instructions="You help users get weather alert information for US states. Use the get_weather_alerts tool to fetch real data from the National Weather Service.",
    tools=[get_weather_alerts]
)

print("[OK] Agent created with weather tool")
print("ℹ️  The agent can now fetch real weather alerts")

api_key = os.getenv("OPENAI_API_KEY")
if not api_key:
    raise RuntimeError("OPENAI_API_KEY is not set. Add it to your .env file.")

masked_key = f"{api_key[:4]}...{api_key[-4:]}" if len(api_key) >= 8 else "***"
print(f"[OK] OPENAI_API_KEY detected: {masked_key}")

async def run_solution():
    query = "Are there any weather alerts for California?"
    print(f"\nQuery: {query}\n")

    # Run the agent using the Agents SDK Runner
    result = await Runner.run(agent, query)

    # Display the final output
    print(f"\n Agent Response:\n{result.final_output}\n")
    print("ℹ️  This used a real API call to the National Weather Service!\n")

# Jupyter-specific async handling (notebooks already have an event loop)
try:
    loop = asyncio.get_running_loop()
    nest_asyncio.apply()
    await run_solution()
except RuntimeError:
    asyncio.run(run_solution())