Add descriptions

langchain-ai · Sep 16, 2024 · 0f68455 · 0f68455
1 parent 65a21dd
commit 0f68455
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 19 deletions.
diff --git 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
@@ -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(