diff --git a/README.md b/README.md index 340bf6a..1812834 100644 --- a/README.md +++ b/README.md @@ -125,6 +125,7 @@ Configuration auto-generated by `langgraph template lock`. DO NOT EDIT MANUALLY. "model_name": { "type": "string", "default": "anthropic/claude-3-5-sonnet-20240620", + "description": "The name of the language model to use for the agent's main interactions. Should be in the form: provider/model-name.", "environment": [ { "value": "anthropic/claude-1.2", @@ -311,6 +312,7 @@ Configuration auto-generated by `langgraph template lock`. DO NOT EDIT MANUALLY. "scraper_tool_model_name": { "type": "string", "default": "accounts/fireworks/models/firefunction-v2", + "description": "The name of the language model to use for the web scraping tool. This model is specifically used for summarizing and extracting information from web pages.", "environment": [ { "value": "anthropic/claude-1.2", diff --git a/src/react_agent/configuration.py b/src/react_agent/configuration.py index a88311c..34f6df5 100644 --- a/src/react_agent/configuration.py +++ b/src/react_agent/configuration.py @@ -14,29 +14,38 @@ class Configuration: """The configuration for the agent.""" - system_prompt: str = field(default=prompts.SYSTEM_PROMPT) - """The system prompt to use for the agent's interactions. - - This prompt sets the context and behavior for the agent. - """ - - model_name: Annotated[str, {"__template_metadata__": {"kind": "llm"}}] = ( - "anthropic/claude-3-5-sonnet-20240620" + 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_name: Annotated[str, {"__template_metadata__": {"kind": "llm"}}] = field( + default="anthropic/claude-3-5-sonnet-20240620", + metadata={ + "description": "The name of the language model to use for the agent's main interactions. " + "Should be in the form: provider/model-name." + }, ) - """The name of the language model to use for the agent's main interactions. - - Should be in the form: provider/model-name. - """ scraper_tool_model_name: Annotated[ str, {"__template_metadata__": {"kind": "llm"}} - ] = "accounts/fireworks/models/firefunction-v2" - """The name of the language model to use for the web scraping tool. - - This model is specifically used for summarizing and extracting information from web pages. - """ - max_search_results: int = 10 - """The maximum number of search results to return for each search query.""" + ] = field( + default="accounts/fireworks/models/firefunction-v2", + metadata={ + "description": "The name of the language model to use for the web scraping tool. " + "This model is specifically used for summarizing and extracting information from web pages." + }, + ) + + max_search_results: int = field( + default=10, + metadata={ + "description": "The maximum number of search results to return for each search query." + }, + ) @classmethod def from_runnable_config(