phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

quick fix to enable Cloudflare Gateway in google-ai-studio

Browse source
This commit is contained in:
Nicolas Wang 2025-04-14 10:06:21 +00:00
parent 64bb89c70f
commit 05c1fd2ac0
1 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/llms/vertex_ai/vertex_llm_base.py
View file

@ -172,12 +172,16 @@ class VertexBase:
if custom_llm_provider == "gemini":
url = "{}:{}".format(api_base, endpoint)
if gemini_api_key is None:
raise ValueError(
"Missing gemini_api_key, please set `GEMINI_API_KEY`"
# Do not raise error as google-ai-studio expect header 'x-goog-api-key' instead of 'Authorization',
# https://developers.cloudflare.com/ai-gateway/providers/google-ai-studio/#curl
# raise ValueError(
# "Missing gemini_api_key, please set `GEMINI_API_KEY`"
# )
pass
else:
auth_header = (
gemini_api_key # cloudflare expects api key as bearer token
)
auth_header = (
gemini_api_key # cloudflare expects api key as bearer token
)
else:
url = "{}:{}".format(api_base, endpoint)