ChatGPT/RefreshToV1Api
1
0
Fork 0
mirror of https://github.com/Yanyutin753/RefreshToV1Api.git synced 2025-10-15 07:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

[feat] 支持使用随机ua头下载gpt-4-vision接口传入的文件

Browse Source
This commit is contained in:
Wizerd
2024-01-05 15:50:04 +08:00
parent a5551d3fc5
commit ad435d05f4
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
main.py
View File

@@ -187,9 +187,9 @@ CORS(app, resources={r"/images/*": {"origins": "*"}})
PANDORA_UPLOAD_URL = 'files.pandoranext.com'
VERSION = '0.4.5'
VERSION = '0.4.6'
# VERSION = 'test'
UPDATE_INFO = '支持使用pandora的文件服务器下载文件'
UPDATE_INFO = '支持使用随机ua头下载gpt-4-vision接口传入的文件'
# UPDATE_INFO = '【仅供临时测试使用】 '
with app.app_context():
@@ -555,10 +555,12 @@ def send_text_prompt_and_get_response(messages, api_key, stream, model):
else:
# 处理普通的文件URL
try:
headers = {
'User-Agent': ua.random
tmp_user_agent = ua.random
logger.debug(f"随机 User-Agent: {tmp_user_agent}")
tmp_headers = {
'User-Agent': tmp_user_agent
}
file_response = requests.get(url=file_url, headers=headers)
file_response = requests.get(url=file_url, headers=tmp_headers)
file_content = file_response.content
mime_type = file_response.headers.get('Content-Type', '').split(';')[0].strip()
except Exception as e: