From d6470aab60351b1e410f66ad481023c37ec22dd2 Mon Sep 17 00:00:00 2001 From: Leonid Ganeline Date: Fri, 19 Apr 2024 18:41:36 -0700 Subject: [PATCH] langchain: `dosctore` import fix (#20678) Cleaned up imports --- libs/langchain/langchain/__init__.py | 4 ++-- libs/langchain/langchain/agents/react/base.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/langchain/langchain/__init__.py b/libs/langchain/langchain/__init__.py index 78afad3a0aa..519fe78d2fa 100644 --- a/libs/langchain/langchain/__init__.py +++ b/libs/langchain/langchain/__init__.py @@ -112,13 +112,13 @@ def __getattr__(name: str) -> Any: return VectorDBQAWithSourcesChain elif name == "InMemoryDocstore": - from langchain.docstore import InMemoryDocstore + from langchain_community.docstore import InMemoryDocstore _warn_on_import(name, replacement="langchain.docstore.InMemoryDocstore") return InMemoryDocstore elif name == "Wikipedia": - from langchain.docstore import Wikipedia + from langchain_community.docstore import Wikipedia _warn_on_import(name, replacement="langchain.docstore.Wikipedia") diff --git a/libs/langchain/langchain/agents/react/base.py b/libs/langchain/langchain/agents/react/base.py index 437eb80e720..0ed388d217c 100644 --- a/libs/langchain/langchain/agents/react/base.py +++ b/libs/langchain/langchain/agents/react/base.py @@ -1,6 +1,7 @@ """Chain that implements the ReAct paper from https://arxiv.org/pdf/2210.03629.pdf.""" from typing import Any, List, Optional, Sequence +from langchain_community.docstore.base import Docstore from langchain_core._api import deprecated from langchain_core.documents import Document from langchain_core.language_models import BaseLanguageModel @@ -14,7 +15,6 @@ from langchain.agents.react.output_parser import ReActOutputParser from langchain.agents.react.textworld_prompt import TEXTWORLD_PROMPT from langchain.agents.react.wiki_prompt import WIKI_PROMPT from langchain.agents.utils import validate_tools_single_input -from langchain.docstore.base import Docstore @deprecated("0.1.0", removal="0.2.0")