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

{langgraph_agent_toolkit-0.7.2 → langgraph_agent_toolkit-0.7.4}/.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.2 → langgraph_agent_toolkit-0.7.4}/CHANGELOG.md

@@ -6,11 +6,23 @@ 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.4]
+
+### Updated
+
+- Langfuse Callback import fix
+
+## [0.7.3]
+
+### Added
+
+- New env variable for model config (base64)
+
 ## [0.7.2]
 
 ### Added
 
-- New env variable for model config
+- New env variable for model config (file)
 
 ## [0.7.1]
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.7.2
+Version: 0.7.4
 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.2 → langgraph_agent_toolkit-0.7.4}/langgraph_agent_toolkit/core/observability/langfuse.py

@@ -3,7 +3,6 @@ import json
 from typing import Any, Dict, Literal, Optional, Tuple, Union
 
 from langfuse import Langfuse
-from langfuse.callback import CallbackHandler
 
 from langgraph_agent_toolkit.core.observability.base import BaseObservabilityPlatform
 from langgraph_agent_toolkit.core.observability.types import PromptReturnType, PromptTemplateType
@@ -11,6 +10,13 @@ from langgraph_agent_toolkit.helper.constants import DEFAULT_CACHE_TTL_SECOND
 from langgraph_agent_toolkit.helper.logging import logger
 
 
+try:
+    from langfuse.callback import CallbackHandler
+except (ModuleNotFoundError, ImportError):
+    # New langfuse version uses langfuse.langchain.CallbackHandler
+    from langfuse.langchain import CallbackHandler
+
+
 class LangfuseObservability(BaseObservabilityPlatform):
     """Langfuse implementation of observability platform."""
 

{langgraph_agent_toolkit-0.7.2 → langgraph_agent_toolkit-0.7.4}/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,7 @@ 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:
@@ -152,6 +154,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_base64_env = os.environ.get("MODEL_CONFIGS_BASE64")
         model_configs_path_env = os.environ.get("MODEL_CONFIGS_PATH")
         if model_configs_env:
             try:
@@ -163,6 +166,17 @@ 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:
@@ -171,7 +185,7 @@ class Settings(BaseSettings):
                         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")
+                        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:

{langgraph_agent_toolkit-0.7.2 → langgraph_agent_toolkit-0.7.4}/langgraph_agent_toolkit/run_agent.py

@@ -5,7 +5,12 @@ from uuid import UUID
 import rootutils
 from dotenv import find_dotenv, load_dotenv
 from langchain_core.runnables import RunnableConfig
-from langfuse.callback import CallbackHandler
+
+
+try:
+    from langfuse.callback import CallbackHandler
+except (ModuleNotFoundError, ImportError):
+    from langfuse.langchain import CallbackHandler
 
 
 _ = rootutils.setup_root(

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-agent-toolkit"
-version = "0.7.2"
+version = "0.7.4"
 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" }]