langgraph-api 0.0.48__py3-none-any.whl → 0.1.0__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.46"
+__version__ = "0.1.0"

langgraph_api/api/__init__.py

@@ -21,7 +21,7 @@ from langgraph_api.auth.middleware import auth_middleware
 from langgraph_api.config import HTTP_CONFIG, MIGRATIONS_PATH
 from langgraph_api.graph import js_bg_tasks
 from langgraph_api.validation import DOCS_HTML
-from langgraph_storage.database import connect, healthcheck
+from langgraph_runtime.database import connect, healthcheck
 
 logger = structlog.stdlib.get_logger(__name__)
 
@@ -138,7 +138,7 @@ else:
 if "inmem" in MIGRATIONS_PATH:
 
     async def truncate(request: Request):
-        from langgraph_storage.checkpoint import Checkpointer
+        from langgraph_runtime.checkpoint import Checkpointer
 
         await asyncio.to_thread(Checkpointer().clear)
         async with connect() as conn:

langgraph_api/api/assistants.py

@@ -21,9 +21,9 @@ from langgraph_api.validation import (
     AssistantVersionChange,
     AssistantVersionsSearchRequest,
 )
-from langgraph_storage.database import connect
-from langgraph_storage.ops import Assistants
-from langgraph_storage.retry import retry_db
+from langgraph_runtime.database import connect
+from langgraph_runtime.ops import Assistants
+from langgraph_runtime.retry import retry_db
 
 logger = structlog.stdlib.get_logger(__name__)
 

langgraph_api/api/meta.py

@@ -5,9 +5,9 @@ from starlette.responses import JSONResponse, PlainTextResponse
 from langgraph_api import config
 from langgraph_api.route import ApiRequest
 from langgraph_license.validation import plus_features_enabled
-from langgraph_storage.database import connect, pool_stats
-from langgraph_storage.ops import Runs
-from langgraph_storage.queue import WORKERS
+from langgraph_runtime.database import connect, pool_stats
+from langgraph_runtime.metrics import get_metrics
+from langgraph_runtime.ops import Runs
 
 METRICS_FORMATS = {"prometheus", "json"}
 
@@ -32,20 +32,18 @@ async def meta_metrics(request: ApiRequest):
         format = "prometheus"
 
     # collect stats
-    workers_max = config.N_JOBS_PER_WORKER
-    workers_active = len(WORKERS)
-    workers_available = workers_max - workers_active
+    metrics = get_metrics()
+    worker_metrics = metrics["workers"]
+    workers_max = worker_metrics["max"]
+    workers_active = worker_metrics["active"]
+    workers_available = worker_metrics["available"]
 
     if format == "json":
         async with connect() as conn:
             return JSONResponse(
                 {
                     **pool_stats(),
-                    "workers": {
-                        "max": workers_max,
-                        "active": workers_active,
-                        "available": workers_available,
-                    },
+                    "workers": worker_metrics,
                     "queue": await Runs.stats(conn),
                 }
             )

langgraph_api/api/runs.py

@@ -22,9 +22,9 @@ from langgraph_api.validation import (
     RunsCancel,
 )
 from langgraph_license.validation import plus_features_enabled
-from langgraph_storage.database import connect
-from langgraph_storage.ops import Crons, Runs, Threads
-from langgraph_storage.retry import retry_db
+from langgraph_runtime.database import connect
+from langgraph_runtime.ops import Crons, Runs, Threads
+from langgraph_runtime.retry import retry_db
 
 
 @retry_db

langgraph_api/api/store.py

@@ -13,8 +13,8 @@ from langgraph_api.validation import (
     StorePutRequest,
     StoreSearchRequest,
 )
-from langgraph_storage.retry import retry_db
-from langgraph_storage.store import Store
+from langgraph_runtime.retry import retry_db
+from langgraph_runtime.store import Store
 
 
 def _validate_namespace(namespace: tuple[str, ...]) -> Response | None:

langgraph_api/api/threads.py

@@ -15,9 +15,9 @@ from langgraph_api.validation import (
     ThreadStateSearch,
     ThreadStateUpdate,
 )
-from langgraph_storage.database import connect
-from langgraph_storage.ops import Threads
-from langgraph_storage.retry import retry_db
+from langgraph_runtime.database import connect
+from langgraph_runtime.ops import Threads
+from langgraph_runtime.retry import retry_db
 
 
 @retry_db

langgraph_api/cli.py

@@ -6,6 +6,7 @@ import pathlib
 import threading
 import typing
 from collections.abc import Mapping, Sequence
+from typing import Literal
 
 from typing_extensions import TypedDict
 
@@ -134,6 +135,7 @@ def run_server(
     studio_url: str | None = None,
     disable_persistence: bool = False,
     allow_blocking: bool = False,
+    runtime_edition: Literal["inmem", "community", "postgres"] = "inmem",
     **kwargs: typing.Any,
 ):
     """Run the LangGraph API server."""
@@ -197,6 +199,7 @@ def run_server(
         LANGGRAPH_UI_BUNDLER="true",
         LANGGRAPH_API_URL=local_url,
         LANGGRAPH_DISABLE_FILE_PERSISTENCE=str(disable_persistence).lower(),
+        LANGGRAPH_RUNTIME_EDITION=runtime_edition,
         # If true, we will not raise on blocking IO calls (via blockbuster)
         LANGGRAPH_ALLOW_BLOCKING=str(allow_blocking).lower(),
         # See https://developer.chrome.com/blog/private-network-access-update-2024-03
@@ -274,7 +277,6 @@ For production use, please use LangGraph Cloud.
 
 """
         logger.info(welcome)
-
         if open_browser:
             threading.Thread(target=_open_browser, daemon=True).start()
         supported_kwargs = {

langgraph_api/config.py

@@ -323,6 +323,9 @@ USES_INDEXING = (
     and STORE_CONFIG.get("index").get("embed")
 )
 USES_CUSTOM_APP = HTTP_CONFIG and HTTP_CONFIG.get("app")
+USES_CUSTOM_AUTH = bool(LANGGRAPH_AUTH)
+USES_THREAD_TTL = bool(THREAD_TTL)
+USES_STORE_TTL = bool(STORE_CONFIG and STORE_CONFIG.get("ttl"))
 
 API_VARIANT = env("LANGSMITH_LANGGRAPH_API_VARIANT", cast=str, default="")
 

langgraph_api/cron_scheduler.py

@@ -7,9 +7,9 @@ from langchain_core.runnables.config import run_in_executor
 from langgraph_api.models.run import create_valid_run
 from langgraph_api.utils import next_cron_date
 from langgraph_api.worker import set_auth_ctx_for_run
-from langgraph_storage.database import connect
-from langgraph_storage.ops import Crons
-from langgraph_storage.retry import retry_db
+from langgraph_runtime.database import connect
+from langgraph_runtime.ops import Crons
+from langgraph_runtime.retry import retry_db
 
 logger = structlog.stdlib.get_logger(__name__)
 

langgraph_api/graph.py

@@ -50,8 +50,8 @@ async def register_graph(
     description: str | None = None,
 ) -> None:
     """Register a graph."""
-    from langgraph_storage.database import connect
-    from langgraph_storage.ops import Assistants
+    from langgraph_runtime.database import connect
+    from langgraph_runtime.ops import Assistants
 
     await logger.ainfo(f"Registering graph with id '{graph_id}'", graph_id=graph_id)
     GRAPHS[graph_id] = graph

langgraph_api/js/remote.py

@@ -365,7 +365,7 @@ async def run_js_process(paths_str: str, watch: bool = False):
 
 
 def _get_passthrough_checkpointer(conn: AsyncConnectionProto):
-    from langgraph_storage.checkpoint import Checkpointer
+    from langgraph_runtime.checkpoint import Checkpointer
 
     class PassthroughSerialiser(SerializerProtocol):
         def dumps(self, obj: Any) -> bytes:
@@ -393,7 +393,7 @@ def _get_passthrough_checkpointer(conn: AsyncConnectionProto):
 
 
 def _get_passthrough_store():
-    from langgraph_storage.store import Store
+    from langgraph_runtime.store import Store
 
     return Store()
 
@@ -401,7 +401,7 @@ def _get_passthrough_store():
 # Setup a HTTP server on top of CHECKPOINTER_SOCKET unix socket
 # used by `client.mts` to communicate with the Python checkpointer
 async def run_remote_checkpointer():
-    from langgraph_storage.database import connect
+    from langgraph_runtime.database import connect
 
     async def checkpointer_list(payload: dict):
         """Search checkpoints"""

langgraph_api/metadata.py

@@ -11,7 +11,10 @@ from langgraph_api.config import (
     LANGSMITH_API_KEY,
     LANGSMITH_AUTH_ENDPOINT,
     USES_CUSTOM_APP,
+    USES_CUSTOM_AUTH,
     USES_INDEXING,
+    USES_STORE_TTL,
+    USES_THREAD_TTL,
 )
 from langgraph_api.http import http_request
 from langgraph_license.validation import plus_features_enabled
@@ -95,8 +98,12 @@ async def metadata_loop() -> None:
                 "langgraph.platform.variant": VARIANT,
                 "langgraph.platform.host": HOST,
                 "langgraph.platform.plan": PLAN,
+                # user app features
                 "user_app.uses_indexing": USES_INDEXING,
                 "user_app.uses_custom_app": USES_CUSTOM_APP,
+                "user_app.uses_custom_auth": USES_CUSTOM_AUTH,
+                "user_app.uses_thread_ttl": USES_THREAD_TTL,
+                "user_app.uses_store_ttl": USES_STORE_TTL,
             },
             "measures": {
                 "langgraph.platform.runs": runs,

langgraph_api/models/run.py

@@ -23,7 +23,7 @@ from langgraph_api.schema import (
     StreamMode,
 )
 from langgraph_api.utils import AsyncConnectionProto, get_auth_ctx
-from langgraph_storage.ops import Runs, logger
+from langgraph_runtime.ops import Runs, logger
 
 
 class RunCreateDict(TypedDict):
@@ -177,6 +177,15 @@ async def create_valid_run(
         payload,
         run_id=run_id,
     )
+    if (
+        thread_id is None
+        and (command := payload.get("command"))
+        and command.get("resume")
+    ):
+        raise HTTPException(
+            status_code=400,
+            detail="You must provide a thread_id when resuming.",
+        )
     temporary = thread_id is None and payload.get("on_completion", "delete") == "delete"
     stream_mode, multitask_strategy, prevent_insert_if_inflight = assign_defaults(
         payload

langgraph_api/queue_entrypoint.py

@@ -8,7 +8,7 @@ import pathlib
 import structlog
 import uvloop
 
-from langgraph_api.lifespan import lifespan
+from langgraph_runtime.lifespan import lifespan
 
 logger = structlog.stdlib.get_logger(__name__)
 

langgraph_api/server.py

@@ -22,12 +22,12 @@ from langgraph_api.errors import (
     validation_error_handler,
     value_error_handler,
 )
-from langgraph_api.lifespan import lifespan
+from langgraph_runtime.lifespan import lifespan
 from langgraph_api.middleware.http_logger import AccessLoggerMiddleware
 from langgraph_api.middleware.private_network import PrivateNetworkMiddleware
 from langgraph_api.utils import SchemaGenerator
 from langgraph_license.middleware import LicenseValidationMiddleware
-from langgraph_storage.retry import OVERLOADED_EXCEPTIONS
+from langgraph_runtime.retry import OVERLOADED_EXCEPTIONS
 from langgraph_sdk.client import configure_loopback_transports
 
 logging.captureWarnings(True)

langgraph_api/stream.py

@@ -30,9 +30,9 @@ from langgraph_api.metadata import HOST, PLAN, USER_API_URL, incr_nodes
 from langgraph_api.schema import Run, StreamMode
 from langgraph_api.serde import json_dumpb
 from langgraph_api.utils import AsyncConnectionProto
-from langgraph_storage.checkpoint import Checkpointer
-from langgraph_storage.ops import Runs
-from langgraph_storage.store import Store
+from langgraph_runtime.checkpoint import Checkpointer
+from langgraph_runtime.ops import Runs
+from langgraph_runtime.store import Store
 
 logger = structlog.stdlib.get_logger(__name__)
 

langgraph_api/thread_ttl.py

@@ -5,7 +5,7 @@ import asyncio
 import structlog
 
 from langgraph_api.config import THREAD_TTL
-from langgraph_storage.database import connect
+from langgraph_runtime.database import connect
 
 logger = structlog.stdlib.get_logger(__name__)
 
@@ -30,7 +30,7 @@ async def thread_ttl_sweep_loop():
         interval_minutes=sweep_interval_minutes,
     )
 
-    from langgraph_storage.ops import Threads
+    from langgraph_runtime.ops import Threads
 
     while True:
         await asyncio.sleep(sweep_interval_minutes * 60)

langgraph_api/worker.py

@@ -26,9 +26,9 @@ from langgraph_api.stream import (
     consume,
 )
 from langgraph_api.utils import set_auth_ctx, with_user
-from langgraph_storage.database import connect
-from langgraph_storage.ops import Runs, Threads
-from langgraph_storage.retry import RETRIABLE_EXCEPTIONS
+from langgraph_runtime.database import connect
+from langgraph_runtime.ops import Runs, Threads
+from langgraph_runtime.retry import RETRIABLE_EXCEPTIONS
 
 try:
     from psycopg.errors import InFailedSqlTransaction

{langgraph_api-0.0.48.dist-info → langgraph_api-0.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langgraph-api
-Version: 0.0.48
+Version: 0.1.0
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos

{langgraph_api-0.0.48.dist-info → langgraph_api-0.1.0.dist-info}/RECORD

@@ -1,13 +1,13 @@
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api/__init__.py,sha256=WvtDrm5o4jAYjOF-kumI_Il7bjdj2Nr-QuEWD_sgdXQ,23
-langgraph_api/api/__init__.py,sha256=qNcg8QJydef0gM-vYJlxITMRZw-9r1vw8zqm2raqqYE,5493
-langgraph_api/api/assistants.py,sha256=WwaBtx1MpGn9gdJ8P9fkorJHMVrJKHt1nvtw0OCZcdw,14353
+langgraph_api/__init__.py,sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0,22
+langgraph_api/api/__init__.py,sha256=ohkuKTIYaWMAnr2pck2XAMrg4srA418VM76GQWRf5tU,5493
+langgraph_api/api/assistants.py,sha256=i-nxkScUB2g8bTVGtQIp1psABXlaY1aVx9pkB_UiRH8,14353
 langgraph_api/api/mcp.py,sha256=KbR19dtFCpJEiKYj3IfepAuJij8YZVELuVp7JY_yu_o,13754
-langgraph_api/api/meta.py,sha256=ifJ_Ki0Qf2DYbmY6OKlqKhLGxbt55gm0lEqH1A0cJbw,2790
+langgraph_api/api/meta.py,sha256=M1JHcuZGEn8E-ZpuEbNBJ6SoAwZBCcPOq0NRHMCplR8,2713
 langgraph_api/api/openapi.py,sha256=f9gfmWN2AMKNUpLCpSgZuw_aeOF9jCXPdOtFT5PaTWM,10960
-langgraph_api/api/runs.py,sha256=qAXfgZjjaMBfwPnlnAogvemtKZZeshNSIMQqcW20tGs,18010
-langgraph_api/api/store.py,sha256=XNNZFRlvgReidq9u7mg-i7pjwz21BdP9Qw3Jr5Ra9Fk,5447
-langgraph_api/api/threads.py,sha256=QbAy7MRupWKDUhmC6_LKU2ExJbLJ6Z-CJG2gSpcMXtc,9163
+langgraph_api/api/runs.py,sha256=dhHZ3xu7V0anftqzXaOYnhVEryJpVeqzu60MFiUw4u8,18010
+langgraph_api/api/store.py,sha256=G4Fm8hgFLlJUW5_ekLS_IOgCfpFy-TK9uq5r9QTOELQ,5447
+langgraph_api/api/threads.py,sha256=EvKJDnZDWutKoB5KoTP40NPEqG5LJ-Iw97TLYHkhLbw,9163
 langgraph_api/api/ui.py,sha256=kdCQ-p0voxAqIFc72aqqzdPGH2v-yEBKzjRE6cUPvpU,2201
 langgraph_api/asyncio.py,sha256=h0eZ7aoDGnJpoxnHLZABVlj1jQ78UxjgiHntTmAEWek,8613
 langgraph_api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -18,12 +18,12 @@ langgraph_api/auth/langsmith/client.py,sha256=eKchvAom7hdkUXauD8vHNceBDDUijrFgdT
 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=W8rf3v4rMwv9W-tK_9yqcc7waQezMMlpBoy64WSGLMw,12682
+langgraph_api/cli.py,sha256=eVX8zGeQAoVMotib2s-QhIQWQX1JInpVMItkTTgG_dU,12833
 langgraph_api/command.py,sha256=3O9v3i0OPa96ARyJ_oJbLXkfO8rPgDhLCswgO9koTFA,768
-langgraph_api/config.py,sha256=OJ6_9UCkPYNpUEB17EIU1nO0zQd3_T3G1Dr-ag0J8hE,10218
-langgraph_api/cron_scheduler.py,sha256=9yzbbGxzNgJdIg4ZT7yu2oTwT_wRuPxD1c2sbbd52xs,2630
+langgraph_api/config.py,sha256=ZxVQitVzc9Rp9_j3myjW_NcSJ463kUAMmbxmmOS-tXY,10357
+langgraph_api/cron_scheduler.py,sha256=i87j4pJrcsmsqMKeKUs69gaAjrGaSM3pM3jnXdN5JDQ,2630
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
-langgraph_api/graph.py,sha256=v7qj60MBQKfEx_0kmw9AsPygCQsOLmrJ1MkVkt4-GiM,21086
+langgraph_api/graph.py,sha256=ba3G0AiI7vyHAMLV7ol8nZnsLivcBRMGGmj9fiaO7PU,21086
 langgraph_api/http.py,sha256=gYbxxjY8aLnsXeJymcJ7G7Nj_yToOGpPYQqmZ1_ggfA,5240
 langgraph_api/js/.gitignore,sha256=l5yI6G_V6F1600I1IjiUKn87f4uYIrBAYU1MOyBBhg4,59
 langgraph_api/js/base.py,sha256=xkBp5bwRrbpMFaAMViEU-qIlnsJuu3X_G8sa1pqNZK0,227
@@ -32,7 +32,7 @@ langgraph_api/js/client.mts,sha256=GJv846jWBRL0W_Yr6XtDu3UdYjdpndQ2v6dOfAhJXi0,2
 langgraph_api/js/errors.py,sha256=Cm1TKWlUCwZReDC5AQ6SgNIVGD27Qov2xcgHyf8-GXo,361
 langgraph_api/js/global.d.ts,sha256=yDusqAyzVYhxfwqqcERUzucu2Pw9ma3-ug4DFyUvQfs,167
 langgraph_api/js/package.json,sha256=oNcWe7UIoJtqzoIVr47px6-1n8KziqwEhyi4wBpzTQ0,1266
-langgraph_api/js/remote.py,sha256=dDrxJhe9Nm1OQREoa811GeXHygiBHTtHjvEVglOGfto,24808
+langgraph_api/js/remote.py,sha256=-f3o85b7bBWhcbgdO4fLmRBdqG_lAN56d9WWcTlEBGo,24808
 langgraph_api/js/schema.py,sha256=7idnv7URlYUdSNMBXQcw7E4SxaPxCq_Oxwnlml8q5ik,408
 langgraph_api/js/src/graph.mts,sha256=otgztTNzNJpeF2IOrpNuuwbSbpAy4eFE5dHtUd7eQwU,3742
 langgraph_api/js/src/hooks.mjs,sha256=XtktgmIHlls_DsknAuwib-z7TqCm0haRoTXvnkgzMuo,601
@@ -63,43 +63,35 @@ langgraph_api/js/tests/parser.test.mts,sha256=dEC8KTqKygeb1u39ZvpPqCT4HtfPD947nL
 langgraph_api/js/tests/utils.mts,sha256=q1V9gvT63v95onlfK9W4iv3n9ZJO3h-0RD9TdDYuRyY,439
 langgraph_api/js/ui.py,sha256=XNT8iBcyT8XmbIqSQUWd-j_00HsaWB2vRTVabwFBkik,2439
 langgraph_api/js/yarn.lock,sha256=m4erGgHd30Enxdu1wKnohnzotZiEDvXh-H83qitf_W4,83812
-langgraph_api/lifespan.py,sha256=SfVZj0SQdsIfYwvN5s4dfxGMb7MMlGe40DghCGDFaTQ,2915
 langgraph_api/logging.py,sha256=JJIzbNIgLCN6ClQ3tA-Mm5ffuBGvpRDSZsEvnIlsuu4,3693
-langgraph_api/metadata.py,sha256=bAeN3NwibBuXUVPjOEbEUJMnhUXe_VdTGw508VNeav4,3655
+langgraph_api/metadata.py,sha256=iM_3qaxFk1ItB4E5VaVb4n_mbqXzqZFMknVcHXnV0mI,3937
 langgraph_api/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/middleware/http_logger.py,sha256=aj4mdisRobFePkD3Iy6-w_Mujwx4TQRaEhPvSd6HgLk,3284
 langgraph_api/middleware/private_network.py,sha256=eYgdyU8AzU2XJu362i1L8aSFoQRiV7_aLBPw7_EgeqI,2111
 langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/models/run.py,sha256=85-pyyvosFAot8WrWl2QjCje-2c4lhattYvoAEMqztA,11000
+langgraph_api/models/run.py,sha256=6zl3t1_LAr5v2AIcrFSWDG5rLvqREeswXJymuGx6AT8,11258
 langgraph_api/patch.py,sha256=Dgs0PXHytekX4SUL6KsjjN0hHcOtGLvv1GRGbh6PswU,1408
-langgraph_api/queue_entrypoint.py,sha256=4xICUxXarNV8DhnaqAMhVi3xCmyVKCL3J5NzHxPA9Xc,1835
+langgraph_api/queue_entrypoint.py,sha256=gjtajZfnDXhsi7JElfmkY-p0ENBiKBDJ4Ugiw-exapw,1839
 langgraph_api/route.py,sha256=fM4qYCGbmH0a3_cV8uKocb1sLklehxO6HhdRXqLK6OM,4421
 langgraph_api/schema.py,sha256=Frh_YOC3S1cDAMPUVanNi78ooSXK2WFpu9YkIVz5h14,5433
 langgraph_api/serde.py,sha256=D5t_HeABMYKRAsoXWaWG0IsdaYM8yOXIez2wJUTIgT8,3963
-langgraph_api/server.py,sha256=bnXOOYztQmqR-QVpEFoRWB5Fzd33PuEIlwBK2R7W8NE,4849
+langgraph_api/server.py,sha256=Chjk-TddIVKqYqulRJb7AGYW6GZS3aU3T8ApNS7azGA,4853
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
-langgraph_api/stream.py,sha256=lhjnom-T8GbquUZry-KSkajnqYjElaIERhPiXPtpw1E,11354
-langgraph_api/thread_ttl.py,sha256=QS9C7cx63KBh82_I6DqOb42-5mLTqTvneL8FJeX3aqU,1765
+langgraph_api/stream.py,sha256=EdFb1ctSzkIWGws4vJMkTAtdGIWMmvNuTSQ7FBazb4I,11354
+langgraph_api/thread_ttl.py,sha256=4vch4nu1UOiYcqYRd7bTHsfs0Ei_lXuy9nBQ0uVJLyo,1765
 langgraph_api/utils.py,sha256=92mSti9GfGdMRRWyESKQW5yV-75Z9icGHnIrBYvdypU,3619
 langgraph_api/validation.py,sha256=Qo3EkSRXzIRe3GRuqRWbElTcUXRMXMyA1w0VbMvdwME,4934
 langgraph_api/webhook.py,sha256=1ncwO0rIZcj-Df9sxSnFEzd1gP1bfS4okeZQS8NSRoE,1382
-langgraph_api/worker.py,sha256=CLmohk3B03OTsKHq_ZPhujMHFSuc694qeEL4VNSxHvg,12679
+langgraph_api/worker.py,sha256=_i1cmpnjMzQtok5CzssjnR5IFxLk4dMyuPwreq8dorU,12679
 langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_license/middleware.py,sha256=_ODIYzQkymr6W9_Fp9wtf1kAQspnpsmr53xuzyF2GA0,612
 langgraph_license/validation.py,sha256=Uu_G8UGO_WTlLsBEY0gTVWjRR4czYGfw5YAD3HLZoj0,203
-langgraph_storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_storage/checkpoint.py,sha256=Qq0y6vdh27qdF3h5nOLT5CcX9Rj2bcFqkVOMeCaGoK4,4036
-langgraph_storage/database.py,sha256=sZjZvMcvbr_6dX0d1YrYEccVuQozIfkiWt8bdlXGVYU,5849
-langgraph_storage/inmem_stream.py,sha256=LjJSAxsh_E0ywqEMzdWJk8Hy_Jn9oQByzycss-fANng,3264
-langgraph_storage/ops.py,sha256=ksJPrjdFRXRGYffrdLvk06Ec-OT1E_Q6qB1_rbdF0w8,75800
-langgraph_storage/queue.py,sha256=IGjzCzYaGhbR9_Y37p1Hbd5uQkN9_IGzYkT_5lcPU54,7595
-langgraph_storage/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
-langgraph_storage/store.py,sha256=JB9jZ87GE19MVN9wgl3-esgR2eIkeipws9q6qsPWkgc,3399
+langgraph_runtime/__init__.py,sha256=3stEi5ECApK7l0Y7q4vN-fQQz8cgJ9lyqAyhqDBqags,1075
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=YW4ND-N3adriEoNwxw7UD9endO2xUZoodCtwVIfa2dU,132261
-langgraph_api-0.0.48.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.48.dist-info/METADATA,sha256=0Bi2FktyQZbXA3aE3Q1YDCmaGh0Vwiswtk_x1xSUtz0,4165
-langgraph_api-0.0.48.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-langgraph_api-0.0.48.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.48.dist-info/RECORD,,
+langgraph_api-0.1.0.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.1.0.dist-info/METADATA,sha256=nFYl9c7QS5BKMRaRLTr1eFhOTPyVh4ouyYgbUalt_K8,4164
+langgraph_api-0.1.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+langgraph_api-0.1.0.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.1.0.dist-info/RECORD,,

langgraph_runtime/__init__.py

@@ -0,0 +1,39 @@
+import importlib.util
+import os
+import sys
+
+import structlog
+
+logger = structlog.stdlib.get_logger(__name__)
+
+try:
+    RUNTIME_EDITION = os.environ["LANGGRAPH_RUNTIME_EDITION"]
+    RUNTIME_PACKAGE = f"langgraph_runtime_{RUNTIME_EDITION}"
+except KeyError:
+    raise ValueError(
+        "LANGGRAPH_RUNTIME_EDITION environment variable is not set."
+        " Expected LANGGRAPH_RUNTIME_EDITION to be set to one of:\n"
+        " - inmem\n"
+        " - postgres\n"
+        " - community\n"
+    ) from None
+if importlib.util.find_spec(RUNTIME_PACKAGE):
+    backend = importlib.import_module(RUNTIME_PACKAGE)
+    logger.info(f"Using {RUNTIME_PACKAGE}")
+else:
+    raise ImportError(
+        "Langgraph runtime backend not found. Please install with "
+        f'`pip install "langgraph-runtime[{RUNTIME_EDITION}"`'
+    ) from None
+
+# All runtime backends share the same API
+for module_name in (
+    "checkpoint",
+    "database",
+    "lifespan",
+    "ops",
+    "retry",
+    "store",
+    "metrics",
+):
+    sys.modules["langgraph_runtime." + module_name] = getattr(backend, module_name)

langgraph_api/lifespan.py

@@ -1,74 +0,0 @@
-import asyncio
-from contextlib import asynccontextmanager
-
-import structlog
-from langchain_core.runnables.config import var_child_runnable_config
-from langgraph.constants import CONF, CONFIG_KEY_STORE
-from starlette.applications import Starlette
-
-import langgraph_api.config as config
-from langgraph_api.asyncio import SimpleTaskGroup, set_event_loop
-from langgraph_api.cron_scheduler import cron_scheduler
-from langgraph_api.graph import collect_graphs_from_env, stop_remote_graphs
-from langgraph_api.http import start_http_client, stop_http_client
-from langgraph_api.js.ui import start_ui_bundler, stop_ui_bundler
-from langgraph_api.metadata import metadata_loop
-from langgraph_api.thread_ttl import thread_ttl_sweep_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
-
-logger = structlog.stdlib.get_logger(__name__)
-
-
-@asynccontextmanager
-async def lifespan(
-    app: Starlette | None = None,
-    with_cron_scheduler: bool = True,
-    taskset: set[asyncio.Task] | None = None,
-):
-    try:
-        current_loop = asyncio.get_running_loop()
-        set_event_loop(current_loop)
-    except RuntimeError:
-        await logger.aerror("Failed to set loop")
-
-    if not await get_license_status():
-        raise ValueError(
-            "License verification failed. Please ensure proper configuration:\n"
-            "- For local development, set a valid LANGSMITH_API_KEY for an account with LangGraph Cloud access "
-            "in the environment defined in your langgraph.json file.\n"
-            "- For production, configure the LANGGRAPH_CLOUD_LICENSE_KEY environment variable "
-            "with your LangGraph Cloud license key.\n"
-            "Review your configuration settings and try again. If issues persist, "
-            "contact support for assistance."
-        )
-    await start_http_client()
-    await start_pool()
-    await collect_graphs_from_env(True)
-    await start_ui_bundler()
-    try:
-        async with SimpleTaskGroup(
-            cancel=True, taskset=taskset, taskgroup_name="Lifespan"
-        ) as tg:
-            tg.create_task(metadata_loop())
-            if config.N_JOBS_PER_WORKER > 0:
-                tg.create_task(queue())
-            if (
-                with_cron_scheduler
-                and config.FF_CRONS_ENABLED
-                and plus_features_enabled()
-            ):
-                tg.create_task(cron_scheduler())
-            store = Store()
-            tg.create_task(Store().start_ttl_sweeper())
-            tg.create_task(thread_ttl_sweep_loop())
-            var_child_runnable_config.set({CONF: {CONFIG_KEY_STORE: store}})
-
-            yield
-    finally:
-        await stop_ui_bundler()
-        await stop_remote_graphs()
-        await stop_http_client()
-        await stop_pool()