langgraph-agent-toolkit 0.7.7__tar.gz → 0.7.8__tar.gz

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/.env.example

@@ -84,6 +84,9 @@ DEEPSEEK_API_KEY=
 OLLAMA_MODEL_NAME=
 OLLAMA_BASE_URL=
 
+# Openrouter
+OPENROUTER_API_KEY=
+
 # 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}}

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/CHANGELOG.md

@@ -6,6 +6,17 @@ 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.8]
+
+### Added
+
+- Prompt manager
+- Utils functions
+
+### Updated
+
+- Dependencies
+
 ## [0.7.7]
 
 ### Fixed

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.7.7
+Version: 0.7.8
 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
@@ -31,11 +31,11 @@ Requires-Dist: jiter~=0.10.0
 Requires-Dist: joblib>=1.4.2
 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-postgres<=2.0.21
 Requires-Dist: langgraph-checkpoint-sqlite~=2.0.7
 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: langgraph~=0.6.2
 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'
@@ -50,7 +50,7 @@ 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.47,>=1.45
+Requires-Dist: streamlit<1.48,>=1.45
 Requires-Dist: uvicorn~=0.34.0
 Requires-Dist: watchdog~=6.0.0
 Provides-Extra: all-backends

langgraph_agent_toolkit-0.7.8/langgraph_agent_toolkit/agents/agent.py

@@ -0,0 +1,35 @@
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Optional
+
+import rootutils
+from langgraph.func import Pregel
+
+from langgraph_agent_toolkit.core.observability.base import BaseObservabilityPlatform
+
+
+@dataclass
+class Agent:
+    name: str
+    description: str
+    graph: Pregel
+    observability: BaseObservabilityPlatform | None = None
+
+
+def draw_agent_graph(agent: Agent, image_path: Optional[str | Path] = None, **kwargs):
+    if image_path:
+        image_path = Path(image_path)
+    else:
+        path_to_root = rootutils.find_root(search_from=__file__, indicator=".project-root")
+
+        image_path = path_to_root / "docs" / "visualizations"
+        image_path.mkdir(exist_ok=True, parents=True)
+
+        image_path = image_path / f"{agent.name}.png"
+
+    graph_image = agent.graph.get_graph(xray=1).draw_mermaid_png(**kwargs)
+    with open(image_path, "wb") as f:
+        f.write(graph_image)
+
+
+__all__ = ["Agent", "draw_agent_graph"]

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/langgraph_agent_toolkit/agents/components/utils.py

@@ -1,5 +1,6 @@
-from typing import TypeVar
+from typing import Sequence, TypeVar
 
+from langchain_core.messages import BaseMessage
 from langchain_core.messages.utils import trim_messages
 from langchain_core.runnables import RunnableConfig
 from langgraph.managed.is_last_step import RemainingSteps
@@ -38,3 +39,34 @@ def pre_model_hook_standard(state: T, config: RunnableConfig):
 
 def default_pre_model_hook(state: T, config: RunnableConfig) -> T:
     return state
+
+
+def trim_messages_wrapper(messages: Sequence[BaseMessage], config: RunnableConfig, **kwargs):
+    """Trim messages to fit within the max token limit.
+
+    Args:
+        messages (Sequence[BaseMessage]): The list of messages to trim.
+        config (RunnableConfig): Configuration containing parameters for trimming.
+        **kwargs: Additional keyword arguments to pass to the trim function.
+
+    Returns:
+        Sequence[BaseMessage]: The trimmed list of messages.
+
+    """
+    _max_messages = config.get("configurable", {}).get("checkpointer_params", {}).get("k", None)
+
+    default_kwargs = dict(
+        token_counter=len,
+        max_tokens=int(_max_messages or DEFAULT_MAX_MESSAGE_HISTORY_LENGTH),
+        strategy="last",
+        start_on="human",
+        end_on=("human", "tool"),
+        include_system=True,
+        allow_partial=False,
+    )
+    default_kwargs.update(kwargs)
+
+    return trim_messages(
+        messages=messages,
+        **default_kwargs,
+    )

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/langgraph_agent_toolkit/core/memory/postgres.py

@@ -34,6 +34,13 @@ class PostgresMemoryBackend(BaseMemoryBackend):
                 f"Missing required PostgreSQL configuration: {', '.join(missing)}. "
                 "These environment variables must be set to use PostgreSQL persistence."
             )
+
+        if settings.POSTGRES_MIN_SIZE > settings.POSTGRES_POOL_SIZE:
+            raise ValueError(
+                f"POSTGRES_MIN_SIZE ({settings.POSTGRES_MIN_SIZE}) must be less than or equal to "
+                f"POSTGRES_POOL_SIZE ({settings.POSTGRES_POOL_SIZE})"
+            )
+
         return True
 
     @staticmethod
@@ -48,25 +55,35 @@ class PostgresMemoryBackend(BaseMemoryBackend):
 
     @asynccontextmanager
     async def _get_connection_context(
-        self, factory_func: Callable[[AsyncConnectionPool], T]
+        self,
+        factory_func: Callable[[AsyncConnectionPool], T],
+        app_prefix: str,
     ) -> AsyncGenerator[T, None]:
         """Yield the result of the factory function.
 
         Args:
             factory_func: Function that creates the appropriate object from the connection pool
+            app_prefix: Prefix for the application name in the connection pool
 
         Yields:
             The object created by the factory_func
 
         """
+        application_name = f"{settings.POSTGRES_APPLICATION_NAME}-{app_prefix}"
+
         logger.info(
             f"Creating PostgreSQL connection pool: min_size={settings.POSTGRES_MIN_SIZE}, "
             f"max_size={settings.POSTGRES_POOL_SIZE}, max_idle={settings.POSTGRES_MAX_IDLE}, "
-            f"schema={settings.POSTGRES_SCHEMA}"
+            f"schema={settings.POSTGRES_SCHEMA}, application_name={application_name}"
         )
 
         # Prepare connection kwargs with schema setting
-        connection_kwargs = {"autocommit": True, "prepare_threshold": 0, "row_factory": dict_row}
+        connection_kwargs = {
+            "autocommit": True,
+            "prepare_threshold": 0,
+            "row_factory": dict_row,
+            "application_name": application_name,
+        }
 
         # Set search_path using options parameter if schema is specified and not default
         if settings.POSTGRES_SCHEMA and settings.POSTGRES_SCHEMA != "public":
@@ -78,6 +95,7 @@ class PostgresMemoryBackend(BaseMemoryBackend):
             min_size=settings.POSTGRES_MIN_SIZE,
             max_size=settings.POSTGRES_POOL_SIZE,
             max_idle=settings.POSTGRES_MAX_IDLE,
+            check=AsyncConnectionPool.check_connection,
             kwargs=connection_kwargs,
         ) as pool:
             logger.info("PostgreSQL connection pool opened successfully")
@@ -95,7 +113,9 @@ class PostgresMemoryBackend(BaseMemoryBackend):
             AsyncPostgresSaver: The database saver instance
 
         """
-        async with self._get_connection_context(lambda pool: AsyncPostgresSaver(conn=pool)) as saver:
+        async with self._get_connection_context(
+            lambda pool: AsyncPostgresSaver(conn=pool), app_prefix="saver"
+        ) as saver:
             yield saver
 
     @asynccontextmanager
@@ -106,7 +126,9 @@ class PostgresMemoryBackend(BaseMemoryBackend):
             AsyncPostgresStore: The database store instance
 
         """
-        async with self._get_connection_context(lambda pool: AsyncPostgresStore(conn=pool)) as store:
+        async with self._get_connection_context(
+            lambda pool: AsyncPostgresStore(conn=pool, app_prefix="store")
+        ) as store:
             yield store
 
     def get_checkpoint_saver(self) -> AbstractAsyncContextManager[AsyncPostgresSaver]:

langgraph_agent_toolkit-0.7.8/langgraph_agent_toolkit/core/prompts/prompt_manager.py

@@ -0,0 +1,126 @@
+from pathlib import Path
+from typing import Dict, List, Optional
+
+from langgraph_agent_toolkit.core.observability.factory import ObservabilityFactory
+from langgraph_agent_toolkit.core.observability.types import ChatMessageDict, MessageRole
+from langgraph_agent_toolkit.core.prompts.chat_prompt_template import ObservabilityChatPromptTemplate
+from langgraph_agent_toolkit.core.settings import settings
+from langgraph_agent_toolkit.helper.utils import read_file
+
+
+class PromptManager:
+    """Manages prompt templates for the price plan assistant agent."""
+
+    def __init__(
+        self,
+        observability_backend: Optional[str] = None,
+        prompts_dir: Optional[Path] = None,
+        force_create_new_version: bool = False,
+        template_format: str = "jinja2",
+        load_at_runtime: bool = True,
+    ):
+        """Initialize the PromptManager with configurable parameters.
+
+        Args:
+            observability_backend: Backend type for observability (defaults to settings.OBSERVABILITY_BACKEND)
+            prompts_dir: Directory containing prompt templates (defaults to ./prompts)
+            force_create_new_version: Whether to force creation of new prompt versions
+            template_format: Format for prompt templates (default: "jinja2")
+            load_at_runtime: Whether to load prompts at runtime (default: True)
+
+        """
+        self._prompt_cache: Dict[str, ObservabilityChatPromptTemplate] = {}
+        self._observability = None
+        self._observability_backend = observability_backend or settings.OBSERVABILITY_BACKEND
+        self._prompts_dir = prompts_dir or (Path(__file__).parent / "prompts")
+        self._force_create_new_version = force_create_new_version
+        self._template_format = template_format
+        self._load_at_runtime = load_at_runtime
+
+    @property
+    def observability(self):
+        """Lazy-loaded observability platform."""
+        if self._observability is None:
+            self._observability = ObservabilityFactory.create(self._observability_backend)
+        return self._observability
+
+    def _create_prompt_template(
+        self,
+        prompt_name: str,
+        template_path: Path,
+        input_variables: List[str],
+        has_messages_placeholder: bool = False,
+        partial_variables: Optional[Dict[str, str]] = None,
+    ) -> ObservabilityChatPromptTemplate:
+        """Create and cache a prompt template from a file."""
+        template_content = read_file(template_path)
+
+        # Build prompt template messages
+        prompt_template = [
+            ChatMessageDict(
+                role=MessageRole.SYSTEM,
+                content=template_content,
+            )
+        ]
+
+        # Add messages placeholder if needed
+        if has_messages_placeholder:
+            prompt_template.append(
+                ChatMessageDict(
+                    role=MessageRole.PLACEHOLDER,
+                    content="messages",
+                )
+            )
+
+        # Push to observability platform
+        self.observability.push_prompt(
+            name=prompt_name,
+            prompt_template=prompt_template,
+            force_create_new_version=self._force_create_new_version,
+        )
+
+        # Create and cache the prompt
+        prompt = ObservabilityChatPromptTemplate.from_observability_platform(
+            prompt_name=prompt_name,
+            observability_platform=self.observability,
+            load_at_runtime=self._load_at_runtime,
+            template_format=self._template_format,
+            input_variables=input_variables,
+            partial_variables=partial_variables or {},
+        )
+
+        self._prompt_cache[prompt_name] = prompt
+        return prompt
+
+    def _get_or_create_prompt(
+        self,
+        prompt_name: str,
+        template_filename: str,
+        input_variables: List[str],
+        has_messages_placeholder: bool = False,
+        partial_variables: Optional[Dict[str, str]] = None,
+    ) -> ObservabilityChatPromptTemplate:
+        """Get cached prompt or create it if it doesn't exist."""
+        if prompt_name not in self._prompt_cache:
+            template_path = self._prompts_dir / template_filename
+            self._create_prompt_template(
+                prompt_name=prompt_name,
+                template_path=template_path,
+                input_variables=input_variables,
+                has_messages_placeholder=has_messages_placeholder,
+                partial_variables=partial_variables,
+            )
+        return self._prompt_cache[prompt_name]
+
+    def clear_cache(self) -> None:
+        """Clear the prompt cache."""
+        self._prompt_cache.clear()
+
+    def get_cached_prompt_names(self) -> List[str]:
+        """Get list of cached prompt names."""
+        return list(self._prompt_cache.keys())
+
+    def set_prompts_directory(self, prompts_dir: Path) -> None:
+        """Update the prompts directory and clear cache."""
+        self._prompts_dir = prompts_dir
+        self.clear_cache()

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/langgraph_agent_toolkit/core/settings.py

@@ -70,6 +70,9 @@ class Settings(BaseSettings):
     OLLAMA_MODEL_NAME: str | None = None
     OLLAMA_BASE_URL: str | None = None
 
+    # OpenRouter Settings
+    OPENROUTER_API_KEY: str | None = None
+
     # Observability platform
     OBSERVABILITY_BACKEND: ObservabilityBackend | None = None
 
@@ -94,6 +97,7 @@ class Settings(BaseSettings):
     SQLITE_DB_PATH: str = "checkpoints.db"
 
     # postgresql Configuration
+    POSTGRES_APPLICATION_NAME: str = "langgraph-agent-toolkit"
     POSTGRES_USER: str | None = None
     POSTGRES_PASSWORD: SecretStr | None = None
     POSTGRES_HOST: str | None = None

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/langgraph_agent_toolkit/helper/utils.py

@@ -1,4 +1,6 @@
 import inspect
+from pathlib import Path
+from typing import Any
 
 from langchain_core.messages import (
     AIMessage,
@@ -100,3 +102,20 @@ def create_ai_message(parts: dict) -> AIMessage:
     valid_keys = set(sig.parameters)
     filtered = {k: v for k, v in parts.items() if k in valid_keys}
     return AIMessage(**filtered)
+
+
+def read_file(file_path: Path | str, mode: str = "r", encoding: str = "utf-8", **kwargs) -> Any:
+    """Read the content of a file and return it as a string.
+
+    Args:
+        file_path (Path | str): The path to the file to read.
+        mode (str): The mode in which to open the file. Default is "r".
+        encoding (str): The encoding to use for reading the file. Default is "utf-8".
+        **kwargs: Additional arguments to pass to the open function.
+
+    Returns:
+        Any: The content of the file as a string.
+
+    """
+    with open(file_path, mode=mode, encoding=encoding, **kwargs) as file:
+        return file.read()

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-agent-toolkit"
-version = "0.7.7"
+version = "0.7.8"
 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" }]
@@ -36,10 +36,11 @@ dependencies = [
     "jiter ~= 0.10.0",
     "langchain-core ~= 0.3.60",
     "langchain-community ~= 0.3.24",
-    "langgraph ~= 0.4.5",
+    "langgraph ~= 0.6.2",
     "langgraph-cli[inmem] >= 0.2.10,< 0.4.0",
     "langgraph-checkpoint-sqlite ~= 2.0.7",
-    "langgraph-checkpoint-postgres ~= 2.0.21",
+    # "langgraph-checkpoint-postgres (~=2.0.21, !=2.0.22)",
+    "langgraph-checkpoint-postgres <= 2.0.21",
     "langgraph-supervisor ~= 0.0.21",
     "numpy ~= 1.26.4; python_version <= '3.12'",
     "numpy >= 2.2.3,< 2.4.0; python_version >= '3.13'",
@@ -52,7 +53,7 @@ dependencies = [
     "pyowm ~= 3.3.0",
     "python-dotenv ~= 1.1.0",
     "setuptools >= 75.6.0",
-    "streamlit >= 1.45,< 1.47",
+    "streamlit >= 1.45, < 1.48",
     "watchdog ~= 6.0.0",
     "joblib >= 1.4.2",
     "uvicorn ~= 0.34.0",

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/tests/client/test_client.py

@@ -248,8 +248,8 @@ async def test_astream(agent_client):
     mock_response.aiter_lines = Mock(return_value=async_events())
     mock_response.__aenter__ = AsyncMock(return_value=mock_response)
     mock_response.__aexit__ = AsyncMock(return_value=None)
-    # Explicitly set raise_for_status as an AsyncMock to make it awaitable
-    mock_response.raise_for_status = AsyncMock()
+    # Use Mock() instead of AsyncMock() since raise_for_status is synchronous
+    mock_response.raise_for_status = Mock()
 
     # Create a mock client that returns the mock_response directly (not as a coroutine)
     mock_client = AsyncMock()

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/tests/core/test_memory.py

@@ -78,6 +78,8 @@ class TestPostgresMemoryBackend(unittest.TestCase):
         mock_settings.POSTGRES_HOST = "localhost"
         mock_settings.POSTGRES_PORT = "5432"
         mock_settings.POSTGRES_DB = "testdb"
+        mock_settings.POSTGRES_MIN_SIZE = 1
+        mock_settings.POSTGRES_POOL_SIZE = 5
 
         self.assertTrue(self.backend.validate_config())
 
@@ -89,6 +91,8 @@ class TestPostgresMemoryBackend(unittest.TestCase):
         mock_settings.POSTGRES_HOST = "localhost"
         mock_settings.POSTGRES_PORT = "5432"
         mock_settings.POSTGRES_DB = "testdb"
+        mock_settings.POSTGRES_MIN_SIZE = 1
+        mock_settings.POSTGRES_POOL_SIZE = 5
 
         with pytest.raises(ValueError, match=r"Missing required PostgreSQL configuration"):
             self.backend.validate_config()

{langgraph_agent_toolkit-0.7.7 → langgraph_agent_toolkit-0.7.8}/tests/service/test_service.py

@@ -5,7 +5,7 @@ import langsmith
 import pytest
 from langchain_core.messages import AIMessage, HumanMessage
 from langgraph.errors import GraphRecursionError
-from langgraph.pregel.types import StateSnapshot
+from langgraph.types import StateSnapshot
 
 from langgraph_agent_toolkit.schema import ChatHistory, ChatMessage, ServiceMetadata
 from langgraph_agent_toolkit.schema.models import ModelProvider