langchain: allow runnables as prompt

langchain-ai · Jan 8, 2025 · 509e9a9 · 509e9a9
1 parent fcc9cdd
commit 509e9a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/langchain/langchain/chains/combine_documents/stuff.py b/libs/langchain/langchain/chains/combine_documents/stuff.py
@@ -5,7 +5,7 @@
 from langchain_core._api import deprecated
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
-from langchain_core.language_models import LanguageModelLike
+from langchain_core.language_models import LanguageModelInput, LanguageModelLike
 from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.prompts import BasePromptTemplate, format_document
 from langchain_core.runnables import Runnable, RunnablePassthrough
@@ -23,7 +23,7 @@
 
 def create_stuff_documents_chain(
     llm: LanguageModelLike,
-    prompt: BasePromptTemplate,
+    prompt: Runnable[dict, LanguageModelInput],
     *,
     output_parser: Optional[BaseOutputParser] = None,
     document_prompt: Optional[BasePromptTemplate] = None,