scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-02 02:11:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

test(s3-vectors): cover embedding dimension model

Browse source
This commit is contained in:
Ganni Galea Curmi 2026-05-26 23:50:17 -04:00
parent 44e96a7c1d
commit d667fe7fca
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/test_litellm/proxy/rag_endpoints/test_s3_vectors_ingestion.py
View file

@ -1,5 +1,5 @@
from types import SimpleNamespace
from unittest.mock import AsyncMock, patch
from unittest.mock import AsyncMock, call, patch
import pytest
@ -36,11 +36,16 @@ async def test_s3_vectors_ingestion_uses_vector_store_embedding_model():
)
embeddings = await ingestion.embed(["hello"])
dimension = await ingestion._get_dimension_from_embedding_request()
mock_aembedding.assert_awaited_once_with(
model="text-embedding-3-large", input=["hello"]
mock_aembedding.assert_has_awaits(
[
call(model="text-embedding-3-large", input=["hello"]),
call(model="text-embedding-3-large", input=["test"]),
]
)
assert embeddings == [[0.1, 0.2, 0.3]]
assert dimension == 3
def test_s3_vectors_ingestion_prefers_top_level_embedding_config():