simcop2387/text-generation-webui-mirror
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-06-07 06:06:20 -04:00
Code Issues Projects Releases Packages Wiki Activity

Better request header

Browse source
This commit is contained in:
oobabooga 2025-05-28 09:17:02 -07:00
parent 2e21b1f5e3
commit c1a47a0b60
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
extensions/openai/completions.py
View file

@ -102,7 +102,10 @@ def process_image_url(url, image_id):
raise ValueError("Unsupported data URL format")
else:
# Handle regular URL - download image
response = requests.get(url, timeout=10)
headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36'
}
response = requests.get(url, timeout=10, headers=headers)
response.raise_for_status()
image_data = base64.b64encode(response.content).decode('utf-8')
@ -111,7 +114,7 @@ def process_image_url(url, image_id):
"type": "image",
"image_data": image_data,
"image_id": image_id,
"file_path": f"api_image_{image_id}", # Add this for consistency with UI
"file_path": f"api_image_{image_id}",
}
except Exception as e:
logger.error(f"Error processing image URL {url}: {e}")