Add stream_subgraphs param

langchain-ai · Sep 20, 2024 · 41cb547 · 41cb547
1 parent eadcec7
commit 41cb547
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/libs/sdk-py/langgraph_sdk/client.py b/libs/sdk-py/langgraph_sdk/client.py
@@ -961,6 +961,7 @@ def stream(
         *,
         input: Optional[dict] = None,
         stream_mode: Union[StreamMode, list[StreamMode]] = "values",
+        stream_subgraphs: bool = False,
         metadata: Optional[dict] = None,
         config: Optional[Config] = None,
         checkpoint_id: Optional[str] = None,
@@ -981,6 +982,7 @@ def stream(
         *,
         input: Optional[dict] = None,
         stream_mode: Union[StreamMode, list[StreamMode]] = "values",
+        stream_subgraphs: bool = False,
         metadata: Optional[dict] = None,
         config: Optional[Config] = None,
         interrupt_before: Optional[list[str]] = None,
@@ -999,6 +1001,7 @@ def stream(
         *,
         input: Optional[dict] = None,
         stream_mode: Union[StreamMode, list[StreamMode]] = "values",
+        stream_subgraphs: bool = False,
         metadata: Optional[dict] = None,
         config: Optional[Config] = None,
         checkpoint_id: Optional[str] = None,
@@ -1020,6 +1023,7 @@ def stream(
                 If using graph name, will default to first assistant created from that graph.
             input: The input to the graph.
             stream_mode: The stream mode(s) to use.
+            stream_subgraphs: Whether to stream output from subgraphs.
             metadata: Metadata to assign to the run.
             config: The configuration for the assistant.
             checkpoint_id: The checkpoint to start streaming from.
@@ -1072,6 +1076,7 @@ def stream(
             "config": config,
             "metadata": metadata,
             "stream_mode": stream_mode,
+            "stream_subgraphs": stream_subgraphs,
             "assistant_id": assistant_id,
             "interrupt_before": interrupt_before,
             "interrupt_after": interrupt_after,
@@ -1099,6 +1104,8 @@ async def create(
         assistant_id: str,
         *,
         input: Optional[dict] = None,
+        stream_mode: Union[StreamMode, list[StreamMode]] = "values",
+        stream_subgraphs: bool = False,
         metadata: Optional[dict] = None,
         config: Optional[Config] = None,
         interrupt_before: Optional[list[str]] = None,
@@ -1115,6 +1122,8 @@ async def create(
         assistant_id: str,
         *,
         input: Optional[dict] = None,
+        stream_mode: Union[StreamMode, list[StreamMode]] = "values",
+        stream_subgraphs: bool = False,
         metadata: Optional[dict] = None,
         config: Optional[Config] = None,
         checkpoint_id: Optional[str] = None,
@@ -1131,6 +1140,8 @@ async def create(
         assistant_id: str,
         *,
         input: Optional[dict] = None,
+        stream_mode: Union[StreamMode, list[StreamMode]] = "values",
+        stream_subgraphs: bool = False,
         metadata: Optional[dict] = None,
         config: Optional[Config] = None,
         checkpoint_id: Optional[str] = None,
@@ -1149,6 +1160,8 @@ async def create(
             assistant_id: The assistant ID or graph name to stream from.
                 If using graph name, will default to first assistant created from that graph.
             input: The input to the graph.
+            stream_mode: The stream mode(s) to use.
+            stream_subgraphs: Whether to stream output from subgraphs.
             metadata: Metadata to assign to the run.
             config: The configuration for the assistant.
             checkpoint_id: The checkpoint to start streaming from.
@@ -1232,6 +1245,8 @@ async def create(
         """  # noqa: E501
         payload = {
             "input": input,
+            "stream_mode": stream_mode,
+            "stream_subgraphs": stream_subgraphs,
             "config": config,
             "metadata": metadata,
             "assistant_id": assistant_id,