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

26
litellm/tests/test_embedding.py
View file

@ -177,18 +177,20 @@ def test_hf_embedding():
# test async embeddings # test async embeddings
def test_aembedding(): def test_aembedding():
import asyncio try:
async def embedding_call(): import asyncio
try: async def embedding_call():
response = await litellm.aembedding( try:
model="text-embedding-ada-002", response = await litellm.aembedding(
input=["good morning from litellm", "this is another item"] model="text-embedding-ada-002",
) input=["good morning from litellm", "this is another item"]
print(response) )
except: print(response)
print(f"error occurred: {traceback.format_exc()}") except Exception as e:
pass pytest.fail(f"Error occurred: {e}")
asyncio.run(embedding_call()) asyncio.run(embedding_call())
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_aembedding() test_aembedding()