langgraph-api 0.2.42__py3-none-any.whl → 0.2.44__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.42"
+__version__ = "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/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/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/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/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.dist-info → langgraph_api-0.2.44.dist-info}/METADATA

@@ -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.42.dist-info → langgraph_api-0.2.44.dist-info}/RECORD

@@ -1,15 +1,15 @@
-langgraph_api/__init__.py,sha256=ZfPaHT9jrZeNvdPfDssvO9QlJXdREiFfpk5CG-mKeQQ,23
+langgraph_api/__init__.py,sha256=U9IXTG0wwP5qkavLVbDEmAfwEJjO-1-R5e__z2cNpz0,23
 langgraph_api/asgi_transport.py,sha256=eqifhHxNnxvI7jJqrY1_8RjL4Fp9NdN4prEub2FWBt8,5091
 langgraph_api/asyncio.py,sha256=nelZwKL7iOjM5GHj1rVjiPE7igUIKLNKtc-3urxmlfo,9250
 langgraph_api/cli.py,sha256=9Ou3tGDDY_VVLt5DFle8UviJdpI4ZigC5hElYvq2-To,14519
 langgraph_api/command.py,sha256=3O9v3i0OPa96ARyJ_oJbLXkfO8rPgDhLCswgO9koTFA,768
-langgraph_api/config.py,sha256=DqINcyfxGD5gT0zH6XTZMpx_ujT3G5xmo74esfnqb_0,11007
+langgraph_api/config.py,sha256=pFIZb4t2Vo7HbX0ZMjUNDR7Q7Bpj-stp_TmSmI026yo,11115
 langgraph_api/cron_scheduler.py,sha256=i87j4pJrcsmsqMKeKUs69gaAjrGaSM3pM3jnXdN5JDQ,2630
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
 langgraph_api/graph.py,sha256=MPm8DvInBQsq2em45c2YD5bW6T_G1LlDkAuWq-19gCQ,23240
 langgraph_api/http.py,sha256=gYbxxjY8aLnsXeJymcJ7G7Nj_yToOGpPYQqmZ1_ggfA,5240
-langgraph_api/logging.py,sha256=VJ-lPwHK6UY2ZcK8LQsYhu7h0zODfVov1laxzyujRcU,4490
-langgraph_api/metadata.py,sha256=ptaxwmzdx2bUBSc1KRhqgF-Xnm-Zh2gqwSiHpl8LD9c,4482
+langgraph_api/logging.py,sha256=bx6kxV7O2FUnu0ubrgeJuj5q2EjYC4ve1BkbrVfzy28,3953
+langgraph_api/metadata.py,sha256=2sz9ECnbnQtgqN6ptDkRmymaVKfQPgaX-JuDJDJB47c,4254
 langgraph_api/patch.py,sha256=Dgs0PXHytekX4SUL6KsjjN0hHcOtGLvv1GRGbh6PswU,1408
 langgraph_api/queue_entrypoint.py,sha256=_41ZveMDdn9bapjA7Ik9FG3r4hyIwXESUM5F1PdlieE,2309
 langgraph_api/route.py,sha256=4VBkJMeusfiZtLzyUaKm1HwLHTq0g15y2CRiRhM6xyA,4773
@@ -19,12 +19,12 @@ langgraph_api/server.py,sha256=Z_VL-kIphybTRDWBIqHMfRhgCmAFyTRqAGlgnHQF0Zg,6973
 langgraph_api/sse.py,sha256=F7swfjKBDrlUmXZ_dWuDVHtp-3o1Cpjq1lwp0bJD-nw,4223
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
 langgraph_api/store.py,sha256=_xGhdwEIMoY1_hIy_oWwxZp4Y7FH833BNJfgFIhT80E,4640
-langgraph_api/stream.py,sha256=T1tkUZFr5KlgqvE-QHEh-m80mWAgp2MUT9gFD7HLen0,12670
+langgraph_api/stream.py,sha256=0qLblAQ406bGZEM1zRQKVyU407c9PjcBcgPCKN78ywc,12847
 langgraph_api/thread_ttl.py,sha256=-Ox8NFHqUH3wGNdEKMIfAXUubY5WGifIgCaJ7npqLgw,1762
 langgraph_api/utils.py,sha256=92mSti9GfGdMRRWyESKQW5yV-75Z9icGHnIrBYvdypU,3619
 langgraph_api/validation.py,sha256=zMuKmwUEBjBgFMwAaeLZmatwGVijKv2sOYtYg7gfRtc,4950
 langgraph_api/webhook.py,sha256=1ncwO0rIZcj-Df9sxSnFEzd1gP1bfS4okeZQS8NSRoE,1382
-langgraph_api/worker.py,sha256=sdUKXr19CSMaptg_3cvzUAy4KOUme1ajwSBt1dvSbsg,14587
+langgraph_api/worker.py,sha256=YjMadekwt8t7Mn-pxZe8H-SJfetT31bl3MO2WFGVzBg,14510
 langgraph_api/api/__init__.py,sha256=YVzpbn5IQotvuuLG9fhS9QMrxXfP4s4EpEMG0n4q3Nw,5625
 langgraph_api/api/assistants.py,sha256=x_V1rnSGFYjNZFJkZKFN9yNFOqXhqkOSqMDSv3I8VeE,15880
 langgraph_api/api/mcp.py,sha256=RvRYgANqRzNQzSmgjNkq4RlKTtoEJYil04ot9lsmEtE,14352
@@ -78,8 +78,8 @@ langgraph_runtime/__init__.py,sha256=O4GgSmu33c-Pr8Xzxj_brcK5vkm70iNTcyxEjICFZxA
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=_4GFDqbq1X9vD4_FxwahuVODJMOHx-U76gkY4rdy3DA,138189
-langgraph_api-0.2.42.dist-info/METADATA,sha256=GK8GedSLk6Bg8NPQhNfQAodCid1rz3C4IsQ8a2aEXGk,3892
-langgraph_api-0.2.42.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langgraph_api-0.2.42.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.2.42.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.2.42.dist-info/RECORD,,
+langgraph_api-0.2.44.dist-info/METADATA,sha256=9OXLai4MqakDiva3wfVGk3vmX4akdCJ1zKuvNpl62t4,3892
+langgraph_api-0.2.44.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langgraph_api-0.2.44.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.2.44.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.2.44.dist-info/RECORD,,