diff --git a/litellm/__pycache__/__init__.cpython-311.pyc b/litellm/__pycache__/__init__.cpython-311.pyc index ff33c5b10a..991b3d6358 100644 Binary files a/litellm/__pycache__/__init__.cpython-311.pyc and b/litellm/__pycache__/__init__.cpython-311.pyc differ diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc index b977a57784..f8e81abe6c 100644 Binary files a/litellm/__pycache__/main.cpython-311.pyc and b/litellm/__pycache__/main.cpython-311.pyc differ diff --git a/litellm/__pycache__/utils.cpython-311.pyc b/litellm/__pycache__/utils.cpython-311.pyc index 4c042a2505..3f33c9e65d 100644 Binary files a/litellm/__pycache__/utils.cpython-311.pyc and b/litellm/__pycache__/utils.cpython-311.pyc differ diff --git a/litellm/utils.py b/litellm/utils.py index e6816cced3..e5886c8e63 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -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 diff --git a/pyproject.toml b/pyproject.toml index 78e3c77422..1d143a7b1d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.421" +version = "0.1.422" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"