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

{langgraph_agent_toolkit-0.7.0 → langgraph_agent_toolkit-0.7.2}/.env.example

@@ -85,31 +85,7 @@ 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={\
-  "gpt4o": {\
-    "provider": "azure_openai",\
-    "name": "gpt-4o",\
-    "api_key": "azure-key-123",\
-    "endpoint": "https://your-resource.openai.azure.com/",\
-    "api_version": "2023-05-15",\
-    "deployment": "gpt4o-deployment",\
-    "temperature": 0.7\
-  },\
-  "gpt4o-mini": {\
-    "provider": "azure_openai",\
-    "name": "gpt-4o-mini",\
-    "api_key": "azure-key-123",\
-    "endpoint": "https://your-resource.openai.azure.com/",\
-    "api_version": "2023-05-15",\
-    "deployment": "gpt4o-mini-deployment"\
-  },\
-  "gemini": {\
-    "provider": "google_genai",\
-    "name": "gemini-pro",\
-    "api_key": "google-key-123",\
-    "temperature": 0.7\
-  }\
-}
+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}}
 
 # Amazon Bedrock Knowledge Base ID
 AWS_KB_ID=

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

@@ -6,6 +6,36 @@ 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
+
+- Agent executor test mock assertion to include additional `environment` and
+  `tags` parameters
+- Async test methods missing `@pytest.mark.asyncio` decorator in prompts tests
+
+### Updated
+
+- Dependencies to latest versions
+- Agent executor `get_callback_handler` method to pass additional parameters:
+  - Added `environment` parameter from settings.ENV_MODE
+  - Added `tags` parameter with agent name for better observability tracking
+
+### Improved
+
+- Langfuse observability prompt hash detection with fallback mechanism:
+  - Enhanced `push_prompt` method to use tags as fallback when commit_message is
+    empty
+  - Added robust `hasattr` checks for both `commit_message` and `tags`
+    attributes
+  - Improved logging to show old vs new hash values for better debugging
+
 ## [0.7.0]
 
 ### Fixed

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.7.0
+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
@@ -33,24 +33,24 @@ Requires-Dist: langchain-community~=0.3.24
 Requires-Dist: langchain-core~=0.3.60
 Requires-Dist: langgraph-checkpoint-postgres~=2.0.21
 Requires-Dist: langgraph-checkpoint-sqlite~=2.0.7
-Requires-Dist: langgraph-cli[inmem]~=0.2.10
+Requires-Dist: langgraph-cli[inmem]<0.4.0,>=0.2.10
 Requires-Dist: langgraph-supervisor~=0.0.21
 Requires-Dist: langgraph~=0.4.5
 Requires-Dist: loguru~=0.7.3
 Requires-Dist: multidict>6.3.2
+Requires-Dist: numpy<2.4.0,>=2.2.3; python_version >= '3.13'
 Requires-Dist: numpy~=1.26.4; python_version <= '3.12'
-Requires-Dist: numpy~=2.2.3; python_version >= '3.13'
 Requires-Dist: onnxruntime~=1.22.0
-Requires-Dist: pandas~=2.2.3
+Requires-Dist: pandas<2.4.0,>=2.2.3
 Requires-Dist: psycopg[binary,pool]~=3.2.4
 Requires-Dist: pyarrow>=19.0.1
-Requires-Dist: pydantic-settings~=2.9.1
+Requires-Dist: pydantic-settings<2.11.0,>=2.9.1
 Requires-Dist: pydantic~=2.11.1
 Requires-Dist: pyowm~=3.3.0
 Requires-Dist: python-dotenv~=1.1.0
 Requires-Dist: rootutils>=1.0.7
 Requires-Dist: setuptools>=75.6.0
-Requires-Dist: streamlit~=1.45.0
+Requires-Dist: streamlit<1.47,>=1.45
 Requires-Dist: uvicorn~=0.34.0
 Requires-Dist: watchdog~=6.0.0
 Provides-Extra: all-backends
@@ -68,8 +68,8 @@ Requires-Dist: langchain-groq~=0.3.2; extra == 'all-llms'
 Requires-Dist: langchain-ollama~=0.3.2; extra == 'all-llms'
 Requires-Dist: langchain-openai~=0.3.16; extra == 'all-llms'
 Provides-Extra: all-observability
-Requires-Dist: langfuse~=2.60.3; extra == 'all-observability'
-Requires-Dist: langsmith~=0.3.31; extra == 'all-observability'
+Requires-Dist: langfuse<3.1.0,>=2.60.3; extra == 'all-observability'
+Requires-Dist: langsmith<0.5.0,>=0.3.31; extra == 'all-observability'
 Provides-Extra: anthropic
 Requires-Dist: langchain-anthropic~=0.3.12; extra == 'anthropic'
 Provides-Extra: aws
@@ -90,9 +90,9 @@ Provides-Extra: gunicorn-backend
 Requires-Dist: gunicorn~=23.0.0; extra == 'gunicorn-backend'
 Requires-Dist: uvicorn[standard]~=0.34.2; extra == 'gunicorn-backend'
 Provides-Extra: langfuse
-Requires-Dist: langfuse~=2.60.3; extra == 'langfuse'
+Requires-Dist: langfuse<3.1.0,>=2.60.3; extra == 'langfuse'
 Provides-Extra: langsmith
-Requires-Dist: langsmith~=0.3.31; extra == 'langsmith'
+Requires-Dist: langsmith<0.5.0,>=0.3.31; extra == 'langsmith'
 Provides-Extra: ollama
 Requires-Dist: langchain-ollama~=0.3.2; extra == 'ollama'
 Provides-Extra: openai
@@ -150,6 +150,9 @@ Whether you're building a simple chatbot or complex multi-agent system, this
 toolkit provides the infrastructure to develop, test, and deploy your
 LangGraph-based agents with confidence.
 
+You can use [DeepWiki](https://deepwiki.com/kryvokhyzha/langgraph-agent-toolkit)
+to learn more about this repository.
+
 ## 📑 Contents
 
 - [Introduction](#-introduction)

{langgraph_agent_toolkit-0.7.0 → langgraph_agent_toolkit-0.7.2}/README.md

@@ -47,6 +47,9 @@ Whether you're building a simple chatbot or complex multi-agent system, this
 toolkit provides the infrastructure to develop, test, and deploy your
 LangGraph-based agents with confidence.
 
+You can use [DeepWiki](https://deepwiki.com/kryvokhyzha/langgraph-agent-toolkit)
+to learn more about this repository.
+
 ## 📑 Contents
 
 - [Introduction](#-introduction)

{langgraph_agent_toolkit-0.7.0 → langgraph_agent_toolkit-0.7.2}/langgraph_agent_toolkit/agents/agent_executor.py

@@ -237,7 +237,12 @@ class AgentExecutor:
         if agent_config:
             configurable.update(agent_config)
 
-        callback = agent.observability.get_callback_handler(session_id=thread_id, user_id=user_id)
+        callback = agent.observability.get_callback_handler(
+            session_id=thread_id,
+            user_id=user_id,
+            environment=settings.ENV_MODE,
+            tags=[agent.name],
+        )
 
         config = RunnableConfig(
             configurable=configurable,

{langgraph_agent_toolkit-0.7.0 → langgraph_agent_toolkit-0.7.2}/langgraph_agent_toolkit/core/observability/langfuse.py

@@ -67,11 +67,21 @@ class LangfuseObservability(BaseObservabilityPlatform):
             existing_prompt = langfuse.get_prompt(name=name)
 
             # Check if content has changed by comparing hashes
-            if existing_prompt.commit_message and existing_prompt.commit_message == prompt_hash:
+            # First try commit_message, then fall back to tags
+            existing_hash = None
+            if hasattr(existing_prompt, "commit_message") and existing_prompt.commit_message:
+                existing_hash = existing_prompt.commit_message
+            elif hasattr(existing_prompt, "tags") and existing_prompt.tags and len(existing_prompt.tags) > 0:
+                # Use the first tag as the hash (assuming it contains the prompt hash)
+                existing_hash = existing_prompt.tags[0]
+
+            if existing_hash and existing_hash == prompt_hash:
                 content_changed = False
-                logger.debug(f"Prompt '{name}' content unchanged")
+                logger.debug(f"Prompt '{name}' content unchanged (hash: {existing_hash})")
             else:
-                logger.debug(f"Prompt '{name}' content changed from previous version")
+                logger.debug(
+                    f"Prompt '{name}' content changed from previous version (old: {existing_hash}, new: {prompt_hash})"
+                )
         except Exception:
             logger.debug(f"Existing prompt '{name}' not found, will create a new one")
 
@@ -95,6 +105,7 @@ class LangfuseObservability(BaseObservabilityPlatform):
                 prompt=prompt_template,
                 labels=labels,
                 type=type_prompt,
+                tags=[prompt_hash],  # for v2 version
                 commit_message=prompt_hash,  # Store hash in commit_message
             )
             if existing_prompt is None:

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

@@ -86,7 +86,7 @@ class Settings(BaseSettings):
 
     LANGFUSE_SECRET_KEY: SecretStr | None = None
     LANGFUSE_PUBLIC_KEY: SecretStr | None = None
-    LANGFUSE_HOST: Annotated[str, BeforeValidator(check_str_is_http)] = "http://localhost:3000"
+    LANGFUSE_HOST: Annotated[str, BeforeValidator(check_str_is_http)] = "https://cloud.langfuse.com"
 
     # Database Configuration
     MEMORY_BACKEND: MemoryBackends = MemoryBackends.SQLITE
@@ -98,12 +98,13 @@ class Settings(BaseSettings):
     POSTGRES_HOST: str | None = None
     POSTGRES_PORT: int | None = None
     POSTGRES_DB: str | None = None
-    POSTGRES_POOL_SIZE: int = Field(default=10, description="Maximum number of connections in the pool")
-    POSTGRES_MIN_SIZE: int = Field(default=3, description="Minimum number of connections in the pool")
-    POSTGRES_MAX_IDLE: int = Field(default=5, description="Maximum number of idle connections")
+    POSTGRES_POOL_SIZE: int = Field(default=200, description="Maximum number of connections in the pool")
+    POSTGRES_MIN_SIZE: int = Field(default=10, description="Minimum number of connections in the pool")
+    POSTGRES_MAX_IDLE: int = Field(default=300, description="Maximum number of idle connections")
 
     # 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.0 → langgraph_agent_toolkit-0.7.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-agent-toolkit"
-version = "0.7.0"
+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" }]
@@ -37,22 +37,22 @@ dependencies = [
     "langchain-core ~= 0.3.60",
     "langchain-community ~= 0.3.24",
     "langgraph ~= 0.4.5",
-    "langgraph-cli[inmem] ~= 0.2.10",
+    "langgraph-cli[inmem] >= 0.2.10,< 0.4.0",
     "langgraph-checkpoint-sqlite ~= 2.0.7",
     "langgraph-checkpoint-postgres ~= 2.0.21",
     "langgraph-supervisor ~= 0.0.21",
     "numpy ~= 1.26.4; python_version <= '3.12'",
-    "numpy ~= 2.2.3; python_version >= '3.13'",
+    "numpy >= 2.2.3,< 2.4.0; python_version >= '3.13'",
     "onnxruntime ~= 1.22.0",
-    "pandas ~= 2.2.3",
+    "pandas >= 2.2.3,< 2.4.0",
     "psycopg[binary,pool] ~= 3.2.4",
     "pyarrow >= 19.0.1",
     "pydantic ~= 2.11.1",
-    "pydantic-settings ~= 2.9.1",
+    "pydantic-settings >= 2.9.1,< 2.11.0",
     "pyowm ~= 3.3.0",
     "python-dotenv ~= 1.1.0",
     "setuptools >= 75.6.0",
-    "streamlit ~= 1.45.0",
+    "streamlit >= 1.45,< 1.47",
     "watchdog ~= 6.0.0",
     "joblib >= 1.4.2",
     "uvicorn ~= 0.34.0",
@@ -81,7 +81,7 @@ client = [
     "httpx ~= 0.28.1",
     "pydantic ~= 2.11.1",
     "python-dotenv ~= 1.1.0",
-    "streamlit ~= 1.45.0",
+    "streamlit >= 1.45,< 1.47",
     "watchdog ~= 6.0.0",
 ]
 
@@ -158,17 +158,17 @@ all-llms = [
 
 # Observability tools and platforms
 langfuse = [
-    "langfuse ~= 2.60.3",
-]
+    "langfuse >= 2.60.3,< 3.1.0",
+ ]
 
-langsmith = [
-    "langsmith ~= 0.3.31",
-]
+ langsmith = [
+    "langsmith >= 0.3.31,< 0.5.0",
+ ]
 
-all-observability = [
-    "langfuse ~= 2.60.3",
-    "langsmith ~= 0.3.31",
-]
+ all-observability = [
+    "langfuse >= 2.60.3,< 3.1.0",
+    "langsmith >= 0.3.31,< 0.5.0",
+ ]
 
 [project.urls]
 documentation = "https://kryvokhyzha.github.io/langgraph-agent-toolkit"

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.0 → langgraph_agent_toolkit-0.7.2}/tests/agents/test_agent_executor.py

@@ -331,38 +331,41 @@ async def test_setup_agent_execution(agent_executor, mock_agent):
     message_content = "Hello, setup!"
     input_obj = MockInput(message=message_content)
 
-    # Call the setup method directly
-    agent, input_data, config, run_id = await agent_executor._setup_agent_execution(
-        agent_id="test-agent",
-        input=input_obj,
-        thread_id="test-thread-123",
-        user_id="test-user-456",
-        model_name="test-model",
-        agent_config={"temperature": 0.7},
-    )
+    with patch("langgraph_agent_toolkit.agents.agent_executor.settings") as mock_settings:
+        mock_settings.ENV_MODE = "test"
+
+        # Call the setup method directly
+        agent, input_data, config, run_id = await agent_executor._setup_agent_execution(
+            agent_id="test-agent",
+            input=input_obj,
+            thread_id="test-thread-123",
+            user_id="test-user-456",
+            model_name="test-model",
+            agent_config={"temperature": 0.7},
+        )
 
-    # Verify agent is correct
-    assert agent is mock_agent
+        # Verify agent is correct
+        assert agent is mock_agent
 
-    # Verify input is a HumanMessage with expected content
-    assert isinstance(input_data, dict)
-    assert "messages" in input_data
-    assert isinstance(input_data["messages"][0], HumanMessage)
-    assert input_data["messages"][0].content == message_content
+        # Verify input is a HumanMessage with expected content
+        assert isinstance(input_data, dict)
+        assert "messages" in input_data
+        assert isinstance(input_data["messages"][0], HumanMessage)
+        assert input_data["messages"][0].content == message_content
 
-    # Verify config is correct - access as a dictionary, not an object
-    assert config["configurable"]["thread_id"] == "test-thread-123"
-    assert config["configurable"]["user_id"] == "test-user-456"
-    assert config["configurable"]["model_name"] == "test-model"
-    assert config["configurable"]["temperature"] == 0.7
+        # Verify config is correct - access as a dictionary, not an object
+        assert config["configurable"]["thread_id"] == "test-thread-123"
+        assert config["configurable"]["user_id"] == "test-user-456"
+        assert config["configurable"]["model_name"] == "test-model"
+        assert config["configurable"]["temperature"] == 0.7
 
-    # Verify run_id is a UUID
-    assert isinstance(run_id, UUID)
+        # Verify run_id is a UUID
+        assert isinstance(run_id, UUID)
 
-    # Verify observability callback was requested with both session_id and user_id
-    mock_agent.observability.get_callback_handler.assert_called_once_with(
-        session_id="test-thread-123", user_id="test-user-456"
-    )
+        # Verify observability callback was requested with both session_id and user_id
+        mock_agent.observability.get_callback_handler.assert_called_once_with(
+            session_id="test-thread-123", user_id="test-user-456", environment="test", tags=["test-agent"]
+        )
 
 
 @pytest.mark.asyncio

{langgraph_agent_toolkit-0.7.0 → langgraph_agent_toolkit-0.7.2}/tests/core/test_prompts.py

@@ -402,6 +402,7 @@ class TestObservabilityChatPromptTemplate:
         template.invoke(input=dict(test_var="test4"))
         assert mock_platform.pull_prompt.call_count == 2
 
+    @pytest.mark.asyncio
     async def test_async_invoke(self):
         """Test asynchronous invocation."""
         template = ObservabilityChatPromptTemplate.from_observability_platform(
@@ -575,6 +576,7 @@ class TestObservabilityChatPromptTemplate:
         assert "Follow-up answer" in messages2[4].content
         assert "Human prompt with human value" in messages2[5].content
 
+    @pytest.mark.asyncio
     async def test_async_message_placeholder(self):
         """Test asynchronous handling of MessagesPlaceholder."""
         # Use the same approach as in test_message_placeholder_handling