Move concatSupported to RunnableConfig and handle when it is false

langchain-ai · Nov 30, 2024 · b605406 · b605406
1 parent 983acd2
commit b605406
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/langchain-core/src/runnables/base.ts b/langchain-core/src/runnables/base.ts
@@ -1989,7 +1989,7 @@ export class RunnableSequence<
       otherOptions?.runName
     );
     const steps = [this.first, ...this.middle, this.last];
-    let concatSupported = true;
+    let concatSupported = options?.concatSupported ?? true;
     let finalOutput;
     async function* inputGenerator() {
       yield input;
@@ -2029,6 +2029,8 @@ export class RunnableSequence<
               concatSupported = false;
             }
           }
+        } else {
+          finalOutput = chunk;
         }
       }
     } catch (e) {

diff --git a/langchain-core/src/runnables/types.ts b/langchain-core/src/runnables/types.ts
@@ -107,4 +107,9 @@ export interface RunnableConfig<
    * @see https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal
    */
   signal?: AbortSignal;
+
+  /**
+   * Whether to concatenate the stream of outputs into a single output or not.
+   */
+  concatSupported?: boolean;
 }