langchain 0.2.11__py3-none-any.whl → 0.2.13__py3-none-any.whl

langchain/agents/agent.py

@@ -156,7 +156,7 @@ class BaseSingleActionAgent(BaseModel):
             llm: Language model to use.
             tools: Tools to use.
             callback_manager: Callback manager to use.
-            **kwargs: Additional arguments.
+            kwargs: Additional arguments.
 
         Returns:
             BaseSingleActionAgent: Agent object.
@@ -420,8 +420,6 @@ class RunnableAgent(BaseSingleActionAgent):
     """
 
     class Config:
-        """Configuration for this pydantic object."""
-
         arbitrary_types_allowed = True
 
     @property
@@ -530,8 +528,6 @@ class RunnableMultiActionAgent(BaseMultiActionAgent):
     """
 
     class Config:
-        """Configuration for this pydantic object."""
-
         arbitrary_types_allowed = True
 
     @property
@@ -939,7 +935,7 @@ class Agent(BaseSingleActionAgent):
             tools: Tools to use.
             callback_manager: Callback manager to use.
             output_parser: Output parser to use.
-            **kwargs: Additional arguments.
+            kwargs: Additional arguments.
 
         Returns:
             Agent: Agent object.
@@ -1110,7 +1106,7 @@ class AgentExecutor(Chain):
             agent: Agent to use.
             tools: Tools to use.
             callbacks: Callbacks to use.
-            **kwargs: Additional arguments.
+            kwargs: Additional arguments.
 
         Returns:
             AgentExecutor: Agent executor object.
@@ -1741,7 +1737,7 @@ class AgentExecutor(Chain):
         Args:
             input: Input to the agent.
             config: Config to use.
-            **kwargs: Additional arguments.
+            kwargs: Additional arguments.
 
         Yields:
             AddableDict: Addable dictionary.
@@ -1772,7 +1768,7 @@ class AgentExecutor(Chain):
         Args:
             input: Input to the agent.
             config: Config to use.
-            **kwargs: Additional arguments.
+            kwargs: Additional arguments.
 
         Yields:
             AddableDict: Addable dictionary.

langchain/agents/agent_toolkits/vectorstore/base.py

@@ -2,6 +2,7 @@
 
 from typing import Any, Dict, Optional
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks.base import BaseCallbackManager
 from langchain_core.language_models import BaseLanguageModel
 
@@ -15,6 +16,16 @@ from langchain.agents.mrkl.base import ZeroShotAgent
 from langchain.chains.llm import LLMChain
 
 
+@deprecated(
+    since="0.2.13",
+    removal="1.0",
+    message=(
+        "See API reference for this function for a replacement implementation: "
+        "https://api.python.langchain.com/en/latest/agents/langchain.agents.agent_toolkits.vectorstore.base.create_vectorstore_agent.html "  # noqa: E501
+        "Read more here on how to create agents that query vector stores: "
+        "https://python.langchain.com/v0.2/docs/how_to/qa_chat_history_how_to/#agents"
+    ),
+)
 def create_vectorstore_agent(
     llm: BaseLanguageModel,
     toolkit: VectorStoreToolkit,
@@ -26,6 +37,44 @@ def create_vectorstore_agent(
 ) -> AgentExecutor:
     """Construct a VectorStore agent from an LLM and tools.
 
+    Note: this class is deprecated. See below for a replacement that uses tool
+    calling methods and LangGraph. Install LangGraph with:
+
+        .. code-block:: bash
+
+            pip install -U langgraph
+
+        .. code-block:: python
+
+            from langchain_core.tools import create_retriever_tool
+            from langchain_core.vectorstores import InMemoryVectorStore
+            from langchain_openai import ChatOpenAI, OpenAIEmbeddings
+            from langgraph.prebuilt import create_react_agent
+
+            llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
+
+            vector_store = InMemoryVectorStore.from_texts(
+                [
+                    "Dogs are great companions, known for their loyalty and friendliness.",
+                    "Cats are independent pets that often enjoy their own space.",
+                ],
+                OpenAIEmbeddings(),
+            )
+
+            tool = create_retriever_tool(
+                vector_store.as_retriever(),
+                "pet_information_retriever",
+                "Fetches information about pets.",
+            )
+
+            agent = create_react_agent(llm, [tool])
+
+            for step in agent.stream(
+                {"messages": [("human", "What are dogs known for?")]},
+                stream_mode="values",
+            ):
+                step["messages"][-1].pretty_print()
+
     Args:
         llm (BaseLanguageModel): LLM that will be used by the agent
         toolkit (VectorStoreToolkit): Set of tools for the agent
@@ -33,7 +82,7 @@ def create_vectorstore_agent(
         prefix (str, optional): The prefix prompt for the agent. If not provided uses default PREFIX.
         verbose (bool, optional): If you want to see the content of the scratchpad. [ Defaults to False ]
         agent_executor_kwargs (Optional[Dict[str, Any]], optional): If there is any other parameter you want to send to the agent. [ Defaults to None ]
-        **kwargs: Additional named parameters to pass to the ZeroShotAgent.
+        kwargs: Additional named parameters to pass to the ZeroShotAgent.
 
     Returns:
         AgentExecutor: Returns a callable AgentExecutor object. Either you can call it or use run method with the query to get the response
@@ -56,6 +105,16 @@ def create_vectorstore_agent(
     )
 
 
+@deprecated(
+    since="0.2.13",
+    removal="1.0",
+    message=(
+        "See API reference for this function for a replacement implementation: "
+        "https://api.python.langchain.com/en/latest/agents/langchain.agents.agent_toolkits.vectorstore.base.create_vectorstore_router_agent.html "  # noqa: E501
+        "Read more here on how to create agents that query vector stores: "
+        "https://python.langchain.com/v0.2/docs/how_to/qa_chat_history_how_to/#agents"
+    ),
+)
 def create_vectorstore_router_agent(
     llm: BaseLanguageModel,
     toolkit: VectorStoreRouterToolkit,
@@ -67,6 +126,59 @@ def create_vectorstore_router_agent(
 ) -> AgentExecutor:
     """Construct a VectorStore router agent from an LLM and tools.
 
+    Note: this class is deprecated. See below for a replacement that uses tool
+    calling methods and LangGraph. Install LangGraph with:
+
+        .. code-block:: bash
+
+            pip install -U langgraph
+
+        .. code-block:: python
+
+            from langchain_core.tools import create_retriever_tool
+            from langchain_core.vectorstores import InMemoryVectorStore
+            from langchain_openai import ChatOpenAI, OpenAIEmbeddings
+            from langgraph.prebuilt import create_react_agent
+
+            llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
+
+            pet_vector_store = InMemoryVectorStore.from_texts(
+                [
+                    "Dogs are great companions, known for their loyalty and friendliness.",
+                    "Cats are independent pets that often enjoy their own space.",
+                ],
+                OpenAIEmbeddings(),
+            )
+
+            food_vector_store = InMemoryVectorStore.from_texts(
+                [
+                    "Carrots are orange and delicious.",
+                    "Apples are red and delicious.",
+                ],
+                OpenAIEmbeddings(),
+            )
+
+            tools = [
+                create_retriever_tool(
+                    pet_vector_store.as_retriever(),
+                    "pet_information_retriever",
+                    "Fetches information about pets.",
+                ),
+                create_retriever_tool(
+                    food_vector_store.as_retriever(),
+                    "food_information_retriever",
+                    "Fetches information about food.",
+                )
+            ]
+
+            agent = create_react_agent(llm, tools)
+
+            for step in agent.stream(
+                {"messages": [("human", "Tell me about carrots.")]},
+                stream_mode="values",
+            ):
+                step["messages"][-1].pretty_print()
+
     Args:
         llm (BaseLanguageModel): LLM that will be used by the agent
         toolkit (VectorStoreRouterToolkit): Set of tools for the agent which have routing capability with multiple vector stores
@@ -74,7 +186,7 @@ def create_vectorstore_router_agent(
         prefix (str, optional): The prefix prompt for the router agent. If not provided uses default ROUTER_PREFIX.
         verbose (bool, optional): If you want to see the content of the scratchpad. [ Defaults to False ]
         agent_executor_kwargs (Optional[Dict[str, Any]], optional): If there is any other parameter you want to send to the agent. [ Defaults to None ]
-        **kwargs: Additional named parameters to pass to the ZeroShotAgent.
+        kwargs: Additional named parameters to pass to the ZeroShotAgent.
 
     Returns:
         AgentExecutor: Returns a callable AgentExecutor object. Either you can call it or use run method with the query to get the response.

langchain/agents/agent_toolkits/vectorstore/toolkit.py

@@ -16,8 +16,6 @@ class VectorStoreInfo(BaseModel):
     description: str
 
     class Config:
-        """Configuration for this pydantic object."""
-
         arbitrary_types_allowed = True
 
 
@@ -28,8 +26,6 @@ class VectorStoreToolkit(BaseToolkit):
     llm: BaseLanguageModel
 
     class Config:
-        """Configuration for this pydantic object."""
-
         arbitrary_types_allowed = True
 
     def get_tools(self) -> List[BaseTool]:
@@ -71,8 +67,6 @@ class VectorStoreRouterToolkit(BaseToolkit):
     llm: BaseLanguageModel
 
     class Config:
-        """Configuration for this pydantic object."""
-
         arbitrary_types_allowed = True
 
     def get_tools(self) -> List[BaseTool]:

langchain/agents/chat/output_parser.py

@@ -1,6 +1,6 @@
 import json
 import re
-from typing import Union
+from typing import Pattern, Union
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.exceptions import OutputParserException
@@ -17,7 +17,7 @@ class ChatOutputParser(AgentOutputParser):
     format_instructions: str = FORMAT_INSTRUCTIONS
     """Default formatting instructions"""
 
-    pattern = re.compile(r"^.*?`{3}(?:json)?\n(.*?)`{3}.*?$", re.DOTALL)
+    pattern: Pattern = re.compile(r"^.*?`{3}(?:json)?\n(.*?)`{3}.*?$", re.DOTALL)
     """Regex pattern to parse the output."""
 
     def get_format_instructions(self) -> str:

langchain/agents/initialize.py

@@ -45,7 +45,7 @@ def initialize_agent(
         agent_kwargs: Additional keyword arguments to pass to the underlying agent.
             Defaults to None.
         tags: Tags to apply to the traced runs. Defaults to None.
-        **kwargs: Additional keyword arguments passed to the agent executor.
+        kwargs: Additional keyword arguments passed to the agent executor.
 
     Returns:
         An agent executor.

langchain/agents/loading.py

@@ -44,7 +44,7 @@ def load_agent_from_config(
         config: Config dict to load agent from.
         llm: Language model to use as the agent.
         tools: List of tools this agent has access to.
-        **kwargs: Additional keyword arguments passed to the agent executor.
+        kwargs: Additional keyword arguments passed to the agent executor.
 
     Returns:
         An agent executor.
@@ -98,7 +98,7 @@ def load_agent(
 
     Args:
         path: Path to the agent file.
-        **kwargs: Additional keyword arguments passed to the agent executor.
+        kwargs: Additional keyword arguments passed to the agent executor.
 
     Returns:
         An agent executor.

langchain/agents/mrkl/base.py

@@ -128,7 +128,7 @@ class ZeroShotAgent(Agent):
             format_instructions: The format instructions to use.
                 Defaults to FORMAT_INSTRUCTIONS.
             input_variables: The input variables to use. Defaults to None.
-            **kwargs: Additional parameters to pass to the agent.
+            kwargs: Additional parameters to pass to the agent.
         """
         cls._validate_tools(tools)
         prompt = cls.create_prompt(

langchain/agents/openai_assistant/base.py

@@ -261,7 +261,7 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]):
             model: Assistant model to use.
             client: OpenAI or AzureOpenAI client.
                 Will create a default OpenAI client if not specified.
-            **kwargs: Additional arguments.
+            kwargs: Additional arguments.
 
         Returns:
             OpenAIAssistantRunnable configured to run using the created assistant.
@@ -418,7 +418,7 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]):
                 tools: Override Assistant tools for this run.
                 run_metadata: Metadata to associate with new run.
             config: Runnable config. Defaults to None.
-            **kwargs: Additional arguments.
+            kwargs: Additional arguments.
 
         Return:
             If self.as_agent, will return

langchain/agents/openai_functions_agent/base.py

@@ -262,7 +262,7 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent):
             extra_prompt_messages: Extra prompt messages to use. Defaults to None.
             system_message: The system message to use.
                 Defaults to a default system message.
-            **kwargs: Additional parameters to pass to the agent.
+            kwargs: Additional parameters to pass to the agent.
         """
         prompt = cls.create_prompt(
             extra_prompt_messages=extra_prompt_messages,

langchain/agents/openai_functions_multi_agent/base.py

@@ -307,7 +307,7 @@ class OpenAIMultiFunctionsAgent(BaseMultiActionAgent):
             extra_prompt_messages: Extra prompt messages to use. Default is None.
             system_message: The system message to use.
                 Default is a default system message.
-            **kwargs: Additional arguments.
+            kwargs: Additional arguments.
         """
         prompt = cls.create_prompt(
             extra_prompt_messages=extra_prompt_messages,

langchain/agents/output_parsers/react_json_single_input.py

@@ -1,6 +1,6 @@
 import json
 import re
-from typing import Union
+from typing import Pattern, Union
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.exceptions import OutputParserException
@@ -42,7 +42,7 @@ class ReActJsonSingleInputOutputParser(AgentOutputParser):
 
     """
 
-    pattern = re.compile(r"^.*?`{3}(?:json)?\n?(.*?)`{3}.*?$", re.DOTALL)
+    pattern: Pattern = re.compile(r"^.*?`{3}(?:json)?\n?(.*?)`{3}.*?$", re.DOTALL)
     """Regex pattern to parse the output."""
 
     def get_format_instructions(self) -> str:

langchain/agents/structured_chat/output_parser.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import json
 import logging
 import re
-from typing import Optional, Union
+from typing import Optional, Pattern, Union
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.exceptions import OutputParserException
@@ -23,7 +23,7 @@ class StructuredChatOutputParser(AgentOutputParser):
     format_instructions: str = FORMAT_INSTRUCTIONS
     """Default formatting instructions"""
 
-    pattern = re.compile(r"```(?:json\s+)?(\W.*?)```", re.DOTALL)
+    pattern: Pattern = re.compile(r"```(?:json\s+)?(\W.*?)```", re.DOTALL)
     """Regex pattern to parse the output."""
 
     def get_format_instructions(self) -> str:

langchain/chains/__init__.py

@@ -59,6 +59,7 @@ _module_lookup = {
     "OpenAIModerationChain": "langchain.chains.moderation",
     "NatBotChain": "langchain.chains.natbot.base",
     "create_citation_fuzzy_match_chain": "langchain.chains.openai_functions",
+    "create_citation_fuzzy_match_runnable": "langchain.chains.openai_functions",
     "create_extraction_chain": "langchain.chains.openai_functions",
     "create_extraction_chain_pydantic": "langchain.chains.openai_functions",
     "create_qa_with_sources_chain": "langchain.chains.openai_functions",

langchain/chains/api/base.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 from typing import Any, Dict, List, Optional, Sequence, Tuple
 from urllib.parse import urlparse
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
@@ -53,6 +54,15 @@ def _check_in_allowed_domain(url: str, limit_to_domains: Sequence[str]) -> bool:
 try:
     from langchain_community.utilities.requests import TextRequestsWrapper
 
+    @deprecated(
+        since="0.2.13",
+        message=(
+            "This class is deprecated and will be removed in langchain 1.0. "
+            "See API reference for replacement: "
+            "https://api.python.langchain.com/en/latest/chains/langchain.chains.api.base.APIChain.html"  # noqa: E501
+        ),
+        removal="1.0",
+    )
     class APIChain(Chain):
         """Chain that makes API calls and summarizes the responses to answer a question.
 
@@ -69,7 +79,117 @@ try:
             what network access it has.
 
             See https://python.langchain.com/docs/security for more information.
-        """
+
+        Note: this class is deprecated. See below for a replacement implementation
+        using LangGraph. The benefits of this implementation are:
+
+        - Uses LLM tool calling features to encourage properly-formatted API requests;
+        - Support for both token-by-token and step-by-step streaming;
+        - Support for checkpointing and memory of chat history;
+        - Easier to modify or extend (e.g., with additional tools, structured responses, etc.)
+
+        Install LangGraph with:
+
+        .. code-block:: bash
+
+            pip install -U langgraph
+
+        .. code-block:: python
+
+            from typing import Annotated, Sequence
+            from typing_extensions import TypedDict
+
+            from langchain.chains.api.prompt import API_URL_PROMPT
+            from langchain_community.agent_toolkits.openapi.toolkit import RequestsToolkit
+            from langchain_community.utilities.requests import TextRequestsWrapper
+            from langchain_core.messages import BaseMessage
+            from langchain_core.prompts import ChatPromptTemplate
+            from langchain_openai import ChatOpenAI
+            from langchain_core.runnables import RunnableConfig
+            from langgraph.graph import END, StateGraph
+            from langgraph.graph.message import add_messages
+            from langgraph.prebuilt.tool_node import ToolNode
+
+            # NOTE: There are inherent risks in giving models discretion
+            # to execute real-world actions. We must "opt-in" to these
+            # risks by setting allow_dangerous_request=True to use these tools.
+            # This can be dangerous for calling unwanted requests. Please make
+            # sure your custom OpenAPI spec (yaml) is safe and that permissions
+            # associated with the tools are narrowly-scoped.
+            ALLOW_DANGEROUS_REQUESTS = True
+
+            # Subset of spec for https://jsonplaceholder.typicode.com
+            api_spec = \"\"\"
+            openapi: 3.0.0
+            info:
+              title: JSONPlaceholder API
+              version: 1.0.0
+            servers:
+              - url: https://jsonplaceholder.typicode.com
+            paths:
+              /posts:
+                get:
+                  summary: Get posts
+                  parameters: &id001
+                    - name: _limit
+                      in: query
+                      required: false
+                      schema:
+                        type: integer
+                      example: 2
+                      description: Limit the number of results
+            \"\"\"
+
+            llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
+            toolkit = RequestsToolkit(
+                requests_wrapper=TextRequestsWrapper(headers={}),  # no auth required
+                allow_dangerous_requests=ALLOW_DANGEROUS_REQUESTS,
+            )
+            tools = toolkit.get_tools()
+
+            api_request_chain = (
+                API_URL_PROMPT.partial(api_docs=api_spec)
+                | llm.bind_tools(tools, tool_choice="any")
+            )
+
+            class ChainState(TypedDict):
+                \"\"\"LangGraph state.\"\"\"
+
+                messages: Annotated[Sequence[BaseMessage], add_messages]
+
+
+            async def acall_request_chain(state: ChainState, config: RunnableConfig):
+                last_message = state["messages"][-1]
+                response = await api_request_chain.ainvoke(
+                    {"question": last_message.content}, config
+                )
+                return {"messages": [response]}
+
+            async def acall_model(state: ChainState, config: RunnableConfig):
+                response = await llm.ainvoke(state["messages"], config)
+                return {"messages": [response]}
+
+            graph_builder = StateGraph(ChainState)
+            graph_builder.add_node("call_tool", acall_request_chain)
+            graph_builder.add_node("execute_tool", ToolNode(tools))
+            graph_builder.add_node("call_model", acall_model)
+            graph_builder.set_entry_point("call_tool")
+            graph_builder.add_edge("call_tool", "execute_tool")
+            graph_builder.add_edge("execute_tool", "call_model")
+            graph_builder.add_edge("call_model", END)
+            chain = graph_builder.compile()
+
+        .. code-block:: python
+
+            example_query = "Fetch the top two posts. What are their titles?"
+
+            events = chain.astream(
+                {"messages": [("user", example_query)]},
+                stream_mode="values",
+            )
+            async for event in events:
+                event["messages"][-1].pretty_print()
+        """  # noqa: E501
 
         api_request_chain: LLMChain
         api_answer_chain: LLMChain

langchain/chains/base.py

@@ -97,8 +97,6 @@ class Chain(RunnableSerializable[Dict[str, Any], Dict[str, Any]], ABC):
     """[DEPRECATED] Use `callbacks` instead."""
 
     class Config:
-        """Configuration for this pydantic object."""
-
         arbitrary_types_allowed = True
 
     def get_input_schema(

langchain/chains/combine_documents/map_reduce.py

@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Optional, Tuple, Type
 
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
-from langchain_core.pydantic_v1 import BaseModel, Extra, root_validator
+from langchain_core.pydantic_v1 import BaseModel, root_validator
 from langchain_core.runnables.config import RunnableConfig
 from langchain_core.runnables.utils import create_model
 
@@ -127,10 +127,8 @@ class MapReduceDocumentsChain(BaseCombineDocumentsChain):
         return _output_keys
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @root_validator(pre=True)
     def get_reduce_chain(cls, values: Dict) -> Dict:

langchain/chains/combine_documents/map_rerank.py

@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Optional, Sequence, Tuple, Type, Union, cast
 
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
-from langchain_core.pydantic_v1 import BaseModel, Extra, root_validator
+from langchain_core.pydantic_v1 import BaseModel, root_validator
 from langchain_core.runnables.config import RunnableConfig
 from langchain_core.runnables.utils import create_model
 
@@ -25,7 +25,7 @@ class MapRerankDocumentsChain(BaseCombineDocumentsChain):
     Example:
         .. code-block:: python
 
-            from langchain.chains import StuffDocumentsChain, LLMChain
+            from langchain.chains import MapRerankDocumentsChain, LLMChain
             from langchain_core.prompts import PromptTemplate
             from langchain_community.llms import OpenAI
             from langchain.output_parsers.regex import RegexParser
@@ -39,7 +39,7 @@ class MapRerankDocumentsChain(BaseCombineDocumentsChain):
             prompt_template = (
                 "Use the following context to tell me the chemical formula "
                 "for water. Output both your answer and a score of how confident "
-                "you are. Context: {content}"
+                "you are. Context: {context}"
             )
             output_parser = RegexParser(
                 regex=r"(.*?)\nScore: (.*)",
@@ -75,10 +75,8 @@ class MapRerankDocumentsChain(BaseCombineDocumentsChain):
     Intermediate steps include the results of calling llm_chain on each document."""
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     def get_output_schema(
         self, config: Optional[RunnableConfig] = None

langchain/chains/combine_documents/reduce.py

@@ -6,7 +6,6 @@ from typing import Any, Callable, List, Optional, Protocol, Tuple
 
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
-from langchain_core.pydantic_v1 import Extra
 
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
 
@@ -206,10 +205,8 @@ class ReduceDocumentsChain(BaseCombineDocumentsChain):
     Otherwise, after it reaches the max number, it will throw an error"""
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @property
     def _collapse_chain(self) -> BaseCombineDocumentsChain:

langchain/chains/combine_documents/refine.py

@@ -8,7 +8,7 @@ from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from langchain_core.prompts import BasePromptTemplate, format_document
 from langchain_core.prompts.prompt import PromptTemplate
-from langchain_core.pydantic_v1 import Extra, Field, root_validator
+from langchain_core.pydantic_v1 import Field, root_validator
 
 from langchain.chains.combine_documents.base import (
     BaseCombineDocumentsChain,
@@ -99,10 +99,8 @@ class RefineDocumentsChain(BaseCombineDocumentsChain):
         return _output_keys
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @root_validator(pre=True)
     def get_return_intermediate_steps(cls, values: Dict) -> Dict: