|
|
@ -9,6 +9,7 @@ from langchain_core.callbacks import (
|
|
|
|
from langchain_core.documents import Document
|
|
|
|
from langchain_core.documents import Document
|
|
|
|
from langchain_core.language_models import BaseLanguageModel
|
|
|
|
from langchain_core.language_models import BaseLanguageModel
|
|
|
|
from langchain_core.output_parsers import BaseOutputParser
|
|
|
|
from langchain_core.output_parsers import BaseOutputParser
|
|
|
|
|
|
|
|
from langchain_core.prompts import BasePromptTemplate
|
|
|
|
from langchain_core.prompts.prompt import PromptTemplate
|
|
|
|
from langchain_core.prompts.prompt import PromptTemplate
|
|
|
|
from langchain_core.retrievers import BaseRetriever
|
|
|
|
from langchain_core.retrievers import BaseRetriever
|
|
|
|
from langchain_core.runnables import Runnable
|
|
|
|
from langchain_core.runnables import Runnable
|
|
|
@ -62,7 +63,7 @@ class MultiQueryRetriever(BaseRetriever):
|
|
|
|
cls,
|
|
|
|
cls,
|
|
|
|
retriever: BaseRetriever,
|
|
|
|
retriever: BaseRetriever,
|
|
|
|
llm: BaseLanguageModel,
|
|
|
|
llm: BaseLanguageModel,
|
|
|
|
prompt: PromptTemplate = DEFAULT_QUERY_PROMPT,
|
|
|
|
prompt: BasePromptTemplate = DEFAULT_QUERY_PROMPT,
|
|
|
|
parser_key: Optional[str] = None,
|
|
|
|
parser_key: Optional[str] = None,
|
|
|
|
include_original: bool = False,
|
|
|
|
include_original: bool = False,
|
|
|
|
) -> "MultiQueryRetriever":
|
|
|
|
) -> "MultiQueryRetriever":
|
|
|
|