langchain 0.3.26__py3-none-any.whl → 0.4.0.dev0__py3-none-any.whl

langchain/agents/output_parsers/xml.py

@@ -1,49 +1,123 @@
-from typing import Union
+import re
+from typing import Literal, Optional, Union
 
 from langchain_core.agents import AgentAction, AgentFinish
+from pydantic import Field
+from typing_extensions import override
 
 from langchain.agents import AgentOutputParser
 
 
+def _unescape(text: str) -> str:
+    """Convert custom tag delimiters back into XML tags."""
+    replacements = {
+        "[[tool]]": "<tool>",
+        "[[/tool]]": "</tool>",
+        "[[tool_input]]": "<tool_input>",
+        "[[/tool_input]]": "</tool_input>",
+        "[[observation]]": "<observation>",
+        "[[/observation]]": "</observation>",
+    }
+    for repl, orig in replacements.items():
+        text = text.replace(repl, orig)
+    return text
+
+
 class XMLAgentOutputParser(AgentOutputParser):
-    """Parses tool invocations and final answers in XML format.
+    """Parses tool invocations and final answers from XML-formatted agent output.
+
+    This parser extracts structured information from XML tags to determine whether
+    an agent should perform a tool action or provide a final answer. It includes
+    built-in escaping support to safely handle tool names and inputs
+    containing XML special characters.
+
+    Args:
+        escape_format: The escaping format to use when parsing XML content.
+            Supports 'minimal' which uses custom delimiters like [[tool]] to replace
+            XML tags within content, preventing parsing conflicts.
+            Use 'minimal' if using a corresponding encoding format that uses
+            the _escape function when formatting the output (e.g., with format_xml).
+
+    Expected formats:
+        Tool invocation (returns AgentAction):
+            <tool>search</tool>
+            <tool_input>what is 2 + 2</tool_input>
 
-    Expects output to be in one of two formats.
+        Final answer (returns AgentFinish):
+            <final_answer>The answer is 4</final_answer>
 
-    If the output signals that an action should be taken,
-    should be in the below format. This will result in an AgentAction
-    being returned.
+    Note:
+        Minimal escaping allows tool names containing XML tags to be safely
+        represented. For example, a tool named "search<tool>nested</tool>" would be
+        escaped as "search[[tool]]nested[[/tool]]" in the XML and automatically
+        unescaped during parsing.
 
-    ```
-    <tool>search</tool>
-    <tool_input>what is 2 + 2</tool_input>
-    ```
+    Raises:
+        ValueError: If the input doesn't match either expected XML format or
+            contains malformed XML structure.
+    """
+
+    escape_format: Optional[Literal["minimal"]] = Field(default="minimal")
+    """The format to use for escaping XML characters.
 
-    If the output signals that a final answer should be given,
-    should be in the below format. This will result in an AgentFinish
-    being returned.
+    minimal - uses custom delimiters to replace XML tags within content,
+    preventing parsing conflicts. This is the only supported format currently.
 
-    ```
-    <final_answer>Foo</final_answer>
-    ```
+    None - no escaping is applied, which may lead to parsing conflicts.
     """
 
+    @override
     def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
-        if "</tool>" in text:
-            tool, tool_input = text.split("</tool>")
-            _tool = tool.split("<tool>")[1]
-            _tool_input = tool_input.split("<tool_input>")[1]
-            if "</tool_input>" in _tool_input:
-                _tool_input = _tool_input.split("</tool_input>")[0]
+        # Check for tool invocation first
+        tool_matches = re.findall(r"<tool>(.*?)</tool>", text, re.DOTALL)
+        if tool_matches:
+            if len(tool_matches) != 1:
+                msg = (
+                    f"Malformed tool invocation: expected exactly one <tool> block, "
+                    f"but found {len(tool_matches)}."
+                )
+                raise ValueError(msg)
+            _tool = tool_matches[0]
+
+            # Match optional tool input
+            input_matches = re.findall(
+                r"<tool_input>(.*?)</tool_input>", text, re.DOTALL
+            )
+            if len(input_matches) > 1:
+                msg = (
+                    f"Malformed tool invocation: expected at most one <tool_input> "
+                    f"block, but found {len(input_matches)}."
+                )
+                raise ValueError(msg)
+            _tool_input = input_matches[0] if input_matches else ""
+
+            # Unescape if minimal escape format is used
+            if self.escape_format == "minimal":
+                _tool = _unescape(_tool)
+                _tool_input = _unescape(_tool_input)
+
             return AgentAction(tool=_tool, tool_input=_tool_input, log=text)
-        elif "<final_answer>" in text:
-            _, answer = text.split("<final_answer>")
-            if "</final_answer>" in answer:
-                answer = answer.split("</final_answer>")[0]
+        # Check for final answer
+        if "<final_answer>" in text and "</final_answer>" in text:
+            matches = re.findall(r"<final_answer>(.*?)</final_answer>", text, re.DOTALL)
+            if len(matches) != 1:
+                msg = (
+                    "Malformed output: expected exactly one "
+                    "<final_answer>...</final_answer> block."
+                )
+                raise ValueError(msg)
+            answer = matches[0]
+            # Unescape custom delimiters in final answer
+            if self.escape_format == "minimal":
+                answer = _unescape(answer)
             return AgentFinish(return_values={"output": answer}, log=text)
-        else:
-            raise ValueError
+        msg = (
+            "Malformed output: expected either a tool invocation "
+            "or a final answer in XML format."
+        )
+        raise ValueError(msg)
 
+    @override
     def get_format_instructions(self) -> str:
         raise NotImplementedError
 

langchain/agents/react/agent.py

@@ -116,12 +116,14 @@ def create_react_agent(
             Thought:{agent_scratchpad}'''
 
             prompt = PromptTemplate.from_template(template)
+
     """  # noqa: E501
     missing_vars = {"tools", "tool_names", "agent_scratchpad"}.difference(
-        prompt.input_variables + list(prompt.partial_variables)
+        prompt.input_variables + list(prompt.partial_variables),
     )
     if missing_vars:
-        raise ValueError(f"Prompt missing required variables: {missing_vars}")
+        msg = f"Prompt missing required variables: {missing_vars}"
+        raise ValueError(msg)
 
     prompt = prompt.partial(
         tools=tools_renderer(list(tools)),
@@ -133,7 +135,7 @@ def create_react_agent(
     else:
         llm_with_stop = llm
     output_parser = output_parser or ReActSingleInputOutputParser()
-    agent = (
+    return (
         RunnablePassthrough.assign(
             agent_scratchpad=lambda x: format_log_to_str(x["intermediate_steps"]),
         )
@@ -141,4 +143,3 @@ def create_react_agent(
         | llm_with_stop
         | output_parser
     )
-    return agent

langchain/agents/react/base.py

@@ -11,6 +11,7 @@ from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import BasePromptTemplate
 from langchain_core.tools import BaseTool, Tool
 from pydantic import Field
+from typing_extensions import override
 
 from langchain._api.deprecation import AGENT_DEPRECATION_WARNING
 from langchain.agents.agent import Agent, AgentExecutor, AgentOutputParser
@@ -24,6 +25,9 @@ if TYPE_CHECKING:
     from langchain_community.docstore.base import Docstore
 
 
+_LOOKUP_AND_SEARCH_TOOLS = {"Lookup", "Search"}
+
+
 @deprecated(
     "0.1.0",
     message=AGENT_DEPRECATION_WARNING,
@@ -35,6 +39,7 @@ class ReActDocstoreAgent(Agent):
     output_parser: AgentOutputParser = Field(default_factory=ReActOutputParser)
 
     @classmethod
+    @override
     def _get_default_output_parser(cls, **kwargs: Any) -> AgentOutputParser:
         return ReActOutputParser()
 
@@ -44,6 +49,7 @@ class ReActDocstoreAgent(Agent):
         return AgentType.REACT_DOCSTORE
 
     @classmethod
+    @override
     def create_prompt(cls, tools: Sequence[BaseTool]) -> BasePromptTemplate:
         """Return default prompt."""
         return WIKI_PROMPT
@@ -52,13 +58,13 @@ class ReActDocstoreAgent(Agent):
     def _validate_tools(cls, tools: Sequence[BaseTool]) -> None:
         validate_tools_single_input(cls.__name__, tools)
         super()._validate_tools(tools)
-        if len(tools) != 2:
-            raise ValueError(f"Exactly two tools must be specified, but got {tools}")
+        if len(tools) != len(_LOOKUP_AND_SEARCH_TOOLS):
+            msg = f"Exactly two tools must be specified, but got {tools}"
+            raise ValueError(msg)
         tool_names = {tool.name for tool in tools}
-        if tool_names != {"Lookup", "Search"}:
-            raise ValueError(
-                f"Tool names should be Lookup and Search, got {tool_names}"
-            )
+        if tool_names != _LOOKUP_AND_SEARCH_TOOLS:
+            msg = f"Tool names should be Lookup and Search, got {tool_names}"
+            raise ValueError(msg)
 
     @property
     def observation_prefix(self) -> str:
@@ -96,14 +102,14 @@ class DocstoreExplorer:
         if isinstance(result, Document):
             self.document = result
             return self._summary
-        else:
-            self.document = None
-            return result
+        self.document = None
+        return result
 
     def lookup(self, term: str) -> str:
         """Lookup a term in document (if saved)."""
         if self.document is None:
-            raise ValueError("Cannot lookup without a successful search first")
+            msg = "Cannot lookup without a successful search first"
+            raise ValueError(msg)
         if term.lower() != self.lookup_str:
             self.lookup_str = term.lower()
             self.lookup_index = 0
@@ -112,11 +118,10 @@ class DocstoreExplorer:
         lookups = [p for p in self._paragraphs if self.lookup_str in p.lower()]
         if len(lookups) == 0:
             return "No Results"
-        elif self.lookup_index >= len(lookups):
+        if self.lookup_index >= len(lookups):
             return "No More Results"
-        else:
-            result_prefix = f"(Result {self.lookup_index + 1}/{len(lookups)})"
-            return f"{result_prefix} {lookups[self.lookup_index]}"
+        result_prefix = f"(Result {self.lookup_index + 1}/{len(lookups)})"
+        return f"{result_prefix} {lookups[self.lookup_index]}"
 
     @property
     def _summary(self) -> str:
@@ -125,7 +130,8 @@ class DocstoreExplorer:
     @property
     def _paragraphs(self) -> list[str]:
         if self.document is None:
-            raise ValueError("Cannot get paragraphs without a document")
+            msg = "Cannot get paragraphs without a document"
+            raise ValueError(msg)
         return self.document.page_content.split("\n\n")
 
 
@@ -138,6 +144,7 @@ class ReActTextWorldAgent(ReActDocstoreAgent):
     """Agent for the ReAct TextWorld chain."""
 
     @classmethod
+    @override
     def create_prompt(cls, tools: Sequence[BaseTool]) -> BasePromptTemplate:
         """Return default prompt."""
         return TEXTWORLD_PROMPT
@@ -147,10 +154,12 @@ class ReActTextWorldAgent(ReActDocstoreAgent):
         validate_tools_single_input(cls.__name__, tools)
         super()._validate_tools(tools)
         if len(tools) != 1:
-            raise ValueError(f"Exactly one tool must be specified, but got {tools}")
+            msg = f"Exactly one tool must be specified, but got {tools}"
+            raise ValueError(msg)
         tool_names = {tool.name for tool in tools}
         if tool_names != {"Play"}:
-            raise ValueError(f"Tool name should be Play, got {tool_names}")
+            msg = f"Tool name should be Play, got {tool_names}"
+            raise ValueError(msg)
 
 
 @deprecated(

langchain/agents/react/output_parser.py

@@ -3,6 +3,7 @@ from typing import Union
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.exceptions import OutputParserException
+from typing_extensions import override
 
 from langchain.agents.agent import AgentOutputParser
 
@@ -10,24 +11,24 @@ from langchain.agents.agent import AgentOutputParser
 class ReActOutputParser(AgentOutputParser):
     """Output parser for the ReAct agent."""
 
+    @override
     def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
         action_prefix = "Action: "
         if not text.strip().split("\n")[-1].startswith(action_prefix):
-            raise OutputParserException(f"Could not parse LLM Output: {text}")
+            msg = f"Could not parse LLM Output: {text}"
+            raise OutputParserException(msg)
         action_block = text.strip().split("\n")[-1]
 
         action_str = action_block[len(action_prefix) :]
         # Parse out the action and the directive.
         re_matches = re.search(r"(.*?)\[(.*?)\]", action_str)
         if re_matches is None:
-            raise OutputParserException(
-                f"Could not parse action directive: {action_str}"
-            )
+            msg = f"Could not parse action directive: {action_str}"
+            raise OutputParserException(msg)
         action, action_input = re_matches.group(1), re_matches.group(2)
         if action == "Finish":
             return AgentFinish({"output": action_input}, text)
-        else:
-            return AgentAction(action, action_input, text)
+        return AgentAction(action, action_input, text)
 
     @property
     def _type(self) -> str:

langchain/agents/react/textworld_prompt.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts.prompt import PromptTemplate
 
 EXAMPLES = [

langchain/agents/react/wiki_prompt.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts.prompt import PromptTemplate
 
 EXAMPLES = [
@@ -16,7 +15,7 @@ Thought: I need to instead search High Plains (United States).
 Action: Search[High Plains (United States)]
 Observation: The High Plains are a subregion of the Great Plains. From east to west, the High Plains rise in elevation from around 1,800 to 7,000 ft (550 to 2,130 m).[3]
 Thought: High Plains rise in elevation from around 1,800 to 7,000 ft, so the answer is 1,800 to 7,000 ft.
-Action: Finish[1,800 to 7,000 ft]""",
+Action: Finish[1,800 to 7,000 ft]""",  # noqa: E501
     """Question: Musician and satirist Allie Goertz wrote a song about the "The Simpsons" character Milhouse, who Matt Groening named after who?
 Thought: The question simplifies to "The Simpsons" character Milhouse is named after who. I only need to search Milhouse and find who it is named after.
 Action: Search[Milhouse]
@@ -25,16 +24,16 @@ Thought: The paragraph does not tell who Milhouse is named after, maybe I can lo
 Action: Lookup[named after]
 Observation: (Result 1 / 1) Milhouse was named after U.S. president Richard Nixon, whose middle name was Milhous.
 Thought: Milhouse was named after U.S. president Richard Nixon, so the answer is Richard Nixon.
-Action: Finish[Richard Nixon]""",
+Action: Finish[Richard Nixon]""",  # noqa: E501
     """Question: Which documentary is about Finnish rock groups, Adam Clayton Powell or The Saimaa Gesture?
 Thought: I need to search Adam Clayton Powell and The Saimaa Gesture, and find which documentary is about Finnish rock groups.
 Action: Search[Adam Clayton Powell]
-Observation: Could not find [Adam Clayton Powell]. Similar: [’Adam Clayton Powell III’, ’Seventh Avenue (Manhattan)’, ’Adam Clayton Powell Jr. State Office Building’, ’Isabel Washington Powell’, ’Adam Powell’, ’Adam Clayton Powell (film)’, ’Giancarlo Esposito’].
+Observation: Could not find [Adam Clayton Powell]. Similar: ['Adam Clayton Powell III', 'Seventh Avenue (Manhattan)', 'Adam Clayton Powell Jr. State Office Building', 'Isabel Washington Powell', 'Adam Powell', 'Adam Clayton Powell (film)', 'Giancarlo Esposito'].
 Thought: To find the documentary, I can search Adam Clayton Powell (film).
 Action: Search[Adam Clayton Powell (film)]
 Observation: Adam Clayton Powell is a 1989 American documentary film directed by Richard Kilberg. The film is about the rise and fall of influential African-American politician Adam Clayton Powell Jr.[3][4] It was later aired as part of the PBS series The American Experience.
 Thought: Adam Clayton Powell (film) is a documentary about an African-American politician, not Finnish rock groups. So the documentary about Finnish rock groups must instead be The Saimaa Gesture.
-Action: Finish[The Saimaa Gesture]""",
+Action: Finish[The Saimaa Gesture]""",  # noqa: E501
     """Question: What profession does Nicholas Ray and Elia Kazan have in common?
 Thought: I need to search Nicholas Ray and Elia Kazan, find their professions, then find the profession they have in common.
 Action: Search[Nicholas Ray]
@@ -43,16 +42,16 @@ Thought: Professions of Nicholas Ray are director, screenwriter, and actor. I ne
 Action: Search[Elia Kazan]
 Observation: Elia Kazan was an American film and theatre director, producer, screenwriter and actor.
 Thought: Professions of Elia Kazan are director, producer, screenwriter, and actor. So profession Nicholas Ray and Elia Kazan have in common is director, screenwriter, and actor.
-Action: Finish[director, screenwriter, actor]""",
-    """Question: Which magazine was started first Arthur’s Magazine or First for Women?
-Thought: I need to search Arthur’s Magazine and First for Women, and find which was started first.
-Action: Search[Arthur’s Magazine]
-Observation: Arthur’s Magazine (1844-1846) was an American literary periodical published in Philadelphia in the 19th century.
-Thought: Arthur’s Magazine was started in 1844. I need to search First for Women next.
+Action: Finish[director, screenwriter, actor]""",  # noqa: E501
+    """Question: Which magazine was started first Arthur's Magazine or First for Women?
+Thought: I need to search Arthur's Magazine and First for Women, and find which was started first.
+Action: Search[Arthur's Magazine]
+Observation: Arthur's Magazine (1844-1846) was an American literary periodical published in Philadelphia in the 19th century.
+Thought: Arthur's Magazine was started in 1844. I need to search First for Women next.
 Action: Search[First for Women]
-Observation: First for Women is a woman’s magazine published by Bauer Media Group in the USA.[1] The magazine was started in 1989.
-Thought: First for Women was started in 1989. 1844 (Arthur’s Magazine) < 1989 (First for Women), so Arthur’s Magazine was started first.
-Action: Finish[Arthur’s Magazine]""",
+Observation: First for Women is a woman's magazine published by Bauer Media Group in the USA.[1] The magazine was started in 1989.
+Thought: First for Women was started in 1989. 1844 (Arthur's Magazine) < 1989 (First for Women), so Arthur's Magazine was started first.
+Action: Finish[Arthur's Magazine]""",  # noqa: E501
     """Question: Were Pavel Urysohn and Leonid Levin known for the same type of work?
 Thought: I need to search Pavel Urysohn and Leonid Levin, find their types of work, then find if they are the same.
 Action: Search[Pavel Urysohn]
@@ -61,7 +60,7 @@ Thought: Pavel Urysohn is a mathematician. I need to search Leonid Levin next an
 Action: Search[Leonid Levin]
 Observation: Leonid Anatolievich Levin is a Soviet-American mathematician and computer scientist.
 Thought: Leonid Levin is a mathematician and computer scientist. So Pavel Urysohn and Leonid Levin have the same type of work.
-Action: Finish[yes]""",
+Action: Finish[yes]""",  # noqa: E501
 ]
 SUFFIX = """\nQuestion: {input}
 {agent_scratchpad}"""

langchain/agents/schema.py

@@ -2,17 +2,20 @@ from typing import Any
 
 from langchain_core.agents import AgentAction
 from langchain_core.prompts.chat import ChatPromptTemplate
+from typing_extensions import override
 
 
 class AgentScratchPadChatPromptTemplate(ChatPromptTemplate):
     """Chat prompt template for the agent scratchpad."""
 
     @classmethod
+    @override
     def is_lc_serializable(cls) -> bool:
         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 ""
@@ -29,6 +32,6 @@ class AgentScratchPadChatPromptTemplate(ChatPromptTemplate):
     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
+            intermediate_steps,
         )
         return kwargs

langchain/agents/self_ask_with_search/base.py

@@ -11,6 +11,7 @@ from langchain_core.prompts import BasePromptTemplate
 from langchain_core.runnables import Runnable, RunnablePassthrough
 from langchain_core.tools import BaseTool, Tool
 from pydantic import Field
+from typing_extensions import override
 
 from langchain.agents.agent import Agent, AgentExecutor, AgentOutputParser
 from langchain.agents.agent_types import AgentType
@@ -32,6 +33,7 @@ class SelfAskWithSearchAgent(Agent):
     output_parser: AgentOutputParser = Field(default_factory=SelfAskOutputParser)
 
     @classmethod
+    @override
     def _get_default_output_parser(cls, **kwargs: Any) -> AgentOutputParser:
         return SelfAskOutputParser()
 
@@ -41,6 +43,7 @@ class SelfAskWithSearchAgent(Agent):
         return AgentType.SELF_ASK_WITH_SEARCH
 
     @classmethod
+    @override
     def create_prompt(cls, tools: Sequence[BaseTool]) -> BasePromptTemplate:
         """Prompt does not depend on tools."""
         return PROMPT
@@ -50,12 +53,12 @@ class SelfAskWithSearchAgent(Agent):
         validate_tools_single_input(cls.__name__, tools)
         super()._validate_tools(tools)
         if len(tools) != 1:
-            raise ValueError(f"Exactly one tool must be specified, but got {tools}")
+            msg = f"Exactly one tool must be specified, but got {tools}"
+            raise ValueError(msg)
         tool_names = {tool.name for tool in tools}
         if tool_names != {"Intermediate Answer"}:
-            raise ValueError(
-                f"Tool name should be Intermediate Answer, got {tool_names}"
-            )
+            msg = f"Tool name should be Intermediate Answer, got {tool_names}"
+            raise ValueError(msg)
 
     @property
     def observation_prefix(self) -> str:
@@ -76,7 +79,9 @@ class SelfAskWithSearchChain(AgentExecutor):
         self,
         llm: BaseLanguageModel,
         search_chain: Union[
-            GoogleSerperAPIWrapper, SearchApiAPIWrapper, SerpAPIWrapper
+            GoogleSerperAPIWrapper,
+            SearchApiAPIWrapper,
+            SerpAPIWrapper,
         ],
         **kwargs: Any,
     ):
@@ -92,7 +97,9 @@ class SelfAskWithSearchChain(AgentExecutor):
 
 
 def create_self_ask_with_search_agent(
-    llm: BaseLanguageModel, tools: Sequence[BaseTool], prompt: BasePromptTemplate
+    llm: BaseLanguageModel,
+    tools: Sequence[BaseTool],
+    prompt: BasePromptTemplate,
 ) -> Runnable:
     """Create an agent that uses self-ask with search prompting.
 
@@ -178,23 +185,25 @@ def create_self_ask_with_search_agent(
             Are followup questions needed here:{agent_scratchpad}'''
 
             prompt = PromptTemplate.from_template(template)
+
     """  # noqa: E501
     missing_vars = {"agent_scratchpad"}.difference(
-        prompt.input_variables + list(prompt.partial_variables)
+        prompt.input_variables + list(prompt.partial_variables),
     )
     if missing_vars:
-        raise ValueError(f"Prompt missing required variables: {missing_vars}")
+        msg = f"Prompt missing required variables: {missing_vars}"
+        raise ValueError(msg)
 
     if len(tools) != 1:
-        raise ValueError("This agent expects exactly one tool")
-    tool = list(tools)[0]
+        msg = "This agent expects exactly one tool"
+        raise ValueError(msg)
+    tool = next(iter(tools))
     if tool.name != "Intermediate Answer":
-        raise ValueError(
-            "This agent expects the tool to be named `Intermediate Answer`"
-        )
+        msg = "This agent expects the tool to be named `Intermediate Answer`"
+        raise ValueError(msg)
 
     llm_with_stop = llm.bind(stop=["\nIntermediate answer:"])
-    agent = (
+    return (
         RunnablePassthrough.assign(
             agent_scratchpad=lambda x: format_log_to_str(
                 x["intermediate_steps"],
@@ -208,4 +217,3 @@ def create_self_ask_with_search_agent(
         | llm_with_stop
         | SelfAskOutputParser()
     )
-    return agent

langchain/agents/self_ask_with_search/prompt.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts.prompt import PromptTemplate
 
 _DEFAULT_TEMPLATE = """Question: Who lived longer, Muhammad Ali or Alan Turing?

langchain/agents/structured_chat/base.py

@@ -16,6 +16,7 @@ from langchain_core.runnables import Runnable, RunnablePassthrough
 from langchain_core.tools import BaseTool
 from langchain_core.tools.render import ToolsRenderer
 from pydantic import Field
+from typing_extensions import override
 
 from langchain.agents.agent import Agent, AgentOutputParser
 from langchain.agents.format_scratchpad import format_log_to_str
@@ -35,7 +36,7 @@ class StructuredChatAgent(Agent):
     """Structured Chat Agent."""
 
     output_parser: AgentOutputParser = Field(
-        default_factory=StructuredChatOutputParserWithRetries
+        default_factory=StructuredChatOutputParserWithRetries,
     )
     """Output parser for the agent."""
 
@@ -50,35 +51,41 @@ 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):
-            raise ValueError("agent_scratchpad should be of type string.")
+            msg = "agent_scratchpad should be of type string."
+            raise ValueError(msg)  # noqa: TRY004
         if agent_scratchpad:
             return (
                 f"This was your previous work "
                 f"(but I haven't seen any of it! I only see what "
                 f"you return as final answer):\n{agent_scratchpad}"
             )
-        else:
-            return agent_scratchpad
+        return agent_scratchpad
 
     @classmethod
     def _validate_tools(cls, tools: Sequence[BaseTool]) -> None:
         pass
 
     @classmethod
+    @override
     def _get_default_output_parser(
-        cls, llm: Optional[BaseLanguageModel] = None, **kwargs: Any
+        cls,
+        llm: Optional[BaseLanguageModel] = None,
+        **kwargs: Any,
     ) -> AgentOutputParser:
         return StructuredChatOutputParserWithRetries.from_llm(llm=llm)
 
     @property
+    @override
     def _stop(self) -> list[str]:
         return ["Observation:"]
 
     @classmethod
+    @override
     def create_prompt(
         cls,
         tools: Sequence[BaseTool],
@@ -96,7 +103,7 @@ class StructuredChatAgent(Agent):
         formatted_tools = "\n".join(tool_strings)
         tool_names = ", ".join([tool.name for tool in tools])
         format_instructions = format_instructions.format(tool_names=tool_names)
-        template = "\n\n".join([prefix, formatted_tools, format_instructions, suffix])
+        template = f"{prefix}\n\n{formatted_tools}\n\n{format_instructions}\n\n{suffix}"
         if input_variables is None:
             input_variables = ["input", "agent_scratchpad"]
         _memory_prompts = memory_prompts or []
@@ -273,12 +280,14 @@ def create_structured_chat_agent(
                     ("human", human),
                 ]
             )
+
     """  # noqa: E501
     missing_vars = {"tools", "tool_names", "agent_scratchpad"}.difference(
-        prompt.input_variables + list(prompt.partial_variables)
+        prompt.input_variables + list(prompt.partial_variables),
     )
     if missing_vars:
-        raise ValueError(f"Prompt missing required variables: {missing_vars}")
+        msg = f"Prompt missing required variables: {missing_vars}"
+        raise ValueError(msg)
 
     prompt = prompt.partial(
         tools=tools_renderer(list(tools)),
@@ -290,7 +299,7 @@ def create_structured_chat_agent(
     else:
         llm_with_stop = llm
 
-    agent = (
+    return (
         RunnablePassthrough.assign(
             agent_scratchpad=lambda x: format_log_to_str(x["intermediate_steps"]),
         )
@@ -298,4 +307,3 @@ def create_structured_chat_agent(
         | llm_with_stop
         | JSONAgentOutputParser()
     )
-    return agent