fix: keep existing header values while merging tracking headers for llm_request.config.http_options in Gemini.generate_content_async

PiperOrigin-RevId: 789013693
This commit is contained in:
Xuan Yang
2025-07-30 13:09:56 -07:00
committed by Copybara-Service
parent 3be1bb37d9
commit 6191412b07
2 changed files with 21 additions and 4 deletions
+1 -1
View File
@@ -403,7 +403,7 @@ async def test_generate_content_async_with_custom_headers(
for key, value in config_arg.http_options.headers.items():
if key in gemini_llm._tracking_headers:
assert value == gemini_llm._tracking_headers[key]
assert value == gemini_llm._tracking_headers[key] + " custom"
else:
assert value == custom_headers[key]