langgraph-api 0.2.46__tar.gz → 0.2.51__tar.gz

{langgraph_api-0.2.46 → langgraph_api-0.2.51}/PKG-INFO

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

langgraph_api-0.2.51/langgraph_api/__init__.py

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

{langgraph_api-0.2.46 → langgraph_api-0.2.51}/langgraph_api/api/meta.py

@@ -2,7 +2,7 @@ import os
 
 from starlette.responses import JSONResponse, PlainTextResponse
 
-from langgraph_api import config, metadata
+from langgraph_api import __version__, config, metadata
 from langgraph_api.route import ApiRequest
 from langgraph_license.validation import plus_features_enabled
 from langgraph_runtime.database import connect, pool_stats
@@ -16,10 +16,12 @@ async def meta_info(request: ApiRequest):
     plus = plus_features_enabled()
     return JSONResponse(
         {
+            "version": __version__,
             "flags": {
                 "assistants": True,
                 "crons": plus and config.FF_CRONS_ENABLED,
                 "langsmith": bool(config.LANGSMITH_API_KEY) and bool(config.TRACING),
+                "langsmith_tracing_replicas": True,
             },
             "host": {
                 "kind": metadata.HOST,

{langgraph_api-0.2.46 → langgraph_api-0.2.51}/langgraph_api/asyncio.py

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

{langgraph_api-0.2.46 → langgraph_api-0.2.51}/langgraph_api/middleware/http_logger.py

@@ -5,8 +5,6 @@ import structlog
 from starlette.requests import ClientDisconnect
 from starlette.types import Message, Receive, Scope, Send
 
-from langgraph_api.logging import LOG_JSON
-
 asgi = structlog.stdlib.get_logger("asgi")
 
 PATHS_IGNORE = {"/ok", "/metrics"}
@@ -28,7 +26,7 @@ class AccessLoggerMiddleware:
             self.debug_enabled = False
 
     async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
-        if scope["type"] != "http" or (LOG_JSON and scope.get("path") in PATHS_IGNORE):
+        if scope["type"] != "http" or scope.get("path") in PATHS_IGNORE:
             return await self.app(scope, receive, send)  # pragma: no cover
 
         loop = asyncio.get_event_loop()

{langgraph_api-0.2.46 → langgraph_api-0.2.51}/langgraph_api/middleware/request_id.py

@@ -21,6 +21,8 @@ class RequestIdMiddleware:
 
     async def __call__(self, scope: Scope, receive: Receive, send: Send):
         if scope["type"] == "http" and self.pattern.match(scope["path"]):
+            from langgraph_api.logging import set_logging_context
+
             request_id = next(
                 (h[1] for h in scope["headers"] if h[0] == b"x-request-id"),
                 None,
@@ -29,4 +31,5 @@ class RequestIdMiddleware:
                 request_id = str(uuid.uuid4()).encode()
                 scope["headers"].append((b"x-request-id", request_id))
             scope["request_start_time_ms"] = int(time.time() * 1000)
+            set_logging_context({"request_id": request_id.decode()})
         await self.app(scope, receive, send)

{langgraph_api-0.2.46 → langgraph_api-0.2.51}/langgraph_api/stream.py

@@ -24,6 +24,7 @@ from langsmith.utils import get_tracer_project
 from pydantic import ValidationError
 from pydantic.v1 import ValidationError as ValidationErrorLegacy
 
+from langgraph_api import __version__
 from langgraph_api import store as api_store
 from langgraph_api.asyncio import ValueEvent, wait_if_not_done
 from langgraph_api.command import map_cmd
@@ -118,6 +119,7 @@ async def astream_state(
     config["metadata"]["run_attempt"] = attempt
     # attach langgraph metadata
     config["metadata"]["langgraph_version"] = langgraph.version.__version__
+    config["metadata"]["langgraph_api_version"] = __version__
     config["metadata"]["langgraph_plan"] = PLAN
     config["metadata"]["langgraph_host"] = HOST
     config["metadata"]["langgraph_api_url"] = USER_API_URL

{langgraph_api-0.2.46 → langgraph_api-0.2.51}/langgraph_api/worker.py

@@ -178,11 +178,16 @@ async def worker(
                         else None
                     ),
                 )
-        except Exception as ee:
-            # Note we don't handle asyncio.CancelledError here, as we want to
-            # let it bubble up and rollback db transaction, thus marking the run
-            # as available to be picked up by another worker
+        except (Exception, asyncio.CancelledError) as ee:
             exception = ee
+        except BaseException as eee:
+            await logger.aerror(
+                "Bubbling failed background run",
+                run_id=str(run_id),
+                exception_type=str(type(eee)),
+                exception=str(eee),
+            )
+            raise
 
         # handle exceptions and set status
         async with connect() as conn:
@@ -286,7 +291,7 @@ async def worker(
                 run_ended_at_dt = datetime.now(UTC)
                 run_ended_at = run_ended_at_dt.isoformat()
                 await logger.awarning(
-                    f"Background run failed, will retry. Exception: {exception}",
+                    f"Background run failed, will retry. Exception: {type(exception)}({exception})",
                     exc_info=True,
                     run_id=str(run_id),
                     run_attempt=attempt,
@@ -301,7 +306,7 @@ async def worker(
                 run_ended_at_dt = datetime.now(UTC)
                 run_ended_at = run_ended_at_dt.isoformat()
                 await logger.aexception(
-                    f"Background run failed. Exception: {exception}",
+                    f"Background run failed. Exception: {type(exception)}({exception})",
                     exc_info=not isinstance(exception, RemoteException),
                     run_id=str(run_id),
                     run_attempt=attempt,

{langgraph_api-0.2.46 → langgraph_api-0.2.51}/openapi.json

@@ -4703,6 +4703,14 @@
             "default": 0,
             "title": "Offset",
             "description": "Number of items to skip before returning results (default is 0)."
+          },
+          "query": {
+            "type": [
+              "string",
+              "null"
+            ],
+            "title": "Query",
+            "description": "Query string for semantic/vector search."
           }
         },
         "title": "StoreSearchRequest",

langgraph_api-0.2.46/langgraph_api/__init__.py

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