chore(langchain): add ruff rules D

Author: cbornet

libs/langchain/langchain/agents/__init__.py

@@ -1,5 +1,4 @@
-"""
-**Agent** is a class that uses an LLM to choose a sequence of actions to take.
+"""**Agent** is a class that uses an LLM to choose a sequence of actions to take.
 
 In Chains, a sequence of actions is hardcoded. In Agents,
 a language model is used as a reasoning engine to determine which actions

libs/langchain/langchain/agents/agent.py

@@ -366,7 +366,6 @@ def save(self, file_path: Union[Path, str]) -> None:
 
     def tool_run_logging_kwargs(self) -> builtins.dict:
         """Return logging kwargs for tool run."""
-
         return {}
 
 
@@ -1807,7 +1806,6 @@ async def astream(
         Yields:
             AddableDict: Addable dictionary.
         """
-
         config = ensure_config(config)
         iterator = AgentExecutorIterator(
             self,

libs/langchain/langchain/agents/agent_iterator.py

@@ -55,7 +55,8 @@ def __init__(
         include_run_info: bool = False,
         yield_actions: bool = False,
     ):
-        """
+        """Initialize the AgentExecutorIterator.
+
         Initialize the AgentExecutorIterator with the given AgentExecutor,
         inputs, and optional callbacks.
 
@@ -129,7 +130,8 @@ def color_mapping(self) -> dict[str, str]:
         )
 
     def reset(self) -> None:
-        """
+        """Reset the iterator to its initial state.
+
         Reset the iterator to its initial state, clearing intermediate steps,
         iterations, and time elapsed.
         """
@@ -141,9 +143,7 @@ def reset(self) -> None:
         self.start_time = time.time()
 
     def update_iterations(self) -> None:
-        """
-        Increment the number of iterations and update the time elapsed.
-        """
+        """Increment the number of iterations and update the time elapsed."""
         self.iterations += 1
         self.time_elapsed = time.time() - self.start_time
         logger.debug(
@@ -242,9 +242,10 @@ def __iter__(self: AgentExecutorIterator) -> Iterator[AddableDict]:
         yield self._stop(run_manager)
 
     async def __aiter__(self) -> AsyncIterator[AddableDict]:
-        """
+        """Create an async iterator for the AgentExecutor.
+
         N.B. __aiter__ must be a normal method, so need to initialize async run manager
-        on first __anext__ call where we can await it
+        on first __anext__ call where we can await it.
         """
         logger.debug("Initialising AgentExecutorIterator (async)")
         self.reset()
@@ -326,7 +327,8 @@ def _process_next_step_output(
         next_step_output: Union[AgentFinish, list[tuple[AgentAction, str]]],
         run_manager: CallbackManagerForChainRun,
     ) -> AddableDict:
-        """
+        """Process the output of the next step.
+
         Process the output of the next step,
         handling AgentFinish and tool return cases.
         """
@@ -354,7 +356,8 @@ async def _aprocess_next_step_output(
         next_step_output: Union[AgentFinish, list[tuple[AgentAction, str]]],
         run_manager: AsyncCallbackManagerForChainRun,
     ) -> AddableDict:
-        """
+        """Process the output of the next async step.
+
         Process the output of the next async step,
         handling AgentFinish and tool return cases.
         """
@@ -378,7 +381,8 @@ async def _aprocess_next_step_output(
         return AddableDict(intermediate_step=next_step_output)
 
     def _stop(self, run_manager: CallbackManagerForChainRun) -> AddableDict:
-        """
+        """Stop the iterator.
+
         Stop the iterator and raise a StopIteration exception with the stopped response.
         """
         logger.warning("Stopping agent prematurely due to triggering stop condition")
@@ -391,7 +395,8 @@ def _stop(self, run_manager: CallbackManagerForChainRun) -> AddableDict:
         return self._return(output, run_manager=run_manager)
 
     async def _astop(self, run_manager: AsyncCallbackManagerForChainRun) -> AddableDict:
-        """
+        """Stop the async iterator.
+
         Stop the async iterator and raise a StopAsyncIteration exception with
         the stopped response.
         """
@@ -408,9 +413,7 @@ def _return(
         output: AgentFinish,
         run_manager: CallbackManagerForChainRun,
     ) -> AddableDict:
-        """
-        Return the final output of the iterator.
-        """
+        """Return the final output of the iterator."""
         returned_output = self.agent_executor._return(  # noqa: SLF001
             output,
             self.intermediate_steps,
@@ -425,9 +428,7 @@ async def _areturn(
         output: AgentFinish,
         run_manager: AsyncCallbackManagerForChainRun,
     ) -> AddableDict:
-        """
-        Return the final output of the async iterator.
-        """
+        """Return the final output of the async iterator."""
         returned_output = await self.agent_executor._areturn(  # noqa: SLF001
             output,
             self.intermediate_steps,

libs/langchain/langchain/agents/agent_toolkits/conversational_retrieval/openai_functions.py

@@ -51,11 +51,11 @@ def create_conversational_retrieval_agent(
             defaults to False.
         max_token_limit: The max number of tokens to keep around in memory.
             Defaults to 2000.
+        **kwargs: Additional keyword arguments to pass to the AgentExecutor.
 
     Returns:
         An agent executor initialized appropriately
     """
-
     if remember_intermediate_steps:
         memory: BaseMemory = AgentTokenBufferMemory(
             memory_key=memory_key,

libs/langchain/langchain/agents/agent_toolkits/csv/__init__.py

@@ -6,7 +6,6 @@
 
 def __getattr__(name: str) -> Any:
     """Get attr name."""
-
     if name == "create_csv_agent":
         # Get directory of langchain package
         HERE = Path(__file__).parents[3]

libs/langchain/langchain/agents/agent_toolkits/nasa/__init__.py

@@ -1 +1 @@
-"""NASA Toolkit"""
+"""NASA Toolkit."""

libs/langchain/langchain/agents/agent_toolkits/pandas/__init__.py

@@ -6,7 +6,6 @@
 
 def __getattr__(name: str) -> Any:
     """Get attr name."""
-
     if name == "create_pandas_dataframe_agent":
         # Get directory of langchain package
         HERE = Path(__file__).parents[3]

libs/langchain/langchain/agents/agent_toolkits/python/__init__.py

@@ -6,7 +6,6 @@
 
 def __getattr__(name: str) -> Any:
     """Get attr name."""
-
     if name == "create_python_agent":
         # Get directory of langchain package
         HERE = Path(__file__).parents[3]

libs/langchain/langchain/agents/agent_toolkits/spark/__init__.py

@@ -6,7 +6,6 @@
 
 def __getattr__(name: str) -> Any:
     """Get attr name."""
-
     if name == "create_spark_dataframe_agent":
         # Get directory of langchain package
         HERE = Path(__file__).parents[3]

libs/langchain/langchain/agents/agent_toolkits/xorbits/__init__.py

@@ -6,7 +6,6 @@
 
 def __getattr__(name: str) -> Any:
     """Get attr name."""
-
     if name == "create_xorbits_agent":
         # Get directory of langchain package
         HERE = Path(__file__).parents[3]