pull/20847/head
Eugene Yurtsev 1 month ago
parent 62dc24dccc
commit 22cfb1ea06

@ -60,6 +60,10 @@ try:
from langchain.chains import LLMChain from langchain.chains import LLMChain
from langchain.chains.prompt_selector import ConditionalPromptSelector from langchain.chains.prompt_selector import ConditionalPromptSelector
DEFAULT_TEXT_SPLITTER = RecursiveCharacterTextSplitter(
chunk_size=1500, chunk_overlap=150
)
class WebResearchRetriever(BaseRetriever): class WebResearchRetriever(BaseRetriever):
"""`Google Search API` retriever.""" """`Google Search API` retriever."""
@ -90,9 +94,7 @@ try:
search: GoogleSearchAPIWrapper, search: GoogleSearchAPIWrapper,
prompt: Optional[BasePromptTemplate] = None, prompt: Optional[BasePromptTemplate] = None,
num_search_results: int = 1, num_search_results: int = 1,
text_splitter: RecursiveCharacterTextSplitter = RecursiveCharacterTextSplitter( text_splitter: RecursiveCharacterTextSplitter = DEFAULT_TEXT_SPLITTER,
chunk_size=1500, chunk_overlap=150
),
) -> "WebResearchRetriever": ) -> "WebResearchRetriever":
"""Initialize from llm using default template. """Initialize from llm using default template.

Loading…
Cancel
Save