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

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)
@@ -42,14 +45,15 @@ class StructuredChatOutputParser(AgentOutputParser):
                     response = response[0]
                 if response["action"] == "Final Answer":
                     return AgentFinish({"output": response["action_input"]}, text)
-                else:
-                    return AgentAction(
-                        response["action"], response.get("action_input", {}), text
-                    )
-            else:
-                return AgentFinish({"output": text}, text)
+                return AgentAction(
+                    response["action"],
+                    response.get("action_input", {}),
+                    text,
+                )
+            return AgentFinish({"output": text}, text)
         except Exception as e:
-            raise OutputParserException(f"Could not parse LLM output: {text}") from e
+            msg = f"Could not parse LLM output: {text}"
+            raise OutputParserException(msg) from e
 
     @property
     def _type(self) -> str:
@@ -64,20 +68,19 @@ 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:
-                parsed_obj: Union[AgentAction, AgentFinish] = (
-                    self.output_fixing_parser.parse(text)
-                )
-            else:
-                parsed_obj = self.base_parser.parse(text)
-            return parsed_obj
+                return self.output_fixing_parser.parse(text)
+            return self.base_parser.parse(text)
         except Exception as e:
-            raise OutputParserException(f"Could not parse LLM output: {text}") from e
+            msg = f"Could not parse LLM output: {text}"
+            raise OutputParserException(msg) from e
 
     @classmethod
     def from_llm(
@@ -85,16 +88,24 @@ 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(
-                llm=llm, parser=base_parser
+                llm=llm,
+                parser=base_parser,
             )
             return cls(output_fixing_parser=output_fixing_parser)
-        elif base_parser is not None:
+        if base_parser is not None:
             return cls(base_parser=base_parser)
-        else:
-            return cls()
+        return cls()
 
     @property
     def _type(self) -> str:

langchain/agents/structured_chat/prompt.py

@@ -1,5 +1,4 @@
-# flake8: noqa
-PREFIX = """Respond to the human as helpfully and accurately as possible. You have access to the following tools:"""
+PREFIX = """Respond to the human as helpfully and accurately as possible. You have access to the following tools:"""  # noqa: E501
 FORMAT_INSTRUCTIONS = """Use a json blob to specify a tool by providing an action key (tool name) and an action_input key (tool input).
 
 Valid "action" values: "Final Answer" or {tool_names}
@@ -30,6 +29,6 @@ Action:
   "action": "Final Answer",
   "action_input": "Final response to human"
 }}}}
-```"""
+```"""  # noqa: E501
 SUFFIX = """Begin! Reminder to ALWAYS respond with a valid json blob of a single action. Use tools if necessary. Respond directly if appropriate. Format is Action:```$JSON_BLOB```then Observation:.
-Thought:"""
+Thought:"""  # noqa: E501

langchain/agents/tool_calling_agent/base.py

@@ -85,25 +85,27 @@ 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)
+        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 not hasattr(llm, "bind_tools"):
+        msg = "This function requires a bind_tools() method be implemented on the LLM."
         raise ValueError(
-            "This function requires a .bind_tools method be implemented on the LLM.",
+            msg,
         )
     llm_with_tools = llm.bind_tools(tools)
 
-    agent = (
+    return (
         RunnablePassthrough.assign(
-            agent_scratchpad=lambda x: message_formatter(x["intermediate_steps"])
+            agent_scratchpad=lambda x: message_formatter(x["intermediate_steps"]),
         )
         | prompt
         | llm_with_tools
         | ToolsAgentOutputParser()
     )
-    return agent

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,
@@ -24,12 +26,13 @@ class InvalidTool(BaseTool):
         run_manager: Optional[CallbackManagerForToolRun] = None,
     ) -> str:
         """Use the tool."""
-        available_tool_names_str = ", ".join([tool for tool in available_tool_names])
+        available_tool_names_str = ", ".join(list(available_tool_names))
         return (
             f"{requested_tool_name} is not a valid tool, "
             f"try one of [{available_tool_names_str}]."
         )
 
+    @override
     async def _arun(
         self,
         requested_tool_name: str,
@@ -37,7 +40,7 @@ class InvalidTool(BaseTool):
         run_manager: Optional[AsyncCallbackManagerForToolRun] = None,
     ) -> str:
         """Use the tool asynchronously."""
-        available_tool_names_str = ", ".join([tool for tool in available_tool_names])
+        available_tool_names_str = ", ".join(list(available_tool_names))
         return (
             f"{requested_tool_name} is not a valid tool, "
             f"try one of [{available_tool_names_str}]."

langchain/agents/utils.py

@@ -15,6 +15,5 @@ def validate_tools_single_input(class_name: str, tools: Sequence[BaseTool]) -> N
     """
     for tool in tools:
         if not tool.is_single_input:
-            raise ValueError(
-                f"{class_name} does not support multi-input tool {tool.name}."
-            )
+            msg = f"{class_name} does not support multi-input tool {tool.name}."
+            raise ValueError(msg)

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,20 +45,24 @@ 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}"
+            "{intermediate_steps}",
         )
 
     @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,12 +208,14 @@ 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)
+        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)),
@@ -220,7 +227,7 @@ def create_xml_agent(
     else:
         llm_with_stop = llm
 
-    agent = (
+    return (
         RunnablePassthrough.assign(
             agent_scratchpad=lambda x: format_xml(x["intermediate_steps"]),
         )
@@ -228,4 +235,3 @@ def create_xml_agent(
         | llm_with_stop
         | XMLAgentOutputParser()
     )
-    return agent

langchain/agents/xml/prompt.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 # TODO: deprecate
 agent_instructions = """You are a helpful assistant. Help the user answer any questions.
 
@@ -19,4 +18,4 @@ When you are done, respond with a final answer between <final_answer></final_ans
 
 Begin!
 
-Question: {question}"""
+Question: {question}"""  # noqa: E501

langchain/cache.py

@@ -53,20 +53,20 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "FullLLMCache",
-    "SQLAlchemyCache",
-    "SQLiteCache",
-    "UpstashRedisCache",
-    "RedisCache",
-    "RedisSemanticCache",
-    "GPTCache",
-    "MomentoCache",
-    "InMemoryCache",
+    "AstraDBCache",
+    "AstraDBSemanticCache",
+    "AzureCosmosDBSemanticCache",
     "CassandraCache",
     "CassandraSemanticCache",
+    "FullLLMCache",
     "FullMd5LLMCache",
+    "GPTCache",
+    "InMemoryCache",
+    "MomentoCache",
+    "RedisCache",
+    "RedisSemanticCache",
+    "SQLAlchemyCache",
     "SQLAlchemyMd5Cache",
-    "AstraDBCache",
-    "AstraDBSemanticCache",
-    "AzureCosmosDBSemanticCache",
+    "SQLiteCache",
+    "UpstashRedisCache",
 ]

langchain/callbacks/__init__.py

@@ -107,33 +107,33 @@ __all__ = [
     "AimCallbackHandler",
     "ArgillaCallbackHandler",
     "ArizeCallbackHandler",
-    "PromptLayerCallbackHandler",
     "ArthurCallbackHandler",
+    "AsyncIteratorCallbackHandler",
     "ClearMLCallbackHandler",
     "CometCallbackHandler",
     "ContextCallbackHandler",
     "FileCallbackHandler",
+    "FinalStreamingStdOutCallbackHandler",
+    "FlyteCallbackHandler",
     "HumanApprovalCallbackHandler",
     "InfinoCallbackHandler",
-    "MlflowCallbackHandler",
+    "LLMThoughtLabeler",
     "LLMonitorCallbackHandler",
+    "LabelStudioCallbackHandler",
+    "LangChainTracer",
+    "MlflowCallbackHandler",
     "OpenAICallbackHandler",
+    "PromptLayerCallbackHandler",
+    "SageMakerCallbackHandler",
     "StdOutCallbackHandler",
-    "AsyncIteratorCallbackHandler",
     "StreamingStdOutCallbackHandler",
-    "FinalStreamingStdOutCallbackHandler",
-    "LLMThoughtLabeler",
-    "LangChainTracer",
     "StreamlitCallbackHandler",
+    "TrubricsCallbackHandler",
     "WandbCallbackHandler",
     "WhyLabsCallbackHandler",
+    "collect_runs",
     "get_openai_callback",
     "tracing_enabled",
     "tracing_v2_enabled",
-    "collect_runs",
     "wandb_tracing_enabled",
-    "FlyteCallbackHandler",
-    "SageMakerCallbackHandler",
-    "LabelStudioCallbackHandler",
-    "TrubricsCallbackHandler",
 ]

langchain/callbacks/aim_callback.py

@@ -27,7 +27,7 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "import_aim",
-    "BaseMetadataCallbackHandler",
     "AimCallbackHandler",
+    "BaseMetadataCallbackHandler",
+    "import_aim",
 ]

langchain/callbacks/argilla_callback.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "ArgillaCallbackHandler": "langchain_community.callbacks.argilla_callback"
+    "ArgillaCallbackHandler": "langchain_community.callbacks.argilla_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/arize_callback.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "ArizeCallbackHandler": "langchain_community.callbacks.arize_callback"
+    "ArizeCallbackHandler": "langchain_community.callbacks.arize_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/arthur_callback.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "ArthurCallbackHandler": "langchain_community.callbacks.arthur_callback"
+    "ArthurCallbackHandler": "langchain_community.callbacks.arthur_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/base.py

@@ -16,14 +16,14 @@ from langchain_core.callbacks import (
 )
 
 __all__ = [
-    "RetrieverManagerMixin",
-    "LLMManagerMixin",
-    "ChainManagerMixin",
-    "ToolManagerMixin",
-    "CallbackManagerMixin",
-    "RunManagerMixin",
-    "BaseCallbackHandler",
     "AsyncCallbackHandler",
+    "BaseCallbackHandler",
     "BaseCallbackManager",
+    "CallbackManagerMixin",
     "Callbacks",
+    "ChainManagerMixin",
+    "LLMManagerMixin",
+    "RetrieverManagerMixin",
+    "RunManagerMixin",
+    "ToolManagerMixin",
 ]

langchain/callbacks/clearml_callback.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "ClearMLCallbackHandler": "langchain_community.callbacks.clearml_callback"
+    "ClearMLCallbackHandler": "langchain_community.callbacks.clearml_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/comet_ml_callback.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "CometCallbackHandler": "langchain_community.callbacks.comet_ml_callback"
+    "CometCallbackHandler": "langchain_community.callbacks.comet_ml_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/confident_callback.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "DeepEvalCallbackHandler": "langchain_community.callbacks.confident_callback"
+    "DeepEvalCallbackHandler": "langchain_community.callbacks.confident_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/context_callback.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "ContextCallbackHandler": "langchain_community.callbacks.context_callback"
+    "ContextCallbackHandler": "langchain_community.callbacks.context_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/flyte_callback.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "FlyteCallbackHandler": "langchain_community.callbacks.flyte_callback"
+    "FlyteCallbackHandler": "langchain_community.callbacks.flyte_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/human.py

@@ -27,7 +27,7 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "HumanRejectedException",
-    "HumanApprovalCallbackHandler",
     "AsyncHumanApprovalCallbackHandler",
+    "HumanApprovalCallbackHandler",
+    "HumanRejectedException",
 ]

langchain/callbacks/infino_callback.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "InfinoCallbackHandler": "langchain_community.callbacks.infino_callback"
+    "InfinoCallbackHandler": "langchain_community.callbacks.infino_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/labelstudio_callback.py

@@ -27,7 +27,7 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
+    "LabelStudioCallbackHandler",
     "LabelStudioMode",
     "get_default_label_configs",
-    "LabelStudioCallbackHandler",
 ]

langchain/callbacks/llmonitor_callback.py

@@ -11,7 +11,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "LLMonitorCallbackHandler": "langchain_community.callbacks.llmonitor_callback"
+    "LLMonitorCallbackHandler": "langchain_community.callbacks.llmonitor_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/manager.py

@@ -58,7 +58,7 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "ahandle_event",
+    "AsyncCallbackManager",
     "AsyncCallbackManagerForChainGroup",
     "AsyncCallbackManagerForChainRun",
     "AsyncCallbackManagerForLLMRun",
@@ -66,7 +66,6 @@ __all__ = [
     "AsyncCallbackManagerForToolRun",
     "AsyncParentRunManager",
     "AsyncRunManager",
-    "atrace_as_chain_group",
     "BaseRunManager",
     "CallbackManager",
     "CallbackManagerForChainGroup",
@@ -75,13 +74,14 @@ __all__ = [
     "CallbackManagerForRetrieverRun",
     "CallbackManagerForToolRun",
     "Callbacks",
-    "AsyncCallbackManager",
+    "ParentRunManager",
+    "RunManager",
+    "ahandle_event",
+    "atrace_as_chain_group",
     "collect_runs",
     "env_var_is_set",
     "get_openai_callback",
     "handle_event",
-    "ParentRunManager",
-    "RunManager",
     "trace_as_chain_group",
     "tracing_enabled",
     "tracing_v2_enabled",

langchain/callbacks/mlflow_callback.py

@@ -31,8 +31,8 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
+    "MlflowCallbackHandler",
+    "MlflowLogger",
     "analyze_text",
     "construct_html_from_prompt_and_generation",
-    "MlflowLogger",
-    "MlflowCallbackHandler",
 ]

langchain/callbacks/openai_info.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "OpenAICallbackHandler": "langchain_community.callbacks.openai_info"
+    "OpenAICallbackHandler": "langchain_community.callbacks.openai_info",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/promptlayer_callback.py

@@ -11,7 +11,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "PromptLayerCallbackHandler": "langchain_community.callbacks.promptlayer_callback"
+    "PromptLayerCallbackHandler": "langchain_community.callbacks.promptlayer_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/callbacks/sagemaker_callback.py

@@ -11,7 +11,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "SageMakerCallbackHandler": "langchain_community.callbacks.sagemaker_callback"
+    "SageMakerCallbackHandler": "langchain_community.callbacks.sagemaker_callback",
 }
 
 _import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)