| """Define the configurable parameters for the agent."""
|
|
|
| from __future__ import annotations
|
|
|
| import os
|
| from dataclasses import dataclass, field, fields
|
| from typing import Annotated
|
|
|
| from . import prompts
|
|
|
|
|
| @dataclass(kw_only=True)
|
| class Context:
|
| """The context for the agent."""
|
|
|
| system_prompt: str = field(
|
| default=prompts.SYSTEM_PROMPT,
|
| metadata={
|
| "description": "The system prompt to use for the agent's interactions. "
|
| "This prompt sets the context and behavior for the agent."
|
| },
|
| )
|
|
|
| model: Annotated[str, {"__template_metadata__": {"kind": "llm"}}] = field(
|
| default="anthropic/claude-sonnet-4-5-20250929",
|
| metadata={
|
| "description": "The name of the language model to use for the agent's main interactions. "
|
| "Should be in the form: provider/model-name."
|
| },
|
| )
|
|
|
| max_search_results: int = field(
|
| default=10,
|
| metadata={
|
| "description": "The maximum number of search results to return for each search query."
|
| },
|
| )
|
|
|
| def __post_init__(self) -> None:
|
| """Fetch env vars for attributes that were not passed as args."""
|
| for f in fields(self):
|
| if not f.init:
|
| continue
|
|
|
| if getattr(self, f.name) == f.default:
|
| setattr(self, f.name, os.environ.get(f.name.upper(), f.default))
|
|
|
