langgraph-api 0.0.14__tar.gz → 0.0.16__tar.gz

{langgraph_api-0.0.14 → langgraph_api-0.0.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.14
+Version: 0.0.16
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -16,11 +16,11 @@ Requires-Dist: jsonschema-rs (>=0.25.0,<0.26.0)
 Requires-Dist: langchain-core (>=0.2.38,<0.4.0)
 Requires-Dist: langgraph (>=0.2.56,<0.3.0)
 Requires-Dist: langgraph-checkpoint (>=2.0.7,<3.0)
-Requires-Dist: langgraph-sdk (>=0.1.48,<0.2.0)
+Requires-Dist: langgraph-sdk (>=0.1.51,<0.2.0)
 Requires-Dist: langsmith (>=0.1.63,<0.3.0)
 Requires-Dist: orjson (>=3.10.1)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)
-Requires-Dist: sse-starlette (>=2.1.0,<3.0.0)
+Requires-Dist: sse-starlette (>=2.1.0,<2.2.0)
 Requires-Dist: starlette (>=0.38.6)
 Requires-Dist: structlog (>=24.4.0,<25.0.0)
 Requires-Dist: tenacity (>=8.3.0,<9.0.0)

{langgraph_api-0.0.14 → langgraph_api-0.0.16}/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-0.0.14 → langgraph_api-0.0.16}/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-0.0.14 → langgraph_api-0.0.16}/langgraph_api/api/store.py

@@ -1,7 +1,12 @@
+from typing import Any
+
+from langgraph_sdk.auth import Auth
 from starlette.responses import Response
 from starlette.routing import BaseRoute
 
+from langgraph_api.auth.custom import handle_event as _handle_event
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
+from langgraph_api.utils import get_auth_ctx
 from langgraph_api.validation import (
     StoreDeleteRequest,
     StoreListNamespacesRequest,
@@ -21,6 +26,24 @@ def _validate_namespace(namespace: tuple[str, ...]) -> Response | None:
             )
 
 
+async def handle_event(
+    action: str,
+    value: Any,
+) -> None:
+    ctx = get_auth_ctx()
+    if not ctx:
+        return
+    await _handle_event(
+        Auth.types.AuthContext(
+            user=ctx.user,
+            permissions=ctx.permissions,
+            resource="store",
+            action=action,
+        ),
+        value,
+    )
+
+
 @retry_db
 async def put_item(request: ApiRequest):
     """Store or update an item."""
@@ -28,9 +51,13 @@ async def put_item(request: ApiRequest):
     namespace = tuple(payload["namespace"]) if payload.get("namespace") else ()
     if err := _validate_namespace(namespace):
         return err
-    key = payload["key"]
-    value = payload["value"]
-    await Store().aput(namespace, key, value)
+    handler_payload = {
+        "namespace": namespace,
+        "key": payload["key"],
+        "value": payload["value"],
+    }
+    await handle_event("put", handler_payload)
+    await Store().aput(namespace, handler_payload["key"], handler_payload["value"])
     return Response(status_code=204)
 
 
@@ -43,6 +70,11 @@ async def get_item(request: ApiRequest):
     key = request.query_params.get("key")
     if not key:
         return ApiResponse({"error": "Key is required"}, status_code=400)
+    handler_payload = {
+        "namespace": namespace,
+        "key": key,
+    }
+    await handle_event("get", handler_payload)
     result = await Store().aget(namespace, key)
     return ApiResponse(result.dict() if result is not None else None)
 
@@ -54,8 +86,12 @@ async def delete_item(request: ApiRequest):
     namespace = tuple(payload["namespace"]) if payload.get("namespace") else ()
     if err := _validate_namespace(namespace):
         return err
-    key = payload["key"]
-    await Store().adelete(namespace, key)
+    handler_payload = {
+        "namespace": namespace,
+        "key": payload["key"],
+    }
+    await handle_event("delete", handler_payload)
+    await Store().adelete(handler_payload["namespace"], handler_payload["key"])
     return Response(status_code=204)
 
 
@@ -70,12 +106,20 @@ async def search_items(request: ApiRequest):
     limit = payload.get("limit") or 10
     offset = payload.get("offset") or 0
     query = payload.get("query")
+    handler_payload = {
+        "namespace": namespace_prefix,
+        "filter": filter,
+        "limit": limit,
+        "offset": offset,
+        "query": query,
+    }
+    await handle_event("search", handler_payload)
     items = await Store().asearch(
-        namespace_prefix,
-        filter=filter,
-        limit=limit,
-        offset=offset,
-        query=query,
+        handler_payload["namespace"],
+        filter=handler_payload["filter"],
+        limit=handler_payload["limit"],
+        offset=handler_payload["offset"],
+        query=handler_payload["query"],
     )
     return ApiResponse({"items": [item.dict() for item in items]})
 
@@ -93,12 +137,20 @@ async def list_namespaces(request: ApiRequest):
     max_depth = payload.get("max_depth")
     limit = payload.get("limit", 100)
     offset = payload.get("offset", 0)
+    handler_payload = {
+        "namespace": prefix,
+        "suffix": suffix,
+        "max_depth": max_depth,
+        "limit": limit,
+        "offset": offset,
+    }
+    await handle_event("list_namespaces", handler_payload)
     result = await Store().alist_namespaces(
-        prefix=prefix,
-        suffix=suffix,
-        max_depth=max_depth,
-        limit=limit,
-        offset=offset,
+        prefix=handler_payload["namespace"],
+        suffix=handler_payload["suffix"],
+        max_depth=handler_payload["max_depth"],
+        limit=handler_payload["limit"],
+        offset=handler_payload["offset"],
     )
     return ApiResponse({"namespaces": result})
 

{langgraph_api-0.0.14 → langgraph_api-0.0.16}/langgraph_api/asyncio.py

@@ -60,6 +60,11 @@ async def wait_if_not_done(coro: Coroutine[Any, Any, T], done: ValueEvent) -> T:
             try:
                 return await coro_task
             except asyncio.CancelledError as e:
+                if e.args and asyncio.isfuture(e.args[-1]):
+                    await logger.ainfo(
+                        "Awaiting future upon cancellation", task=str(e.args[-1])
+                    )
+                    await e.args[-1]
                 if e.args and isinstance(e.args[0], Exception):
                     raise e.args[0] from None
                 raise

{langgraph_api-0.0.14 → langgraph_api-0.0.16}/langgraph_api/auth/custom.py

@@ -398,6 +398,16 @@ class DotDict:
             raise AttributeError(f"'DotDict' object has no attribute '{name}'")
         return self._dict[name]
 
+    def __getitem__(self, key):
+        return self._dict[key]
+
+    def __setitem__(self, key, value):
+        self._dict[key] = value
+        if isinstance(value, dict):
+            setattr(self, key, DotDict(value))
+        else:
+            setattr(self, key, value)
+
     def __deepcopy__(self, memo):
         return DotDict(copy.deepcopy(self._dict))
 
@@ -457,6 +467,12 @@ class ProxyUser(BaseUser):
             **d,
         }
 
+    def __getitem__(self, key):
+        return self._user[key]
+
+    def __setitem__(self, key, value):
+        self._user[key] = value
+
     def __getattr__(self, name: str) -> Any:
         """Proxy any other attributes to the underlying user object."""
         return getattr(self._user, name)
@@ -481,10 +497,10 @@ def _normalize_auth_response(
         user = response
         permissions = []
 
-    return AuthCredentials(permissions), _normalize_user(user)
+    return AuthCredentials(permissions), normalize_user(user)
 
 
-def _normalize_user(user: Any) -> BaseUser:
+def normalize_user(user: Any) -> BaseUser:
     """Normalize user into a BaseUser instance."""
     if isinstance(user, BaseUser):
         return user
@@ -571,9 +587,8 @@ def _get_handler(auth: Auth, ctx: Auth.types.AuthContext) -> Auth.types.Handler
     ]
     for key in keys:
         if key in auth._handlers:
-            result = auth._handlers[key][
-                -1
-            ]  # Get the last defined, most specific handler
+            # Get the last defined, most specific handler
+            result = auth._handlers[key][-1]
             auth._handler_cache[key] = result
             return result
     if auth._global_handlers:

{langgraph_api-0.0.14 → langgraph_api-0.0.16}/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-0.0.14 → langgraph_api-0.0.16}/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-0.0.16/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-0.0.14 → langgraph_api-0.0.16}/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";