mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
fix: add more type hints to init methods
This commit is contained in:
parent
f6fadee10e
commit
da343c7aca
2 changed files with 2 additions and 2 deletions
|
@ -10,7 +10,7 @@ import traceback
|
|||
|
||||
class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback#callback-class
|
||||
# Class variables or attributes
|
||||
def __init__(self):
|
||||
def __init__(self) -> None:
|
||||
pass
|
||||
|
||||
def log_pre_api_call(self, model, messages, kwargs):
|
||||
|
|
|
@ -528,7 +528,7 @@ class ModelResponse(OpenAIObject):
|
|||
response_ms=None,
|
||||
hidden_params=None,
|
||||
**params,
|
||||
):
|
||||
) -> None:
|
||||
if stream is not None and stream is True:
|
||||
object = "chat.completion.chunk"
|
||||
if choices is not None and isinstance(choices, list):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue