langchain 0.3.22__py3-none-any.whl → 0.3.24__py3-none-any.whl

langchain/agents/openai_functions_agent/agent_token_buffer_memory.py

@@ -1,6 +1,6 @@
 """Memory used to save agent output AND intermediate steps."""
 
-from typing import Any, Dict, List
+from typing import Any
 
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.messages import BaseMessage, get_buffer_string
@@ -12,7 +12,7 @@ from langchain.agents.format_scratchpad import (
 from langchain.memory.chat_memory import BaseChatMemory
 
 
-class AgentTokenBufferMemory(BaseChatMemory):  # type: ignore[override]
+class AgentTokenBufferMemory(BaseChatMemory):
     """Memory used to save agent output AND intermediate steps.
 
     Parameters:
@@ -43,19 +43,19 @@ class AgentTokenBufferMemory(BaseChatMemory):  # type: ignore[override]
     format_as_tools: bool = False
 
     @property
-    def buffer(self) -> List[BaseMessage]:
+    def buffer(self) -> list[BaseMessage]:
         """String buffer of memory."""
         return self.chat_memory.messages
 
     @property
-    def memory_variables(self) -> List[str]:
+    def memory_variables(self) -> list[str]:
         """Always return list of memory variables.
 
         :meta private:
         """
         return [self.memory_key]
 
-    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+    def load_memory_variables(self, inputs: dict[str, Any]) -> dict[str, Any]:
         """Return history buffer.
 
         Args:
@@ -74,7 +74,7 @@ class AgentTokenBufferMemory(BaseChatMemory):  # type: ignore[override]
             )
         return {self.memory_key: final_buffer}
 
-    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, Any]) -> None:
+    def save_context(self, inputs: dict[str, Any], outputs: dict[str, Any]) -> None:
         """Save context from this conversation to buffer. Pruned.
 
         Args:

langchain/agents/openai_functions_agent/base.py

@@ -1,6 +1,7 @@
 """Module implements an agent that uses OpenAI's APIs function enabled API."""
 
-from typing import Any, List, Optional, Sequence, Tuple, Type, Union
+from collections.abc import Sequence
+from typing import Any, Optional, Union
 
 from langchain_core._api import deprecated
 from langchain_core.agents import AgentAction, AgentFinish
@@ -51,11 +52,11 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent):
     llm: BaseLanguageModel
     tools: Sequence[BaseTool]
     prompt: BasePromptTemplate
-    output_parser: Type[OpenAIFunctionsAgentOutputParser] = (
+    output_parser: type[OpenAIFunctionsAgentOutputParser] = (
         OpenAIFunctionsAgentOutputParser
     )
 
-    def get_allowed_tools(self) -> List[str]:
+    def get_allowed_tools(self) -> list[str]:
         """Get allowed tools."""
         return [t.name for t in self.tools]
 
@@ -81,19 +82,19 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent):
         return self
 
     @property
-    def input_keys(self) -> List[str]:
+    def input_keys(self) -> list[str]:
         """Get input keys. Input refers to user input here."""
         return ["input"]
 
     @property
-    def functions(self) -> List[dict]:
+    def functions(self) -> list[dict]:
         """Get functions."""
 
         return [dict(convert_to_openai_function(t)) for t in self.tools]
 
     def plan(
         self,
-        intermediate_steps: List[Tuple[AgentAction, str]],
+        intermediate_steps: list[tuple[AgentAction, str]],
         callbacks: Callbacks = None,
         with_functions: bool = True,
         **kwargs: Any,
@@ -135,7 +136,7 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent):
 
     async def aplan(
         self,
-        intermediate_steps: List[Tuple[AgentAction, str]],
+        intermediate_steps: list[tuple[AgentAction, str]],
         callbacks: Callbacks = None,
         **kwargs: Any,
     ) -> Union[AgentAction, AgentFinish]:
@@ -168,7 +169,7 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent):
     def return_stopped_response(
         self,
         early_stopping_method: str,
-        intermediate_steps: List[Tuple[AgentAction, str]],
+        intermediate_steps: list[tuple[AgentAction, str]],
         **kwargs: Any,
     ) -> AgentFinish:
         """Return response when agent has been stopped due to max iterations.
@@ -213,7 +214,7 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent):
         system_message: Optional[SystemMessage] = SystemMessage(
             content="You are a helpful AI assistant."
         ),
-        extra_prompt_messages: Optional[List[BaseMessagePromptTemplate]] = None,
+        extra_prompt_messages: Optional[list[BaseMessagePromptTemplate]] = None,
     ) -> ChatPromptTemplate:
         """Create prompt for this agent.
 
@@ -227,7 +228,7 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent):
             A prompt template to pass into this agent.
         """
         _prompts = extra_prompt_messages or []
-        messages: List[Union[BaseMessagePromptTemplate, BaseMessage]]
+        messages: list[Union[BaseMessagePromptTemplate, BaseMessage]]
         if system_message:
             messages = [system_message]
         else:
@@ -240,7 +241,7 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent):
                 MessagesPlaceholder(variable_name="agent_scratchpad"),
             ]
         )
-        return ChatPromptTemplate(messages=messages)  # type: ignore[arg-type, call-arg]
+        return ChatPromptTemplate(messages=messages)
 
     @classmethod
     def from_llm_and_tools(
@@ -248,7 +249,7 @@ class OpenAIFunctionsAgent(BaseSingleActionAgent):
         llm: BaseLanguageModel,
         tools: Sequence[BaseTool],
         callback_manager: Optional[BaseCallbackManager] = None,
-        extra_prompt_messages: Optional[List[BaseMessagePromptTemplate]] = None,
+        extra_prompt_messages: Optional[list[BaseMessagePromptTemplate]] = None,
         system_message: Optional[SystemMessage] = SystemMessage(
             content="You are a helpful AI assistant."
         ),

langchain/agents/openai_functions_multi_agent/base.py

@@ -1,8 +1,9 @@
 """Module implements an agent that uses OpenAI's APIs function enabled API."""
 
 import json
+from collections.abc import Sequence
 from json import JSONDecodeError
-from typing import Any, List, Optional, Sequence, Tuple, Union
+from typing import Any, Optional, Union
 
 from langchain_core._api import deprecated
 from langchain_core.agents import AgentAction, AgentActionMessageLog, AgentFinish
@@ -34,7 +35,7 @@ from langchain.agents.format_scratchpad.openai_functions import (
 _FunctionsAgentAction = AgentActionMessageLog
 
 
-def _parse_ai_message(message: BaseMessage) -> Union[List[AgentAction], AgentFinish]:
+def _parse_ai_message(message: BaseMessage) -> Union[list[AgentAction], AgentFinish]:
     """Parse an AI message."""
     if not isinstance(message, AIMessage):
         raise TypeError(f"Expected an AI message got {type(message)}")
@@ -58,7 +59,7 @@ def _parse_ai_message(message: BaseMessage) -> Union[List[AgentAction], AgentFin
                 f"the `arguments` JSON does not contain `actions` key."
             )
 
-        final_tools: List[AgentAction] = []
+        final_tools: list[AgentAction] = []
         for tool_schema in tools:
             if "action" in tool_schema:
                 _tool_input = tool_schema["action"]
@@ -112,7 +113,7 @@ class OpenAIMultiFunctionsAgent(BaseMultiActionAgent):
     tools: Sequence[BaseTool]
     prompt: BasePromptTemplate
 
-    def get_allowed_tools(self) -> List[str]:
+    def get_allowed_tools(self) -> list[str]:
         """Get allowed tools."""
         return [t.name for t in self.tools]
 
@@ -127,12 +128,12 @@ class OpenAIMultiFunctionsAgent(BaseMultiActionAgent):
         return self
 
     @property
-    def input_keys(self) -> List[str]:
+    def input_keys(self) -> list[str]:
         """Get input keys. Input refers to user input here."""
         return ["input"]
 
     @property
-    def functions(self) -> List[dict]:
+    def functions(self) -> list[dict]:
         """Get the functions for the agent."""
         enum_vals = [t.name for t in self.tools]
         tool_selection = {
@@ -194,10 +195,10 @@ class OpenAIMultiFunctionsAgent(BaseMultiActionAgent):
 
     def plan(
         self,
-        intermediate_steps: List[Tuple[AgentAction, str]],
+        intermediate_steps: list[tuple[AgentAction, str]],
         callbacks: Callbacks = None,
         **kwargs: Any,
-    ) -> Union[List[AgentAction], AgentFinish]:
+    ) -> Union[list[AgentAction], AgentFinish]:
         """Given input, decided what to do.
 
         Args:
@@ -224,10 +225,10 @@ class OpenAIMultiFunctionsAgent(BaseMultiActionAgent):
 
     async def aplan(
         self,
-        intermediate_steps: List[Tuple[AgentAction, str]],
+        intermediate_steps: list[tuple[AgentAction, str]],
         callbacks: Callbacks = None,
         **kwargs: Any,
-    ) -> Union[List[AgentAction], AgentFinish]:
+    ) -> Union[list[AgentAction], AgentFinish]:
         """Async given input, decided what to do.
 
         Args:
@@ -258,7 +259,7 @@ class OpenAIMultiFunctionsAgent(BaseMultiActionAgent):
         system_message: Optional[SystemMessage] = SystemMessage(
             content="You are a helpful AI assistant."
         ),
-        extra_prompt_messages: Optional[List[BaseMessagePromptTemplate]] = None,
+        extra_prompt_messages: Optional[list[BaseMessagePromptTemplate]] = None,
     ) -> BasePromptTemplate:
         """Create prompt for this agent.
 
@@ -272,7 +273,7 @@ class OpenAIMultiFunctionsAgent(BaseMultiActionAgent):
             A prompt template to pass into this agent.
         """
         _prompts = extra_prompt_messages or []
-        messages: List[Union[BaseMessagePromptTemplate, BaseMessage]]
+        messages: list[Union[BaseMessagePromptTemplate, BaseMessage]]
         if system_message:
             messages = [system_message]
         else:
@@ -285,7 +286,7 @@ class OpenAIMultiFunctionsAgent(BaseMultiActionAgent):
                 MessagesPlaceholder(variable_name="agent_scratchpad"),
             ]
         )
-        return ChatPromptTemplate(messages=messages)  # type: ignore[arg-type, call-arg]
+        return ChatPromptTemplate(messages=messages)
 
     @classmethod
     def from_llm_and_tools(
@@ -293,7 +294,7 @@ class OpenAIMultiFunctionsAgent(BaseMultiActionAgent):
         llm: BaseLanguageModel,
         tools: Sequence[BaseTool],
         callback_manager: Optional[BaseCallbackManager] = None,
-        extra_prompt_messages: Optional[List[BaseMessagePromptTemplate]] = None,
+        extra_prompt_messages: Optional[list[BaseMessagePromptTemplate]] = None,
         system_message: Optional[SystemMessage] = SystemMessage(
             content="You are a helpful AI assistant."
         ),

langchain/agents/openai_tools/base.py

@@ -1,4 +1,5 @@
-from typing import Optional, Sequence
+from collections.abc import Sequence
+from typing import Optional
 
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts.chat import ChatPromptTemplate

langchain/agents/output_parsers/openai_functions.py

@@ -1,6 +1,6 @@
 import json
 from json import JSONDecodeError
-from typing import List, Union
+from typing import Union
 
 from langchain_core.agents import AgentAction, AgentActionMessageLog, AgentFinish
 from langchain_core.exceptions import OutputParserException
@@ -77,7 +77,7 @@ class OpenAIFunctionsAgentOutputParser(AgentOutputParser):
         )
 
     def parse_result(
-        self, result: List[Generation], *, partial: bool = False
+        self, result: list[Generation], *, partial: bool = False
     ) -> Union[AgentAction, AgentFinish]:
         if not isinstance(result[0], ChatGeneration):
             raise ValueError("This output parser only works on ChatGeneration output")

langchain/agents/output_parsers/openai_tools.py

@@ -1,4 +1,4 @@
-from typing import List, Union
+from typing import Union
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.messages import BaseMessage
@@ -15,12 +15,12 @@ OpenAIToolAgentAction = ToolAgentAction
 
 def parse_ai_message_to_openai_tool_action(
     message: BaseMessage,
-) -> Union[List[AgentAction], AgentFinish]:
+) -> Union[list[AgentAction], AgentFinish]:
     """Parse an AI message potentially containing tool_calls."""
     tool_actions = parse_ai_message_to_tool_action(message)
     if isinstance(tool_actions, AgentFinish):
         return tool_actions
-    final_actions: List[AgentAction] = []
+    final_actions: list[AgentAction] = []
     for action in tool_actions:
         if isinstance(action, ToolAgentAction):
             final_actions.append(
@@ -54,12 +54,12 @@ class OpenAIToolsAgentOutputParser(MultiActionAgentOutputParser):
         return "openai-tools-agent-output-parser"
 
     def parse_result(
-        self, result: List[Generation], *, partial: bool = False
-    ) -> Union[List[AgentAction], AgentFinish]:
+        self, result: list[Generation], *, partial: bool = False
+    ) -> Union[list[AgentAction], AgentFinish]:
         if not isinstance(result[0], ChatGeneration):
             raise ValueError("This output parser only works on ChatGeneration output")
         message = result[0].message
         return parse_ai_message_to_openai_tool_action(message)
 
-    def parse(self, text: str) -> Union[List[AgentAction], AgentFinish]:
+    def parse(self, text: str) -> Union[list[AgentAction], AgentFinish]:
         raise ValueError("Can only parse messages")

langchain/agents/output_parsers/react_json_single_input.py

@@ -1,6 +1,7 @@
 import json
 import re
-from typing import Pattern, Union
+from re import Pattern
+from typing import Union
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.exceptions import OutputParserException

langchain/agents/output_parsers/self_ask.py

@@ -1,4 +1,5 @@
-from typing import Sequence, Union
+from collections.abc import Sequence
+from typing import Union
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.exceptions import OutputParserException

langchain/agents/output_parsers/tools.py

@@ -1,6 +1,6 @@
 import json
 from json import JSONDecodeError
-from typing import List, Union
+from typing import Union
 
 from langchain_core.agents import AgentAction, AgentActionMessageLog, AgentFinish
 from langchain_core.exceptions import OutputParserException
@@ -14,19 +14,19 @@ from langchain_core.outputs import ChatGeneration, Generation
 from langchain.agents.agent import MultiActionAgentOutputParser
 
 
-class ToolAgentAction(AgentActionMessageLog):  # type: ignore[override]
+class ToolAgentAction(AgentActionMessageLog):
     tool_call_id: str
     """Tool call that this message is responding to."""
 
 
 def parse_ai_message_to_tool_action(
     message: BaseMessage,
-) -> Union[List[AgentAction], AgentFinish]:
+) -> Union[list[AgentAction], AgentFinish]:
     """Parse an AI message potentially containing tool_calls."""
     if not isinstance(message, AIMessage):
         raise TypeError(f"Expected an AI message got {type(message)}")
 
-    actions: List = []
+    actions: list = []
     if message.tool_calls:
         tool_calls = message.tool_calls
     else:
@@ -91,12 +91,12 @@ class ToolsAgentOutputParser(MultiActionAgentOutputParser):
         return "tools-agent-output-parser"
 
     def parse_result(
-        self, result: List[Generation], *, partial: bool = False
-    ) -> Union[List[AgentAction], AgentFinish]:
+        self, result: list[Generation], *, partial: bool = False
+    ) -> Union[list[AgentAction], AgentFinish]:
         if not isinstance(result[0], ChatGeneration):
             raise ValueError("This output parser only works on ChatGeneration output")
         message = result[0].message
         return parse_ai_message_to_tool_action(message)
 
-    def parse(self, text: str) -> Union[List[AgentAction], AgentFinish]:
+    def parse(self, text: str) -> Union[list[AgentAction], AgentFinish]:
         raise ValueError("Can only parse messages")

langchain/agents/react/agent.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
-from typing import List, Optional, Sequence, Union
+from collections.abc import Sequence
+from typing import Optional, Union
 
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import BasePromptTemplate
@@ -20,7 +21,7 @@ def create_react_agent(
     output_parser: Optional[AgentOutputParser] = None,
     tools_renderer: ToolsRenderer = render_text_description,
     *,
-    stop_sequence: Union[bool, List[str]] = True,
+    stop_sequence: Union[bool, list[str]] = True,
 ) -> Runnable:
     """Create an agent that uses ReAct prompting.
 

langchain/agents/react/base.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, List, Optional, Sequence
+from collections.abc import Sequence
+from typing import TYPE_CHECKING, Any, Optional
 
 from langchain_core._api import deprecated
 from langchain_core.documents import Document
@@ -65,7 +66,7 @@ class ReActDocstoreAgent(Agent):
         return "Observation: "
 
     @property
-    def _stop(self) -> List[str]:
+    def _stop(self) -> list[str]:
         return ["\nObservation:"]
 
     @property
@@ -122,7 +123,7 @@ class DocstoreExplorer:
         return self._paragraphs[0]
 
     @property
-    def _paragraphs(self) -> List[str]:
+    def _paragraphs(self) -> list[str]:
         if self.document is None:
             raise ValueError("Cannot get paragraphs without a document")
         return self.document.page_content.split("\n\n")

langchain/agents/schema.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Tuple
+from typing import Any
 
 from langchain_core.agents import AgentAction
 from langchain_core.prompts.chat import ChatPromptTemplate
@@ -12,7 +12,7 @@ class AgentScratchPadChatPromptTemplate(ChatPromptTemplate):
         return False
 
     def _construct_agent_scratchpad(
-        self, intermediate_steps: List[Tuple[AgentAction, str]]
+        self, intermediate_steps: list[tuple[AgentAction, str]]
     ) -> str:
         if len(intermediate_steps) == 0:
             return ""
@@ -26,7 +26,7 @@ class AgentScratchPadChatPromptTemplate(ChatPromptTemplate):
             f"you return as final answer):\n{thoughts}"
         )
 
-    def _merge_partial_and_user_variables(self, **kwargs: Any) -> Dict[str, Any]:
+    def _merge_partial_and_user_variables(self, **kwargs: Any) -> dict[str, Any]:
         intermediate_steps = kwargs.pop("intermediate_steps")
         kwargs["agent_scratchpad"] = self._construct_agent_scratchpad(
             intermediate_steps

langchain/agents/self_ask_with_search/base.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Sequence, Union
+from collections.abc import Sequence
+from typing import TYPE_CHECKING, Any, Union
 
 from langchain_core._api import deprecated
 from langchain_core.language_models import BaseLanguageModel

langchain/agents/structured_chat/base.py

@@ -1,5 +1,6 @@
 import re
-from typing import Any, List, Optional, Sequence, Tuple, Union
+from collections.abc import Sequence
+from typing import Any, Optional, Union
 
 from langchain_core._api import deprecated
 from langchain_core.agents import AgentAction
@@ -49,7 +50,7 @@ class StructuredChatAgent(Agent):
         return "Thought:"
 
     def _construct_scratchpad(
-        self, intermediate_steps: List[Tuple[AgentAction, str]]
+        self, intermediate_steps: list[tuple[AgentAction, str]]
     ) -> str:
         agent_scratchpad = super()._construct_scratchpad(intermediate_steps)
         if not isinstance(agent_scratchpad, str):
@@ -74,7 +75,7 @@ class StructuredChatAgent(Agent):
         return StructuredChatOutputParserWithRetries.from_llm(llm=llm)
 
     @property
-    def _stop(self) -> List[str]:
+    def _stop(self) -> list[str]:
         return ["Observation:"]
 
     @classmethod
@@ -85,8 +86,8 @@ class StructuredChatAgent(Agent):
         suffix: str = SUFFIX,
         human_message_template: str = HUMAN_MESSAGE_TEMPLATE,
         format_instructions: str = FORMAT_INSTRUCTIONS,
-        input_variables: Optional[List[str]] = None,
-        memory_prompts: Optional[List[BasePromptTemplate]] = None,
+        input_variables: Optional[list[str]] = None,
+        memory_prompts: Optional[list[BasePromptTemplate]] = None,
     ) -> BasePromptTemplate:
         tool_strings = []
         for tool in tools:
@@ -117,8 +118,8 @@ class StructuredChatAgent(Agent):
         suffix: str = SUFFIX,
         human_message_template: str = HUMAN_MESSAGE_TEMPLATE,
         format_instructions: str = FORMAT_INSTRUCTIONS,
-        input_variables: Optional[List[str]] = None,
-        memory_prompts: Optional[List[BasePromptTemplate]] = None,
+        input_variables: Optional[list[str]] = None,
+        memory_prompts: Optional[list[BasePromptTemplate]] = None,
         **kwargs: Any,
     ) -> Agent:
         """Construct an agent from an LLM and tools."""
@@ -157,7 +158,7 @@ def create_structured_chat_agent(
     prompt: ChatPromptTemplate,
     tools_renderer: ToolsRenderer = render_text_description_and_args,
     *,
-    stop_sequence: Union[bool, List[str]] = True,
+    stop_sequence: Union[bool, list[str]] = True,
 ) -> Runnable:
     """Create an agent aimed at supporting tools with multiple inputs.
 

langchain/agents/structured_chat/output_parser.py

@@ -3,7 +3,8 @@ from __future__ import annotations
 import json
 import logging
 import re
-from typing import Optional, Pattern, Union
+from re import Pattern
+from typing import Optional, Union
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.exceptions import OutputParserException

langchain/agents/tool_calling_agent/base.py

@@ -1,4 +1,5 @@
-from typing import Callable, List, Sequence, Tuple
+from collections.abc import Sequence
+from typing import Callable
 
 from langchain_core.agents import AgentAction
 from langchain_core.language_models import BaseLanguageModel
@@ -12,7 +13,7 @@ from langchain.agents.format_scratchpad.tools import (
 )
 from langchain.agents.output_parsers.tools import ToolsAgentOutputParser
 
-MessageFormatter = Callable[[Sequence[Tuple[AgentAction, str]]], List[BaseMessage]]
+MessageFormatter = Callable[[Sequence[tuple[AgentAction, str]]], list[BaseMessage]]
 
 
 def create_tool_calling_agent(

langchain/agents/tools.py

@@ -1,6 +1,6 @@
 """Interface for tools."""
 
-from typing import List, Optional
+from typing import Optional
 
 from langchain_core.callbacks import (
     AsyncCallbackManagerForToolRun,
@@ -9,7 +9,7 @@ from langchain_core.callbacks import (
 from langchain_core.tools import BaseTool, tool
 
 
-class InvalidTool(BaseTool):  # type: ignore[override]
+class InvalidTool(BaseTool):
     """Tool that is run when invalid tool name is encountered by agent."""
 
     name: str = "invalid_tool"
@@ -20,7 +20,7 @@ class InvalidTool(BaseTool):  # type: ignore[override]
     def _run(
         self,
         requested_tool_name: str,
-        available_tool_names: List[str],
+        available_tool_names: list[str],
         run_manager: Optional[CallbackManagerForToolRun] = None,
     ) -> str:
         """Use the tool."""
@@ -33,7 +33,7 @@ class InvalidTool(BaseTool):  # type: ignore[override]
     async def _arun(
         self,
         requested_tool_name: str,
-        available_tool_names: List[str],
+        available_tool_names: list[str],
         run_manager: Optional[AsyncCallbackManagerForToolRun] = None,
     ) -> str:
         """Use the tool asynchronously."""

langchain/agents/types.py

@@ -1,4 +1,4 @@
-from typing import Dict, Type, Union
+from typing import Union
 
 from langchain.agents.agent import BaseSingleActionAgent
 from langchain.agents.agent_types import AgentType
@@ -12,9 +12,9 @@ from langchain.agents.react.base import ReActDocstoreAgent
 from langchain.agents.self_ask_with_search.base import SelfAskWithSearchAgent
 from langchain.agents.structured_chat.base import StructuredChatAgent
 
-AGENT_TYPE = Union[Type[BaseSingleActionAgent], Type[OpenAIMultiFunctionsAgent]]
+AGENT_TYPE = Union[type[BaseSingleActionAgent], type[OpenAIMultiFunctionsAgent]]
 
-AGENT_TO_CLASS: Dict[AgentType, AGENT_TYPE] = {
+AGENT_TO_CLASS: dict[AgentType, AGENT_TYPE] = {
     AgentType.ZERO_SHOT_REACT_DESCRIPTION: ZeroShotAgent,
     AgentType.REACT_DOCSTORE: ReActDocstoreAgent,
     AgentType.SELF_ASK_WITH_SEARCH: SelfAskWithSearchAgent,

langchain/agents/utils.py

@@ -1,4 +1,4 @@
-from typing import Sequence
+from collections.abc import Sequence
 
 from langchain_core.tools import BaseTool
 

langchain/agents/xml/base.py

@@ -1,4 +1,5 @@
-from typing import Any, List, Sequence, Tuple, Union
+from collections.abc import Sequence
+from typing import Any, Union
 
 from langchain_core._api import deprecated
 from langchain_core.agents import AgentAction, AgentFinish
@@ -38,13 +39,13 @@ class XMLAgent(BaseSingleActionAgent):
 
     """
 
-    tools: List[BaseTool]
+    tools: list[BaseTool]
     """List of tools this agent has access to."""
     llm_chain: LLMChain
     """Chain to use to predict action."""
 
     @property
-    def input_keys(self) -> List[str]:
+    def input_keys(self) -> list[str]:
         return ["input"]
 
     @staticmethod
@@ -60,7 +61,7 @@ class XMLAgent(BaseSingleActionAgent):
 
     def plan(
         self,
-        intermediate_steps: List[Tuple[AgentAction, str]],
+        intermediate_steps: list[tuple[AgentAction, str]],
         callbacks: Callbacks = None,
         **kwargs: Any,
     ) -> Union[AgentAction, AgentFinish]:
@@ -84,7 +85,7 @@ class XMLAgent(BaseSingleActionAgent):
 
     async def aplan(
         self,
-        intermediate_steps: List[Tuple[AgentAction, str]],
+        intermediate_steps: list[tuple[AgentAction, str]],
         callbacks: Callbacks = None,
         **kwargs: Any,
     ) -> Union[AgentAction, AgentFinish]:
@@ -113,7 +114,7 @@ def create_xml_agent(
     prompt: BasePromptTemplate,
     tools_renderer: ToolsRenderer = render_text_description,
     *,
-    stop_sequence: Union[bool, List[str]] = True,
+    stop_sequence: Union[bool, list[str]] = True,
 ) -> Runnable:
     """Create an agent that uses XML to format its logic.