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

{langgraph_agent_toolkit-0.7.0 → langgraph_agent_toolkit-0.7.1}/.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.1}/CHANGELOG.md

@@ -6,6 +6,30 @@ 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.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.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.7.0
+Version: 0.7.1
 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.1}/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.1}/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.1}/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.1}/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,9 +98,9 @@ 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)

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-agent-toolkit"
-version = "0.7.0"
+version = "0.7.1"
 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.0 → langgraph_agent_toolkit-0.7.1}/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.1}/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