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

langchain/agents/output_parsers/xml.py

@@ -3,6 +3,7 @@ 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
 
@@ -65,6 +66,7 @@ class XMLAgentOutputParser(AgentOutputParser):
     None - no escaping is applied, which may lead to parsing conflicts.
     """
 
+    @override
     def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
         # Check for tool invocation first
         tool_matches = re.findall(r"<tool>(.*?)</tool>", text, re.DOTALL)
@@ -115,6 +117,7 @@ class XMLAgentOutputParser(AgentOutputParser):
         )
         raise ValueError(msg)
 
+    @override
     def get_format_instructions(self) -> str:
         raise NotImplementedError
 

langchain/agents/react/agent.py

@@ -116,6 +116,7 @@ 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),

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
@@ -38,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()
 
@@ -47,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
@@ -141,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

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,6 +11,7 @@ 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):

langchain/agents/schema.py

@@ -2,12 +2,14 @@ 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
 

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
@@ -182,6 +185,7 @@ 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),

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
@@ -70,6 +71,7 @@ class StructuredChatAgent(Agent):
         pass
 
     @classmethod
+    @override
     def _get_default_output_parser(
         cls,
         llm: Optional[BaseLanguageModel] = None,
@@ -78,10 +80,12 @@ class StructuredChatAgent(Agent):
         return StructuredChatOutputParserWithRetries.from_llm(llm=llm)
 
     @property
+    @override
     def _stop(self) -> list[str]:
         return ["Observation:"]
 
     @classmethod
+    @override
     def create_prompt(
         cls,
         tools: Sequence[BaseTool],
@@ -276,6 +280,7 @@ def create_structured_chat_agent(
                     ("human", human),
                 ]
             )
+
     """  # noqa: E501
     missing_vars = {"tools", "tool_names", "agent_scratchpad"}.difference(
         prompt.input_variables + list(prompt.partial_variables),

langchain/agents/structured_chat/output_parser.py

@@ -10,6 +10,7 @@ from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.exceptions import OutputParserException
 from langchain_core.language_models import BaseLanguageModel
 from pydantic import Field
+from typing_extensions import override
 
 from langchain.agents.agent import AgentOutputParser
 from langchain.agents.structured_chat.prompt import FORMAT_INSTRUCTIONS
@@ -27,10 +28,12 @@ class StructuredChatOutputParser(AgentOutputParser):
     pattern: Pattern = re.compile(r"```(?:json\s+)?(\W.*?)```", re.DOTALL)
     """Regex pattern to parse the output."""
 
+    @override
     def get_format_instructions(self) -> str:
         """Returns formatting instructions for the given output parser."""
         return self.format_instructions
 
+    @override
     def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
         try:
             action_match = self.pattern.search(text)
@@ -65,9 +68,11 @@ class StructuredChatOutputParserWithRetries(AgentOutputParser):
     output_fixing_parser: Optional[OutputFixingParser] = None
     """The output fixing parser to use."""
 
+    @override
     def get_format_instructions(self) -> str:
         return FORMAT_INSTRUCTIONS
 
+    @override
     def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
         try:
             if self.output_fixing_parser is not None:
@@ -83,6 +88,14 @@ class StructuredChatOutputParserWithRetries(AgentOutputParser):
         llm: Optional[BaseLanguageModel] = None,
         base_parser: Optional[StructuredChatOutputParser] = None,
     ) -> StructuredChatOutputParserWithRetries:
+        """Create a StructuredChatOutputParserWithRetries from a language model.
+
+        Args:
+            llm: The language model to use.
+            base_parser: An optional StructuredChatOutputParser to use.
+        Returns:
+            An instance of StructuredChatOutputParserWithRetries.
+        """
         if llm is not None:
             base_parser = base_parser or StructuredChatOutputParser()
             output_fixing_parser: OutputFixingParser = OutputFixingParser.from_llm(

langchain/agents/tool_calling_agent/base.py

@@ -85,6 +85,7 @@ def create_tool_calling_agent(
         The agent prompt must have an `agent_scratchpad` key that is a
             ``MessagesPlaceholder``. Intermediate agent actions and tool output
             messages will be passed in here.
+
     """
     missing_vars = {"agent_scratchpad"}.difference(
         prompt.input_variables + list(prompt.partial_variables),

langchain/agents/tools.py

@@ -7,6 +7,7 @@ from langchain_core.callbacks import (
     CallbackManagerForToolRun,
 )
 from langchain_core.tools import BaseTool, tool
+from typing_extensions import override
 
 
 class InvalidTool(BaseTool):
@@ -17,6 +18,7 @@ class InvalidTool(BaseTool):
     description: str = "Called when tool name is invalid. Suggests valid tool names."
     """Description of the tool."""
 
+    @override
     def _run(
         self,
         requested_tool_name: str,
@@ -30,6 +32,7 @@ class InvalidTool(BaseTool):
             f"try one of [{available_tool_names_str}]."
         )
 
+    @override
     async def _arun(
         self,
         requested_tool_name: str,

langchain/agents/xml/base.py

@@ -10,6 +10,7 @@ from langchain_core.prompts.chat import AIMessagePromptTemplate, ChatPromptTempl
 from langchain_core.runnables import Runnable, RunnablePassthrough
 from langchain_core.tools import BaseTool
 from langchain_core.tools.render import ToolsRenderer, render_text_description
+from typing_extensions import override
 
 from langchain.agents.agent import BaseSingleActionAgent
 from langchain.agents.format_scratchpad import format_xml
@@ -36,7 +37,6 @@ class XMLAgent(BaseSingleActionAgent):
             tools = ...
             model =
 
-
     """
 
     tools: list[BaseTool]
@@ -45,11 +45,13 @@ class XMLAgent(BaseSingleActionAgent):
     """Chain to use to predict action."""
 
     @property
+    @override
     def input_keys(self) -> list[str]:
         return ["input"]
 
     @staticmethod
     def get_default_prompt() -> ChatPromptTemplate:
+        """Return the default prompt for the XML agent."""
         base_prompt = ChatPromptTemplate.from_template(agent_instructions)
         return base_prompt + AIMessagePromptTemplate.from_template(
             "{intermediate_steps}",
@@ -57,8 +59,10 @@ class XMLAgent(BaseSingleActionAgent):
 
     @staticmethod
     def get_default_output_parser() -> XMLAgentOutputParser:
+        """Return an XMLAgentOutputParser."""
         return XMLAgentOutputParser()
 
+    @override
     def plan(
         self,
         intermediate_steps: list[tuple[AgentAction, str]],
@@ -83,6 +87,7 @@ class XMLAgent(BaseSingleActionAgent):
         response = self.llm_chain(inputs, callbacks=callbacks)
         return response[self.llm_chain.output_key]
 
+    @override
     async def aplan(
         self,
         intermediate_steps: list[tuple[AgentAction, str]],
@@ -203,6 +208,7 @@ def create_xml_agent(
             Question: {input}
             {agent_scratchpad}'''
             prompt = PromptTemplate.from_template(template)
+
     """  # noqa: E501
     missing_vars = {"tools", "agent_scratchpad"}.difference(
         prompt.input_variables + list(prompt.partial_variables),

langchain/callbacks/streaming_aiter.py

@@ -6,6 +6,8 @@ from typing import Any, Literal, Union, cast
 
 from langchain_core.callbacks import AsyncCallbackHandler
 from langchain_core.outputs import LLMResult
+from langchain_core.v1.messages import AIMessage
+from typing_extensions import override
 
 # TODO If used by two LLM runs in parallel this won't work as expected
 
@@ -19,12 +21,15 @@ class AsyncIteratorCallbackHandler(AsyncCallbackHandler):
 
     @property
     def always_verbose(self) -> bool:
+        """Always verbose."""
         return True
 
     def __init__(self) -> None:
+        """Instantiate AsyncIteratorCallbackHandler."""
         self.queue = asyncio.Queue()
         self.done = asyncio.Event()
 
+    @override
     async def on_llm_start(
         self,
         serialized: dict[str, Any],
@@ -34,19 +39,25 @@ class AsyncIteratorCallbackHandler(AsyncCallbackHandler):
         # If two calls are made in a row, this resets the state
         self.done.clear()
 
+    @override
     async def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
         if token is not None and token != "":
             self.queue.put_nowait(token)
 
-    async def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None:
+    @override
+    async def on_llm_end(
+        self, response: Union[LLMResult, AIMessage], **kwargs: Any
+    ) -> None:
         self.done.set()
 
+    @override
     async def on_llm_error(self, error: BaseException, **kwargs: Any) -> None:
         self.done.set()
 
     # TODO implement the other methods
 
     async def aiter(self) -> AsyncIterator[str]:
+        """Asynchronous iterator that yields tokens."""
         while not self.queue.empty() or not self.done.is_set():
             # Wait for the next token in the queue,
             # but stop waiting if the done event is set
@@ -65,7 +76,7 @@ class AsyncIteratorCallbackHandler(AsyncCallbackHandler):
                 other.pop().cancel()
 
             # Extract the value of the first completed task
-            token_or_done = cast(Union[str, Literal[True]], done.pop().result())
+            token_or_done = cast("Union[str, Literal[True]]", done.pop().result())
 
             # If the extracted value is the boolean True, the done event was set
             if token_or_done is True:

langchain/callbacks/streaming_aiter_final_only.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
 
-from typing import Any, Optional
+from typing import Any, Optional, Union
 
 from langchain_core.outputs import LLMResult
+from langchain_core.v1.messages import AIMessage
+from typing_extensions import override
 
 from langchain.callbacks.streaming_aiter import AsyncIteratorCallbackHandler
 
@@ -15,6 +17,7 @@ class AsyncFinalIteratorCallbackHandler(AsyncIteratorCallbackHandler):
     """
 
     def append_to_last_tokens(self, token: str) -> None:
+        """Append token to the last tokens."""
         self.last_tokens.append(token)
         self.last_tokens_stripped.append(token.strip())
         if len(self.last_tokens) > len(self.answer_prefix_tokens):
@@ -22,6 +25,7 @@ class AsyncFinalIteratorCallbackHandler(AsyncIteratorCallbackHandler):
             self.last_tokens_stripped.pop(0)
 
     def check_if_answer_reached(self) -> bool:
+        """Check if the answer has been reached."""
         if self.strip_tokens:
             return self.last_tokens_stripped == self.answer_prefix_tokens_stripped
         return self.last_tokens == self.answer_prefix_tokens
@@ -60,6 +64,7 @@ class AsyncFinalIteratorCallbackHandler(AsyncIteratorCallbackHandler):
         self.stream_prefix = stream_prefix
         self.answer_reached = False
 
+    @override
     async def on_llm_start(
         self,
         serialized: dict[str, Any],
@@ -70,10 +75,14 @@ class AsyncFinalIteratorCallbackHandler(AsyncIteratorCallbackHandler):
         self.done.clear()
         self.answer_reached = False
 
-    async def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None:
+    @override
+    async def on_llm_end(
+        self, response: Union[LLMResult, AIMessage], **kwargs: Any
+    ) -> None:
         if self.answer_reached:
             self.done.set()
 
+    @override
     async def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
         # Remember the last n tokens, where n = len(answer_prefix_tokens)
         self.append_to_last_tokens(token)

langchain/callbacks/streaming_stdout_final_only.py

@@ -4,6 +4,7 @@ import sys
 from typing import Any, Optional
 
 from langchain_core.callbacks import StreamingStdOutCallbackHandler
+from typing_extensions import override
 
 DEFAULT_ANSWER_PREFIX_TOKENS = ["Final", "Answer", ":"]
 
@@ -16,6 +17,7 @@ class FinalStreamingStdOutCallbackHandler(StreamingStdOutCallbackHandler):
     """
 
     def append_to_last_tokens(self, token: str) -> None:
+        """Append token to the last tokens."""
         self.last_tokens.append(token)
         self.last_tokens_stripped.append(token.strip())
         if len(self.last_tokens) > len(self.answer_prefix_tokens):
@@ -23,6 +25,7 @@ class FinalStreamingStdOutCallbackHandler(StreamingStdOutCallbackHandler):
             self.last_tokens_stripped.pop(0)
 
     def check_if_answer_reached(self) -> bool:
+        """Check if the answer has been reached."""
         if self.strip_tokens:
             return self.last_tokens_stripped == self.answer_prefix_tokens_stripped
         return self.last_tokens == self.answer_prefix_tokens
@@ -61,6 +64,7 @@ class FinalStreamingStdOutCallbackHandler(StreamingStdOutCallbackHandler):
         self.stream_prefix = stream_prefix
         self.answer_reached = False
 
+    @override
     def on_llm_start(
         self,
         serialized: dict[str, Any],
@@ -70,6 +74,7 @@ class FinalStreamingStdOutCallbackHandler(StreamingStdOutCallbackHandler):
         """Run when LLM starts running."""
         self.answer_reached = False
 
+    @override
     def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
         """Run on new LLM token. Only available when streaming is enabled."""
 

langchain/callbacks/tracers/logging.py

@@ -7,6 +7,7 @@ from uuid import UUID
 from langchain_core.exceptions import TracerException
 from langchain_core.tracers.stdout import FunctionCallbackHandler
 from langchain_core.utils.input import get_bolded_text, get_colored_text
+from typing_extensions import override
 
 
 class LoggingCallbackHandler(FunctionCallbackHandler):
@@ -21,6 +22,15 @@ class LoggingCallbackHandler(FunctionCallbackHandler):
         extra: Optional[dict] = None,
         **kwargs: Any,
     ) -> None:
+        """
+        Initialize the LoggingCallbackHandler.
+
+        Args:
+            logger: the logger to use for logging
+            log_level: the logging level (default: logging.INFO)
+            extra: the extra context to log (default: None)
+            **kwargs:
+        """
         log_method = getattr(logger, logging.getLevelName(level=log_level).lower())
 
         def callback(text: str) -> None:
@@ -28,6 +38,7 @@ class LoggingCallbackHandler(FunctionCallbackHandler):
 
         super().__init__(function=callback, **kwargs)
 
+    @override
     def on_text(
         self,
         text: str,

langchain/chains/api/base.py

@@ -191,6 +191,7 @@ try:
             )
             async for event in events:
                 event["messages"][-1].pretty_print()
+
         """  # noqa: E501
 
         api_request_chain: LLMChain
@@ -386,7 +387,10 @@ try:
 except ImportError:
 
     class APIChain:  # type: ignore[no-redef]
-        def __init__(self, *args: Any, **kwargs: Any) -> None:
+        """Raise an ImportError if APIChain is used without langchain_community."""
+
+        def __init__(self, *_: Any, **__: Any) -> None:
+            """Raise an ImportError if APIChain is used without langchain_community."""
             msg = (
                 "To use the APIChain, you must install the langchain_community package."
                 "pip install langchain_community"

langchain/chains/base.py

@@ -108,6 +108,7 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC):
         arbitrary_types_allowed=True,
     )
 
+    @override
     def get_input_schema(
         self,
         config: Optional[RunnableConfig] = None,
@@ -115,6 +116,7 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC):
         # This is correct, but pydantic typings/mypy don't think so.
         return create_model("ChainInput", **dict.fromkeys(self.input_keys, (Any, None)))
 
+    @override
     def get_output_schema(
         self,
         config: Optional[RunnableConfig] = None,
@@ -409,7 +411,7 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC):
 
         return self.invoke(
             inputs,
-            cast(RunnableConfig, {k: v for k, v in config.items() if v is not None}),
+            cast("RunnableConfig", {k: v for k, v in config.items() if v is not None}),
             return_only_outputs=return_only_outputs,
             include_run_info=include_run_info,
         )
@@ -459,7 +461,7 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC):
         }
         return await self.ainvoke(
             inputs,
-            cast(RunnableConfig, {k: v for k, v in config.items() if k is not None}),
+            cast("RunnableConfig", {k: v for k, v in config.items() if k is not None}),
             return_only_outputs=return_only_outputs,
             include_run_info=include_run_info,
         )
@@ -616,6 +618,7 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC):
                 context = "Weather report for Boise, Idaho on 07/03/23..."
                 chain.run(question=question, context=context)
                 # -> "The temperature in Boise is..."
+
         """
         # Run at start to make sure this is possible/defined
         _output_key = self._run_output_key
@@ -690,6 +693,7 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC):
                 context = "Weather report for Boise, Idaho on 07/03/23..."
                 await chain.arun(question=question, context=context)
                 # -> "The temperature in Boise is..."
+
         """
         if len(self.output_keys) != 1:
             msg = (
@@ -744,6 +748,7 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC):
 
                 chain.dict(exclude_unset=True)
                 # -> {"_type": "foo", "verbose": False, ...}
+
         """
         _dict = super().dict(**kwargs)
         with contextlib.suppress(NotImplementedError):
@@ -763,6 +768,7 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC):
             .. code-block:: python
 
                 chain.save(file_path="path/chain.yaml")
+
         """
         if self.memory is not None:
             msg = "Saving of memory is not yet supported."

langchain/chains/combine_documents/base.py

@@ -14,6 +14,7 @@ from langchain_core.runnables.config import RunnableConfig
 from langchain_core.utils.pydantic import create_model
 from langchain_text_splitters import RecursiveCharacterTextSplitter, TextSplitter
 from pydantic import BaseModel, Field
+from typing_extensions import override
 
 from langchain.chains.base import Chain
 
@@ -46,6 +47,7 @@ class BaseCombineDocumentsChain(Chain, ABC):
     input_key: str = "input_documents"  #: :meta private:
     output_key: str = "output_text"  #: :meta private:
 
+    @override
     def get_input_schema(
         self,
         config: Optional[RunnableConfig] = None,
@@ -55,6 +57,7 @@ class BaseCombineDocumentsChain(Chain, ABC):
             **{self.input_key: (list[Document], None)},
         )
 
+    @override
     def get_output_schema(
         self,
         config: Optional[RunnableConfig] = None,
@@ -80,7 +83,7 @@ class BaseCombineDocumentsChain(Chain, ABC):
         """
         return [self.output_key]
 
-    def prompt_length(self, docs: list[Document], **kwargs: Any) -> Optional[int]:
+    def prompt_length(self, docs: list[Document], **kwargs: Any) -> Optional[int]:  # noqa: ARG002
         """Return the prompt length given the documents passed in.
 
         This can be used by a caller to determine whether passing in a list
@@ -231,6 +234,7 @@ class AnalyzeDocumentChain(Chain):
                     input_documents=itemgetter("input_document") | split_text,
                 ) | chain.pick("output_text")
             )
+
     """
 
     input_key: str = "input_document"  #: :meta private:
@@ -253,6 +257,7 @@ class AnalyzeDocumentChain(Chain):
         """
         return self.combine_docs_chain.output_keys
 
+    @override
     def get_input_schema(
         self,
         config: Optional[RunnableConfig] = None,
@@ -262,6 +267,7 @@ class AnalyzeDocumentChain(Chain):
             **{self.input_key: (str, None)},
         )
 
+    @override
     def get_output_schema(
         self,
         config: Optional[RunnableConfig] = None,

langchain/chains/combine_documents/map_reduce.py

@@ -10,6 +10,7 @@ from langchain_core.documents import Document
 from langchain_core.runnables.config import RunnableConfig
 from langchain_core.utils.pydantic import create_model
 from pydantic import BaseModel, ConfigDict, model_validator
+from typing_extensions import override
 
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
 from langchain.chains.combine_documents.reduce import ReduceDocumentsChain
@@ -98,6 +99,7 @@ class MapReduceDocumentsChain(BaseCombineDocumentsChain):
                 llm_chain=llm_chain,
                 reduce_documents_chain=reduce_documents_chain,
             )
+
     """
 
     llm_chain: LLMChain
@@ -111,6 +113,7 @@ class MapReduceDocumentsChain(BaseCombineDocumentsChain):
     return_intermediate_steps: bool = False
     """Return the results of the map steps in the output."""
 
+    @override
     def get_output_schema(
         self,
         config: Optional[RunnableConfig] = None,