langgraph-api 0.12.0.dev19__tar.gz → 0.12.0.dev20__tar.gz

{langgraph_api-0.12.0.dev19 → langgraph_api-0.12.0.dev20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.12.0.dev19
+Version: 0.12.0.dev20
 Author-email: Will Fu-Hinthorn <will@langchain.dev>, Josh Rogers <josh@langchain.dev>, Parker Rule <parker@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE

langgraph_api-0.12.0.dev20/langgraph_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.12.0.dev20"

{langgraph_api-0.12.0.dev19 → langgraph_api-0.12.0.dev20}/langgraph_api/api/runs.py

@@ -787,7 +787,9 @@ async def create_cron(request: ApiRequest):
 
     enabled = payload.get("enabled", True)
 
-    async with connect(supports_core_api=False) as conn:
+    # gRPC/postgres Crons route through core-api and ignore ``conn`` (the default
+    # ``connect()`` yields None there); inmem uses the yielded connection.
+    async with connect() as conn:
         cron = await Crons.put(
             conn,
             thread_id=None,
@@ -834,7 +836,7 @@ async def create_thread_cron(request: ApiRequest):
             CRON_PAYLOAD_ENCRYPTION_SUBFIELDS,
         )
 
-    async with connect(supports_core_api=False) as conn:
+    async with connect() as conn:
         cron = await Crons.put(
             conn,
             thread_id=thread_id,
@@ -890,7 +892,7 @@ async def patch_cron(request: ApiRequest):
             CRON_PAYLOAD_ENCRYPTION_SUBFIELDS,
         )
 
-    async with connect(supports_core_api=False) as conn:
+    async with connect() as conn:
         cron = await Crons.update(
             conn,
             cron_id=cron_id,
@@ -917,7 +919,7 @@ async def delete_cron(request: ApiRequest):
     validate_uuid(cron_id, "Invalid cron ID: must be a UUID")
 
     try:
-        async with connect(supports_core_api=False) as conn:
+        async with connect() as conn:
             cid = await Crons.delete(
                 conn,
                 cron_id=cron_id,
@@ -940,7 +942,7 @@ async def search_crons(request: ApiRequest):
         validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
 
     offset = int(payload.get("offset", 0))
-    async with connect(supports_core_api=False) as conn:
+    async with connect() as conn:
         crons_iter, next_offset = await Crons.search(
             conn,
             assistant_id=assistant_id,
@@ -972,7 +974,7 @@ async def count_crons(request: ApiRequest):
     if thread_id := payload.get("thread_id"):
         validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
 
-    async with connect(supports_core_api=False) as conn:
+    async with connect() as conn:
         count = await Crons.count(
             conn,
             assistant_id=assistant_id,

{langgraph_api-0.12.0.dev19 → langgraph_api-0.12.0.dev20}/langgraph_api/api/threads.py

@@ -13,7 +13,10 @@ from langgraph_api.encryption.shared import (
     using_aes_encryption,
     using_custom_encryption,
 )
-from langgraph_api.feature_flags import IS_POSTGRES_OR_GRPC_BACKEND
+from langgraph_api.feature_flags import (
+    IS_POSTGRES_OR_GRPC_BACKEND,
+    PREFER_GRPC_CHECKPOINTER,
+)
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
 from langgraph_api.schema import (
     THREAD_ENCRYPTION_FIELDS,
@@ -82,7 +85,7 @@ async def create_thread(
         )
     else:
         # Need connection for inmem put or gRPC State.bulk
-        async with connect(supports_core_api=False) as conn:
+        async with connect(supports_core_api=PREFER_GRPC_CHECKPOINTER) as conn:
             iter = await Threads.put(
                 conn,
                 thread_id,
@@ -226,7 +229,7 @@ async def get_thread_state(
     thread_id = request.path_params["thread_id"]
     validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
     subgraphs = request.query_params.get("subgraphs") in ("true", "True")
-    async with connect(supports_core_api=False) as conn:
+    async with connect(supports_core_api=PREFER_GRPC_CHECKPOINTER) as conn:
         config = {
             "configurable": {
                 **get_configurable_headers(request.headers),
@@ -247,7 +250,7 @@ async def get_thread_state_at_checkpoint(
     thread_id = request.path_params["thread_id"]
     validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
     checkpoint_id = request.path_params["checkpoint_id"]
-    async with connect(supports_core_api=False) as conn:
+    async with connect(supports_core_api=PREFER_GRPC_CHECKPOINTER) as conn:
         config = {
             "configurable": {
                 **get_configurable_headers(request.headers),
@@ -273,7 +276,7 @@ async def get_thread_state_at_checkpoint_post(
     thread_id = request.path_params["thread_id"]
     validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
     payload = await request.json(ThreadStateCheckpointRequest)
-    async with connect(supports_core_api=False) as conn:
+    async with connect(supports_core_api=PREFER_GRPC_CHECKPOINTER) as conn:
         config = {
             "configurable": {
                 **payload["checkpoint"],
@@ -311,7 +314,7 @@ async def update_thread_state(
         pass
     config["configurable"].update(get_configurable_headers(request.headers))
     config["configurable"]["thread_id"] = thread_id
-    async with connect(supports_core_api=False) as conn:
+    async with connect(supports_core_api=PREFER_GRPC_CHECKPOINTER) as conn:
         inserted = await Threads.State.post(
             conn,
             config,
@@ -341,7 +344,7 @@ async def get_thread_history(
             **get_configurable_headers(request.headers),
         }
     }
-    async with connect(supports_core_api=False) as conn:
+    async with connect(supports_core_api=PREFER_GRPC_CHECKPOINTER) as conn:
         states = [
             state_snapshot_to_thread_state(c)
             for c in await Threads.State.list(
@@ -363,7 +366,7 @@ async def get_thread_history_post(
     config["configurable"].update(payload.get("checkpoint", {}))
     config["configurable"].update(get_configurable_headers(request.headers))
     config["configurable"]["thread_id"] = thread_id
-    async with connect(supports_core_api=False) as conn:
+    async with connect(supports_core_api=PREFER_GRPC_CHECKPOINTER) as conn:
         states = [
             state_snapshot_to_thread_state(c)
             for c in await Threads.State.list(

{langgraph_api-0.12.0.dev19 → langgraph_api-0.12.0.dev20}/langgraph_api/config/__init__.py

@@ -4,7 +4,7 @@ from typing import TYPE_CHECKING, Annotated, Literal, TypeVar, cast
 
 import structlog
 from pydantic.functional_validators import AfterValidator
-from starlette.config import Config, undefined
+from starlette.config import Config
 from starlette.datastructures import CommaSeparatedStrings
 
 from langgraph_api import traceblock
@@ -41,7 +41,14 @@ STATS_INTERVAL_SECS = env("STATS_INTERVAL_SECS", cast=int, default=60)
 # storage
 
 # Not in public docs: infrastructure, set by platform
-DATABASE_URI = env("DATABASE_URI", cast=str, default=getenv("POSTGRES_URI", undefined))
+# Optional: persistence can go through the Go core over gRPC, so the Python
+# runtime can start without a direct DB connection string. When unset, the
+# Postgres runtime skips creating a psycopg pool (see database._startup_needs /
+# database.start_pool). Callers that still require a direct connection fail
+# loudly via connect()'s "Postgres pool not initialized" guard.
+DATABASE_URI: str | None = env(
+    "DATABASE_URI", cast=str, default=getenv("POSTGRES_URI", None)
+)
 # Not in public docs: infrastructure, set by platform
 MIGRATIONS_PATH = env("MIGRATIONS_PATH", cast=str, default="/storage/migrations")
 POSTGRES_POOL_MAX_SIZE = env("LANGGRAPH_POSTGRES_POOL_MAX_SIZE", cast=int, default=150)
@@ -246,6 +253,10 @@ STORE_CONFIG = env(
 LANGGRAPH_STORE_BACKEND = env("LANGGRAPH_STORE_BACKEND", cast=str, default="python")
 USE_GRPC_STORE = LANGGRAPH_STORE_BACKEND == "grpc"
 
+# When true, the Go core-server owns Postgres schema setup (DB migrations +
+# vector index)
+DB_MIGRATION_BY_CORE_API = env("DB_MIGRATION_BY_CORE_API", cast=bool, default=False)
+
 
 def _validate_mount_prefix(mount_prefix: str | None) -> str | None:
     if not mount_prefix:
@@ -642,6 +653,7 @@ __all__ = [
     "CRON_SCHEDULER_SLEEP_TIME",
     "DATABASE_URI",
     "DATADOG_METRICS_ENABLED",
+    "DB_MIGRATION_BY_CORE_API",
     "EXPOSE_INTERNAL_METRICS_PROMETHEUS",
     "FF_CRONS_ENABLED",
     "FF_LOG_DROPPED_EVENTS",

{langgraph_api-0.12.0.dev19 → langgraph_api-0.12.0.dev20}/langgraph_api/event_streaming/service.py

@@ -43,7 +43,7 @@ from langgraph_api.event_streaming.session import (
     _is_supported_channel,
 )
 from langgraph_api.event_streaming.types import Subscription
-from langgraph_api.feature_flags import FF_V2_EVENT_STREAMING
+from langgraph_api.feature_flags import FF_V2_EVENT_STREAMING, PREFER_GRPC_CHECKPOINTER
 from langgraph_api.schema import MultitaskStrategy
 
 logger = structlog.stdlib.get_logger(__name__)
@@ -962,11 +962,11 @@ class ThreadRunManager:
         async def get_thread_state() -> dict[str, Any] | None:
             from langgraph_runtime.database import connect  # noqa: PLC0415
 
-            # ``supports_core_api=False`` so the postgres backend yields a
-            # real connection. ``State.get`` uses the local checkpointer
-            # and needs a usable ``conn`` — the default flag yields None.
+            # When ``PREFER_GRPC_CHECKPOINTER`` is false the postgres backend
+            # yields a real connection for the Python checkpointer; when true
+            # it yields a no-op stand-in and checkpoint I/O goes through gRPC.
             try:
-                async with connect(supports_core_api=False) as conn:
+                async with connect(supports_core_api=PREFER_GRPC_CHECKPOINTER) as conn:
                     return await self._threads.State.get(
                         conn,
                         {"configurable": {"thread_id": thread_id}},

{langgraph_api-0.12.0.dev19 → langgraph_api-0.12.0.dev20}/langgraph_api/worker.py

@@ -25,7 +25,10 @@ from langgraph_api.encryption.middleware import (
     extract_blob_encryption_context,
 )
 from langgraph_api.errors import UserInterrupt, UserRollback, UserTimeout
-from langgraph_api.feature_flags import IS_POSTGRES_OR_GRPC_BACKEND
+from langgraph_api.feature_flags import (
+    IS_POSTGRES_OR_GRPC_BACKEND,
+    PREFER_GRPC_CHECKPOINTER,
+)
 from langgraph_api.graph import restore_dd_trace_context
 from langgraph_api.js.errors import RemoteException
 from langgraph_api.metadata import incr_runs
@@ -319,9 +322,11 @@ async def worker(
                         run_attempt=attempt,
                     )
                     try:
-                        # We actually still need a connection for the python checkpointer
-                        # This only doubles up connections for inmem which is fine
-                        async with connect(supports_core_api=False) as conn:
+                        # Python checkpointer needs a real PG conn; gRPC checkpointer
+                        # does not. This only doubles up connections for inmem.
+                        async with connect(
+                            supports_core_api=PREFER_GRPC_CHECKPOINTER
+                        ) as conn:
                             state_snapshot = await Threads.State.get(
                                 conn, run["kwargs"]["config"], subgraphs=False
                             )

{langgraph_api-0.12.0.dev19 → langgraph_api-0.12.0.dev20}/uv.lock

@@ -901,7 +901,7 @@ requires-dist = [
 [package.metadata.requires-dev]
 dev = [
     { name = "grpcio-health-checking", specifier = ">=1.80.0,<1.82.0" },
-    { name = "grpcio-tools", specifier = "==1.80.0" },
+    { name = "grpcio-tools", specifier = "==1.81.1" },
     { name = "langchain", specifier = ">=0.3" },
     { name = "langchain-core", specifier = ">=0.3" },
     { name = "mypy-protobuf", specifier = ">=5.0.0,<6" },

langgraph_api-0.12.0.dev19/langgraph_api/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.12.0.dev19"