Skip to content

Commit

Permalink
fix linter
Browse files Browse the repository at this point in the history
  • Loading branch information
diegofiori committed Aug 14, 2024
1 parent ef5b3b8 commit c6d32da
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions nebuly/providers/langchain.py
Original file line number Diff line number Diff line change
Expand Up @@ -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={
Expand All @@ -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
)
Expand All @@ -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={
Expand All @@ -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
)
Expand All @@ -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={
Expand Down Expand Up @@ -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)

Expand Down Expand Up @@ -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={
Expand All @@ -545,20 +545,20 @@ 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
)
self._events_storage.events[run_id].set_end_time()

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
Expand Down

0 comments on commit c6d32da

Please sign in to comment.