phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) text completion response

Browse source
This commit is contained in:
ishaan-jaff 2023-11-13 10:29:23 -08:00
parent 18b694f01a
commit 4439109658
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/utils.py
View file

@ -385,6 +385,7 @@ class TextCompletionResponse(OpenAIObject):
}
"""
def __init__(self, id=None, choices=None, created=None, model=None, usage=None, stream=False, response_ms=None, **params):
super(TextCompletionResponse, self).__init__(**params)
if stream:
self.object = "text_completion.chunk"
self.choices = [TextChoices()]
@ -409,7 +410,7 @@ class TextCompletionResponse(OpenAIObject):
else:
self.usage = Usage()
self._hidden_params = {} # used in case users want to access the original model response
super(TextCompletionResponse, self).__init__(**params)
def __contains__(self, key):
# Define custom behavior for the 'in' operator