fix(traceloop.py): add additional openllmetry traces

This commit is contained in:
Krrish Dholakia 2023-12-16 19:21:39 -08:00
parent 7c2fad2d57
commit 7847ae1e23
2 changed files with 36 additions and 9 deletions

View file

@ -222,7 +222,7 @@ class Delta(OpenAIObject):
class Choices(OpenAIObject):
def __init__(self, finish_reason=None, index=0, message=None, **params):
super(Choices, self).__init__(**params)
self.finish_reason = map_finish_reason(finish_reason) # set finish_reason for all responses
self.finish_reason = map_finish_reason(finish_reason) or "stop" # set finish_reason for all responses
self.index = index
if message is None:
self.message = Message(content=None)