Fix combining runnable sequences (#8557)

Combining runnable sequences was dropping a step in the middle. @nfcampos @baskaryan
langchain-ai · Aug 1, 2023 · 2a26cc6 · 2a26cc6
1 parent 3fbb737
commit 2a26cc6
Show file tree

Hide file tree

Showing 3 changed files with 511 additions and 2 deletions.
diff --git a/libs/langchain/langchain/schema/runnable.py b/libs/langchain/langchain/schema/runnable.py
@@ -214,7 +214,7 @@ def __or__(
         if isinstance(other, RunnableSequence):
             return RunnableSequence(
                 first=self.first,
-                middle=self.middle + [self.last] + other.middle,
+                middle=self.middle + [self.last] + [other.first] + other.middle,
                 last=other.last,
             )
         else:
@@ -235,7 +235,7 @@ def __ror__(
         if isinstance(other, RunnableSequence):
             return RunnableSequence(
                 first=other.first,
-                middle=other.middle + [other.last] + self.middle,
+                middle=other.middle + [other.last] + [self.first] + self.middle,
                 last=self.last,
             )
         else: