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

(test) aembedding

Browse source
This commit is contained in:
ishaan-jaff 2023-11-29 19:36:42 -08:00
parent 10e21ae978
commit 53554bae85
1 changed files with 14 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_embedding.py
View file

@ -177,6 +177,7 @@ def test_hf_embedding():
# test async embeddings
def test_aembedding():
try:
import asyncio
async def embedding_call():
try:
@ -185,10 +186,11 @@ def test_aembedding():
input=["good morning from litellm", "this is another item"]
)
print(response)
except:
print(f"error occurred: {traceback.format_exc()}")
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
asyncio.run(embedding_call())
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_aembedding()