From d667fe7fcadbf88a581593d3480cff834db28415 Mon Sep 17 00:00:00 2001 From: Ganni Galea Curmi Date: Tue, 26 May 2026 23:50:17 -0400 Subject: [PATCH] test(s3-vectors): cover embedding dimension model --- .../proxy/rag_endpoints/test_s3_vectors_ingestion.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/tests/test_litellm/proxy/rag_endpoints/test_s3_vectors_ingestion.py b/tests/test_litellm/proxy/rag_endpoints/test_s3_vectors_ingestion.py index b96a4d4ff65..85750531b45 100644 --- a/tests/test_litellm/proxy/rag_endpoints/test_s3_vectors_ingestion.py +++ b/tests/test_litellm/proxy/rag_endpoints/test_s3_vectors_ingestion.py @@ -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():