RunnableWithMessageHistory: Fix input schema (langchain-ai#14516)

Input schema should not have history key
0xken · Dec 11, 2023 · c0f4b95 · c0f4b95
1 parent d9bfdc9
commit c0f4b95
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/libs/core/langchain_core/runnables/history.py b/libs/core/langchain_core/runnables/history.py
@@ -197,8 +197,6 @@ def get_input_schema(
                 fields[self.input_messages_key] = (Sequence[BaseMessage], ...)
             else:
                 fields["__root__"] = (Sequence[BaseMessage], ...)
-            if self.history_messages_key:
-                fields[self.history_messages_key] = (Sequence[BaseMessage], ...)
             return create_model(  # type: ignore[call-overload]
                 "RunnableWithChatHistoryInput",
                 **fields,

diff --git a/libs/core/tests/unit_tests/runnables/test_history.py b/libs/core/tests/unit_tests/runnables/test_history.py
@@ -178,7 +178,6 @@ def test_output_dict() -> None:
 def test_get_input_schema_input_dict() -> None:
     class RunnableWithChatHistoryInput(BaseModel):
         input: Union[str, BaseMessage, Sequence[BaseMessage]]
-        history: Sequence[BaseMessage]
 
     runnable = RunnableLambda(
         lambda input: {