From c6d32da307b5e769fb52fb32f3635a2a7853cfaf Mon Sep 17 00:00:00 2001 From: diegofiori Date: Wed, 14 Aug 2024 16:49:38 +0200 Subject: [PATCH] fix linter --- nebuly/providers/langchain.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/nebuly/providers/langchain.py b/nebuly/providers/langchain.py index 4e4c192..0883e57 100644 --- a/nebuly/providers/langchain.py +++ b/nebuly/providers/langchain.py @@ -394,7 +394,7 @@ def on_tool_start( # pylint: disable=arguments-differ **kwargs: Any, ) -> None: if self.verbose: - logger.info(f"Tool started with input: {input_str}") + logger.info("Tool started with input: %s", input_str) data = EventData( type=EventType.TOOL, kwargs={ @@ -414,7 +414,7 @@ def on_tool_end( # pylint: disable=arguments-differ **kwargs: Any, ) -> None: if self.verbose: - logger.info(f"Tool ended with output: {output}") + logger.info("Tool ended with output: %s", output) self._events_storage.events[run_id].data.add_end_event_data( kwargs=kwargs, output=output ) @@ -429,7 +429,7 @@ def on_retriever_start( # pylint: disable=arguments-differ **kwargs: Any, ) -> None: if self.verbose: - logger.info(f"Retriever started with query: {query}") + logger.info("Retriever started with query: %s", query) data = EventData( type=EventType.RETRIEVAL, kwargs={ @@ -449,7 +449,7 @@ def on_retriever_end( # pylint: disable=arguments-differ **kwargs: Any, ) -> None: if self.verbose: - logger.info(f"Retriever ended with {len(documents)} documents") + logger.info("Retriever ended with %d documents", len(documents)) self._events_storage.events[run_id].data.add_end_event_data( kwargs=kwargs, output=documents ) @@ -464,7 +464,7 @@ def on_llm_start( # pylint: disable=arguments-differ **kwargs: Any, ) -> None: if self.verbose: - logger.info(f"LLM model started with {len(prompts)} prompts") + logger.info("LLM model started with %d prompts", len(prompts)) data = EventData( type=EventType.LLM_MODEL, kwargs={ @@ -492,7 +492,7 @@ def on_llm_end( # pylint: disable=arguments-differ if len(self._events_storage.events) == 1: if self.verbose: - logger.info("Sending interaction for id: {run_id}") + logger.info("Sending interaction for id: %s", str(run_id)) self._send_interaction(run_id) self._events_storage.delete_events(run_id) @@ -525,7 +525,7 @@ def on_chain_start( # pylint: disable=arguments-differ **kwargs: Any, ) -> None: if self.verbose: - logger.info(f"Chain started with run_id: {run_id}") + logger.info("Chain started with run_id: %s", str(run_id)) data = EventData( type=EventType.CHAIN, kwargs={ @@ -545,7 +545,7 @@ def on_chain_end( # pylint: disable=arguments-differ **kwargs: Any, ) -> None: if self.verbose: - logger.info(f"Chain ended with run_id: {run_id}") + logger.info("Chain ended with run_id: %s", str(run_id)) self._events_storage.events[run_id].data.add_end_event_data( kwargs=kwargs, output=outputs ) @@ -553,12 +553,12 @@ def on_chain_end( # pylint: disable=arguments-differ if self._events_storage.events[run_id].hierarchy is None: if self.verbose: - logger.info("Sending interaction for id: {run_id}") + logger.info("Sending interaction for id: %s", str(run_id)) self._send_interaction(run_id) self._events_storage.delete_events(run_id) def send_pending_interaction(self, output: dict[str, str]) -> None: - for run_id in self._events_storage.events: + for run_id in self._events_storage.events: # pylint: disable=consider-using-dict-items if self._events_storage.events[run_id].hierarchy is None: self._events_storage.events[run_id].data.add_end_event_data( kwargs={}, output=output