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

fix: add more type hints to init methods

Browse source
This commit is contained in:
Nejc Habjan 2024-06-18 12:09:39 +02:00
parent c91ebb6a64
commit 2ecd614a73
No known key found for this signature in database
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/integrations/custom_logger.py
View file

@ -10,7 +10,7 @@ import traceback
class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback#callback-class class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback#callback-class
# Class variables or attributes # Class variables or attributes
def __init__(self): def __init__(self) -> None:
pass pass
def log_pre_api_call(self, model, messages, kwargs): def log_pre_api_call(self, model, messages, kwargs):

2
litellm/types/utils.py
View file

@ -528,7 +528,7 @@ class ModelResponse(OpenAIObject):
response_ms=None, response_ms=None,
hidden_params=None, hidden_params=None,
**params, **params,
): ) -> None:
if stream is not None and stream is True: if stream is not None and stream is True:
object = "chat.completion.chunk" object = "chat.completion.chunk"
if choices is not None and isinstance(choices, list): if choices is not None and isinstance(choices, list):