UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-22 13:18:25 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: Use x-goog-api-key header for Gemini image generation

Browse Source 
Place the API key in a header instead of a query parameter. This avoids leaking the API key in logs on request failure, etc...
This commit is contained in:
JoaoCostaIFG 2025-02-19 00:00:54 +00:00
parent e56b5c063c
commit 918764a4f7
No known key found for this signature in database
GPG Key ID: E85A047FEBBA3E93
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/open_webui/routers/images.py
View File

@ -515,7 +515,8 @@ async def image_generations(
elif request.app.state.config.IMAGE_GENERATION_ENGINE == "gemini": elif request.app.state.config.IMAGE_GENERATION_ENGINE == "gemini":
headers = {} headers = {}
headers["Content-Type"] = "application/json" headers["Content-Type"] = "application/json"
api_key = request.app.state.config.IMAGES_GEMINI_API_KEY headers["x-goog-api-key"] = request.app.state.config.IMAGES_GEMINI_API_KEY
model = get_image_model(request) model = get_image_model(request)
data = { data = {
"instances": {"prompt": form_data.prompt}, "instances": {"prompt": form_data.prompt},
@ -528,7 +529,7 @@ async def image_generations(
# Use asyncio.to_thread for the requests.post call # Use asyncio.to_thread for the requests.post call
r = await asyncio.to_thread( r = await asyncio.to_thread(
requests.post, requests.post,
url=f"{request.app.state.config.IMAGES_GEMINI_API_BASE_URL}/models/{model}:predict?key={api_key}", url=f"{request.app.state.config.IMAGES_GEMINI_API_BASE_URL}/models/{model}:predict",
json=data, json=data,
headers=headers, headers=headers,
) )