pull/20847/head
Eugene Yurtsev 3 weeks ago
parent 62dc24dccc
commit 22cfb1ea06

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

Loading…
Cancel
Save