phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_completion.py): cleanup tests

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-13 11:23:38 -08:00
parent 4340749ea3
commit aa8ca781ba
3 changed files with 29 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_caching.py
View file

@ -127,6 +127,7 @@ def test_embedding_caching():
start_time = time.time()
embedding2 = embedding(model="text-embedding-ada-002", input=text_to_embed, caching=True)
end_time = time.time()
print(f"embedding2: {embedding2}")
print(f"Embedding 2 response time: {end_time - start_time} seconds")
litellm.cache = None
@ -136,7 +137,7 @@ def test_embedding_caching():
print(f"embedding2: {embedding2}")
pytest.fail("Error occurred: Embedding caching failed")
# test_embedding_caching()
test_embedding_caching()
def test_embedding_caching_azure():