langgraph-api 0.13.0.dev2__tar.gz → 0.13.0.dev4__tar.gz

{langgraph_api-0.13.0.dev2 → langgraph_api-0.13.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.13.0.dev2
+Version: 0.13.0.dev4
 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.13.0.dev4/langgraph_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.13.0.dev4"

{langgraph_api-0.13.0.dev2 → langgraph_api-0.13.0.dev4}/langgraph_api/api/assistants.py

@@ -311,29 +311,30 @@ async def get_assistant_graph(
     configurable = config.setdefault("configurable", {})
     configurable.update(get_configurable_headers(request.headers))
 
+    xray: bool | int = False
+    xray_query = request.query_params.get("xray")
+    if xray_query:
+        if xray_query in ("true", "True"):
+            xray = True
+        elif xray_query in ("false", "False"):
+            xray = False
+        else:
+            try:
+                xray = int(xray_query)
+            except ValueError:
+                raise HTTPException(422, detail="Invalid xray value") from None
+
+            if xray <= 0:
+                raise HTTPException(422, detail="Invalid xray value") from None
+
     async with get_graph(
         assistant["graph_id"],
         config,
         checkpointer=(await api_checkpointer.get_checkpointer()),
         store=(await api_store.get_store()),
         access_context="assistants.read",
+        use_langgraph_runtime=bool(xray),
     ) as graph:
-        xray: bool | int = False
-        xray_query = request.query_params.get("xray")
-        if xray_query:
-            if xray_query in ("true", "True"):
-                xray = True
-            elif xray_query in ("false", "False"):
-                xray = False
-            else:
-                try:
-                    xray = int(xray_query)
-                except ValueError:
-                    raise HTTPException(422, detail="Invalid xray value") from None
-
-                if xray <= 0:
-                    raise HTTPException(422, detail="Invalid xray value") from None
-
         if isinstance(graph, BaseRemotePregel):
             drawable_graph = await graph.fetch_graph(xray=xray)
             json_graph = drawable_graph.to_json()

{langgraph_api-0.13.0.dev2 → langgraph_api-0.13.0.dev4}/langgraph_api/graph.py

@@ -357,6 +357,7 @@ async def get_graph(
     access_context: AccessContext,
     run_id: str | None = None,
     context: Any | None = None,
+    use_langgraph_runtime: bool = False,
 ) -> AsyncIterator[Pregel]:
     """Return the runnable."""
     from langgraph_api.utils import config as lg_config  # noqa: PLC0415
@@ -373,7 +374,13 @@ async def get_graph(
                 CONFIG_KEY_RUNTIME,
             )
 
-            config["configurable"][CONFIG_KEY_RUNTIME] = server_runtime
+            if use_langgraph_runtime:
+                # xray draw calls runtime.override(), which the SDK ServerRuntime lacks (LSD-1713).
+                from langgraph.runtime import Runtime  # noqa: PLC0415
+
+                config["configurable"][CONFIG_KEY_RUNTIME] = Runtime(store=store)
+            else:
+                config["configurable"][CONFIG_KEY_RUNTIME] = server_runtime
         elif store is not None:
             from langgraph.constants import CONFIG_KEY_STORE  # noqa: PLC0415
 

{langgraph_api-0.13.0.dev2 → langgraph_api-0.13.0.dev4}/langgraph_api/queue_entrypoint.py

@@ -19,7 +19,6 @@ from contextlib import suppress
 
 import structlog
 
-from langgraph_api.api.meta import meta_pool_stats
 from langgraph_api.utils.errors import GraphLoadError, HealthServerStartupError
 from langgraph_api.utils.network import format_hostport, normalize_host
 from langgraph_runtime import lifespan
@@ -72,7 +71,7 @@ async def health_and_metrics_server():
     from starlette.routing import Mount, Route  # noqa: PLC0415
 
     from langgraph_api import config as lc_config  # noqa: PLC0415
-    from langgraph_api.api.meta import METRICS_FORMATS  # noqa: PLC0415
+    from langgraph_api.api.meta import METRICS_FORMATS, meta_pool_stats  # noqa: PLC0415
 
     port = int(os.getenv("PORT", "8080"))
     # Not in public docs: LANGGRAPH_SERVER_HOST is internal
@@ -298,9 +297,14 @@ async def main(entrypoint_name: str = "python-queue"):
 
 
 if __name__ == "__main__":
+    # Set before any langgraph_api.api import. That package's __init__ pulls in
+    # auth.middleware, which builds the auth backend at import time. Custom JS
+    # auth then imports js.remote and freezes REMOTE_PORT from this flag; if we
+    # set it too late the colocated queue worker binds 5555 instead of 5565.
     from langgraph_api import config
 
     config.IS_QUEUE_ENTRYPOINT = True
+
     with open(pathlib.Path(__file__).parent.parent / "logging.json") as file:
         loaded_config = json.load(file)
         logging.config.dictConfig(loaded_config)

{langgraph_api-0.13.0.dev2 → langgraph_api-0.13.0.dev4}/pyproject.toml

@@ -147,6 +147,7 @@ path = "langgraph_api/__init__.py"
 #                       section of the configuration file raise errors.
 # --snapshot-warn-unused   by default pytest will fail if any snapshots are unused, but we have a bunch of branching
 addopts = "--strict-markers --strict-config --durations=5 -vv --snapshot-warn-unused"
+pythonpath = ["../.github/scripts"]
 timeout = 30
 markers = [
     "grpc: include test in gRPC persistence test suite",

langgraph_api-0.13.0.dev2/langgraph_api/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.13.0.dev2"