langgraph-api 0.0.28rc1__py3-none-any.whl → 0.0.31__py3-none-any.whl

langgraph_api/api/__init__.py

@@ -64,6 +64,8 @@ if HTTP_CONFIG:
         protected_routes.extend(threads_routes)
     if not HTTP_CONFIG.get("disable_store"):
         protected_routes.extend(store_routes)
+    if not HTTP_CONFIG.get("disable_ui"):
+        protected_routes.extend(ui_routes)
 else:
     protected_routes.extend(assistants_routes)
     protected_routes.extend(runs_routes)

langgraph_api/api/assistants.py

@@ -184,7 +184,8 @@ async def get_assistant_graph(
             return ApiResponse(drawable_graph.to_json())
 
         try:
-            return ApiResponse(graph.get_graph(xray=xray).to_json())
+            drawable_graph = await graph.aget_graph(xray=xray)
+            return ApiResponse(drawable_graph.to_json())
         except NotImplementedError:
             raise HTTPException(
                 422, detail="The graph does not support visualization"

langgraph_api/api/runs.py

@@ -354,11 +354,13 @@ async def join_run_stream_endpoint(request: ApiRequest):
     cancel_on_disconnect = cancel_on_disconnect_str.lower() in {"true", "yes", "1"}
     validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
     validate_uuid(run_id, "Invalid run ID: must be a UUID")
+    stream_mode = request.query_params.get("stream_mode") or None
     return EventSourceResponse(
         Runs.Stream.join(
             run_id,
             thread_id=thread_id,
             cancel_on_disconnect=cancel_on_disconnect,
+            stream_mode=stream_mode,
         )
     )
 

langgraph_api/auth/custom.py

@@ -18,6 +18,7 @@ from starlette.authentication import (
     BaseUser,
 )
 from starlette.concurrency import run_in_threadpool
+from starlette.datastructures import QueryParams
 from starlette.exceptions import HTTPException
 from starlette.requests import HTTPConnection, Request
 from starlette.responses import Response
@@ -37,7 +38,7 @@ SUPPORTED_PARAMETERS = {
     "scopes": list[str],
     "path_params": dict[str, str] | None,
     "query_params": dict[str, str] | None,
-    "headers": dict[str, bytes] | None,
+    "headers": dict[bytes, bytes] | None,
     "authorization": str | None,
     "scope": dict[str, Any],
 }
@@ -271,7 +272,7 @@ def _extract_arguments_from_scope(
     if "path" in param_names:
         args["path"] = scope["path"]
     if "query_params" in param_names:
-        args["query_params"] = scope.get("query_params", {})
+        args["query_params"] = QueryParams(scope.get("query_string"))
     if "headers" in param_names:
         args["headers"] = dict(scope.get("headers", {}))
     if "authorization" in param_names:
@@ -478,6 +479,9 @@ class ProxyUser(BaseUser):
         """Proxy any other attributes to the underlying user object."""
         return getattr(self._user, name)
 
+    def __str__(self) -> str:
+        return f"{self._user}"
+
 
 class SimpleUser(ProxyUser):
     def __init__(self, username: str):

langgraph_api/auth/middleware.py

@@ -42,6 +42,11 @@ class ConditionalAuthenticationMiddleware(AuthenticationMiddleware):
             await self.app(scope, receive, send)
             return
 
+        if scope["path"].startswith("/ui") and scope["method"] == "GET":
+            # disable auth for UI asset requests
+            await self.app(scope, receive, send)
+            return
+
         return await super().__call__(scope, receive, send)
 
 

langgraph_api/cli.py

@@ -10,7 +10,7 @@ from collections.abc import Mapping, Sequence
 from typing_extensions import TypedDict
 
 if typing.TYPE_CHECKING:
-    from langgraph_api.config import HttpConfig
+    from langgraph_api.config import HttpConfig, StoreConfig
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -75,40 +75,6 @@ def patch_environment(**kwargs):
                 os.environ[key] = value
 
 
-class IndexConfig(TypedDict, total=False):
-    """Configuration for indexing documents for semantic search in the store."""
-
-    dims: int
-    """Number of dimensions in the embedding vectors.
-    
-    Common embedding models have the following dimensions:
-        - OpenAI text-embedding-3-large: 256, 1024, or 3072
-        - OpenAI text-embedding-3-small: 512 or 1536
-        - OpenAI text-embedding-ada-002: 1536
-        - Cohere embed-english-v3.0: 1024
-        - Cohere embed-english-light-v3.0: 384
-        - Cohere embed-multilingual-v3.0: 1024
-        - Cohere embed-multilingual-light-v3.0: 384
-    """
-
-    embed: str
-    """Either a path to an embedding model (./path/to/file.py:embedding_model)
-    or a name of an embedding model (openai:text-embedding-3-small)
-    
-    Note: LangChain is required to use the model format specification.
-    """
-
-    fields: list[str] | None
-    """Fields to extract text from for embedding generation.
-    
-    Defaults to the root ["$"], which embeds the json object as a whole.
-    """
-
-
-class StoreConfig(TypedDict, total=False):
-    index: IndexConfig
-
-
 class SecurityConfig(TypedDict, total=False):
     securitySchemes: dict
     security: list
@@ -160,7 +126,7 @@ def run_server(
     env: str | pathlib.Path | Mapping[str, str] | None = None,
     reload_includes: Sequence[str] | None = None,
     reload_excludes: Sequence[str] | None = None,
-    store: StoreConfig | None = None,
+    store: typing.Optional["StoreConfig"] = None,
     auth: AuthConfig | None = None,
     http: typing.Optional["HttpConfig"] = None,
     **kwargs: typing.Any,

langgraph_api/config.py

@@ -1,5 +1,5 @@
 from os import environ, getenv
-from typing import TypedDict
+from typing import Literal, TypedDict
 
 import orjson
 from starlette.config import Config, undefined
@@ -34,6 +34,66 @@ class HttpConfig(TypedDict, total=False):
     cors: CorsConfig | None
 
 
+class IndexConfig(TypedDict, total=False):
+    """Configuration for indexing documents for semantic search in the store."""
+
+    dims: int
+    """Number of dimensions in the embedding vectors.
+    
+    Common embedding models have the following dimensions:
+        - OpenAI text-embedding-3-large: 256, 1024, or 3072
+        - OpenAI text-embedding-3-small: 512 or 1536
+        - OpenAI text-embedding-ada-002: 1536
+        - Cohere embed-english-v3.0: 1024
+        - Cohere embed-english-light-v3.0: 384
+        - Cohere embed-multilingual-v3.0: 1024
+        - Cohere embed-multilingual-light-v3.0: 384
+    """
+
+    embed: str
+    """Either a path to an embedding model (./path/to/file.py:embedding_model)
+    or a name of an embedding model (openai:text-embedding-3-small)
+    
+    Note: LangChain is required to use the model format specification.
+    """
+
+    fields: list[str] | None
+    """Fields to extract text from for embedding generation.
+    
+    Defaults to the root ["$"], which embeds the json object as a whole.
+    """
+
+
+class TTLConfig(TypedDict, total=False):
+    """Configuration for TTL (time-to-live) behavior in the store."""
+
+    refresh_on_read: bool
+    """Default behavior for refreshing TTLs on read operations (GET and SEARCH).
+    
+    If True, TTLs will be refreshed on read operations (get/search) by default.
+    This can be overridden per-operation by explicitly setting refresh_ttl.
+    Defaults to True if not configured.
+    """
+    default_ttl: float | None
+    """Default TTL (time-to-live) in minutes for new items.
+    
+    If provided, new items will expire after this many minutes after their last access.
+    The expiration timer refreshes on both read and write operations.
+    Defaults to None (no expiration).
+    """
+    sweep_interval_minutes: int | None
+    """Interval in minutes between TTL sweep operations.
+    
+    If provided, the store will periodically delete expired items based on TTL.
+    Defaults to None (no sweeping).
+    """
+
+
+class StoreConfig(TypedDict, total=False):
+    index: IndexConfig
+    ttl: TTLConfig
+
+
 # env
 
 env = Config()
@@ -55,6 +115,18 @@ STATS_INTERVAL_SECS = env("STATS_INTERVAL_SECS", cast=int, default=60)
 DATABASE_URI = env("DATABASE_URI", cast=str, default=getenv("POSTGRES_URI", undefined))
 MIGRATIONS_PATH = env("MIGRATIONS_PATH", cast=str, default="/storage/migrations")
 
+
+def _get_encryption_key(key_str: str | None):
+    if not key_str:
+        return None
+    key = key_str.encode(encoding="utf-8")
+    if len(key) not in (16, 24, 32):
+        raise ValueError("LANGGRAPH_AES_KEY must be 16, 24, or 32 bytes long.")
+    return key
+
+
+LANGGRAPH_AES_KEY = env("LANGGRAPH_AES_KEY", default=None, cast=_get_encryption_key)
+
 # redis
 REDIS_URI = env("REDIS_URI", cast=str)
 REDIS_CLUSTER = env("REDIS_CLUSTER", cast=bool, default=False)
@@ -69,7 +141,9 @@ See https://developer.chrome.com/blog/private-network-access-update-2024-03
 """
 
 HTTP_CONFIG: HttpConfig | None = env("LANGGRAPH_HTTP", cast=_parse_json, default=None)
-STORE_CONFIG: dict | None = env("LANGGRAPH_STORE", cast=_parse_json, default=None)
+STORE_CONFIG: StoreConfig | None = env(
+    "LANGGRAPH_STORE", cast=_parse_json, default=None
+)
 CORS_ALLOW_ORIGINS = env("CORS_ALLOW_ORIGINS", cast=CommaSeparatedStrings, default="*")
 if HTTP_CONFIG and HTTP_CONFIG.get("cors"):
     CORS_CONFIG = HTTP_CONFIG["cors"]
@@ -132,6 +206,13 @@ FF_CRONS_ENABLED = env("FF_CRONS_ENABLED", cast=bool, default=True)
 # auth
 
 LANGGRAPH_AUTH_TYPE = env("LANGGRAPH_AUTH_TYPE", cast=str, default="noop")
+LANGGRAPH_POSTGRES_EXTENSIONS: Literal["standard", "lite"] = env(
+    "LANGGRAPH_POSTGRES_EXTENSIONS", cast=str, default="standard"
+)
+if LANGGRAPH_POSTGRES_EXTENSIONS not in ("standard", "lite"):
+    raise ValueError(
+        f"Unknown LANGGRAPH_POSTGRES_EXTENSIONS value: {LANGGRAPH_POSTGRES_EXTENSIONS}"
+    )
 LANGGRAPH_AUTH = env("LANGGRAPH_AUTH", cast=_parse_json, default=None)
 LANGSMITH_TENANT_ID = env("LANGSMITH_TENANT_ID", cast=str, default=None)
 LANGSMITH_AUTH_VERIFY_TENANT_ID = env(

langgraph_api/js/client.mts

@@ -493,6 +493,7 @@ const StreamModeSchema = z.union([
   z.literal("updates"),
   z.literal("debug"),
   z.literal("values"),
+  z.literal("custom"),
 ]);
 
 const ExtraStreamModeSchema = z.union([
@@ -529,8 +530,9 @@ async function* streamEventsRequest(
         ? payload.stream_mode
         : [payload.stream_mode];
 
-  const graphStreamMode: Set<"updates" | "debug" | "values" | "messages"> =
-    new Set();
+  const graphStreamMode: Set<
+    "updates" | "debug" | "values" | "messages" | "custom"
+  > = new Set();
   if (payload.stream_mode) {
     for (const mode of userStreamMode) {
       if (mode === "messages") {

langgraph_api/js/src/parser/parser.mts

@@ -348,6 +348,11 @@ export class SubgraphExtractor {
 
     const fsMap = new Map<string, string>();
     const system = vfs.createFSBackedSystem(fsMap, dirname, ts);
+
+    // TODO: investigate if we should create a PR in @typescript/vfs
+    const oldReadFile = system.readFile.bind(system);
+    system.readFile = (fileName) => oldReadFile(fileName) ?? "// Non-existent file";
+
     const host = vfs.createVirtualCompilerHost(system, compilerOptions, ts);
 
     const targetPath =

langgraph_api/js/src/schema/types.mts

@@ -2014,123 +2014,3 @@ export function buildGenerator(
     settings
   );
 }
-
-export async function extractGraphSchema(
-  id: string,
-  userPath: string,
-  exportName: string
-) {
-  const filePath = path.resolve(process.cwd(), userPath);
-  const parentPath = path.dirname(filePath);
-
-  const typePath = path.resolve(parentPath, `__$0${id}.mts`);
-  const importPath = path.relative(parentPath, filePath);
-
-  try {
-    await fs.writeFile(
-      typePath,
-      dedent`
-        import { ${exportName} as __graph } from "./${importPath}";
-        import type { BaseMessage } from "@langchain/core/messages";
-        import type {
-          StateType,
-          UpdateType,
-          StateDefinition,
-        } from "@langchain/langgraph";
-
-        type Wrap<T> = (a: T) => void;
-        type MatchBaseMessage<T> = T extends BaseMessage ? BaseMessage : never;
-        type MatchBaseMessageArray<T> =
-          T extends Array<infer C>
-            ? Wrap<MatchBaseMessage<C>> extends Wrap<BaseMessage>
-              ? BaseMessage[]
-              : never
-            : never;
-
-        type Defactorify<T> = T extends (...args: any[]) => infer R
-          ? Awaited<R>
-          : Awaited<T>;
-
-        type Inspect<T> = T extends unknown
-          ? {
-              [K in keyof T]: 0 extends 1 & T[K]
-                ? T[K]
-                : Wrap<MatchBaseMessageArray<T[K]>> extends Wrap<BaseMessage[]>
-                  ? BaseMessage[]
-                  : Wrap<MatchBaseMessage<T[K]>> extends Wrap<BaseMessage>
-                    ? BaseMessage
-                    : Inspect<T[K]>;
-            }
-          : never;
-
-        type ReflectCompiled<T> = T extends { RunInput: infer S; RunOutput: infer U }
-          ? { state: S; update: U }
-          : never;
-
-        type Reflect<T> =
-          Defactorify<T> extends infer DT
-            ? DT extends {
-                compile(...args: any[]): infer Compiled;
-              }
-              ? ReflectCompiled<Compiled>
-              : ReflectCompiled<DT>
-            : never;
-
-        type __reflect = Reflect<typeof __graph>;
-        export type __state = Inspect<__reflect["state"]>;
-        export type __update = Inspect<__reflect["update"]>;
-
-        type BuilderReflectCompiled<T> = T extends {
-          builder: {
-            _inputDefinition: infer I extends StateDefinition;
-            _outputDefinition: infer O extends StateDefinition;
-            _configSchema?: infer C extends StateDefinition | undefined;
-          };
-        }
-          ? { input: UpdateType<I>; output: StateType<O>; config: UpdateType<C> }
-          : never;
-
-        type BuilderReflect<T> =
-          Defactorify<T> extends infer DT
-            ? DT extends {
-                compile(...args: any[]): infer Compiled;
-              }
-              ? BuilderReflectCompiled<Compiled>
-              : BuilderReflectCompiled<DT>
-            : never;
-
-        type __builder = BuilderReflect<typeof __graph>;
-        type FilterAny<T> = 0 extends 1 & T ? never : T;
-        export type __input = Inspect<FilterAny<__builder["input"]>>;
-        export type __output = Inspect<FilterAny<__builder["output"]>>;
-        export type __config = Inspect<FilterAny<__builder["config"]>>;
-      `
-    );
-    const program = ts.createProgram([typePath], {
-      noEmit: true,
-      strict: true,
-      allowUnusedLabels: true,
-    });
-
-    const schema = buildGenerator(program);
-
-    const trySymbol = (schema: JsonSchemaGenerator | null, symbol: string) => {
-      try {
-        return schema?.getSchemaForSymbol(symbol) ?? undefined;
-      } catch (e) {
-        console.error(`Failed to obtain symbol "${symbol}":`, e?.message);
-      }
-      return undefined;
-    };
-
-    return {
-      state: trySymbol(schema, "__state"),
-      update: trySymbol(schema, "__update"),
-      input: trySymbol(schema, "__input"),
-      output: trySymbol(schema, "__output"),
-      config: trySymbol(schema, "__config"),
-    };
-  } finally {
-    await fs.unlink(typePath);
-  }
-}

langgraph_api/js/src/schema/types.template.mts

@@ -23,17 +23,19 @@ type Defactorify<T> = T extends (...args: any[]) => infer R
   ? Awaited<R>
   : Awaited<T>;
 
-type Inspect<T> = T extends unknown
-  ? {
-      [K in keyof T]: 0 extends 1 & T[K]
-        ? T[K]
-        : Wrap<MatchBaseMessageArray<T[K]>> extends Wrap<BaseMessage[]>
-          ? BaseMessage[]
-          : Wrap<MatchBaseMessage<T[K]>> extends Wrap<BaseMessage>
-            ? BaseMessage
-            : Inspect<T[K]>;
-    }
-  : never;
+type Inspect<T, CNT extends Array<0> = []> = CNT extends [0, 0, 0]
+  ? any
+  : T extends unknown
+    ? {
+        [K in keyof T]: 0 extends 1 & T[K]
+          ? T[K]
+          : Wrap<MatchBaseMessageArray<T[K]>> extends Wrap<BaseMessage[]>
+            ? BaseMessage[]
+            : Wrap<MatchBaseMessage<T[K]>> extends Wrap<BaseMessage>
+              ? BaseMessage
+              : Inspect<T[K], [0, ...CNT]>;
+      }
+    : never;
 
 type ReflectCompiled<T> = T extends { RunInput: infer S; RunOutput: infer U }
   ? { state: S; update: U }

langgraph_api/js/tests/api.test.mts

@@ -765,7 +765,7 @@ describe("runs", () => {
 
   it.concurrent(
     "human in the loop - no modification",
-    { retry: 0 },
+    { retry: 3 },
     async () => {
       const assistant = await client.assistants.create({ graphId: "agent" });
       const thread = await client.threads.create();

langgraph_api/lifespan.py

@@ -12,6 +12,7 @@ from langgraph_api.metadata import metadata_loop
 from langgraph_license.validation import get_license_status, plus_features_enabled
 from langgraph_storage.database import start_pool, stop_pool
 from langgraph_storage.queue import queue
+from langgraph_storage.store import Store
 
 
 @asynccontextmanager
@@ -44,6 +45,8 @@ async def lifespan(
                 and plus_features_enabled()
             ):
                 tg.create_task(cron_scheduler())
+            if config.STORE_CONFIG and config.STORE_CONFIG.get("ttl"):
+                tg.create_task(Store().start_ttl_sweeper())
             yield
     finally:
         await stop_remote_graphs()

langgraph_api/logging.py

@@ -6,9 +6,6 @@ import structlog
 from starlette.config import Config
 from structlog.typing import EventDict
 
-from langgraph_api.metadata import append_log
-from langgraph_api.serde import json_dumpb
-
 # env
 
 log_env = Config()
@@ -52,6 +49,8 @@ class JSONRenderer:
         """
         The return type of this depends on the return type of self._dumps.
         """
+        from langgraph_api.serde import json_dumpb
+
         return json_dumpb(event_dict).decode()
 
 
@@ -69,6 +68,8 @@ class TapForMetadata:
             event_dict["logger"].startswith("langgraph")
             and LEVELS[event_dict["level"].upper()] > LEVELS["INFO"]
         ):
+            from langgraph_api.metadata import append_log
+
             append_log(event_dict.copy())
         return event_dict
 

{langgraph_api-0.0.28rc1.dist-info → langgraph_api-0.0.31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.28rc1
+Version: 0.0.31
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -12,11 +12,11 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: cryptography (>=43.0.3,<44.0.0)
 Requires-Dist: httpx (>=0.25.0)
-Requires-Dist: jsonschema-rs (>=0.20.0,<0.21.0)
+Requires-Dist: jsonschema-rs (>=0.20.0,<0.30)
 Requires-Dist: langchain-core (>=0.2.38,<0.4.0)
 Requires-Dist: langgraph (>=0.2.56,<0.4.0)
-Requires-Dist: langgraph-checkpoint (>=2.0.15,<3.0)
-Requires-Dist: langgraph-sdk (>=0.1.53,<0.2.0)
+Requires-Dist: langgraph-checkpoint (>=2.0.21,<3.0)
+Requires-Dist: langgraph-sdk (>=0.1.57,<0.2.0)
 Requires-Dist: langsmith (>=0.1.63,<0.4.0)
 Requires-Dist: orjson (>=3.9.7)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)

{langgraph_api-0.0.28rc1.dist-info → langgraph_api-0.0.31.dist-info}/RECORD

@@ -1,24 +1,24 @@
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 langgraph_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/api/__init__.py,sha256=vwV4VnV7wA7tHqgcsSzeaHHQz9aTNZmTKLO6yT5l9UE,5044
-langgraph_api/api/assistants.py,sha256=6K5HTUjl7lQv0sVRe4RURPrLzTW-BblUjbSYHeN2PfQ,13992
+langgraph_api/api/__init__.py,sha256=B2kHOBhKjBRdNiWPljYuJFxP__wYPCvgtgTaWYKZGTQ,5129
+langgraph_api/api/assistants.py,sha256=2HEIcxaOJJ4_JQ00TKT1ehx7DMbYCGpvhShK0ZiWsSE,14043
 langgraph_api/api/meta.py,sha256=ifJ_Ki0Qf2DYbmY6OKlqKhLGxbt55gm0lEqH1A0cJbw,2790
 langgraph_api/api/openapi.py,sha256=f9gfmWN2AMKNUpLCpSgZuw_aeOF9jCXPdOtFT5PaTWM,10960
-langgraph_api/api/runs.py,sha256=HWZ7QPRJwPlvtFtC8cyG5mKd3GJIjRkdR0KWlMssQ84,16574
+langgraph_api/api/runs.py,sha256=_RWKtmjD89ALnTk56dwo2rJwEi2oghk2Tqp0l1aCcZg,16677
 langgraph_api/api/store.py,sha256=VzAJVOwO0IxosBB7km5TTf2rhlWGyPkVz_LpvbxetVY,5437
 langgraph_api/api/threads.py,sha256=taU61XPcCEhBPCYPZcMDsgVDwwWUWJs8p-PrXFXWY48,8661
 langgraph_api/api/ui.py,sha256=LiOZVewKOPbKEykCm30hCEaOA7vuS_Ti5hB32EEy4vw,2082
 langgraph_api/asyncio.py,sha256=ipxOGL0CuKZeHw8895ojtfoBU2fj0iJOp48uhiLAmss,7786
 langgraph_api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/auth/custom.py,sha256=oyXz7UONY1YeTJ5IBGRAbFUz_K6krkptNxW8zzjb0uk,21088
+langgraph_api/auth/custom.py,sha256=z69UvpG7wPSXtRmp_UFmA8fY7HeQD4AEIWu3qlpJOBc,21210
 langgraph_api/auth/langsmith/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/auth/langsmith/backend.py,sha256=InScaL-HYCnxYEauhxU198gRZV9pJn9SzzBoR9Edn7g,2654
 langgraph_api/auth/langsmith/client.py,sha256=eKchvAom7hdkUXauD8vHNceBDDUijrFgdTV8bKd7x4Q,3998
-langgraph_api/auth/middleware.py,sha256=5VEC3KE0DrEgXHbEta95zpL1S6PYyD9ETXn3lWKVhqI,1684
+langgraph_api/auth/middleware.py,sha256=jU8aDSIZHdzCGdifejRF7ndHkSjBtqIHcBwFIuUdHEA,1875
 langgraph_api/auth/noop.py,sha256=Bk6Nf3p8D_iMVy_OyfPlyiJp_aEwzL-sHrbxoXpCbac,586
 langgraph_api/auth/studio_user.py,sha256=FzFQRROKDlA9JjtBuwyZvk6Mbwno5M9RVYjDO6FU3F8,186
-langgraph_api/cli.py,sha256=YIiuS-2D0fyqLd8BzJBIUAM3mmAJPtLUdPEMS6Awj9o,12752
-langgraph_api/config.py,sha256=iIekoGAQL-v--hVrCzSd2StrelzuHqb1vbpJAoHgxkU,5809
+langgraph_api/cli.py,sha256=3SAwPv8EDKqJx52oD3CB3h0vN7m-LI3lT0RmYFMfkNM,11671
+langgraph_api/config.py,sha256=XMJEVfPe1LrmVKGM5sisQs1eZOUDyU3JwscwJOm0q_k,8572
 langgraph_api/cron_scheduler.py,sha256=9yzbbGxzNgJdIg4ZT7yu2oTwT_wRuPxD1c2sbbd52xs,2630
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
 langgraph_api/graph.py,sha256=viFyQa8-BRRNRZqNNUzLB31cB4IypEMkMucQEJQYLWY,16556
@@ -26,7 +26,7 @@ langgraph_api/http.py,sha256=gYbxxjY8aLnsXeJymcJ7G7Nj_yToOGpPYQqmZ1_ggfA,5240
 langgraph_api/js/.gitignore,sha256=l5yI6G_V6F1600I1IjiUKn87f4uYIrBAYU1MOyBBhg4,59
 langgraph_api/js/base.py,sha256=BpE8-xkUp8HFPRjSKx1tfUQubvoV4jYl6OwZdre3veI,209
 langgraph_api/js/build.mts,sha256=ArAEn74HHY_KXt0JCne76FTyGP34VQCHMe6OfPyxLM8,2915
-langgraph_api/js/client.mts,sha256=heQ5LMpK-hWfSrByNeINZn0JTJ3EYaJKCepzfpiXccU,24356
+langgraph_api/js/client.mts,sha256=2dptAX8fMowV9OC4DU4khjpZUgALBLVBTu3jTQbeUJY,24394
 langgraph_api/js/errors.py,sha256=Cm1TKWlUCwZReDC5AQ6SgNIVGD27Qov2xcgHyf8-GXo,361
 langgraph_api/js/global.d.ts,sha256=cLJRZfYVGmgQ6o_xFevVNNTIi918ZUdxVRnpLVSjiAY,133
 langgraph_api/js/package.json,sha256=j6DMoVgwRqWqTwdd7R1f-kvmiTUAbO3HaUhM8K64lbE,1224
@@ -34,15 +34,15 @@ langgraph_api/js/remote.py,sha256=jUUgdskUbDR_tuhZFrJo4OewbndQ76joczBv-mZHG9k,22
 langgraph_api/js/schema.py,sha256=7idnv7URlYUdSNMBXQcw7E4SxaPxCq_Oxwnlml8q5ik,408
 langgraph_api/js/src/graph.mts,sha256=mRyMUp03Fwd5DlmNIFl3RiUCQuJ5XwmFp1AfAeKDfVc,3169
 langgraph_api/js/src/hooks.mjs,sha256=XtktgmIHlls_DsknAuwib-z7TqCm0haRoTXvnkgzMuo,601
-langgraph_api/js/src/parser/parser.mts,sha256=wXre7teh8N8RYmGcyhZp4vMJd0kNnnFgoSGEyMVPzpQ,13007
+langgraph_api/js/src/parser/parser.mts,sha256=c7WLEtMMwtVzir9O8VpJFVHKuLjtI86rde-In2lvaM0,13217
 langgraph_api/js/src/parser/parser.worker.mjs,sha256=2K6D0GlUmkk7LE39I8mryB8VZVE3-N9Cblji-ArPhFo,386
-langgraph_api/js/src/schema/types.mts,sha256=SUj0vpvWVbID1mnGr2SUtumDBQkJ9zjfvJdoFP7DIzk,66536
-langgraph_api/js/src/schema/types.template.mts,sha256=oAwnwWOgkEAQ3EouB8dG5Mdg4H75-aj5av40Mxfq7Xw,2141
+langgraph_api/js/src/schema/types.mts,sha256=SRCYZTWjxyc7528DaapR_DCm3G5bfDSh4vf-JsYpk0w,62633
+langgraph_api/js/src/schema/types.template.mts,sha256=Dbjj_8d-OubqH4QY_OaxSu8ocZ4dVjI94oncL20fqtk,2235
 langgraph_api/js/src/utils/importMap.mts,sha256=pX4TGOyUpuuWF82kXcxcv3-8mgusRezOGe6Uklm2O5A,1644
 langgraph_api/js/src/utils/pythonSchemas.mts,sha256=98IW7Z_VP7L_CHNRMb3_MsiV3BgLE2JsWQY_PQcRR3o,685
 langgraph_api/js/src/utils/serde.mts,sha256=OuyyO9btvwWd55rU_H4x91dFEJiaPxL-lL9O6Zgo908,742
 langgraph_api/js/sse.py,sha256=lsfp4nyJyA1COmlKG9e2gJnTttf_HGCB5wyH8OZBER8,4105
-langgraph_api/js/tests/api.test.mts,sha256=7B7uIiUsi4X3tN5ji7loaU4PGkoJyHhTIbqPN2V-7kY,57709
+langgraph_api/js/tests/api.test.mts,sha256=RTJb9ZMX_IhqUPcShK61LZG9v_-xzdIZ0oEmOIjbJ4c,57709
 langgraph_api/js/tests/compose-postgres.yml,sha256=wV1Kws7WwUWVIudPkB--v58MOPL9hOcV0MUK-cvNrpA,1738
 langgraph_api/js/tests/graphs/.gitignore,sha256=26J8MarZNXh7snXD5eTpV3CPFTht5Znv8dtHYCLNfkw,12
 langgraph_api/js/tests/graphs/agent.css,sha256=QgcOC0W7IBsrg4pSqqpull-WTgtULZfx_lF_5ZxLdag,23
@@ -58,8 +58,8 @@ langgraph_api/js/tests/graphs/yarn.lock,sha256=i2AAIgXA3XBLM8-oU45wgUefCSG-Tne4g
 langgraph_api/js/tests/parser.test.mts,sha256=dEC8KTqKygeb1u39ZvpPqCT4HtfPD947nLmITt2buxA,27883
 langgraph_api/js/tests/utils.mts,sha256=2kTybJ3O7Yfe1q3ehDouqV54ibXkNzsPZ_wBZLJvY-4,421
 langgraph_api/js/yarn.lock,sha256=W89dVYZMThcec08lJMcYnvEEnQK7VM5cPglvwpIdRv0,82773
-langgraph_api/lifespan.py,sha256=tcW68-Vxh10y15NDn0TJG2qAbU9RwsrQmumT5xs2HBA,2046
-langgraph_api/logging.py,sha256=rnd_pJsnTd-QbxVJQq-FswEDkd9tNzhoqp7g3_ReIzA,3598
+langgraph_api/lifespan.py,sha256=u5Fv9I4JqfLk30p5cNqkcEqeB2vTRMGSM5WlFHlE0mU,2219
+langgraph_api/logging.py,sha256=KB1ciduIWcMFfG0q9c5_SZemGrscht1RZXLZfeJSS00,3619
 langgraph_api/metadata.py,sha256=jPLNIRxHi7taZ0g60UdOEXenkvDwoYdI11tsmHenb28,3443
 langgraph_api/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/middleware/http_logger.py,sha256=yuFPNFIWwn-4AE1CogBfWlo8KytzywLi_Bd4ccsyVQE,3150
@@ -86,15 +86,15 @@ langgraph_storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 langgraph_storage/checkpoint.py,sha256=V4t2GwYEJdPCHbhq_4Udhlv0TWKDzlMu_rlNPdTDc50,3589
 langgraph_storage/database.py,sha256=I0AgFeJ-NSTT34vxKxQBUf1z2syFP0S8QpKCqTixrzY,5652
 langgraph_storage/inmem_stream.py,sha256=8bxkILIuFpr7P7RQ37SQAxrpRKvmbHdRB_nbfFiomlk,3263
-langgraph_storage/ops.py,sha256=BB1UabucGRPg_j-S-KktUbBypvrIj9xWQ596mkZbaQQ,69504
+langgraph_storage/ops.py,sha256=6qj3zh3T7KZco_vz9FW1cI82sbrefo6u2BRmrUDXvZI,69529
 langgraph_storage/queue.py,sha256=UDgsUTtUMfBSRDrQ8Onis-FJO4n7KTsX6sdpbY8Hs0A,5055
 langgraph_storage/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
-langgraph_storage/store.py,sha256=SL7VYCO1wI3O1WP20A9jD3lsxXIN990XuuyVrrj6RVY,2855
+langgraph_storage/store.py,sha256=33-J5-Xvobb9ArSa-GezP5KtfXgzWkHUHPyjRYmdw-E,2985
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=8NhoZn3Dse6Ia0i7fXCB46sdGnEg5qAhov_8of-6m0o,125282
-langgraph_api-0.0.28rc1.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.28rc1.dist-info/METADATA,sha256=vW8Ga7YG0_KFwuQFUCdM-beXbWqwYwfy_9MKdsDwBE4,4032
-langgraph_api-0.0.28rc1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.28rc1.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.28rc1.dist-info/RECORD,,
+langgraph_api-0.0.31.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.31.dist-info/METADATA,sha256=lSR24lckxz9BgmrjjqG7b04Q22m5m1W49dHtggG6BeY,4027
+langgraph_api-0.0.31.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.31.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.31.dist-info/RECORD,,

langgraph_storage/ops.py

@@ -1757,7 +1757,7 @@ class Runs(Authenticated):
             queue = (
                 stream_mode
                 if isinstance(stream_mode, asyncio.Queue)
-                else await Runs.Stream.subscribe(run_id)
+                else await Runs.Stream.subscribe(run_id, stream_mode=stream_mode)
             )
 
             try:

langgraph_storage/store.py

@@ -1,3 +1,4 @@
+import asyncio
 import os
 import threading
 from collections import defaultdict
@@ -42,6 +43,9 @@ class DiskBackedInMemStore(InMemoryStore):
                 f"Unexpected error loading store {which} from {container.filename}: {str(e)}"
             ) from e
 
+    async def start_ttl_sweeper(self) -> asyncio.Task[None]:
+        return asyncio.create_task(asyncio.sleep(0))
+
     def close(self) -> None:
         self._data.close()
         self._vectors.close()