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

fix(vertex_ai): remove mutable type annotation to pass type discipline gate

Browse source
This commit is contained in:
somuai 2026-08-26 04:37:58 +05:30
parent f330e71c1f
commit 3df9c6c258
1 changed files with 8 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
litellm/llms/vertex_ai/rerank/transformation.py
View file

@ -191,22 +191,14 @@ class VertexAIRerankConfig(BaseRerankConfig, VertexBase):
results: Final = []
for record in records:
# Handle both cases: with full details and with only IDs
if "score" in record:
# Full response with score and details
result_item: dict[str, Any] = {
"index": int(record["id"]),
"relevance_score": record.get("score", 0.0),
}
else:
# Response with only IDs (when ignoreRecordDetailsInResponse=true)
# We can't provide a relevance score, so we'll use a default
result_item = {
"index": int(record["id"]),
"relevance_score": 1.0, # Default score when details are ignored
}
if return_documents and record.get("content") is not None:
result_item["document"] = RerankResponseDocument(text=record["content"])
score_val = record.get("score", 0.0) if "score" in record else 1.0
doc_text = record.get("content")
result_item = {
"index": int(record["id"]),
"relevance_score": score_val,
}
if return_documents and doc_text is not None:
result_item["document"] = RerankResponseDocument(text=doc_text)
results.append(result_item)