github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-25 08:03:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain: allow runnables as prompt

Browse Source
This commit is contained in:
Erick Friis 2025-01-08 10:35:45 -08:00
parent fcc9cdd100
commit 509e9a9821
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/chains/combine_documents/stuff.py
View File

@ -5,7 +5,7 @@ from typing import Any, Dict, List, Optional, Tuple
from langchain_core._api import deprecated from langchain_core._api import deprecated
from langchain_core.callbacks import Callbacks from langchain_core.callbacks import Callbacks
from langchain_core.documents import Document 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.output_parsers import BaseOutputParser, StrOutputParser
from langchain_core.prompts import BasePromptTemplate, format_document from langchain_core.prompts import BasePromptTemplate, format_document
from langchain_core.runnables import Runnable, RunnablePassthrough from langchain_core.runnables import Runnable, RunnablePassthrough
@ -23,7 +23,7 @@ from langchain.chains.llm import LLMChain
def create_stuff_documents_chain( def create_stuff_documents_chain(
llm: LanguageModelLike, llm: LanguageModelLike,
prompt: BasePromptTemplate, prompt: Runnable[dict, LanguageModelInput],
*, *,
output_parser: Optional[BaseOutputParser] = None, output_parser: Optional[BaseOutputParser] = None,
document_prompt: Optional[BasePromptTemplate] = None, document_prompt: Optional[BasePromptTemplate] = None,