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

langchain/agents/agent_iterator.py

@@ -147,7 +147,9 @@ class AgentExecutorIterator:
         self.iterations += 1
         self.time_elapsed = time.time() - self.start_time
         logger.debug(
-            f"Agent Iterations: {self.iterations} ({self.time_elapsed:.2f}s elapsed)"
+            "Agent Iterations: %s (%.2fs elapsed)",
+            self.iterations,
+            self.time_elapsed,
         )
 
     def make_final_outputs(
@@ -155,19 +157,28 @@ class AgentExecutorIterator:
         outputs: dict[str, Any],
         run_manager: Union[CallbackManagerForChainRun, AsyncCallbackManagerForChainRun],
     ) -> AddableDict:
+        """Make final outputs for the iterator.
+
+        Args:
+            outputs: The outputs from the agent executor.
+            run_manager: The run manager to use for callbacks.
+        """
         # have access to intermediate steps by design in iterator,
         # so return only outputs may as well always be true.
 
         prepared_outputs = AddableDict(
             self.agent_executor.prep_outputs(
-                self.inputs, outputs, return_only_outputs=True
-            )
+                self.inputs,
+                outputs,
+                return_only_outputs=True,
+            ),
         )
         if self.include_run_info:
             prepared_outputs[RUN_KEY] = RunInfo(run_id=run_manager.run_id)
         return prepared_outputs
 
     def __iter__(self: AgentExecutorIterator) -> Iterator[AddableDict]:
+        """Create an async iterator for the AgentExecutor."""
         logger.debug("Initialising AgentExecutorIterator")
         self.reset()
         callback_manager = CallbackManager.configure(
@@ -186,13 +197,14 @@ class AgentExecutorIterator:
             name=self.run_name,
         )
         try:
-            while self.agent_executor._should_continue(
-                self.iterations, self.time_elapsed
+            while self.agent_executor._should_continue(  # noqa: SLF001
+                self.iterations,
+                self.time_elapsed,
             ):
                 # take the next step: this plans next action, executes it,
                 # yielding action and observation as they are generated
                 next_step_seq: NextStepOutput = []
-                for chunk in self.agent_executor._iter_next_step(
+                for chunk in self.agent_executor._iter_next_step(  # noqa: SLF001
                     self.name_to_tool_map,
                     self.color_mapping,
                     self.inputs,
@@ -209,7 +221,7 @@ class AgentExecutorIterator:
                             yield AddableDict(steps=[chunk], messages=chunk.messages)
 
                 # convert iterator output to format handled by _process_next_step_output
-                next_step = self.agent_executor._consume_next_step(next_step_seq)
+                next_step = self.agent_executor._consume_next_step(next_step_seq)  # noqa: SLF001
                 # update iterations and time elapsed
                 self.update_iterations()
                 # decide if this is the final output
@@ -253,13 +265,14 @@ class AgentExecutorIterator:
         )
         try:
             async with asyncio_timeout(self.agent_executor.max_execution_time):
-                while self.agent_executor._should_continue(
-                    self.iterations, self.time_elapsed
+                while self.agent_executor._should_continue(  # noqa: SLF001
+                    self.iterations,
+                    self.time_elapsed,
                 ):
                     # take the next step: this plans next action, executes it,
                     # yielding action and observation as they are generated
                     next_step_seq: NextStepOutput = []
-                    async for chunk in self.agent_executor._aiter_next_step(
+                    async for chunk in self.agent_executor._aiter_next_step(  # noqa: SLF001
                         self.name_to_tool_map,
                         self.color_mapping,
                         self.inputs,
@@ -272,20 +285,23 @@ class AgentExecutorIterator:
                         if self.yield_actions:
                             if isinstance(chunk, AgentAction):
                                 yield AddableDict(
-                                    actions=[chunk], messages=chunk.messages
+                                    actions=[chunk],
+                                    messages=chunk.messages,
                                 )
                             elif isinstance(chunk, AgentStep):
                                 yield AddableDict(
-                                    steps=[chunk], messages=chunk.messages
+                                    steps=[chunk],
+                                    messages=chunk.messages,
                                 )
 
                     # convert iterator output to format handled by _process_next_step
-                    next_step = self.agent_executor._consume_next_step(next_step_seq)
+                    next_step = self.agent_executor._consume_next_step(next_step_seq)  # noqa: SLF001
                     # update iterations and time elapsed
                     self.update_iterations()
                     # decide if this is the final output
                     output = await self._aprocess_next_step_output(
-                        next_step, run_manager
+                        next_step,
+                        run_manager,
                     )
                     is_final = "intermediate_step" not in output
                     # yield the final output always
@@ -317,7 +333,7 @@ class AgentExecutorIterator:
         logger.debug("Processing output of Agent loop step")
         if isinstance(next_step_output, AgentFinish):
             logger.debug(
-                "Hit AgentFinish: _return -> on_chain_end -> run final output logic"
+                "Hit AgentFinish: _return -> on_chain_end -> run final output logic",
             )
             return self._return(next_step_output, run_manager=run_manager)
 
@@ -327,7 +343,7 @@ class AgentExecutorIterator:
         # Check for tool return
         if len(next_step_output) == 1:
             next_step_action = next_step_output[0]
-            tool_return = self.agent_executor._get_tool_return(next_step_action)
+            tool_return = self.agent_executor._get_tool_return(next_step_action)  # noqa: SLF001
             if tool_return is not None:
                 return self._return(tool_return, run_manager=run_manager)
 
@@ -345,7 +361,7 @@ class AgentExecutorIterator:
         logger.debug("Processing output of async Agent loop step")
         if isinstance(next_step_output, AgentFinish):
             logger.debug(
-                "Hit AgentFinish: _areturn -> on_chain_end -> run final output logic"
+                "Hit AgentFinish: _areturn -> on_chain_end -> run final output logic",
             )
             return await self._areturn(next_step_output, run_manager=run_manager)
 
@@ -355,7 +371,7 @@ class AgentExecutorIterator:
         # Check for tool return
         if len(next_step_output) == 1:
             next_step_action = next_step_output[0]
-            tool_return = self.agent_executor._get_tool_return(next_step_action)
+            tool_return = self.agent_executor._get_tool_return(next_step_action)  # noqa: SLF001
             if tool_return is not None:
                 return await self._areturn(tool_return, run_manager=run_manager)
 
@@ -367,7 +383,7 @@ class AgentExecutorIterator:
         """
         logger.warning("Stopping agent prematurely due to triggering stop condition")
         # this manually constructs agent finish with output key
-        output = self.agent_executor._action_agent.return_stopped_response(
+        output = self.agent_executor._action_agent.return_stopped_response(  # noqa: SLF001
             self.agent_executor.early_stopping_method,
             self.intermediate_steps,
             **self.inputs,
@@ -380,7 +396,7 @@ class AgentExecutorIterator:
         the stopped response.
         """
         logger.warning("Stopping agent prematurely due to triggering stop condition")
-        output = self.agent_executor._action_agent.return_stopped_response(
+        output = self.agent_executor._action_agent.return_stopped_response(  # noqa: SLF001
             self.agent_executor.early_stopping_method,
             self.intermediate_steps,
             **self.inputs,
@@ -388,26 +404,34 @@ class AgentExecutorIterator:
         return await self._areturn(output, run_manager=run_manager)
 
     def _return(
-        self, output: AgentFinish, run_manager: CallbackManagerForChainRun
+        self,
+        output: AgentFinish,
+        run_manager: CallbackManagerForChainRun,
     ) -> AddableDict:
         """
         Return the final output of the iterator.
         """
-        returned_output = self.agent_executor._return(
-            output, self.intermediate_steps, run_manager=run_manager
+        returned_output = self.agent_executor._return(  # noqa: SLF001
+            output,
+            self.intermediate_steps,
+            run_manager=run_manager,
         )
         returned_output["messages"] = output.messages
         run_manager.on_chain_end(returned_output)
         return self.make_final_outputs(returned_output, run_manager)
 
     async def _areturn(
-        self, output: AgentFinish, run_manager: AsyncCallbackManagerForChainRun
+        self,
+        output: AgentFinish,
+        run_manager: AsyncCallbackManagerForChainRun,
     ) -> AddableDict:
         """
         Return the final output of the async iterator.
         """
-        returned_output = await self.agent_executor._areturn(
-            output, self.intermediate_steps, run_manager=run_manager
+        returned_output = await self.agent_executor._areturn(  # noqa: SLF001
+            output,
+            self.intermediate_steps,
+            run_manager=run_manager,
         )
         returned_output["messages"] = output.messages
         await run_manager.on_chain_end(returned_output)

langchain/agents/agent_toolkits/__init__.py

@@ -121,12 +121,13 @@ def __getattr__(name: str) -> Any:
         relative_path = as_import_path(Path(__file__).parent, suffix=name)
         old_path = "langchain." + relative_path
         new_path = "langchain_experimental." + relative_path
-        raise ImportError(
+        msg = (
             f"{name} has been moved to langchain experimental. "
             "See https://github.com/langchain-ai/langchain/discussions/11680"
             "for more information.\n"
             f"Please update your import statement from: `{old_path}` to `{new_path}`."
         )
+        raise ImportError(msg)
     return _import_attribute(name)
 
 
@@ -139,28 +140,28 @@ __all__ = [
     "JiraToolkit",
     "JsonToolkit",
     "MultionToolkit",
-    "NasaToolkit",
     "NLAToolkit",
+    "NasaToolkit",
     "O365Toolkit",
     "OpenAPIToolkit",
     "PlayWrightBrowserToolkit",
     "PowerBIToolkit",
-    "SlackToolkit",
-    "SteamToolkit",
     "SQLDatabaseToolkit",
+    "SlackToolkit",
     "SparkSQLToolkit",
+    "SteamToolkit",
     "VectorStoreInfo",
     "VectorStoreRouterToolkit",
     "VectorStoreToolkit",
     "ZapierToolkit",
+    "create_conversational_retrieval_agent",
     "create_json_agent",
     "create_openapi_agent",
     "create_pbi_agent",
     "create_pbi_chat_agent",
+    "create_retriever_tool",
     "create_spark_sql_agent",
     "create_sql_agent",
     "create_vectorstore_agent",
     "create_vectorstore_router_agent",
-    "create_conversational_retrieval_agent",
-    "create_retriever_tool",
 ]

langchain/agents/agent_toolkits/ainetwork/toolkit.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "AINetworkToolkit": "langchain_community.agent_toolkits.ainetwork.toolkit"
+    "AINetworkToolkit": "langchain_community.agent_toolkits.ainetwork.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/amadeus/toolkit.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "AmadeusToolkit": "langchain_community.agent_toolkits.amadeus.toolkit"
+    "AmadeusToolkit": "langchain_community.agent_toolkits.amadeus.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/azure_cognitive_services.py

@@ -13,7 +13,7 @@ if TYPE_CHECKING:
 DEPRECATED_LOOKUP = {
     "AzureCognitiveServicesToolkit": (
         "langchain_community.agent_toolkits.azure_cognitive_services"
-    )
+    ),
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/clickup/toolkit.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "ClickupToolkit": "langchain_community.agent_toolkits.clickup.toolkit"
+    "ClickupToolkit": "langchain_community.agent_toolkits.clickup.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/conversational_retrieval/openai_functions.py

@@ -20,17 +20,17 @@ def _get_default_system_message() -> SystemMessage:
             "Do your best to answer the questions. "
             "Feel free to use any tools available to look up "
             "relevant information, only if necessary"
-        )
+        ),
     )
 
 
 def create_conversational_retrieval_agent(
     llm: BaseLanguageModel,
     tools: list[BaseTool],
-    remember_intermediate_steps: bool = True,
+    remember_intermediate_steps: bool = True,  # noqa: FBT001,FBT002
     memory_key: str = "chat_history",
     system_message: Optional[SystemMessage] = None,
-    verbose: bool = False,
+    verbose: bool = False,  # noqa: FBT001,FBT002
     max_token_limit: int = 2000,
     **kwargs: Any,
 ) -> AgentExecutor:
@@ -58,7 +58,9 @@ def create_conversational_retrieval_agent(
 
     if remember_intermediate_steps:
         memory: BaseMemory = AgentTokenBufferMemory(
-            memory_key=memory_key, llm=llm, max_token_limit=max_token_limit
+            memory_key=memory_key,
+            llm=llm,
+            max_token_limit=max_token_limit,
         )
     else:
         memory = ConversationTokenBufferMemory(

langchain/agents/agent_toolkits/csv/__init__.py

@@ -14,7 +14,7 @@ def __getattr__(name: str) -> Any:
 
         old_path = "langchain." + here + "." + name
         new_path = "langchain_experimental." + here + "." + name
-        raise ImportError(
+        msg = (
             "This agent has been moved to langchain experiment. "
             "This agent relies on python REPL tool under the hood, so to use it "
             "safely please sandbox the python REPL. "
@@ -23,4 +23,6 @@ def __getattr__(name: str) -> Any:
             "To keep using this code as is, install langchain experimental and "
             f"update your import statement from:\n `{old_path}` to `{new_path}`."
         )
-    raise AttributeError(f"{name} does not exist")
+        raise ImportError(msg)
+    msg = f"{name} does not exist"
+    raise AttributeError(msg)

langchain/agents/agent_toolkits/file_management/__init__.py

@@ -15,7 +15,7 @@ if TYPE_CHECKING:
 DEPRECATED_LOOKUP = {
     "FileManagementToolkit": (
         "langchain_community.agent_toolkits.file_management.toolkit"
-    )
+    ),
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/file_management/toolkit.py

@@ -13,7 +13,7 @@ if TYPE_CHECKING:
 DEPRECATED_LOOKUP = {
     "FileManagementToolkit": (
         "langchain_community.agent_toolkits.file_management.toolkit"
-    )
+    ),
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/github/toolkit.py

@@ -51,19 +51,19 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "NoInput",
-    "GetIssue",
+    "BranchName",
     "CommentOnIssue",
-    "GetPR",
-    "CreatePR",
     "CreateFile",
-    "ReadFile",
-    "UpdateFile",
+    "CreatePR",
+    "CreateReviewRequest",
     "DeleteFile",
     "DirectoryPath",
-    "BranchName",
+    "GetIssue",
+    "GetPR",
+    "GitHubToolkit",
+    "NoInput",
+    "ReadFile",
     "SearchCode",
-    "CreateReviewRequest",
     "SearchIssuesAndPRs",
-    "GitHubToolkit",
+    "UpdateFile",
 ]

langchain/agents/agent_toolkits/gitlab/toolkit.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "GitLabToolkit": "langchain_community.agent_toolkits.gitlab.toolkit"
+    "GitLabToolkit": "langchain_community.agent_toolkits.gitlab.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/json/base.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "create_json_agent": "langchain_community.agent_toolkits.json.base"
+    "create_json_agent": "langchain_community.agent_toolkits.json.base",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/multion/toolkit.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "MultionToolkit": "langchain_community.agent_toolkits.multion.toolkit"
+    "MultionToolkit": "langchain_community.agent_toolkits.multion.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/office365/toolkit.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "O365Toolkit": "langchain_community.agent_toolkits.office365.toolkit"
+    "O365Toolkit": "langchain_community.agent_toolkits.office365.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/openapi/base.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "create_openapi_agent": "langchain_community.agent_toolkits.openapi.base"
+    "create_openapi_agent": "langchain_community.agent_toolkits.openapi.base",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/openapi/planner.py

@@ -43,10 +43,10 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
+    "RequestsDeleteToolWithParsing",
     "RequestsGetToolWithParsing",
-    "RequestsPostToolWithParsing",
     "RequestsPatchToolWithParsing",
+    "RequestsPostToolWithParsing",
     "RequestsPutToolWithParsing",
-    "RequestsDeleteToolWithParsing",
     "create_openapi_agent",
 ]

langchain/agents/agent_toolkits/openapi/planner_prompt.py

@@ -83,21 +83,21 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "API_PLANNER_PROMPT",
-    "API_PLANNER_TOOL_NAME",
-    "API_PLANNER_TOOL_DESCRIPTION",
     "API_CONTROLLER_PROMPT",
-    "API_CONTROLLER_TOOL_NAME",
     "API_CONTROLLER_TOOL_DESCRIPTION",
+    "API_CONTROLLER_TOOL_NAME",
     "API_ORCHESTRATOR_PROMPT",
-    "REQUESTS_GET_TOOL_DESCRIPTION",
+    "API_PLANNER_PROMPT",
+    "API_PLANNER_TOOL_DESCRIPTION",
+    "API_PLANNER_TOOL_NAME",
+    "PARSING_DELETE_PROMPT",
     "PARSING_GET_PROMPT",
-    "REQUESTS_POST_TOOL_DESCRIPTION",
-    "PARSING_POST_PROMPT",
-    "REQUESTS_PATCH_TOOL_DESCRIPTION",
     "PARSING_PATCH_PROMPT",
-    "REQUESTS_PUT_TOOL_DESCRIPTION",
+    "PARSING_POST_PROMPT",
     "PARSING_PUT_PROMPT",
     "REQUESTS_DELETE_TOOL_DESCRIPTION",
-    "PARSING_DELETE_PROMPT",
+    "REQUESTS_GET_TOOL_DESCRIPTION",
+    "REQUESTS_PATCH_TOOL_DESCRIPTION",
+    "REQUESTS_POST_TOOL_DESCRIPTION",
+    "REQUESTS_PUT_TOOL_DESCRIPTION",
 ]

langchain/agents/agent_toolkits/openapi/prompt.py

@@ -26,4 +26,4 @@ def __getattr__(name: str) -> Any:
     return _import_attribute(name)
 
 
-__all__ = ["OPENAPI_PREFIX", "OPENAPI_SUFFIX", "DESCRIPTION"]
+__all__ = ["DESCRIPTION", "OPENAPI_PREFIX", "OPENAPI_SUFFIX"]

langchain/agents/agent_toolkits/openapi/toolkit.py

@@ -25,6 +25,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "RequestsToolkit",
     "OpenAPIToolkit",
+    "RequestsToolkit",
 ]

langchain/agents/agent_toolkits/pandas/__init__.py

@@ -14,7 +14,7 @@ def __getattr__(name: str) -> Any:
 
         old_path = "langchain." + here + "." + name
         new_path = "langchain_experimental." + here + "." + name
-        raise ImportError(
+        msg = (
             "This agent has been moved to langchain experiment. "
             "This agent relies on python REPL tool under the hood, so to use it "
             "safely please sandbox the python REPL. "
@@ -23,4 +23,6 @@ def __getattr__(name: str) -> Any:
             "To keep using this code as is, install langchain experimental and "
             f"update your import statement from:\n `{old_path}` to `{new_path}`."
         )
-    raise AttributeError(f"{name} does not exist")
+        raise ImportError(msg)
+    msg = f"{name} does not exist"
+    raise AttributeError(msg)

langchain/agents/agent_toolkits/playwright/__init__.py

@@ -13,7 +13,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "PlayWrightBrowserToolkit": "langchain_community.agent_toolkits.playwright.toolkit"
+    "PlayWrightBrowserToolkit": "langchain_community.agent_toolkits.playwright.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/playwright/toolkit.py

@@ -11,7 +11,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "PlayWrightBrowserToolkit": "langchain_community.agent_toolkits.playwright.toolkit"
+    "PlayWrightBrowserToolkit": "langchain_community.agent_toolkits.playwright.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/powerbi/base.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "create_pbi_agent": "langchain_community.agent_toolkits.powerbi.base"
+    "create_pbi_agent": "langchain_community.agent_toolkits.powerbi.base",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/powerbi/chat_base.py

@@ -11,7 +11,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "create_pbi_chat_agent": "langchain_community.agent_toolkits.powerbi.chat_base"
+    "create_pbi_chat_agent": "langchain_community.agent_toolkits.powerbi.chat_base",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/powerbi/prompt.py

@@ -29,8 +29,8 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "POWERBI_PREFIX",
-    "POWERBI_SUFFIX",
     "POWERBI_CHAT_PREFIX",
     "POWERBI_CHAT_SUFFIX",
+    "POWERBI_PREFIX",
+    "POWERBI_SUFFIX",
 ]

langchain/agents/agent_toolkits/powerbi/toolkit.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "PowerBIToolkit": "langchain_community.agent_toolkits.powerbi.toolkit"
+    "PowerBIToolkit": "langchain_community.agent_toolkits.powerbi.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/python/__init__.py

@@ -14,7 +14,7 @@ def __getattr__(name: str) -> Any:
 
         old_path = "langchain." + here + "." + name
         new_path = "langchain_experimental." + here + "." + name
-        raise ImportError(
+        msg = (
             "This agent has been moved to langchain experiment. "
             "This agent relies on python REPL tool under the hood, so to use it "
             "safely please sandbox the python REPL. "
@@ -23,4 +23,6 @@ def __getattr__(name: str) -> Any:
             "To keep using this code as is, install langchain experimental and "
             f"update your import statement from:\n `{old_path}` to `{new_path}`."
         )
-    raise AttributeError(f"{name} does not exist")
+        raise ImportError(msg)
+    msg = f"{name} does not exist"
+    raise AttributeError(msg)

langchain/agents/agent_toolkits/spark/__init__.py

@@ -14,7 +14,7 @@ def __getattr__(name: str) -> Any:
 
         old_path = "langchain." + here + "." + name
         new_path = "langchain_experimental." + here + "." + name
-        raise ImportError(
+        msg = (
             "This agent has been moved to langchain experiment. "
             "This agent relies on python REPL tool under the hood, so to use it "
             "safely please sandbox the python REPL. "
@@ -23,4 +23,6 @@ def __getattr__(name: str) -> Any:
             "To keep using this code as is, install langchain experimental and "
             f"update your import statement from:\n `{old_path}` to `{new_path}`."
         )
-    raise AttributeError(f"{name} does not exist")
+        raise ImportError(msg)
+    msg = f"{name} does not exist"
+    raise AttributeError(msg)

langchain/agents/agent_toolkits/spark_sql/base.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "create_spark_sql_agent": "langchain_community.agent_toolkits.spark_sql.base"
+    "create_spark_sql_agent": "langchain_community.agent_toolkits.spark_sql.base",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)

langchain/agents/agent_toolkits/spark_sql/toolkit.py

@@ -9,7 +9,7 @@ if TYPE_CHECKING:
 # Used to consolidate logic for raising deprecation warnings and
 # handling optional imports.
 DEPRECATED_LOOKUP = {
-    "SparkSQLToolkit": "langchain_community.agent_toolkits.spark_sql.toolkit"
+    "SparkSQLToolkit": "langchain_community.agent_toolkits.spark_sql.toolkit",
 }
 
 _import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)