phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests 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

2
litellm/tests/test_custom_callback_input.py
View file

@ -840,6 +840,7 @@ def test_sync_embedding():
) )
print(f"customHandler_success.errors: {customHandler_success.errors}") print(f"customHandler_success.errors: {customHandler_success.errors}")
print(f"customHandler_success.states: {customHandler_success.states}") print(f"customHandler_success.states: {customHandler_success.states}")
time.sleep(2)
assert len(customHandler_success.errors) == 0 assert len(customHandler_success.errors) == 0
assert len(customHandler_success.states) == 3 # pre, post, success assert len(customHandler_success.states) == 3 # pre, post, success
# test failure callback # test failure callback
@ -854,6 +855,7 @@ def test_sync_embedding():
pass pass
print(f"customHandler_failure.errors: {customHandler_failure.errors}") print(f"customHandler_failure.errors: {customHandler_failure.errors}")
print(f"customHandler_failure.states: {customHandler_failure.states}") print(f"customHandler_failure.states: {customHandler_failure.states}")
time.sleep(2)
assert len(customHandler_failure.errors) == 1 assert len(customHandler_failure.errors) == 1
assert len(customHandler_failure.states) == 3 # pre, post, failure assert len(customHandler_failure.states) == 3 # pre, post, failure
except Exception as e: except Exception as e: