langgraph-agent-toolkit 0.4.1__tar.gz → 0.4.3__tar.gz

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/.github/workflows/deploy.yml

@@ -5,16 +5,29 @@ on:
     branches:
       - main
   workflow_dispatch:
+    inputs:
+      deploy_enabled:
+        description: 'Enable deployment (set to "true" to deploy)'
+        required: true
+        default: "false"
+        type: choice
+        options:
+          - "false"
+          - "true"
 
 jobs:
   test:
     # Don't try to run deployment in forks and repo copies
-    if: github.repository == 'JoshuaC215/agent-service-toolkit'
+    if: github.repository == 'kryvokhyzha/langgraph-agent-toolkit'
     uses: ./.github/workflows/test.yml
 
   build:
     runs-on: ubuntu-latest
     needs: [test]
+    # Skip build if not manually triggered with deploy_enabled=true
+    if:
+      github.event_name == 'workflow_dispatch' &&
+      github.event.inputs.deploy_enabled == 'true'
 
     steps:
       - uses: actions/checkout@v4
@@ -42,6 +55,8 @@ jobs:
   deploy:
     runs-on: ubuntu-latest
     needs: build
+    # This job will only run if build job runs and succeeds
+    if: always() && needs.build.result == 'success'
 
     steps:
       - name: Deploy to Azure Web App

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/.github/workflows/sphinx.yml

@@ -129,7 +129,7 @@ jobs:
           OBSERVABILITY_BACKEND: "none"
 
       - name: Deploy to GitHub Pages
-        uses: peaceiris/actions-gh-pages@v3
+        uses: peaceiris/actions-gh-pages@v4
         with:
           github_token: ${{ secrets.GITHUB_TOKEN }}
           publish_dir: docs/_build/html

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/CHANGELOG.md

@@ -6,6 +6,21 @@ 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.4.3]
+
+### Updated
+
+- Project dependencies
+
+## [0.4.2]
+
+### Fixed
+
+- Streaming bug
+- Steamlit welcom message display
+- Client handling error
+- Package dependencies
+
 ## [0.4.1]
 
 ### Updated

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.4.1
+Version: 0.4.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
@@ -28,14 +28,14 @@ Requires-Dist: grpcio>=1.68.0
 Requires-Dist: httpx~=0.28.1
 Requires-Dist: jinja2~=3.1.6
 Requires-Dist: jiter~=0.9.0
-Requires-Dist: joblib~=1.4.2
+Requires-Dist: joblib>=1.4.2
 Requires-Dist: langchain-community~=0.3.21
 Requires-Dist: langchain-core~=0.3.52
 Requires-Dist: langgraph-checkpoint-postgres~=2.0.19
 Requires-Dist: langgraph-checkpoint-sqlite~=2.0.6
 Requires-Dist: langgraph-cli[inmem]~=0.2.3
 Requires-Dist: langgraph-supervisor~=0.0.16
-Requires-Dist: langgraph~=0.3.30
+Requires-Dist: langgraph~=0.4.1
 Requires-Dist: loguru~=0.7.3
 Requires-Dist: multidict>6.3.2
 Requires-Dist: numpy~=1.26.4; python_version <= '3.12'
@@ -49,7 +49,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.44.1
+Requires-Dist: streamlit~=1.45.0
 Requires-Dist: uvicorn~=0.34.0
 Requires-Dist: watchdog~=6.0.0
 Provides-Extra: all-backends
@@ -101,18 +101,18 @@ Requires-Dist: uvicorn~=0.34.2; extra == 'uvicorn-backend'
 Description-Content-Type: text/markdown
 
 <div align="center">
-   <img alt="LangGraph Agent Toolkit Logo" src="https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/media/logo.svg?raw=true" width="300">
+   <img alt="LangGraph Agent Toolkit Logo" src="https://raw.githubusercontent.com/kryvokhyzha/langgraph-agent-toolkit/main/docs/media/logo.svg" width="300">
 </div>
 
 ---
 
-# 🧰 Langgraph Agent Toolkit
+# 🧰 LangGraph Agent Toolkit
 
-|            |                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                    |
-| ---------- | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
-| CI/Testing | [![build status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml) [![docs status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/sphinx.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/sphinx.yml) [![codecov](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit/graph/badge.svg?token=OHSACTNSWZ)](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit)                                                                                                                                                         |
-| Package    | [![GitHub release](https://img.shields.io/github/v/release/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/releases) [![PyPI version](https://img.shields.io/pypi/v/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) [![PyPI Downloads](https://img.shields.io/pypi/dm/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) [![Python Version](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Fkryvokhyzha%2Flanggraph-agent-toolkit%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/pyproject.toml) |
-| Meta       | [![Code style: ruff](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff) [![GitHub License](https://img.shields.io/github/license/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE)                                                                                                                                                                                                                                                                                                                                                                                                                                       |
+|            |                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                            |
+| ---------- | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| CI/Testing | [![build status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml) [![docs status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/sphinx.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/sphinx.yml) [![codecov](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit/graph/badge.svg?token=OHSACTNSWZ)](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit) |
+| Package    | [![PyPI version](https://img.shields.io/pypi/v/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) [![PyPI Downloads](https://img.shields.io/pypi/dm/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) [![Python Version](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Fkryvokhyzha%2Flanggraph-agent-toolkit%2Fmain%2Fpyproject.toml)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/pyproject.toml)                          |
+| Meta       | [![Code style: ruff](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff) [![GitHub License](https://img.shields.io/github/license/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE)                                                                                                                                                                                                                                                                               |
 
 <!-- [![Streamlit App](https://static.streamlit.io/badges/streamlit_badge_black_red.svg)](https://langgraph-agent-toolkit.streamlit.app/) -->
 
@@ -123,7 +123,7 @@ LangGraph, FastAPI, and Streamlit. It provides a production-ready framework for
 creating conversational AI agents with features like multi-provider LLM support,
 streaming responses, observability, memory and prompt management.
 
-### What is langgraph-agent-toolkit?
+### What is langGraph-agent-toolkit?
 
 The langgraph-agent-toolkit is a full-featured framework for developing and
 deploying AI agent services. Built on the foundation of:
@@ -213,7 +213,7 @@ For detailed installation instructions and available extras, see the
 
 ## 🏗️ Architecture
 
-<img src="https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/media/agent_architecture.png?raw=true" width="800">
+<img src="https://raw.githubusercontent.com/kryvokhyzha/langgraph-agent-toolkit/main/docs/media/agent_architecture.png" width="800">
 
 <a name="key-features"></a>
 
@@ -284,11 +284,13 @@ Docker setup, using the AgentClient, and local development, see the
 
 ## 📚 Documentation
 
-Full documentation is available in the `docs/` directory and includes:
+Full documentation is available at
+[GitHub repository](https://github.com/kryvokhyzha/langgraph-agent-toolkit/tree/main/docs/)
+and includes:
 
-- [Installation Guide](docs/installation.rst)
-- [Environment Setup](docs/environment_setup.rst)
-- [Usage Guide](docs/usage.rst)
+- [Installation Guide](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/installation.rst)
+- [Environment Setup](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/environment_setup.rst)
+- [Usage Guide](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/usage.rst)
 
 <a name="useful-resources"></a>
 

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/README.md

@@ -1,16 +1,16 @@
 <div align="center">
-   <img alt="LangGraph Agent Toolkit Logo" src="https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/media/logo.svg?raw=true" width="300">
+   <img alt="LangGraph Agent Toolkit Logo" src="https://raw.githubusercontent.com/kryvokhyzha/langgraph-agent-toolkit/main/docs/media/logo.svg" width="300">
 </div>
 
 ---
 
-# 🧰 Langgraph Agent Toolkit
+# 🧰 LangGraph Agent Toolkit
 
-|            |                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                    |
-| ---------- | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
-| CI/Testing | [![build status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml) [![docs status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/sphinx.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/sphinx.yml) [![codecov](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit/graph/badge.svg?token=OHSACTNSWZ)](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit)                                                                                                                                                         |
-| Package    | [![GitHub release](https://img.shields.io/github/v/release/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/releases) [![PyPI version](https://img.shields.io/pypi/v/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) [![PyPI Downloads](https://img.shields.io/pypi/dm/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) [![Python Version](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Fkryvokhyzha%2Flanggraph-agent-toolkit%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/pyproject.toml) |
-| Meta       | [![Code style: ruff](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff) [![GitHub License](https://img.shields.io/github/license/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE)                                                                                                                                                                                                                                                                                                                                                                                                                                       |
+|            |                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                            |
+| ---------- | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| CI/Testing | [![build status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml) [![docs status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/sphinx.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/sphinx.yml) [![codecov](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit/graph/badge.svg?token=OHSACTNSWZ)](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit) |
+| Package    | [![PyPI version](https://img.shields.io/pypi/v/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) [![PyPI Downloads](https://img.shields.io/pypi/dm/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) [![Python Version](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Fkryvokhyzha%2Flanggraph-agent-toolkit%2Fmain%2Fpyproject.toml)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/pyproject.toml)                          |
+| Meta       | [![Code style: ruff](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff) [![GitHub License](https://img.shields.io/github/license/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE)                                                                                                                                                                                                                                                                               |
 
 <!-- [![Streamlit App](https://static.streamlit.io/badges/streamlit_badge_black_red.svg)](https://langgraph-agent-toolkit.streamlit.app/) -->
 
@@ -21,7 +21,7 @@ LangGraph, FastAPI, and Streamlit. It provides a production-ready framework for
 creating conversational AI agents with features like multi-provider LLM support,
 streaming responses, observability, memory and prompt management.
 
-### What is langgraph-agent-toolkit?
+### What is langGraph-agent-toolkit?
 
 The langgraph-agent-toolkit is a full-featured framework for developing and
 deploying AI agent services. Built on the foundation of:
@@ -111,7 +111,7 @@ For detailed installation instructions and available extras, see the
 
 ## 🏗️ Architecture
 
-<img src="https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/media/agent_architecture.png?raw=true" width="800">
+<img src="https://raw.githubusercontent.com/kryvokhyzha/langgraph-agent-toolkit/main/docs/media/agent_architecture.png" width="800">
 
 <a name="key-features"></a>
 
@@ -182,11 +182,13 @@ Docker setup, using the AgentClient, and local development, see the
 
 ## 📚 Documentation
 
-Full documentation is available in the `docs/` directory and includes:
+Full documentation is available at
+[GitHub repository](https://github.com/kryvokhyzha/langgraph-agent-toolkit/tree/main/docs/)
+and includes:
 
-- [Installation Guide](docs/installation.rst)
-- [Environment Setup](docs/environment_setup.rst)
-- [Usage Guide](docs/usage.rst)
+- [Installation Guide](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/installation.rst)
+- [Environment Setup](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/environment_setup.rst)
+- [Usage Guide](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/usage.rst)
 
 <a name="useful-resources"></a>
 

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/docker/app/Dockerfile

@@ -6,10 +6,10 @@ ENV UV_PROJECT_ENVIRONMENT="/usr/local/"
 ENV UV_COMPILE_BYTECODE=1
 ENV PYTHONPATH="/app"
 
-COPY pyproject.toml .
-COPY uv.lock .
+COPY pyproject.toml uv.lock ./
+
 RUN pip install --no-cache-dir uv
-RUN uv sync --frozen --only-group client \
+RUN uv sync --frozen --no-install-project --only-group client \
     && rm -rf /root/.cache
 
 COPY langgraph_agent_toolkit/ ./langgraph_agent_toolkit/

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/langgraph_agent_toolkit/agents/agent_executor.py

@@ -387,8 +387,7 @@ class AgentExecutor:
                             new_messages.append(AIMessage(content=interrupt.value))
                         continue
 
-                    updates = updates or {}
-                    update_messages = updates.get("messages", [])
+                    update_messages = (updates or {}).get("messages", [])
 
                     # Special case for supervisor agent
                     if node == "supervisor":
@@ -430,9 +429,9 @@ class AgentExecutor:
             # More info at: https://langchain-ai.github.io/langgraph/cloud/how-tos/stream_messages/
             processed_messages = []
             current_message: dict[str, Any] = {}
-            for message in new_messages:
-                if isinstance(message, tuple):
-                    key, value = message
+            for msg in new_messages:
+                if isinstance(msg, tuple):
+                    key, value = msg
                     # Store parts in temporary dict
                     current_message[key] = value
                 else:
@@ -440,18 +439,18 @@ class AgentExecutor:
                     if current_message:
                         processed_messages.append(create_ai_message(current_message))
                         current_message = {}
-                    processed_messages.append(message)
+                    processed_messages.append(msg)
 
             # Add any remaining message parts
             if current_message:
                 processed_messages.append(create_ai_message(current_message))
 
-            for message in processed_messages:
+            for msg in new_messages:
                 try:
                     chat_message = langchain_to_chat_message(msg)
                     chat_message.run_id = str(run_id)
                     # Skip the input message if it's repeated by LangGraph
-                    if chat_message.type == "human" and chat_message.content == message:
+                    if chat_message.type == "human" and chat_message.content == msg:
                         continue
                     yield chat_message
                 except Exception as e:

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/langgraph_agent_toolkit/client/client.py

@@ -370,10 +370,14 @@ class AgentClient:
                     response.raise_for_status()
                     async for line in response.aiter_lines():
                         if line.strip():
-                            parsed = self._parse_stream_line(line)
-                            if parsed is None:
+                            try:
+                                parsed = self._parse_stream_line(line)
+                                if parsed is None:
+                                    break
+                                yield parsed
+                            except GeneratorExit:
+                                # Handle GeneratorExit properly to close the stream gracefully
                                 break
-                            yield parsed
             except httpx.HTTPError as e:
                 raise AgentClientError(f"Error: {e}")
 

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/langgraph_agent_toolkit/core/memory/postgres.py

@@ -59,8 +59,6 @@ class PostgresMemoryBackend(BaseMemoryBackend):
             The object created by the factory_func
 
         """
-        conn_string = self.get_connection_string()
-
         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}"
@@ -68,7 +66,7 @@ class PostgresMemoryBackend(BaseMemoryBackend):
 
         # Use AsyncConnectionPool as an async context manager
         async with AsyncConnectionPool(
-            conn_string,
+            self.get_connection_string(),
             min_size=settings.POSTGRES_MIN_SIZE,
             max_size=settings.POSTGRES_POOL_SIZE,
             max_idle=settings.POSTGRES_MAX_IDLE,

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/langgraph_agent_toolkit/streamlit_app.py

@@ -75,6 +75,19 @@ async def main() -> None:
         if not thread_id:
             thread_id = str(uuid.uuid4())
             messages = []
+            # Add welcome message to messages when creating a new thread
+            welcome_message = None
+            match agent_client.agent:
+                case "chatbot":
+                    welcome_content = "Hello! I'm a simple chatbot. Ask me anything!"
+                case "interrupt-agent":
+                    welcome_content = (
+                        "Hello! I'm an interrupt agent. Tell me your birthday and I will predict your personality!"
+                    )
+                case _:
+                    welcome_content = "Hello! I'm an AI agent. Ask me anything!"
+            welcome_message = ChatMessage(type="ai", content=welcome_content)
+            messages.append(welcome_message)
         else:
             try:
                 messages: List[ChatMessage] = agent_client.get_history(
@@ -155,17 +168,6 @@ async def main() -> None:
     # Draw existing messages
     messages: list[ChatMessage] = st.session_state.messages
 
-    if len(messages) == 0:
-        match agent_client.agent:
-            case "chatbot":
-                WELCOME = "Hello! I'm a simple chatbot. Ask me anything!"
-            case "interrupt-agent":
-                WELCOME = "Hello! I'm an interrupt agent. Tell me your birthday and I will predict your personality!"
-            case _:
-                WELCOME = "Hello! I'm an AI agent. Ask me anything!"
-        with st.chat_message("assistant"):
-            st.write(WELCOME)
-
     # draw_messages() expects an async iterator over messages
     async def amessage_iter() -> AsyncGenerator[ChatMessage, None]:
         for m in messages:
@@ -198,7 +200,7 @@ async def main() -> None:
             st.error(f"Error generating response: {e}")
             st.stop()
 
-    # If messages have been generated, show feedback widget
+    # If messages have been generated, show feedback widget only for AI messages
     if len(messages) > 0 and st.session_state.last_message:
         with st.session_state.last_message:
             await handle_feedback()
@@ -248,6 +250,7 @@ async def draw_messages(
             streaming_content += msg
             streaming_placeholder.write(streaming_content)
             continue
+
         if not isinstance(msg, ChatMessage):
             st.error(f"Unexpected message type: {type(msg)}")
             st.write(msg)

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-agent-toolkit"
-version = "0.4.1"
+version = "0.4.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" }]
@@ -36,7 +36,7 @@ dependencies = [
     "jiter ~= 0.9.0",
     "langchain-core ~= 0.3.52",
     "langchain-community ~= 0.3.21",
-    "langgraph ~= 0.3.30",
+    "langgraph ~= 0.4.1",
     "langgraph-cli[inmem] ~= 0.2.3",
     "langgraph-checkpoint-sqlite ~= 2.0.6",
     "langgraph-checkpoint-postgres ~= 2.0.19",
@@ -51,9 +51,9 @@ dependencies = [
     "pyowm ~= 3.3.0",
     "python-dotenv ~= 1.1.0",
     "setuptools >= 75.6.0",
-    "streamlit ~= 1.44.1",
+    "streamlit ~= 1.45.0",
     "watchdog ~= 6.0.0",
-    "joblib ~= 1.4.2",
+    "joblib >= 1.4.2",
     "uvicorn ~= 0.34.0",
     "loguru ~= 0.7.3",
     "fire ~= 0.7.0",
@@ -80,7 +80,7 @@ client = [
     "httpx ~= 0.28.1",
     "pydantic ~= 2.11.1",
     "python-dotenv ~= 1.1.0",
-    "streamlit ~= 1.44.1",
+    "streamlit ~= 1.45.0",
     "watchdog ~= 6.0.0",
 ]
 

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/tests/app/test_streamlit_app.py

@@ -15,7 +15,14 @@ class MockUUID:
         return "test session id"
 
 
-def test_app_simple_non_streaming(mock_agent_client):
+# Add a fixture to patch st.feedback to avoid issues with the feedback component
+@pytest.fixture
+def patch_feedback():
+    with patch("streamlit.feedback", return_value=None) as mock_feedback:
+        yield mock_feedback
+
+
+def test_app_simple_non_streaming(mock_agent_client, patch_feedback):
     """Test the full app - happy path."""
     at = AppTest.from_file("../../langgraph_agent_toolkit/streamlit_app.py").run()
 
@@ -32,15 +39,16 @@ def test_app_simple_non_streaming(mock_agent_client):
 
     at.sidebar.toggle[0].set_value(False)  # Use Streaming = False
     at.chat_input[0].set_value(PROMPT).run()
-    print(at)
-    assert at.chat_message[0].avatar == "user"
-    assert at.chat_message[0].markdown[0].value == PROMPT
-    assert at.chat_message[1].avatar == "assistant"
-    assert at.chat_message[1].markdown[0].value == RESPONSE
+
+    # Fix indices - welcome message is at index 0, user message at index 1, response at index 2
+    assert at.chat_message[1].avatar == "user"
+    assert at.chat_message[1].markdown[0].value == PROMPT
+    assert at.chat_message[2].avatar == "assistant"
+    assert at.chat_message[2].markdown[0].value == RESPONSE
     assert not at.exception
 
 
-def test_app_settings(mock_agent_client):
+def test_app_settings(mock_agent_client, patch_feedback):
     """Test the full app - happy path."""
     at = AppTest.from_file("../../langgraph_agent_toolkit/streamlit_app.py").run()
 
@@ -64,11 +72,11 @@ def test_app_settings(mock_agent_client):
     at.sidebar.selectbox[0].set_value("chatbot")
     at.chat_input[0].set_value(PROMPT).run()
 
-    # Basic checks
-    assert at.chat_message[0].avatar == "user"
-    assert at.chat_message[0].markdown[0].value == PROMPT
-    assert at.chat_message[1].avatar == "assistant"
-    assert at.chat_message[1].markdown[0].value == RESPONSE
+    # Basic checks - adjusted for welcome message at index 0
+    assert at.chat_message[1].avatar == "user"
+    assert at.chat_message[1].markdown[0].value == PROMPT
+    assert at.chat_message[2].avatar == "assistant"
+    assert at.chat_message[2].markdown[0].value == RESPONSE
 
     # Check the args match the settings
     assert mock_agent_client.agent == "chatbot"
@@ -107,7 +115,7 @@ def test_app_thread_id_history(mock_handle_feedback, mock_draw_messages, mock_ag
 
 
 @pytest.mark.asyncio
-async def test_app_streaming(mock_agent_client):
+async def test_app_streaming(mock_agent_client, patch_feedback):
     """Test the app with streaming enabled - including tool messages."""
     at = AppTest.from_file("../../langgraph_agent_toolkit/streamlit_app.py").run()
 
@@ -131,11 +139,11 @@ async def test_app_streaming(mock_agent_client):
 
     at.toggle[0].set_value(True)  # Use Streaming = True
     at.chat_input[0].set_value(PROMPT).run()
-    print(at)
 
-    assert at.chat_message[0].avatar == "user"
-    assert at.chat_message[0].markdown[0].value == PROMPT
-    response = at.chat_message[1]
+    # Fix indices - welcome message is at index 0, user message at index 1, response at index 2
+    assert at.chat_message[1].avatar == "user"
+    assert at.chat_message[1].markdown[0].value == PROMPT
+    response = at.chat_message[2]
     tool_status = response.status[0]
     assert response.avatar == "assistant"
     assert tool_status.label == "Tool Call: calculator"
@@ -149,7 +157,7 @@ async def test_app_streaming(mock_agent_client):
 
 
 @pytest.mark.asyncio
-async def test_app_init_error(mock_agent_client):
+async def test_app_init_error(mock_agent_client, patch_feedback):
     """Test the app with an error in the agent initialization."""
     at = AppTest.from_file("../../langgraph_agent_toolkit/streamlit_app.py").run()
 
@@ -159,7 +167,6 @@ async def test_app_init_error(mock_agent_client):
 
     at.toggle[0].set_value(True)  # Use Streaming = True
     at.chat_input[0].set_value(PROMPT).run()
-    print(at)
 
     assert at.chat_message[0].avatar == "assistant"
     assert at.chat_message[1].avatar == "user"
@@ -168,7 +175,7 @@ async def test_app_init_error(mock_agent_client):
     assert not at.exception
 
 
-def test_app_new_chat_btn(mock_agent_client):
+def test_app_new_chat_btn(mock_agent_client, patch_feedback):
     at = AppTest.from_file("../../langgraph_agent_toolkit/streamlit_app.py").run()
     thread_id_a = at.session_state.thread_id
 

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.3}/tests/service/test_service.py

@@ -482,6 +482,7 @@ def test_history(test_client, mock_agent, mock_agent_executor) -> None:
         created_at=None,
         parent_config=None,
         tasks=(),
+        interrupts=(),
     )
 
     # Mock get_state to return our state snapshot with messages