Web Search Agent

from polos import (
    Agent, max_steps, MaxStepsConfig,
    create_web_search_tool, WebSearchToolConfig,
    create_ask_user_tool,
)

# Web search tool -- uses Tavily API via TAVILY_API_KEY env var.
# The API key is resolved lazily at call time, not at import time.
web_search = create_web_search_tool(
    WebSearchToolConfig(
        max_results=5,
        search_depth="basic",
        include_answer=True,
        approval="always",
    )
)

# Ask-user tool -- lets the agent ask the user for clarification
ask_user = create_ask_user_tool()

import { defineAgent, maxSteps, createWebSearchTool, createAskUserTool } from '@polos/sdk';
import { anthropic } from '@ai-sdk/anthropic';

// Web search tool — uses Tavily API via TAVILY_API_KEY env var.
// The API key is resolved lazily at call time, not at import time.
export const webSearch = createWebSearchTool({
  maxResults: 5,
  searchDepth: 'basic',
  includeAnswer: true,
  approval: 'always',
});

// Ask-user tool — lets the agent ask the user for clarification
export const askUser = createAskUserTool();

research_agent = Agent(
    id="research_agent",
    provider="anthropic",
    model="claude-sonnet-4-5",
    system_prompt=(
        "You are a research assistant with access to web search. "
        "When the user asks a question, search the web for current information and "
        "synthesize a well-sourced answer. Include URLs from your search results as references. "
        "If the user's question is ambiguous, use the ask_user tool to clarify before searching. "
        "You may perform multiple searches to gather comprehensive information. "
        "Always cite your sources with URLs in the final answer."
    ),
    tools=[web_search, ask_user],
    stop_conditions=[max_steps(MaxStepsConfig(count=30))],
)

export const researchAgent = defineAgent({
  id: 'research_agent',
  model: anthropic('claude-sonnet-4-5'),
  systemPrompt:
    'You are a research assistant with access to web search. ' +
    'When the user asks a question, search the web for current information and ' +
    'synthesize a well-sourced answer. Include URLs from your search results as references. ' +
    'If the user\'s question is ambiguous, use the ask_user tool to clarify before searching. ' +
    'You may perform multiple searches to gather comprehensive information. ' +
    'Always cite your sources with URLs in the final answer.',
  tools: [webSearch, askUser],
  stopConditions: [maxSteps({ count: 30 })],
});

git clone https://github.com/polos-dev/polos.git
cd polos/python-examples/20-web-search-agent
cp .env.example .env  # Add your POLOS_PROJECT_ID and API key
uv sync
python main.py

git clone https://github.com/polos-dev/polos.git
cd polos/typescript-examples/20-web-search-agent
cp .env.example .env  # Add your POLOS_PROJECT_ID and API key
npm install
npx tsx main.ts