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

(fix) linting

Browse source
This commit is contained in:
ishaan-jaff 2023-12-06 20:26:57 -08:00
parent 900b8d66f3
commit 095aa94e71
1 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

22
litellm/tests/test_custom_logger.py
View file

@ -10,19 +10,19 @@ from litellm.integrations.custom_logger import CustomLogger
async_success = False async_success = False
class MyCustomHandler(CustomLogger): class MyCustomHandler(CustomLogger):
def __init__(self): def __init__(self):
self.success: bool = False self.success: bool = False # type: ignore
self.failure: bool = False self.failure: bool = False # type: ignore
self.async_success: bool = False self.async_success: bool = False # type: ignore
self.async_success_embedding: bool = False self.async_success_embedding: bool = False # type: ignore
self.async_failure: bool = False self.async_failure: bool = False # type: ignore
self.async_failure_embedding: bool = False self.async_failure_embedding: bool = False # type: ignore
self.async_completion_kwargs = None # test if the kwargs are available, for async_succcess for completion/embedding self.async_completion_kwargs = None # type: ignore
self.async_embedding_kwargs = None self.async_embedding_kwargs = None # type: ignore
self.async_embedding_response = None self.async_embedding_response = None # type: ignore
self.async_completion_kwargs_fail = None self.async_completion_kwargs_fail = None # type: ignore
self.async_embedding_kwargs_fail = None self.async_embedding_kwargs_fail = None # type: ignore
def log_pre_api_call(self, model, messages, kwargs): def log_pre_api_call(self, model, messages, kwargs):
print(f"Pre-API Call") print(f"Pre-API Call")