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

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.4.1
+Version: 0.4.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
@@ -106,13 +106,13 @@ Description-Content-Type: text/markdown
 
 ---
 
-# 🧰 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%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)                                                                                                                                                                                                                                                                               |
 
 <!-- [![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:

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

@@ -4,13 +4,13 @@
 
 ---
 
-# 🧰 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%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)                                                                                                                                                                                                                                                                               |
 
 <!-- [![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:

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.2}/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.2}/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.2}/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.2}/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.2}/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.2}/pyproject.toml

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

{langgraph_agent_toolkit-0.4.1 → langgraph_agent_toolkit-0.4.2}/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