langgraph-api 0.12.0.dev18__py3-none-any.whl → 0.12.0.dev20__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.12.0.dev18"
+__version__ = "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/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/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/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/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.dev18.dist-info → langgraph_api-0.12.0.dev20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.12.0.dev18
+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.dev18.dist-info → langgraph_api-0.12.0.dev20.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langgraph_api/__init__.py,sha256=xTAqwHUE03gcPAzIvvqKQymgYred-UsTnKPtZxRYLq0,29
+langgraph_api/__init__.py,sha256=TKIcjxx-9aUyZmSvMeXz-Ek1FK_EguTANVH7RKLIqQA,29
 langgraph_api/_factory_utils.py,sha256=5JsiJbg_YocVSryN2jwoZTg03-eyymlWMK6sKCmXwz0,5756
 langgraph_api/asgi_transport.py,sha256=XApY3lIWBZTMbbsl8dDJzl0cLGirmAGE0SifqZUnXvs,11896
 langgraph_api/asyncio.py,sha256=smqyAoO9nIyPPQRw1IVSCh85fwoZP4PEo-5dDhxGruc,10676
@@ -35,7 +35,7 @@ langgraph_api/traceblock.py,sha256=GhgAtLhAQ8Z8LHtRQq3par-rfPpRBQp2Q7WBACJq1NI,6
 langgraph_api/tracing_session.py,sha256=gmfv-BW1XRar6suVur1P4rTVKWkjElDEb2hhkBduWOs,728
 langgraph_api/validation.py,sha256=XyeKyt7jAICmIlT_b0J0mv2YbwIbNoe4m6zEmfk9gOA,14657
 langgraph_api/webhook.py,sha256=qXEtkE6orek2POeOQmPRsEarJgXIYp-LBrZB-OwITxc,9572
-langgraph_api/worker.py,sha256=HGirbQ1i1hxaayORoihsQ6JMVAq3VsrC8CUxAHwY9t0,21151
+langgraph_api/worker.py,sha256=zWd14iZ9m9TLb9yD_QwTMxHcfM8dd3bhKtM1lz86M80,21256
 langgraph_api/_checkpointer/__init__.py,sha256=ofJTJLGy7Hsuzhj-2dpfDvrDloM0BzlhTzvZOdR9K8U,2223
 langgraph_api/_checkpointer/_adapter.py,sha256=1jJi0vXoNKeCwNP26W8bmYrwe-fOWFqepnovYFt14W0,21782
 langgraph_api/_checkpointer/protocol.py,sha256=udgYKMNtKWG_eLDwYkHXV3b2bZLZg8Rsfm3fjkhU-rU,3635
@@ -46,9 +46,9 @@ langgraph_api/api/event_streaming.py,sha256=nvoaKz4QGklX5YUmY9WQ3vSwhQ1Q81QeQWNR
 langgraph_api/api/meta.py,sha256=5s017GfMs-Ghq6Z5K6yW4UjbVFzCsmeXKn4UEJmfs98,4757
 langgraph_api/api/openapi.py,sha256=Zkdlb9mjrQyHro1TtrDIWVuaBDovxx-uGWJ1fZMOg54,12604
 langgraph_api/api/profile.py,sha256=CA1ZkHALOuP8orYTICnEhcG_JnnA2wnyjbWyeb117jA,3455
-langgraph_api/api/runs.py,sha256=h5droLgaz_aAyILCRJIpbj2KH1PbijCeXcggOSa3Zww,35178
+langgraph_api/api/runs.py,sha256=eqaeSUXg0mKROEyGTG7ur6RdSFHKIq6hMDEiYxO2R7Q,35197
 langgraph_api/api/store.py,sha256=Y5bMwi03nljlVCiiN91TGkiZj5iUITO68uyctBdYLhE,6656
-langgraph_api/api/threads.py,sha256=fNFRiXckglF_HyX9poJ2h0on9QeA-YRiVpZ0g-Z9b3w,19606
+langgraph_api/api/threads.py,sha256=atUJnC49tpdhx80OdyzT6NNV3J0pN1R1e4nq7qeXD6c,19778
 langgraph_api/api/ui.py,sha256=RalnZm1HArbC2X7fYCFtNeTJpSOQCzkLrdA0lcVmWQw,2695
 langgraph_api/api/mcp/__init__.py,sha256=T9W88PH9OqQtnjP7eFFa_163pFulDGU_AXk_PYBfEJA,469
 langgraph_api/api/mcp/_constants.py,sha256=jmMn_H77KdXP3c5NXIGifFk72B8tK2pNE5ufvJ7RdtY,555
@@ -65,7 +65,7 @@ langgraph_api/auth/studio_user.py,sha256=gNCicIo6cYaHmFj2sEdsvDYkKW7NWfGXGS2tTAM
 langgraph_api/auth/langsmith/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/auth/langsmith/backend.py,sha256=Y6-VxD7zfV1jzGdjmQ66CgNa3SenLbo3d_375CcKZ9U,3770
 langgraph_api/auth/langsmith/client.py,sha256=79kwCVeHU64nsHsxWipfZhf44lM6vfs2nlfTxlJF6LU,4142
-langgraph_api/config/__init__.py,sha256=NYJWKJ-PbfeI6k5KnF3W_HWLE8EOS0rJKaQQqKTCAus,25717
+langgraph_api/config/__init__.py,sha256=lSIVNfG_9LrJm2mKM1m46e1A8aaMwMqOEbGMgHcWY_Q,26300
 langgraph_api/config/_parse.py,sha256=RpfZgFAPJlRMv13AzGr3kAYbIrqHcgjzO8IgeboTw4A,3922
 langgraph_api/config/schemas.py,sha256=cHzVvepthpD7DDeWE2ytkOHah-iDNH7xRx9dSWUatQI,20864
 langgraph_api/encryption/__init__.py,sha256=gaCZ00CocSbqSqrDn6XJHaSp2CZCnC8qnrD9G4fbzyI,363
@@ -79,7 +79,7 @@ langgraph_api/event_streaming/capabilities.py,sha256=qjVbhCjl1VEQPGeiDxeJAhYGI_7
 langgraph_api/event_streaming/constants.py,sha256=eGsm-NvOlqV3gNxDO5vlr00FdngmgEQf59zuHSi_74E,1378
 langgraph_api/event_streaming/event_normalizers.py,sha256=5bVSqGPW-Uh7WX91qgTfwpK433pCSv_wchpbzW8TLi4,2794
 langgraph_api/event_streaming/namespace.py,sha256=aJDFt45Or2_bQdRpKJgdFhBgTDj6PYl7Coz5GzfbM84,1509
-langgraph_api/event_streaming/service.py,sha256=axQ2sDawAtsE96ZeJcBB8q8DbzS2VN-OqsdCM1rDF58,50063
+langgraph_api/event_streaming/service.py,sha256=Ipm6LHwQFfEKT4BjeUex7R55bvKb97M8W9gCsKTNgmw,50119
 langgraph_api/event_streaming/session.py,sha256=8kx5nxJvJlfIu5fFzQCxEl-8UBJEtxftBLhCYgXMNwo,75947
 langgraph_api/event_streaming/state_normalizers.py,sha256=vgT4O4tJPr9VDBMn1EP994ieDGDYP43sROnOyLjkEAE,13659
 langgraph_api/event_streaming/types.py,sha256=RyZqfqgH-jmmmmAFQj5f6nH9M1rGK93zVG7nlmvqZgc,3647
@@ -236,8 +236,8 @@ langgraph_grpc_common/proto/store_pb2.py,sha256=uAYM7OcdEOK12iRcPYihC2suSeCo_Qdb
 langgraph_grpc_common/proto/store_pb2.pyi,sha256=_dcxyxV8moR6H67qU42r2h2p_Ub-PZSHs3OoS9KMdSY,27782
 langgraph_grpc_common/proto/store_pb2_grpc.py,sha256=wSD65QgFkYTkPTHcf7VhyI2FQ7UH_RgIf3vQW3VYlsM,3324
 langgraph_grpc_common/proto/store_pb2_grpc.pyi,sha256=oS1h2cDoq2OjyM8xazcmc8LwRDC5oCAD0zto2QUmPQw,2027
-langgraph_api-0.12.0.dev18.dist-info/METADATA,sha256=LFdiGtoAmXB1C3oFB4mUPPKkF75X2h9xvZFiT_iAbew,4631
-langgraph_api-0.12.0.dev18.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
-langgraph_api-0.12.0.dev18.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.12.0.dev18.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.12.0.dev18.dist-info/RECORD,,
+langgraph_api-0.12.0.dev20.dist-info/METADATA,sha256=gj9GSR38Lrc1p1cb1u0BJKG1oXzuYvvnxSWfdI3y99I,4631
+langgraph_api-0.12.0.dev20.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+langgraph_api-0.12.0.dev20.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.12.0.dev20.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.12.0.dev20.dist-info/RECORD,,

{langgraph_api-0.12.0.dev18.dist-info → langgraph_api-0.12.0.dev20.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.30.1
+Generator: hatchling 1.31.0
 Root-Is-Purelib: true
 Tag: py3-none-any