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
def test_aembedding():
import asyncio
async def embedding_call():
try:
response = await litellm.aembedding(
model="text-embedding-ada-002",
input=["good morning from litellm", "this is another item"]
)
print(response)
except:
print(f"error occurred: {traceback.format_exc()}")
pass
asyncio.run(embedding_call())
try:
import asyncio
async def embedding_call():
try:
response = await litellm.aembedding(
model="text-embedding-ada-002",
input=["good morning from litellm", "this is another item"]
)
print(response)
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()