langgraph-api 0.2.42__tar.gz → 0.2.44__tar.gz

{langgraph_api-0.2.42 → langgraph_api-0.2.44}/PKG-INFO

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

langgraph_api-0.2.44/langgraph_api/__init__.py

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

{langgraph_api-0.2.42 → langgraph_api-0.2.44}/langgraph_api/config.py

@@ -241,6 +241,9 @@ BG_JOB_ISOLATED_LOOPS = env("BG_JOB_ISOLATED_LOOPS", cast=bool, default=False)
 BG_JOB_SHUTDOWN_GRACE_PERIOD_SECS = env(
     "BG_JOB_SHUTDOWN_GRACE_PERIOD_SECS", cast=int, default=3600
 )
+MAX_STREAM_CHUNK_SIZE_BYTES = env(
+    "MAX_STREAM_CHUNK_SIZE_BYTES", cast=int, default=1024 * 1024 * 128
+)
 
 
 def _parse_thread_ttl(value: str | None) -> ThreadTTLConfig | None:

{langgraph_api-0.2.42 → langgraph_api-0.2.44}/langgraph_api/logging.py

@@ -81,23 +81,6 @@ class JSONRenderer:
 LEVELS = logging.getLevelNamesMapping()
 
 
-class TapForMetadata:
-    def __call__(
-        self, logger: logging.Logger, method_name: str, event_dict: EventDict
-    ) -> str:
-        """
-        Tap WARN and above logs for metadata. Exclude user loggers.
-        """
-        if (
-            event_dict["logger"].startswith("langgraph")
-            and LEVELS[event_dict["level"].upper()] > LEVELS["INFO"]
-        ):
-            from langgraph_api.metadata import append_log
-
-            append_log(event_dict.copy())
-        return event_dict
-
-
 # shared config, for both logging and structlog
 
 shared_processors = [
@@ -136,7 +119,6 @@ class Formatter(structlog.stdlib.ProcessorFormatter):
         super().__init__(
             processors=[
                 structlog.stdlib.ProcessorFormatter.remove_processors_meta,
-                TapForMetadata(),
                 renderer,
             ],
             foreign_pre_chain=shared_processors,

{langgraph_api-0.2.42 → langgraph_api-0.2.44}/langgraph_api/metadata.py

@@ -35,7 +35,6 @@ else:
 PLAN = "enterprise" if plus_features_enabled() else "developer"
 USER_API_URL = os.getenv("LANGGRAPH_API_URL", None)
 
-LOGS: list[dict] = []
 RUN_COUNTER = 0
 NODE_COUNTER = 0
 FROM_TIMESTAMP = datetime.now(UTC).isoformat()
@@ -59,14 +58,6 @@ def incr_nodes(_, *, incr: int = 1) -> None:
     NODE_COUNTER += incr
 
 
-def append_log(log: dict) -> None:
-    if not LANGGRAPH_CLOUD_LICENSE_KEY and not LANGSMITH_API_KEY:
-        return
-
-    global LOGS
-    LOGS.append(log)
-
-
 async def metadata_loop() -> None:
     try:
         from langgraph_api import __version__
@@ -91,8 +82,6 @@ async def metadata_loop() -> None:
         to_timestamp = datetime.now(UTC).isoformat()
         nodes = NODE_COUNTER
         runs = RUN_COUNTER
-        logs = LOGS.copy()
-        LOGS.clear()
         RUN_COUNTER = 0
         NODE_COUNTER = 0
         FROM_TIMESTAMP = to_timestamp
@@ -123,7 +112,7 @@ async def metadata_loop() -> None:
                 "langgraph.platform.runs": runs,
                 "langgraph.platform.nodes": nodes,
             },
-            "logs": logs,
+            "logs": [],
         }
         try:
             await http_request(

{langgraph_api-0.2.42 → langgraph_api-0.2.44}/langgraph_api/stream.py

@@ -207,11 +207,16 @@ async def astream_state(
                 elif "events" in stream_mode:
                     yield "events", event
     else:
+        output_keys = kwargs.pop("output_keys", graph.output_channels)
         async with (
             stack,
             aclosing(
                 graph.astream(
-                    input, config, stream_mode=list(stream_modes_set), **kwargs
+                    input,
+                    config,
+                    stream_mode=list(stream_modes_set),
+                    output_keys=output_keys,
+                    **kwargs,
                 )
             ) as stream,
         ):

{langgraph_api-0.2.42 → langgraph_api-0.2.44}/langgraph_api/worker.py

@@ -90,11 +90,11 @@ async def worker(
         if request_created_at is not None
         else None
     )
-
     async with (
         connect() as conn,
         set_auth_ctx_for_run(run["kwargs"]),
         Runs.enter(run_id, main_loop) as done,
+        AsyncExitStack() as exit_stack,
     ):
         temporary = run["kwargs"].get("temporary", False)
         resumable = run["kwargs"].get("resumable", False)
@@ -156,12 +156,10 @@ async def worker(
 
                 raise RuntimeError(error_message)
             if temporary:
-                stream = astream_state(
-                    AsyncExitStack(), conn, cast(Run, run), attempt, done
-                )
+                stream = astream_state(exit_stack, conn, cast(Run, run), attempt, done)
             else:
                 stream = astream_state(
-                    AsyncExitStack(),
+                    exit_stack,
                     conn,
                     cast(Run, run),
                     attempt,
@@ -216,6 +214,8 @@ async def worker(
             run_ended_at_dt = datetime.now(UTC)
             run_ended_at = run_ended_at_dt.isoformat()
             try:
+                # Need to close the pipeline to re-use the connection
+                await exit_stack.aclose()
                 await Runs.delete(conn, run_id, thread_id=run["thread_id"])
                 await logger.ainfo(
                     "Background run rolled back",
@@ -240,9 +240,6 @@ async def worker(
                     run_created_at=run_created_at,
                     exc=str(e),
                 )
-                # We need to clean up the transaction early if we want to
-                # update the thread status with the same connection
-                await exit.aclose()
             except HTTPException as e:
                 if e.status_code == 404:
                     await logger.ainfo(

langgraph_api-0.2.42/langgraph_api/__init__.py

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