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

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

12
.pre-commit-config.yaml
View file

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

4
litellm/types/utils.py
View file

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