From 27ccc444715ff9c8e98ddc2d102ff92c4e703962 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Wed, 29 Jul 2026 20:44:31 -0700
Subject: [PATCH] fix(batches): forward gcs_bucket_name so vertex batch cost
 logging can read the output file

---
 litellm/batches/batch_utils.py                 | 2 ++
 tests/test_litellm/batches/test_batch_utils.py | 8 ++++++++
 2 files changed, 10 insertions(+)

diff --git a/litellm/batches/batch_utils.py b/litellm/batches/batch_utils.py
index 36076d59a35..b62331b904b 100644
--- a/litellm/batches/batch_utils.py
+++ b/litellm/batches/batch_utils.py
@@ -267,6 +267,8 @@ def _extract_file_access_credentials(litellm_params: Optional[dict]) -> dict:
             "vertex_project",
             "vertex_location",
             "vertex_credentials",
+            "gcs_bucket_name",
+            "bucket_name",
             "timeout",
             "max_retries",
         ]
diff --git a/tests/test_litellm/batches/test_batch_utils.py b/tests/test_litellm/batches/test_batch_utils.py
index fea897e9372..70c89f70ceb 100644
--- a/tests/test_litellm/batches/test_batch_utils.py
+++ b/tests/test_litellm/batches/test_batch_utils.py
@@ -236,6 +236,8 @@ def test_extract_credentials_only_known_keys():
         "api_key": "sk-1",
         "api_base": "https://b",
         "vertex_project": "proj",
+        "gcs_bucket_name": "my-bucket",
+        "bucket_name": "my-alias-bucket",
         "model": "gpt-4o",  # not a credential key
         "unrelated": "x",
     }
@@ -243,6 +245,8 @@ def test_extract_credentials_only_known_keys():
         "api_key": "sk-1",
         "api_base": "https://b",
         "vertex_project": "proj",
+        "gcs_bucket_name": "my-bucket",
+        "bucket_name": "my-alias-bucket",
     }
 
 
@@ -262,6 +266,8 @@ def test_extract_credentials_all_supported_keys():
         "vertex_project",
         "vertex_location",
         "vertex_credentials",
+        "gcs_bucket_name",
+        "bucket_name",
         "timeout",
         "max_retries",
     }
@@ -665,6 +671,7 @@ async def test_output_file_content_vertex_fetches_via_afile_content(monkeypatch)
             "vertex_project": "proj-1",
             "vertex_location": "us-central1",
             "vertex_credentials": "/path/to/creds.json",
+            "gcs_bucket_name": "litellm-bucket",
             "model": "vertex_ai/gemini-3.6-flash",
         },
     )
@@ -675,6 +682,7 @@ async def test_output_file_content_vertex_fetches_via_afile_content(monkeypatch)
     assert captured["vertex_project"] == "proj-1"
     assert captured["vertex_location"] == "us-central1"
     assert captured["vertex_credentials"] == "/path/to/creds.json"
+    assert captured["gcs_bucket_name"] == "litellm-bucket"
     assert "model" not in captured