How to Use

import asyncio
import os
from pathlib import Path

from context_agent.agent import AgentDeps, agent

async def with_custom_tools():
    """Example where agent creates its own tools."""
    exec_dir = "/tmp/my-agent-workspace"
    Path(exec_dir).mkdir(parents=True, exist_ok=True)

    deps = AgentDeps(
        exec_dir=exec_dir,
        allowed_domains=["api.stlouisfed.org"],  # For FRED API example
    )

    # Set API key via environment (agent can access CT_PY_* vars)
    os.environ["CT_PY_FRED_API_KEY"] = "your-api-key-here"

    # First turn: agent creates a tool
    result1 = await agent.run(
        """
        Write a tools.js file with a function called fetchGDP that:
        1. Gets the CT_PY_FRED_API_KEY env var
        2. Fetches GDP data from FRED API
        3. Returns the last 5 observations
        """,
        deps=deps,
    )
    print("Tool created:", result1.output)

    # Second turn: agent uses the tool it just created
    result2 = await agent.run(
        "Use the fetchGDP function to get the latest GDP data",
        deps=deps,
        message_history=result1.new_messages(),
    )
    print("Tool result:", result2.output)