github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-19 19:11:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

FLARE Example notebook: switch to named arg to pass pydantic validation (#7267)

Browse Source 
Adding the name of the parameter to comply with latest requirements by
Pydantic usage for BaseModels.
This commit is contained in:
Stefano Lottini 2023-07-06 15:32:00 +02:00 committed by GitHub
parent 0c7a5cb206
commit e61cfb6e99
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
docs/extras/modules/chains/additional/flare.ipynb
View File

@ -56,7 +56,8 @@
"source": [ "source": [
"import os\n", "import os\n",
"\n", "\n",
"os.environ[\"SERPER_API_KEY\"] = \"\"" "os.environ[\"SERPER_API_KEY\"] = \"\"",
"os.environ[\"OPENAI_API_KEY\"] = \"\""
] ]
}, },
{ {
@ -77,7 +78,7 @@
"from langchain.chat_models import ChatOpenAI\n", "from langchain.chat_models import ChatOpenAI\n",
"from langchain.llms import OpenAI\n", "from langchain.llms import OpenAI\n",
"from langchain.schema import Document\n", "from langchain.schema import Document\n",
"from typing import Any" "from typing import Any, List"
] ]
}, },
{ {
@ -96,8 +97,8 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"class SerperSearchRetriever(BaseRetriever):\n", "class SerperSearchRetriever(BaseRetriever):\n",
" def __init__(self, search):\n", "\n",
" self.search = search\n", " search: GoogleSerperAPIWrapper = None\n",
"\n", "\n",
" def _get_relevant_documents(self, query: str, *, run_manager: CallbackManagerForRetrieverRun, **kwargs: Any) -> List[Document]:\n", " def _get_relevant_documents(self, query: str, *, run_manager: CallbackManagerForRetrieverRun, **kwargs: Any) -> List[Document]:\n",
" return [Document(page_content=self.search.run(query))]\n", " return [Document(page_content=self.search.run(query))]\n",
@ -111,7 +112,7 @@
" raise NotImplementedError()\n", " raise NotImplementedError()\n",
"\n", "\n",
"\n", "\n",
"retriever = SerperSearchRetriever(GoogleSerperAPIWrapper())" "retriever = SerperSearchRetriever(search=GoogleSerperAPIWrapper())"
] ]
}, },
{ {