tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-20 00:23:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix gemini web search requests count (#17921)

Browse Source 
* fix gemini web search requests count

* filter queries
This commit is contained in:
Kerem Turgutlu
2025-12-14 08:18:26 +05:30
committed by GitHub
parent 88d5efb2b7
commit 1da0bdd33d
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py
+2 -2
View File
@@ -1553,9 +1553,9 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
for grounding_metadata_item in grounding_metadata:
web_search_queries = grounding_metadata_item.get("webSearchQueries")
if web_search_queries and web_search_requests:
web_search_requests += len(web_search_queries)
web_search_requests += len([q for q in web_search_queries if q])
elif web_search_queries:
web_search_requests = len(grounding_metadata)
web_search_requests = len([q for q in web_search_queries if q])
return web_search_requests
@staticmethod
tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py
+2 -2
View File
@@ -806,7 +806,7 @@ def test_vertex_ai_streaming_usage_web_search_calculation():
{
"content": {"parts": [{"text": "Hello"}]},
"groundingMetadata": [
{"webSearchQueries": ["What is the capital of France?"]}
{"webSearchQueries": ["", "What is the capital of France?", "Capital of France"]}
],
}
],
@@ -821,7 +821,7 @@ def test_vertex_ai_streaming_usage_web_search_calculation():
usage: Usage = completed_response.usage
assert usage.prompt_tokens_details.web_search_requests is not None
assert usage.prompt_tokens_details.web_search_requests == 1
assert usage.prompt_tokens_details.web_search_requests == 2
def test_vertex_ai_transform_parts():