langgraph-api 0.2.23__py3-none-any.whl → 0.2.25__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.23"
+__version__ = "0.2.25"

langgraph_api/serde.py

@@ -1,7 +1,9 @@
 import asyncio
+import re
 import uuid
 from base64 import b64encode
 from collections import deque
+from collections.abc import Mapping
 from datetime import timedelta, timezone
 from decimal import Decimal
 from ipaddress import (
@@ -101,11 +103,36 @@ def default(obj):
 
 _option = orjson.OPT_SERIALIZE_NUMPY | orjson.OPT_NON_STR_KEYS
 
+_SURROGATE_RE = re.compile(r"[\ud800-\udfff]")
+
+
+def _strip_surr(s: str) -> str:
+    return s if _SURROGATE_RE.search(s) is None else _SURROGATE_RE.sub("", s)
+
+
+def _sanitise(o: Any) -> Any:
+    if isinstance(o, str):
+        return _strip_surr(o)
+    if isinstance(o, Mapping):
+        return {_sanitise(k): _sanitise(v) for k, v in o.items()}
+    if isinstance(o, list | tuple | set):
+        ctor = list if isinstance(o, list) else type(o)
+        return ctor(_sanitise(x) for x in o)
+    return o
+
 
 def json_dumpb(obj) -> bytes:
-    return orjson.dumps(obj, default=default, option=_option).replace(
-        rb"\u0000", b""
-    )  # null unicode char not allowed in json
+    try:
+        return orjson.dumps(obj, default=default, option=_option).replace(
+            rb"\u0000", b""
+        )  # null unicode char not allowed in json
+    except TypeError as e:
+        if "surrogates not allowed" not in str(e):
+            raise
+        surrogate_sanitized = _sanitise(obj)
+        return orjson.dumps(
+            surrogate_sanitized, default=default, option=_option
+        ).replace(rb"\u0000", b"")
 
 
 def json_loads(content: bytes | Fragment | dict) -> Any:

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.23.dist-info → langgraph_api-0.2.25.dist-info}/METADATA

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

{langgraph_api-0.2.23.dist-info → langgraph_api-0.2.25.dist-info}/RECORD

@@ -1,5 +1,5 @@
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api/__init__.py,sha256=drfF0qMRjoB2RacyqP-NkNmDOIBrys6RWHvqSTpZLuc,23
+langgraph_api/__init__.py,sha256=kx2V3mL-5OE7Q8rQ4rG5_RJEWOoZ9gDri5puJRGkRm8,23
 langgraph_api/api/__init__.py,sha256=YVzpbn5IQotvuuLG9fhS9QMrxXfP4s4EpEMG0n4q3Nw,5625
 langgraph_api/api/assistants.py,sha256=4_tZCXlOmFg-YwgtsCZPc1wr-3fexqJ6A3_2a5giF90,15811
 langgraph_api/api/mcp.py,sha256=RvRYgANqRzNQzSmgjNkq4RlKTtoEJYil04ot9lsmEtE,14352
@@ -81,11 +81,11 @@ langgraph_api/patch.py,sha256=Dgs0PXHytekX4SUL6KsjjN0hHcOtGLvv1GRGbh6PswU,1408
 langgraph_api/queue_entrypoint.py,sha256=Hw6W66EEY_V7GHvnRzijsbvoNy4TM7IQVoZJ-DRbSfU,2173
 langgraph_api/route.py,sha256=uN311KjIugyNHG3rmVw_ms61QO1W1l16jJx03rf0R_s,4630
 langgraph_api/schema.py,sha256=2711t4PIBk5dky4gmMndrTRC9CVvAgH47C9FKDxhkBo,5444
-langgraph_api/serde.py,sha256=TVsx2QQtepf8Wsgsabcku1NV4Vbugu4Oujmdnq4qMS0,3964
+langgraph_api/serde.py,sha256=8fQXg7T7RVUqj_jgOoSOJrWVpQDW0qJKjAjSsEhPHo4,4803
 langgraph_api/server.py,sha256=4P7GpXbE9m-sAV7rBQ4Gd3oFk6htNbL-tRQfICAFc2k,6837
 langgraph_api/sse.py,sha256=3jG_FZj8FI9r7xGWTqaAyDkmqf6P1NOu0EzGrcSOGYc,4033
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
-langgraph_api/stream.py,sha256=4wKeoLuxqy7m7iamONmP9Lw4vcHEMWPeQFV_PrJC5AY,12042
+langgraph_api/stream.py,sha256=vQXHb2f8g-VvTRjPcgq4MEQ2mR4tYdQ8nxr4V3ogQkQ,12433
 langgraph_api/thread_ttl.py,sha256=-Ox8NFHqUH3wGNdEKMIfAXUubY5WGifIgCaJ7npqLgw,1762
 langgraph_api/tunneling/cloudflare.py,sha256=iKb6tj-VWPlDchHFjuQyep2Dpb-w2NGfJKt-WJG9LH0,3650
 langgraph_api/utils.py,sha256=92mSti9GfGdMRRWyESKQW5yV-75Z9icGHnIrBYvdypU,3619
@@ -96,9 +96,9 @@ langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 langgraph_license/validation.py,sha256=ZKraAVJArAABKqrmHN-EN18ncoNUmRm500Yt1Sc7tUA,537
 langgraph_runtime/__init__.py,sha256=O4GgSmu33c-Pr8Xzxj_brcK5vkm70iNTcyxEjICFZxA,1075
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
-openapi.json,sha256=h2zgGTQD9houdruwRdNwZuAFo-w3eP_f6huQ6jRFp84,133801
-langgraph_api-0.2.23.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.23.dist-info/METADATA,sha256=17XvJmMLdBhzcwrl06VdUWmqIFUgDbCb5_rUjfDrPwY,4275
-langgraph_api-0.2.23.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-langgraph_api-0.2.23.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.2.23.dist-info/RECORD,,
+openapi.json,sha256=ZMY9UXZJYiFK59z8QmDxVZ7LV6KonQbHzG-D5h-ZTYE,135412
+langgraph_api-0.2.25.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.25.dist-info/METADATA,sha256=F1iyqIgf6SPCfAhhV1VG3T1ku730vjotATOHCf820rE,4275
+langgraph_api-0.2.25.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+langgraph_api-0.2.25.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.2.25.dist-info/RECORD,,

openapi.json

@@ -779,56 +779,6 @@
         }
       }
     },
-    "/threads/state/bulk": {
-      "post": {
-        "tags": ["Threads"],
-        "summary": "Bulk Update Thread State",
-        "description": "Create a new thread from a batch of state updates.",
-        "operationId": "bulk_update_thread_state_post",
-        "requestBody": {
-          "content": {
-            "application/json": {
-              "schema": {
-                "$ref": "#/components/schemas/ThreadStateBulkUpdate"
-              }
-            }
-          },
-          "required": true
-        },
-        "responses": {
-          "200": {
-            "description": "Success",
-            "content": {
-              "application/json": {
-                "schema": {
-                  "$ref": "#/components/schemas/Thread"
-                }
-              }
-            }
-          },
-          "409": {
-            "description": "Conflict",
-            "content": {
-              "application/json": {
-                "schema": {
-                  "$ref": "#/components/schemas/ErrorResponse"
-                }
-              }
-            }
-          },
-          "422": {
-            "description": "Validation Error",
-            "content": {
-              "application/json": {
-                "schema": {
-                  "$ref": "#/components/schemas/ErrorResponse"
-                }
-              }
-            }
-          }
-        }
-      }
-    },
     "/threads/{thread_id}/state": {
       "get": {
         "tags": ["Threads"],
@@ -847,6 +797,17 @@
             },
             "name": "thread_id",
             "in": "path"
+          },
+          {
+            "description": "Whether to include subgraphs in the response.",
+            "required": false,
+            "schema": {
+              "type": "boolean",
+              "title": "Subgraphs",
+              "description": "Whether to include subgraphs in the response."
+            },
+            "name": "subgraphs",
+            "in": "query"
           }
         ],
         "responses": {
@@ -925,12 +886,103 @@
         }
       }
     },
+  "/threads/{thread_id}/state/{checkpoint_id}": {
+    "get": {
+      "tags": ["Threads"],
+      "summary": "Get Thread State At Checkpoint",
+      "description": "Get state for a thread at a specific checkpoint.",
+      "operationId": "get_thread_state_at_checkpoint_threads__thread_id__state__checkpoint_id__get",
+      "parameters": [
+        {
+          "description": "The ID of the thread.",
+          "required": true,
+          "schema": {
+            "type": "string",
+            "format": "uuid",
+            "title": "Thread Id",
+            "description": "The ID of the thread."
+          },
+          "name": "thread_id",
+          "in": "path"
+        },
+        {
+          "description": "The ID of the checkpoint.",
+          "required": true,
+          "schema": {
+            "type": "string",
+            "format": "uuid",
+            "title": "Checkpoint Id",
+            "description": "The ID of the checkpoint."
+          },
+          "name": "checkpoint_id",
+          "in": "path"
+        },
+        {
+          "description": "Whether to include subgraphs in the response.",
+          "required": false,
+          "schema": {
+            "type": "boolean",
+            "title": "Subgraphs",
+            "description": "Whether to include subgraphs in the response."
+          },
+          "name": "subgraphs",
+          "in": "query"
+        }
+      ],
+      "responses": {
+        "200": {
+          "description": "Success",
+          "content": {
+            "application/json": {
+              "schema": {
+                "$ref": "#/components/schemas/ThreadState"
+              }
+            }
+          }
+        },
+        "422": {
+          "description": "Validation Error",
+          "content": {
+            "application/json": {
+              "schema": {
+                "$ref": "#/components/schemas/ErrorResponse"
+              }
+            }
+          }
+        }
+      }
+    }
+  },
     "/threads/{thread_id}/state/checkpoint": {
       "post": {
         "tags": ["Threads"],
         "summary": "Get Thread State At Checkpoint",
         "description": "Get state for a thread at a specific checkpoint.",
         "operationId": "post_thread_state_at_checkpoint_threads__thread_id__state__checkpoint_id__get",
+        "parameters": [
+          {
+            "description": "The ID of the thread.",
+            "required": true,
+            "schema": {
+              "type": "string",
+              "format": "uuid",
+              "title": "Thread Id",
+              "description": "The ID of the thread."
+            },
+            "name": "thread_id",
+            "in": "path"
+          },
+          {
+            "required": false,
+            "schema": {
+              "type": "boolean",
+              "title": "Subgraphs",
+              "description": "If true, includes subgraph states."
+            },
+            "name": "subgraphs",
+            "in": "query"
+          }
+        ],
         "requestBody": {
           "content": {
             "application/json": {