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

(feat) add async embeddings

Browse source
This commit is contained in:
ishaan-jaff 2023-10-23 13:59:37 -07:00
parent 34c75b8c0c
commit 6373f6bddd
2 changed files with 31 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/tests/test_embedding.py
View file

@ -91,4 +91,20 @@ def test_cohere_embedding():
# pytest.fail(f"Error occurred: {e}")
# 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())
# test_aembedding()