langgraph-api 0.2.61__tar.gz → 0.2.63__tar.gz

{langgraph_api-0.2.61 → langgraph_api-0.2.63}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.2.61
+Version: 0.2.63
 Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE

langgraph_api-0.2.63/langgraph_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.63"

{langgraph_api-0.2.61 → langgraph_api-0.2.63}/langgraph_api/api/meta.py

@@ -48,31 +48,45 @@ async def meta_metrics(request: ApiRequest):
 
     if metrics_format == "json":
         async with connect() as conn:
-            return JSONResponse(
-                {
-                    **pool_stats(),
-                    "workers": worker_metrics,
-                    "queue": await Runs.stats(conn),
-                }
-            )
+            resp = {
+                **pool_stats(),
+                "queue": await Runs.stats(conn),
+            }
+            if config.N_JOBS_PER_WORKER > 0:
+                resp["workers"] = worker_metrics
+            return JSONResponse(resp)
     elif metrics_format == "prometheus":
         # LANGSMITH_HOST_PROJECT_ID and LANGSMITH_HOST_REVISION_ID are injected
         # into the deployed image by host-backend.
         project_id = os.getenv("LANGSMITH_HOST_PROJECT_ID")
         revision_id = os.getenv("LANGSMITH_HOST_REVISION_ID")
 
-        metrics = [
-            "# HELP lg_api_workers_max The maximum number of workers available.",
-            "# TYPE lg_api_workers_max gauge",
-            f'lg_api_workers_max{{project_id="{project_id}", revision_id="{revision_id}"}} {workers_max}',
-            "# HELP lg_api_workers_active The number of currently active workers.",
-            "# TYPE lg_api_workers_active gauge",
-            f'lg_api_workers_active{{project_id="{project_id}", revision_id="{revision_id}"}} {workers_active}',
-            "# HELP lg_api_workers_available The number of available (idle) workers.",
-            "# TYPE lg_api_workers_available gauge",
-            f'lg_api_workers_available{{project_id="{project_id}", revision_id="{revision_id}"}} {workers_available}',
-            # In the future, we can add more metrics to be scraped by Prometheus.
-        ]
+        async with connect() as conn:
+            queue_stats = await Runs.stats(conn)
+
+            metrics = [
+                "# HELP lg_api_num_pending_runs The number of runs currently pending.",
+                "# TYPE lg_api_num_pending_runs gauge",
+                f'lg_api_num_pending_runs{{project_id="{project_id}", revision_id="{revision_id}"}} {queue_stats["n_pending"]}',
+                "# HELP lg_api_num_running_runs The number of runs currently running.",
+                "# TYPE lg_api_num_running_runs gauge",
+                f'lg_api_num_running_runs{{project_id="{project_id}", revision_id="{revision_id}"}} {queue_stats["n_running"]}',
+            ]
+
+            if config.N_JOBS_PER_WORKER > 0:
+                metrics.extend(
+                    [
+                        "# HELP lg_api_workers_max The maximum number of workers available.",
+                        "# TYPE lg_api_workers_max gauge",
+                        f'lg_api_workers_max{{project_id="{project_id}", revision_id="{revision_id}"}} {workers_max}',
+                        "# HELP lg_api_workers_active The number of currently active workers.",
+                        "# TYPE lg_api_workers_active gauge",
+                        f'lg_api_workers_active{{project_id="{project_id}", revision_id="{revision_id}"}} {workers_active}',
+                        "# HELP lg_api_workers_available The number of available (idle) workers.",
+                        "# TYPE lg_api_workers_available gauge",
+                        f'lg_api_workers_available{{project_id="{project_id}", revision_id="{revision_id}"}} {workers_available}',
+                    ]
+                )
 
         metrics_response = "\n".join(metrics)
         return PlainTextResponse(metrics_response)

langgraph_api-0.2.63/langgraph_api/auth/studio_user.py

@@ -0,0 +1,15 @@
+from langgraph_sdk.auth.types import StudioUser as StudioUserBase
+from starlette.authentication import BaseUser
+
+
+class StudioUser(StudioUserBase, BaseUser):
+    """StudioUser class."""
+
+    def dict(self):
+        return {
+            "kind": "StudioUser",
+            "is_authenticated": self.is_authenticated,
+            "display_name": self.display_name,
+            "identity": self.identity,
+            "permissions": self.permissions,
+        }

{langgraph_api-0.2.61 → langgraph_api-0.2.63}/langgraph_api/js/remote.py

@@ -858,6 +858,16 @@ async def handle_js_auth_event(
     ctx: Auth.types.AuthContext | None,
     value: dict,
 ) -> Auth.types.FilterType | None:
+    if hasattr(ctx.user, "dict") and callable(ctx.user.dict):
+        user = ctx.user.dict()
+    else:
+        user = {
+            "is_authenticated": ctx.user.is_authenticated,
+            "display_name": ctx.user.display_name,
+            "identity": ctx.user.identity,
+            "permissions": ctx.user.permissions,
+        }
+
     res = await _client.post(
         "/auth/authorize",
         headers={"Content-Type": "application/json"},
@@ -868,7 +878,7 @@ async def handle_js_auth_event(
                 "value": value,
                 "context": (
                     {
-                        "user": cast(DotDict, ctx.user).dict(),
+                        "user": user,
                         "scopes": ctx.permissions,
                     }
                     if ctx

langgraph_api-0.2.61/langgraph_api/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.2.61"

langgraph_api-0.2.61/langgraph_api/auth/studio_user.py

@@ -1,6 +0,0 @@
-from langgraph_sdk.auth.types import StudioUser as StudioUserBase
-from starlette.authentication import BaseUser
-
-
-class StudioUser(StudioUserBase, BaseUser):
-    """StudioUser class."""