fix(gemini): strip native self prefixes before generateContent (#36141)

Strip `google/` and `gemini/` self-prefixes before native Gemini generateContent calls, and keep provider-normalization expectations aligned.
This commit is contained in:
Teknium
2026-06-13 13:47:08 -07:00
committed by GitHub
parent 7d11fa4e9e
commit c8e5f34f24
5 changed files with 58 additions and 3 deletions
+2 -1
View File
@@ -143,7 +143,8 @@ class TestGeminiModelNormalization:
assert normalize_model_for_provider("gemini-2.5-flash", "gemini") == "gemini-2.5-flash"
def test_strip_vendor_prefix(self):
assert normalize_model_for_provider("google/gemini-2.5-flash", "gemini") == "google/gemini-2.5-flash"
assert normalize_model_for_provider("google/gemini-2.5-flash", "gemini") == "gemini-2.5-flash"
assert normalize_model_for_provider("gemini/gemini-2.5-flash", "gemini") == "gemini-2.5-flash"
def test_gemma_vendor_detection(self):
assert detect_vendor("gemma-4-31b-it") == "google"