langgraph-agent-toolkit 0.7.1__tar.gz → 0.7.2__tar.gz

{langgraph_agent_toolkit-0.7.1 → langgraph_agent_toolkit-0.7.2}/CHANGELOG.md

@@ -6,6 +6,12 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to
 [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.7.2]
+
+### Added
+
+- New env variable for model config
+
 ## [0.7.1]
 
 ### Fixed

{langgraph_agent_toolkit-0.7.1 → langgraph_agent_toolkit-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.7.1
+Version: 0.7.2
 Summary: Full toolkit for running an AI agent service built with LangGraph, FastAPI and Streamlit
 Project-URL: documentation, https://kryvokhyzha.github.io/langgraph-agent-toolkit
 Project-URL: repository, https://github.com/kryvokhyzha/langgraph-agent-toolkit

{langgraph_agent_toolkit-0.7.1 → langgraph_agent_toolkit-0.7.2}/langgraph_agent_toolkit/core/settings.py

@@ -104,6 +104,7 @@ class Settings(BaseSettings):
 
     # Model configurations dictionary
     MODEL_CONFIGS: Dict[str, Dict[str, Any]] = Field(default_factory=dict)
+    MODEL_CONFIGS_PATH: str | None = None
 
     def _apply_langgraph_env_overrides(self) -> None:
         """Apply any LANGGRAPH_ prefixed environment variables to override settings."""
@@ -151,6 +152,7 @@ class Settings(BaseSettings):
     def _initialize_model_configs(self) -> None:
         """Initialize model configurations from environment variables."""
         model_configs_env = os.environ.get("MODEL_CONFIGS")
+        model_configs_path_env = os.environ.get("MODEL_CONFIGS_PATH")
         if model_configs_env:
             try:
                 configs = json.loads(model_configs_env)
@@ -161,6 +163,19 @@ class Settings(BaseSettings):
                     logger.warning("MODEL_CONFIGS environment variable is not a valid JSON object")
             except json.JSONDecodeError:
                 logger.error("Failed to parse MODEL_CONFIGS environment variable as JSON")
+        elif model_configs_path_env:
+            try:
+                with open(model_configs_path_env, "r", encoding="utf-8") as f:
+                    configs = json.load(f)
+                    if isinstance(configs, dict):
+                        self.MODEL_CONFIGS = configs
+                        logger.info(f"Loaded {len(configs)} model configurations from {model_configs_path_env}")
+                    else:
+                        logger.warning("MODEL_CONFIGS_PATH file is not a valid JSON object")
+            except (FileNotFoundError, json.JSONDecodeError) as e:
+                logger.error(f"Failed to load model configurations from {model_configs_path_env}: {e}")
+        else:
+            logger.info("No MODEL_CONFIGS found in environment variables or file")
 
     def get_model_config(self, config_key: str) -> Optional[Dict[str, Any]]:
         """Get a model configuration by key.

{langgraph_agent_toolkit-0.7.1 → langgraph_agent_toolkit-0.7.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-agent-toolkit"
-version = "0.7.1"
+version = "0.7.2"
 description = "Full toolkit for running an AI agent service built with LangGraph, FastAPI and Streamlit"
 readme = "README.md"
 authors = [{ name = "Roman Kryvokhyzha", email = "kriwohizha@gmail.com" }, { name = "Joshua Carroll", email = "carroll.joshk@gmail.com" }]

langgraph_agent_toolkit-0.7.2/scripts/python/099-t.py

@@ -0,0 +1,52 @@
+import os
+
+from dotenv import find_dotenv, load_dotenv
+from langchain.chat_models.base import init_chat_model
+from pydantic import BaseModel, Field
+
+
+load_dotenv(find_dotenv(".local.env"), override=True)
+
+
+class ResponseSchema(BaseModel):
+    response: str = Field(
+        description="The response on user query.",
+    )
+    alternative_response: str = Field(
+        description="The alternative response on user query.",
+    )
+
+
+if __name__ == "__main__":
+    DEFAULT_CONFIG_PREFIX = "agent"
+    DEFAULT_CONFIGURABLE_FIELDS = ("temperature", "max_tokens", "top_p", "streaming")
+    DEFAULT_MODEL_PARAMETER_VALUES = dict(
+        temperature=0.0,
+        max_tokens=1024,
+        top_p=0.7,
+        streaming=False,
+    )
+
+    configurable_model = init_chat_model(
+        model=os.getenv("OPENAI_MODEL_NAME"),
+        model_provider="openai",
+        config_prefix=DEFAULT_CONFIG_PREFIX,
+        configurable_fields=DEFAULT_CONFIGURABLE_FIELDS,
+        openai_api_key=os.getenv("OPENAI_API_KEY"),
+        openai_api_base=os.getenv("OPENAI_API_BASE_URL"),
+        **DEFAULT_MODEL_PARAMETER_VALUES,
+    )
+
+    print(f"Configurable model type: {type(configurable_model)}")
+    print("=" * 25)
+
+    model_so = configurable_model.with_structured_output(ResponseSchema)
+
+    print(f"Model with SO type: {type(configurable_model)}")
+    print("=" * 25)
+
+    result = model_so.invoke("Who is the president of Brasil?", config={"temperature": 0.5})
+
+    print(f"Result type: {type(result)}")
+    print(f"Result: {result}")
+    print("=" * 25)

{langgraph_agent_toolkit-0.7.1 → langgraph_agent_toolkit-0.7.2}/uv.lock

@@ -1669,7 +1669,7 @@ wheels = [
 
 [[package]]
 name = "langgraph-agent-toolkit"
-version = "0.7.1"
+version = "0.7.2"
 source = { editable = "." }
 dependencies = [
     { name = "duckduckgo-search" },