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

bug fixes for response object

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-17 20:24:38 -07:00
parent 24666a30fe
commit f0b104f37f
5 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/utils.py
View file

@ -12,6 +12,7 @@ from .integrations.aispend import AISpendLogger
from .integrations.berrispend import BerriSpendLogger
from .integrations.supabase import Supabase
from openai.error import OpenAIError as OriginalError
from openai.openai_object import OpenAIObject
from .exceptions import AuthenticationError, InvalidRequestError, RateLimitError, ServiceUnavailableError, OpenAIError
from typing import List, Dict, Union
####### ENVIRONMENT VARIABLES ###################
@ -87,7 +88,7 @@ class Choices:
result = f"{{\n 'finish_reason': '{self.finish_reason}',\n 'index': {self.index},\n 'message': {self.message}\n}}"
return result
class ModelResponse:
class ModelResponse(dict):
def __init__(self):
self.choices: List[Choices] = [Choices()]
self.created: str = None