langgraph-api 0.2.21__tar.gz → 0.2.24__tar.gz

{langgraph_api-0.2.21 → langgraph_api-0.2.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langgraph-api
-Version: 0.2.21
+Version: 0.2.24
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos

langgraph_api-0.2.24/langgraph_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.24"

{langgraph_api-0.2.21 → langgraph_api-0.2.24}/langgraph_api/api/assistants.py

@@ -3,13 +3,6 @@ from uuid import uuid4
 
 import structlog
 from langchain_core.runnables.utils import create_model
-from langgraph.constants import (
-    CONFIG_KEY_CHECKPOINT_ID,
-    CONFIG_KEY_CHECKPOINT_NS,
-    CONFIG_KEY_CHECKPOINTER,
-    CONFIG_KEY_STORE,
-    CONFIG_KEY_THREAD_ID,
-)
 from langgraph.pregel import Pregel
 from pydantic import TypeAdapter
 from starlette.exceptions import HTTPException
@@ -38,11 +31,11 @@ logger = structlog.stdlib.get_logger(__name__)
 
 
 EXCLUDED_CONFIG_SCHEMA = (
-    CONFIG_KEY_CHECKPOINTER,
-    CONFIG_KEY_STORE,
-    CONFIG_KEY_CHECKPOINT_ID,
-    CONFIG_KEY_CHECKPOINT_NS,
-    CONFIG_KEY_THREAD_ID,
+    "__pregel_checkpointer",
+    "__pregel_store",
+    "checkpoint_id",
+    "checkpoint_ns",
+    "thread_id",
 )
 
 

{langgraph_api-0.2.21 → langgraph_api-0.2.24}/langgraph_api/stream.py

@@ -9,6 +9,7 @@ import structlog
 from langchain_core.messages import (
     BaseMessage,
     BaseMessageChunk,
+    convert_to_messages,
     message_chunk_to_message,
 )
 from langchain_core.runnables.config import run_in_executor
@@ -178,7 +179,11 @@ async def astream_state(
                             else:
                                 yield "messages", chunk
                         else:
-                            msg, meta = cast(tuple[BaseMessage, dict[str, Any]], chunk)
+                            msg, meta = cast(
+                                tuple[BaseMessage | dict, dict[str, Any]], chunk
+                            )
+                            if isinstance(msg, dict):
+                                msg = convert_to_messages([msg])[0]
                             if msg.id in messages:
                                 messages[msg.id] += msg
                             else:
@@ -233,7 +238,11 @@ async def astream_state(
                         else:
                             yield "messages", chunk
                     else:
-                        msg, meta = cast(tuple[BaseMessage, dict[str, Any]], chunk)
+                        msg, meta = cast(
+                            tuple[BaseMessage | dict, dict[str, Any]], chunk
+                        )
+                        if isinstance(msg, dict):
+                            msg = convert_to_messages([msg])[0]
                         if msg.id in messages:
                             messages[msg.id] += msg
                         else:

{langgraph_api-0.2.21 → langgraph_api-0.2.24}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.2.21"
+version = "0.2.24"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",

langgraph_api-0.2.21/langgraph_api/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.2.21"