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

{langgraph_agent_toolkit-0.7.1 → langgraph_agent_toolkit-0.7.3}/.env.example

@@ -86,6 +86,8 @@ OLLAMA_BASE_URL=
 # Model configurations - use multiline format with line continuation for readability
 # It can be useful if you want to use different models for different agents or use different models from different providers.
 MODEL_CONFIGS={"router":{"provider":"azure_openai","model_name":"gpt-4o","openai_api_key":"your-azure-key-here","azure_endpoint":"https://your-resource.openai.azure.com/","openai_api_version":"2024-12-01-preview","deployment_name":"gpt-4o-deployment"},"assistant":{"provider":"azure_openai","model_name":"gpt-4o-mini","openai_api_key":"your-azure-key-here","azure_endpoint":"https://your-resource.openai.azure.com/","openai_api_version":"2024-12-01-preview","deployment_name":"gpt-4o-mini-deployment"},"analyzer":{"provider":"google_genai","model_name":"gemini-pro","api_key":"your-google-key-here","temperature":0.7}}
+MODEL_CONFIGS_BASE64=
+MODEL_CONFIGS_PATH=
 
 # Amazon Bedrock Knowledge Base ID
 AWS_KB_ID=

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

@@ -6,6 +6,18 @@ 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.3]
+
+### Added
+
+- New env variable for model config (base64)
+
+## [0.7.2]
+
+### Added
+
+- New env variable for model config (file)
+
 ## [0.7.1]
 
 ### Fixed

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.7.1
+Version: 0.7.3
 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.3}/langgraph_agent_toolkit/core/settings.py

@@ -1,3 +1,4 @@
+import base64
 import json
 import os
 from typing import Annotated, Any, Dict, Optional
@@ -104,6 +105,8 @@ class Settings(BaseSettings):
 
     # Model configurations dictionary
     MODEL_CONFIGS: Dict[str, Dict[str, Any]] = Field(default_factory=dict)
+    MODEL_CONFIGS_BASE64: str | None = None
+    MODEL_CONFIGS_PATH: str | None = None
 
     def _apply_langgraph_env_overrides(self) -> None:
         """Apply any LANGGRAPH_ prefixed environment variables to override settings."""
@@ -151,6 +154,8 @@ 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_base64_env = os.environ.get("MODEL_CONFIGS_BASE64")
+        model_configs_path_env = os.environ.get("MODEL_CONFIGS_PATH")
         if model_configs_env:
             try:
                 configs = json.loads(model_configs_env)
@@ -161,6 +166,30 @@ 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_base64_env:
+            try:
+                decoded_configs = base64.b64decode(model_configs_base64_env).decode("utf-8")
+                configs = json.loads(decoded_configs)
+                if isinstance(configs, dict):
+                    self.MODEL_CONFIGS = configs
+                    logger.info(f"Loaded {len(configs)} model configurations from MODEL_CONFIGS_BASE64")
+                else:
+                    logger.warning("MODEL_CONFIGS_BASE64 cannot be parsed as a valid JSON object")
+            except (ValueError, json.JSONDecodeError) as e:
+                logger.error(f"Failed to decode MODEL_CONFIGS_BASE64: {e}")
+        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 cannot be parsed as 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.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-agent-toolkit"
-version = "0.7.1"
+version = "0.7.3"
 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.3/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.3}/uv.lock

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