Skip to content

Commit

Permalink
ruff
Browse files Browse the repository at this point in the history
  • Loading branch information
lmolkova committed Feb 3, 2025
1 parent 00a470f commit 31633f3
Showing 1 changed file with 3 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -74,9 +74,7 @@ def traced_method(wrapped, instance, args, kwargs):
span, result, event_logger, capture_content
)
for choice in getattr(result, "choices", []):
event_logger.emit(
choice_to_event(choice, capture_content)
)
event_logger.emit(choice_to_event(choice, capture_content))

span.end()
return result
Expand Down Expand Up @@ -133,10 +131,8 @@ async def traced_method(wrapped, instance, args, kwargs):
_set_response_attributes(
span, result, event_logger, capture_content
)
for choice in getattr(result, "choices", []):
event_logger.emit(
choice_to_event(choice, capture_content)
)
for choice in getattr(result, "choices", []):
event_logger.emit(choice_to_event(choice, capture_content))

span.end()
return result
Expand Down

0 comments on commit 31633f3

Please sign in to comment.