scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

test(vertex_ai): update rerank integration test for input-based search_units and unique id

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
shivam 2026-07-30 02:47:47 +00:00
parent ec1d1efc4b
commit 512c41a8fe
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/test_litellm/llms/vertex_ai/rerank/test_vertex_ai_rerank_integration.py
View file

@ -104,10 +104,12 @@ class TestVertexAIRerankIntegration:
raw_response=mock_response,
model_response=model_response,
logging_obj=mock_logging,
request_data=request_data,
)
# Verify response structure
assert result.id == f"vertex_ai_rerank_{self.model}"
assert result.id.startswith("vertex_ai_rerank_")
assert result.id != f"vertex_ai_rerank_{self.model}"
assert len(result.results) == 2
# Results should be sorted by relevance score (descending)
@ -116,8 +118,8 @@ class TestVertexAIRerankIntegration:
assert result.results[1]["index"] == 0 # Second highest score
assert result.results[1]["relevance_score"] == 0.92
# Verify metadata
assert result.meta["billed_units"]["search_units"] == 2
# Verify metadata: 4 input records bill as 1 search unit (ceil(4/100))
assert result.meta["billed_units"]["search_units"] == 1
def test_return_documents_false_flow(self):
"""Test rerank flow when return_documents=False (ID-only response)."""