Skip to content

Commit

Permalink
langchain[small]: Change type to BasePromptTemplate (langchain-ai#23083)
Browse files Browse the repository at this point in the history
```python
Change from_llm(
 prompt: PromptTemplate 
 ...
 )
```
 to
```python
Change from_llm(
 prompt: BasePromptTemplate 
 ...
 )
```
  • Loading branch information
pprados authored Jun 19, 2024
1 parent 94452a9 commit db6f46c
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
3 changes: 2 additions & 1 deletion libs/langchain/langchain/retrievers/multi_query.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
from langchain_core.documents import Document
from langchain_core.language_models import BaseLanguageModel
from langchain_core.output_parsers import BaseOutputParser
from langchain_core.prompts import BasePromptTemplate
from langchain_core.prompts.prompt import PromptTemplate
from langchain_core.retrievers import BaseRetriever
from langchain_core.runnables import Runnable
Expand Down Expand Up @@ -62,7 +63,7 @@ def from_llm(
cls,
retriever: BaseRetriever,
llm: BaseLanguageModel,
prompt: PromptTemplate = DEFAULT_QUERY_PROMPT,
prompt: BasePromptTemplate = DEFAULT_QUERY_PROMPT,
parser_key: Optional[str] = None,
include_original: bool = False,
) -> "MultiQueryRetriever":
Expand Down
3 changes: 2 additions & 1 deletion libs/langchain/langchain/retrievers/re_phraser.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
)
from langchain_core.documents import Document
from langchain_core.language_models import BaseLLM
from langchain_core.prompts import BasePromptTemplate
from langchain_core.prompts.prompt import PromptTemplate
from langchain_core.retrievers import BaseRetriever

Expand Down Expand Up @@ -36,7 +37,7 @@ def from_llm(
cls,
retriever: BaseRetriever,
llm: BaseLLM,
prompt: PromptTemplate = DEFAULT_QUERY_PROMPT,
prompt: BasePromptTemplate = DEFAULT_QUERY_PROMPT,
) -> "RePhraseQueryRetriever":
"""Initialize from llm using default template.
Expand Down

0 comments on commit db6f46c

Please sign in to comment.