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

fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-14 17:24:43 -07:00
parent 0b436cf767
commit 722958a4cc
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -34,7 +34,7 @@ from .exceptions import (
APIError, APIError,
BudgetExceededError BudgetExceededError
) )
from typing import List, Dict, Union, Optional from typing import cast, List, Dict, Union, Optional
from .caching import Cache from .caching import Cache
@ -103,7 +103,7 @@ class Choices(OpenAIObject):
self.message = message self.message = message
class StreamingChoices(OpenAIObject): class StreamingChoices(OpenAIObject):
def __init__(self, finish_reason=None, index=0, delta: Optional[Delta]={}, **params): def __init__(self, finish_reason=None, index=0, delta: Optional[Union[Dict, Delta]]={}, **params):
super(StreamingChoices, self).__init__(**params) super(StreamingChoices, self).__init__(**params)
self.finish_reason = finish_reason self.finish_reason = finish_reason
self.index = index self.index = index