tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-24 10:27:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test_sync_embedding

Browse Source
This commit is contained in:
Ishaan Jaff
2024-07-15 20:51:29 -07:00
parent 254ac37f65
commit 4baa48ba4b
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_custom_callback_input.py
+2
View File
@@ -840,6 +840,7 @@ def test_sync_embedding():
)
print(f"customHandler_success.errors: {customHandler_success.errors}")
print(f"customHandler_success.states: {customHandler_success.states}")
time.sleep(2)
assert len(customHandler_success.errors) == 0
assert len(customHandler_success.states) == 3 # pre, post, success
# test failure callback
@@ -854,6 +855,7 @@ def test_sync_embedding():
pass
print(f"customHandler_failure.errors: {customHandler_failure.errors}")
print(f"customHandler_failure.states: {customHandler_failure.states}")
time.sleep(2)
assert len(customHandler_failure.errors) == 1
assert len(customHandler_failure.states) == 3 # pre, post, failure
except Exception as e: