04 - Conversational Chat

from polos import Agent, max_steps, MaxStepsConfig

chat_assistant = Agent(
    id="chat_assistant",
    provider="openai",
    model="gpt-4o-mini",
    system_prompt="""You are a friendly assistant. You can:
- Tell the current time using get_current_time
- Get weather using get_weather
- Perform calculations using calculator""",
    tools=[get_current_time, get_weather, calculator],
    stop_conditions=[max_steps(MaxStepsConfig(limit=10))],
)

# Use conversation_id to maintain history across messages
conversation_id = "user-123-session"

# First message
result1 = await chat_assistant.stream(
    client,
    "What's the weather in Tokyo?",
    conversation_id=conversation_id,
)

# Follow-up uses same context
result2 = await chat_assistant.stream(
    client,
    "How about London?",  # Agent remembers we're asking about weather
    conversation_id=conversation_id,
)

git clone https://github.com/polos-dev/polos.git
cd polos/python-examples/04-conversational-chat
cp .env.example .env
uv sync
python worker.py      # Terminal 1
python chat.py        # Terminal 2 (interactive)