Skip to content

Commit

Permalink
style: Apply ruff-format changes to improve code formatting
Browse files Browse the repository at this point in the history
Co-Authored-By: Alex Reibman <[email protected]>
  • Loading branch information
devin-ai-integration[bot] and areibman committed Dec 18, 2024
1 parent bb3b056 commit 26c71cf
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 3 deletions.
8 changes: 6 additions & 2 deletions agentops/llms/providers/fireworks.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,9 @@ async def async_generator(stream):
stream_event.end_timestamp = get_ISO_time()

Check warning on line 73 in agentops/llms/providers/fireworks.py

View check run for this annotation

Codecov / codecov/patch

agentops/llms/providers/fireworks.py#L72-L73

Added lines #L72 - L73 were not covered by tests
if self._session:
self._session.record(stream_event)
logger.debug(f"Recorded streaming response for session {self._session.session_id}")
logger.debug(

Check warning on line 76 in agentops/llms/providers/fireworks.py

View check run for this annotation

Codecov / codecov/patch

agentops/llms/providers/fireworks.py#L75-L76

Added lines #L75 - L76 were not covered by tests
f"Recorded streaming response for session {self._session.session_id}"
)
yield content
except Exception as e:
logger.error(f"Error processing streaming chunk: {str(e)}")
Expand All @@ -100,7 +102,9 @@ def generator(stream):
stream_event.end_timestamp = get_ISO_time()

Check warning on line 102 in agentops/llms/providers/fireworks.py

View check run for this annotation

Codecov / codecov/patch

agentops/llms/providers/fireworks.py#L101-L102

Added lines #L101 - L102 were not covered by tests
if self._session:
self._session.record(stream_event)
logger.debug(f"Recorded streaming response for session {self._session.session_id}")
logger.debug(

Check warning on line 105 in agentops/llms/providers/fireworks.py

View check run for this annotation

Codecov / codecov/patch

agentops/llms/providers/fireworks.py#L104-L105

Added lines #L104 - L105 were not covered by tests
f"Recorded streaming response for session {self._session.session_id}"
)
yield content
except Exception as e:
logger.error(f"Error processing streaming chunk: {str(e)}")
Expand Down
1 change: 0 additions & 1 deletion examples/fireworks_examples/fireworks_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,4 +111,3 @@

finally:
print("\nScript execution completed.")

0 comments on commit 26c71cf

Please sign in to comment.