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 e9a5a34 commit 0f10f0a
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions agentops/llms/providers/fireworks.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ async def async_generator(stream_response):
event.completion = accumulated_content
event.end_timestamp = time.time()

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

View check run for this annotation

Codecov / codecov/patch

agentops/llms/providers/fireworks.py#L70-L71

Added lines #L70 - L71 were not covered by tests
if current_session:
current_session.record_event(event)
current_session.record(event)
logger.info("Recorded streaming LLM event")
except Exception as e:
logger.error(f"Error in async_generator: {str(e)}")
Expand All @@ -89,7 +89,7 @@ def generator(stream_response):
event.completion = accumulated_content
event.end_timestamp = time.time()

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

View check run for this annotation

Codecov / codecov/patch

agentops/llms/providers/fireworks.py#L89-L90

Added lines #L89 - L90 were not covered by tests
if current_session:
current_session.record_event(event)
current_session.record(event)
logger.info("Recorded streaming LLM event")
except Exception as e:
logger.error(f"Error in generator: {str(e)}")
Expand All @@ -105,7 +105,7 @@ def generator(stream_response):
event.completion = response.choices[0].message.content
event.end_timestamp = time.time()

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

View check run for this annotation

Codecov / codecov/patch

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

Added lines #L105 - L106 were not covered by tests
if current_session:
current_session.record_event(event)
current_session.record(event)
logger.info("Recorded non-streaming LLM event")

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

View check run for this annotation

Codecov / codecov/patch

agentops/llms/providers/fireworks.py#L108-L109

Added lines #L108 - L109 were not covered by tests

return response

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

View check run for this annotation

Codecov / codecov/patch

agentops/llms/providers/fireworks.py#L111

Added line #L111 was not covered by tests
Expand Down

0 comments on commit 0f10f0a

Please sign in to comment.