tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-21 08:26:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(test_proxy_reject_logging.py): fix test

Browse Source
This commit is contained in:
Krrish Dholakia
2024-07-05 19:09:37 -07:00
parent fd11f63f71
commit 6090fd37fa
3 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/litellm_core_utils/litellm_logging.py
+1
View File
@@ -1607,6 +1607,7 @@ class Logging:
)
== False
): # custom logger class
callback.log_failure_event(
start_time=start_time,
end_time=end_time,
litellm/proxy/utils.py
+17 -16
View File
@@ -585,23 +585,24 @@ class ProxyLogging:
**request_data,
)
# log the custom exception
await litellm_logging_obj.async_failure_handler(
exception=original_exception,
traceback_exception=traceback.format_exc(),
start_time=time.time(),
end_time=time.time(),
)
if litellm_logging_obj is not None:
# log the custom exception
await litellm_logging_obj.async_failure_handler(
exception=original_exception,
traceback_exception=traceback.format_exc(),
start_time=time.time(),
end_time=time.time(),
)
threading.Thread(
target=litellm_logging_obj.failure_handler,
args=(
original_exception,
traceback.format_exc(),
time.time(),
time.time(),
),
).start()
threading.Thread(
target=litellm_logging_obj.failure_handler,
args=(
original_exception,
traceback.format_exc(),
time.time(),
time.time(),
),
).start()
for callback in litellm.callbacks:
try:
litellm/tests/test_proxy_reject_logging.py
+2 -1
View File
@@ -76,7 +76,7 @@ class testLogger(CustomLogger):
self.reaches_async_failure_event = True
def log_failure_event(self, kwargs, response_obj, start_time, end_time):
self.reaches_failure_event = True
self.reaches_sync_failure_event = True
router = Router(
@@ -186,4 +186,5 @@ async def test_chat_completion_request_with_redaction(route, body):
await asyncio.sleep(3)
assert _test_logger.reaches_async_failure_event is True
assert _test_logger.reaches_sync_failure_event is True