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

fix: initial commit

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-27 17:35:56 -07:00
parent d29a7087f1
commit 77e6da78a1
11 changed files with 192 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/tests/test_embedding.py
View file

@ -686,6 +686,22 @@ async def test_triton_embeddings():
pytest.fail(f"Error occurred: {e}")
@pytest.mark.asyncio
async def test_gemini_embeddings():
try:
litellm.set_verbose = True
response = await litellm.aembedding(
model="gemini/text-embedding-004",
input=["good morning from litellm"],
)
print(f"response: {response}")
# stubbed endpoint is setup to return this
assert response.data[0]["embedding"] == [0.1, 0.2]
except Exception as e:
pytest.fail(f"Error occurred: {e}")
@pytest.mark.parametrize("sync_mode", [True, False])
@pytest.mark.asyncio
async def test_databricks_embeddings(sync_mode):