langgraph-api 0.0.26__py3-none-any.whl → 0.0.28__py3-none-any.whl

langgraph_api/js/sse.py

@@ -0,0 +1,138 @@
+"""Adapted from httpx_sse to split lines on \n, \r, \r\n per the SSE spec."""
+
+from collections.abc import AsyncIterator
+
+import httpx
+import orjson
+from langgraph_sdk.schema import StreamPart
+
+BytesLike = bytes | bytearray | memoryview
+
+
+class BytesLineDecoder:
+    """
+    Handles incrementally reading lines from text.
+
+    Has the same behaviour as the stdllib bytes splitlines,
+    but handling the input iteratively.
+    """
+
+    def __init__(self) -> None:
+        self.buffer = bytearray()
+        self.trailing_cr: bool = False
+
+    def decode(self, text: bytes) -> list[BytesLike]:
+        # See https://docs.python.org/3/glossary.html#term-universal-newlines
+        NEWLINE_CHARS = b"\n\r"
+
+        # We always push a trailing `\r` into the next decode iteration.
+        if self.trailing_cr:
+            text = b"\r" + text
+            self.trailing_cr = False
+        if text.endswith(b"\r"):
+            self.trailing_cr = True
+            text = text[:-1]
+
+        if not text:
+            # NOTE: the edge case input of empty text doesn't occur in practice,
+            # because other httpx internals filter out this value
+            return []  # pragma: no cover
+
+        trailing_newline = text[-1] in NEWLINE_CHARS
+        lines = text.splitlines()
+
+        if len(lines) == 1 and not trailing_newline:
+            # No new lines, buffer the input and continue.
+            self.buffer.extend(lines[0])
+            return []
+
+        if self.buffer:
+            # Include any existing buffer in the first portion of the
+            # splitlines result.
+            self.buffer.extend(lines[0])
+            lines = [self.buffer] + lines[1:]
+            self.buffer = bytearray()
+
+        if not trailing_newline:
+            # If the last segment of splitlines is not newline terminated,
+            # then drop it from our output and start a new buffer.
+            self.buffer.extend(lines.pop())
+
+        return lines
+
+    def flush(self) -> list[BytesLike]:
+        if not self.buffer and not self.trailing_cr:
+            return []
+
+        lines = [self.buffer]
+        self.buffer = bytearray()
+        self.trailing_cr = False
+        return lines
+
+
+class SSEDecoder:
+    def __init__(self) -> None:
+        self._event = ""
+        self._data = bytearray()
+        self._last_event_id = ""
+        self._retry: int | None = None
+
+    def decode(self, line: bytes) -> StreamPart | None:
+        # See: https://html.spec.whatwg.org/multipage/server-sent-events.html#event-stream-interpretation  # noqa: E501
+
+        if not line:
+            if (
+                not self._event
+                and not self._data
+                and not self._last_event_id
+                and self._retry is None
+            ):
+                return None
+
+            sse = StreamPart(
+                event=self._event,
+                data=orjson.loads(self._data) if self._data else None,
+            )
+
+            # NOTE: as per the SSE spec, do not reset last_event_id.
+            self._event = ""
+            self._data = bytearray()
+            self._retry = None
+
+            return sse
+
+        if line.startswith(b":"):
+            return None
+
+        fieldname, _, value = line.partition(b":")
+
+        if value.startswith(b" "):
+            value = value[1:]
+
+        if fieldname == b"event":
+            self._event = value.decode()
+        elif fieldname == b"data":
+            self._data.extend(value)
+        elif fieldname == b"id":
+            if b"\0" in value:
+                pass
+            else:
+                self._last_event_id = value.decode()
+        elif fieldname == b"retry":
+            try:
+                self._retry = int(value)
+            except (TypeError, ValueError):
+                pass
+        else:
+            pass  # Field is ignored.
+
+        return None
+
+
+async def aiter_lines_raw(response: httpx.Response) -> AsyncIterator[BytesLike]:
+    decoder = BytesLineDecoder()
+    async for chunk in response.aiter_bytes():
+        for line in decoder.decode(chunk):
+            yield line
+    for line in decoder.flush():
+        yield line

langgraph_api/js/tests/api.test.mts

@@ -461,6 +461,7 @@ describe("runs", () => {
       {
         input: { messages: [{ type: "human", content: "bar" }] },
         config: globalConfig,
+        afterSeconds: 10,
       }
     );
 
@@ -1740,6 +1741,33 @@ describe("long running tasks", () => {
   );
 });
 
+it("unusual newline termination characters", async () => {
+  const thread = await client.threads.create({
+    metadata: { graph_id: "agent" },
+  });
+
+  await client.threads.updateState(thread.thread_id, {
+    values: {
+      messages: [
+        {
+          type: "human",
+          content:
+            "Page break characters: \n\r\x0b\x0c\x1c\x1d\x1e\x85\u2028\u2029",
+        },
+      ],
+    },
+  });
+
+  const history = await client.threads.getHistory<{
+    messages: { type: string; content: string }[];
+  }>(thread.thread_id);
+  expect(history.length).toBe(1);
+  expect(history[0].values.messages.length).toBe(1);
+  expect(history[0].values.messages[0].content).toBe(
+    "Page break characters: \n\r\x0b\x0c\x1c\x1d\x1e\x85\u2028\u2029"
+  );
+});
+
 // Not implemented in JS yet
 describe.skip("command update state", () => {
   it("updates state via commands", async () => {

langgraph_api/js/tests/compose-postgres.yml

@@ -30,7 +30,7 @@ services:
     build:
       context: graphs
       dockerfile_inline: |
-        FROM langchain/langgraphjs-api:20
+        FROM langchain/langgraphjs-api:${NODE_VERSION:-20}
         ADD . /deps/graphs
         WORKDIR /deps/graphs
         RUN yarn install --frozen-lockfile
@@ -52,6 +52,6 @@ services:
     environment:
       REDIS_URI: redis://langgraph-redis:6379
       DATABASE_URI: postgres://postgres:postgres@langgraph-postgres:5432/postgres?sslmode=disable
-      N_JOBS_PER_WORKER: "2"
+      N_JOBS_PER_WORKER: "5"
       LANGGRAPH_CLOUD_LICENSE_KEY: ${LANGGRAPH_CLOUD_LICENSE_KEY}
       FF_JS_ZEROMQ_ENABLED: ${FF_JS_ZEROMQ_ENABLED}

langgraph_api/js/tests/graphs/agent.css

@@ -0,0 +1 @@
+@import "tailwindcss";

langgraph_api/js/tests/graphs/agent.ui.tsx

@@ -0,0 +1,10 @@
+import "./agent.css";
+import React from "react";
+
+export function StockPrice() {
+  return <div>Stock Price</div>;
+}
+
+export default {
+  "stock-price": StockPrice,
+} as const;

langgraph_api/js/tests/graphs/package.json

@@ -1,7 +1,7 @@
 {
   "private": true,
   "dependencies": {
-    "@langchain/core": "^0.3.37",
-    "@langchain/langgraph": "^0.2.43"
+    "@langchain/core": "^0.3.40",
+    "@langchain/langgraph": "^0.2.49"
   }
 }

langgraph_api/js/tests/graphs/yarn.lock

@@ -7,10 +7,10 @@
   resolved "https://registry.yarnpkg.com/@cfworker/json-schema/-/json-schema-4.1.0.tgz#cc114da98c23b12f4cd4673ce8a076be24e0233c"
   integrity sha512-/vYKi/qMxwNsuIJ9WGWwM2rflY40ZenK3Kh4uR5vB9/Nz12Y7IUN/Xf4wDA7vzPfw0VNh3b/jz4+MjcVgARKJg==
 
-"@langchain/core@^0.3.37":
-  version "0.3.37"
-  resolved "https://registry.yarnpkg.com/@langchain/core/-/core-0.3.37.tgz#9ea7985c6cdaa075b02241ba3e6ba76c47454daf"
-  integrity sha512-LFk9GqHxcyCFx0oXvCBP7vDZIOUHYzzNU7JR+2ofIMnfkBLzcCKzBLySQDfPtd13PrpGHkaeOeLq8H1Tqi9lSw==
+"@langchain/core@^0.3.40":
+  version "0.3.42"
+  resolved "https://registry.yarnpkg.com/@langchain/core/-/core-0.3.42.tgz#f1fa38425626d8efe9fe2ee51d36c91506632363"
+  integrity sha512-pT/jC5lqWK3YGDq8dQwgKoa6anqAhMtG1x5JbnrOj9NdaLeBbCKBDQ+/Ykzk3nZ8o+0UMsaXNZo7IVL83VVjHg==
   dependencies:
     "@cfworker/json-schema" "^4.0.2"
     ansi-styles "^5.0.0"
@@ -25,10 +25,10 @@
     zod "^3.22.4"
     zod-to-json-schema "^3.22.3"
 
-"@langchain/langgraph-checkpoint@~0.0.14":
-  version "0.0.14"
-  resolved "https://registry.yarnpkg.com/@langchain/langgraph-checkpoint/-/langgraph-checkpoint-0.0.14.tgz#7f66454438a906283d7f4e6179784f312a5a9cbe"
-  integrity sha512-UoJc1IZqtnn+AiRhygo1yjNZiXTwdOY6NSb7yrXrN96CAW3LPu8cFe7VihKg5OBv9qaGz1GCvnrNdNAB48HuKQ==
+"@langchain/langgraph-checkpoint@~0.0.16":
+  version "0.0.16"
+  resolved "https://registry.yarnpkg.com/@langchain/langgraph-checkpoint/-/langgraph-checkpoint-0.0.16.tgz#e996f31d5da8ce67b2a9bf3dc64c4c0e05f01d72"
+  integrity sha512-B50l7w9o9353drHsdsD01vhQrCJw0eqvYeXid7oKeoj1Yye+qY90r97xuhiflaYCZHM5VEo2oaizs8oknerZsQ==
   dependencies:
     uuid "^10.0.0"
 
@@ -42,12 +42,12 @@
     p-retry "4"
     uuid "^9.0.0"
 
-"@langchain/langgraph@^0.2.43":
-  version "0.2.43"
-  resolved "https://registry.yarnpkg.com/@langchain/langgraph/-/langgraph-0.2.43.tgz#bf24f8b26bc606cd3be7230364314a227382eb1e"
-  integrity sha512-uhdbzm3psUIEqxQUQPXeafLC5dxTzALrVGRnnGZi9gt0qlDueRfopZoh7uWJy+Zol+yN/E2mM3M6ZztSsfUEuQ==
+"@langchain/langgraph@^0.2.49":
+  version "0.2.54"
+  resolved "https://registry.yarnpkg.com/@langchain/langgraph/-/langgraph-0.2.54.tgz#f57a9b471808c122ee5ae4506ed05cc75f1578bd"
+  integrity sha512-+P2rU0Qz6bBCNPXOSV8WeUpLRTvhu8fQuzMYR2MqWsbbfmZrfmLxqtVWPHkmr5khx/txxFy1vOBAy+KwZ94mrg==
   dependencies:
-    "@langchain/langgraph-checkpoint" "~0.0.14"
+    "@langchain/langgraph-checkpoint" "~0.0.16"
     "@langchain/langgraph-sdk" "~0.0.32"
     uuid "^10.0.0"
     zod "^3.23.8"