langgraph-api 0.0.14__py3-none-any.whl → 0.0.15__py3-none-any.whl

langgraph_api/api/__init__.py

@@ -13,7 +13,6 @@ from langgraph_api.api.threads import threads_routes
 from langgraph_api.auth.middleware import auth_middleware
 from langgraph_api.config import MIGRATIONS_PATH
 from langgraph_api.graph import js_bg_tasks
-from langgraph_api.js.remote import js_healthcheck
 from langgraph_api.validation import DOCS_HTML
 from langgraph_storage.database import connect, healthcheck
 
@@ -23,6 +22,8 @@ async def ok(request: Request):
     if check_db:
         await healthcheck()
     if js_bg_tasks:
+        from langgraph_api.js.remote import js_healthcheck
+
         await js_healthcheck()
     return JSONResponse({"ok": True})
 

langgraph_api/api/assistants.py

@@ -8,7 +8,7 @@ from starlette.responses import Response
 from starlette.routing import BaseRoute
 
 from langgraph_api.graph import get_assistant_id, get_graph
-from langgraph_api.js.remote import RemotePregel
+from langgraph_api.js.base import BaseRemotePregel
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
 from langgraph_api.serde import ajson_loads
 from langgraph_api.utils import fetchone, validate_uuid
@@ -138,7 +138,7 @@ async def get_assistant_graph(
                 if xray <= 0:
                     raise HTTPException(422, detail="Invalid xray value") from None
 
-        if isinstance(graph, RemotePregel):
+        if isinstance(graph, BaseRemotePregel):
             drawable_graph = await graph.fetch_graph(xray=xray)
             return ApiResponse(drawable_graph.to_json())
         return ApiResponse(graph.get_graph(xray=xray).to_json())
@@ -158,7 +158,7 @@ async def get_assistant_subgraphs(
     async with get_graph(assistant["graph_id"], config) as graph:
         namespace = request.path_params.get("namespace")
 
-        if isinstance(graph, RemotePregel):
+        if isinstance(graph, BaseRemotePregel):
             return ApiResponse(
                 await graph.fetch_subgraphs(
                     namespace=namespace,
@@ -191,7 +191,7 @@ async def get_assistant_schemas(
     assistant = await fetchone(assistant_)
     config = await ajson_loads(assistant["config"])
     async with get_graph(assistant["graph_id"], config) as graph:
-        if isinstance(graph, RemotePregel):
+        if isinstance(graph, BaseRemotePregel):
             schemas = await graph.fetch_state_schema()
             return ApiResponse(
                 {

langgraph_api/config.py

@@ -28,6 +28,7 @@ BG_JOB_NO_DELAY = env("BG_JOB_NO_DELAY", cast=bool, default=False)
 N_JOBS_PER_WORKER = env("N_JOBS_PER_WORKER", cast=int, default=10)
 BG_JOB_TIMEOUT_SECS = env("BG_JOB_TIMEOUT_SECS", cast=float, default=3600)
 FF_CRONS_ENABLED = env("FF_CRONS_ENABLED", cast=bool, default=True)
+FF_JS_ZEROMQ_ENABLED = env("FF_JS_ZEROMQ_ENABLED", cast=bool, default=False)
 
 # auth
 

langgraph_api/graph.py

@@ -21,7 +21,7 @@ from langgraph.pregel import Pregel
 from langgraph.store.base import BaseStore
 from starlette.exceptions import HTTPException
 
-from langgraph_api.js.remote import RemotePregel
+from langgraph_api.js.base import BaseRemotePregel
 from langgraph_api.schema import Config
 
 if TYPE_CHECKING:
@@ -63,7 +63,7 @@ async def register_graph(graph_id: str, graph: GraphValue, config: dict | None)
 @asynccontextmanager
 async def _generate_graph(value: Any) -> AsyncIterator[Any]:
     """Yield a graph object regardless of its type."""
-    if isinstance(value, Pregel | RemotePregel):
+    if isinstance(value, Pregel | BaseRemotePregel):
         yield value
     elif hasattr(value, "__aenter__") and hasattr(value, "__aexit__"):
         async with value as ctx_value:
@@ -94,12 +94,12 @@ async def get_graph(
     async with _generate_graph(value) as graph_obj:
         if isinstance(graph_obj, Graph):
             graph_obj = graph_obj.compile()
-        if not isinstance(graph_obj, Pregel | RemotePregel):
+        if not isinstance(graph_obj, Pregel | BaseRemotePregel):
             raise HTTPException(
                 status_code=424,
                 detail=f"Graph '{graph_id}' is not valid. Review graph registration.",
             )
-        if isinstance(graph_obj, RemotePregel):
+        if isinstance(graph_obj, BaseRemotePregel):
             graph_obj.checkpointer = checkpointer
             graph_obj.name = graph_id
             yield graph_obj
@@ -228,20 +228,13 @@ async def collect_graphs_from_env(register: bool = False) -> None:
             RemotePregel,
             run_js_process,
             run_remote_checkpointer,
-            run_remote_store,
             wait_until_js_ready,
         )
 
         js_bg_tasks.add(
             asyncio.create_task(
                 run_remote_checkpointer(),
-                name="remote-checkpointer",
-            )
-        )
-        js_bg_tasks.add(
-            asyncio.create_task(
-                run_remote_store(),
-                name="remote-store",
+                name="remote-socket-poller",
             )
         )
         js_bg_tasks.add(
@@ -256,7 +249,7 @@ async def collect_graphs_from_env(register: bool = False) -> None:
         await wait_until_js_ready()
 
         for spec in js_specs:
-            graph = await RemotePregel.load(graph_id=spec.id)
+            graph = RemotePregel.load(graph_id=spec.id)
             if register:
                 await register_graph(spec.id, graph, spec.config)
 

langgraph_api/js/base.py

@@ -0,0 +1,9 @@
+from langchain_core.runnables import Runnable
+
+
+class BaseRemotePregel(Runnable):
+    # TODO: implement name overriding
+    name: str = "LangGraph"
+
+    # TODO: implement graph_id overriding
+    graph_id: str

langgraph_api/js/build.mts

@@ -1,3 +1,5 @@
+/// <reference types="./global.d.ts" />
+
 import { z } from "zod";
 import * as fs from "node:fs/promises";
 import * as path from "node:path";