langgraph-api 0.0.30__tar.gz → 0.0.31__tar.gz

{langgraph_api-0.0.30 → langgraph_api-0.0.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.30
+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.30 → langgraph_api-0.0.31}/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-0.0.30 → langgraph_api-0.0.31}/langgraph_api/api/runs.py

@@ -354,7 +354,7 @@ 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")
+    stream_mode = request.query_params.get("stream_mode") or None
     return EventSourceResponse(
         Runs.Stream.join(
             run_id,

{langgraph_api-0.0.30 → langgraph_api-0.0.31}/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-0.0.30 → langgraph_api-0.0.31}/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-0.0.30 → langgraph_api-0.0.31}/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-0.0.30 → langgraph_api-0.0.31}/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-0.0.30 → langgraph_api-0.0.31}/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()

{langgraph_api-0.0.30 → langgraph_api-0.0.31}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.0.30"
+version = "0.0.31"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",
@@ -24,18 +24,18 @@ sse-starlette = ">=2.1.0,<2.2.0"
 starlette = ">=0.38.6"
 watchfiles = ">=0.13"
 langgraph = ">=0.2.56,<0.4.0"
-langgraph-checkpoint = ">=2.0.15,<3.0"
+langgraph-checkpoint = ">=2.0.21,<3.0"
 orjson = ">=3.9.7"
 uvicorn = ">=0.26.0"
 langsmith = ">=0.1.63,<0.4.0"
 httpx = ">=0.25.0"
 langchain-core = ">=0.2.38,<0.4.0"
 tenacity = ">=8.0.0"
-jsonschema-rs = "^0.20.0"
+jsonschema-rs = ">=0.20.0,<0.30"
 structlog = ">=24.1.0,<26"
 pyjwt = "^2.9.0"
 cryptography = "^43.0.3"
-langgraph-sdk = "^0.1.53"
+langgraph-sdk = "^0.1.57"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.6.2"
@@ -49,6 +49,7 @@ pytest-retry = "^1.6.3"
 pytest-httpserver = "^1.1.0"
 fastapi = "^0.115.8"
 langgraph = ">=0.3.9"
+pycryptodome = "^3.22.0"
 
 [tool.pytest.ini_options]
 # --strict-markers will raise errors on unknown marks.