fix(types/utils.py): fix linting errors

This commit is contained in:
Krrish Dholakia 2024-08-03 11:48:33 -07:00
parent c982ec88d8
commit 5add6687cc
2 changed files with 7 additions and 9 deletions

View file

@ -1,12 +1,12 @@
repos:
- repo: local
hooks:
# - id: mypy
# name: mypy
# entry: python3 -m mypy --ignore-missing-imports
# language: system
# types: [python]
# files: ^litellm/
- id: mypy
name: mypy
entry: python3 -m mypy --ignore-missing-imports
language: system
types: [python]
files: ^litellm/
- id: isort
name: isort
entry: isort

View file

@ -976,9 +976,7 @@ class ImageResponse(OpenAIImageResponse):
else:
created = int(time.time())
_data = {"data": data, "created": created}
super().__init__(**_data)
super().__init__(created=created, data=data)
self.usage = {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0}
def __contains__(self, key):