ChatGPT/RefreshToV1Api
1
0
Fork 0
mirror of https://github.com/Yanyutin753/RefreshToV1Api.git synced 2025-12-10 01:03:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
Clivia
2024-09-13 20:12:42 +08:00
parent 26e6af6317
commit 2d421b0a71
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
main.py
View File

@@ -1870,14 +1870,19 @@ import time
def chat_completions():
logger.info(f"New Request")
proxy_api_prefix = getPROXY_API_PREFIX(lock)
if proxy_api_prefix == None:
return jsonify({"error": "PROXY_API_PREFIX is not accessible"}), 401
data = request.json
messages = data.get('messages')
model = data.get('model')
model = data.get('model', "gpt-3.5-turbo")
ori_model_name = model
accessible_model_list = get_accessible_model_list()
if model not in accessible_model_list and not 'gpt-4-gizmo-' in model:
return jsonify({"error": "model is not accessible"}), 401
model_config = find_model_config(model)
if model_config:
ori_model_name = model_config.get('ori_name', model)
if "o1-" in ori_model_name:
# 使用列表推导式过滤系统角色
messages = [message for message in messages if message["role"] in ["user", "assistant"]]
@@ -2049,12 +2054,17 @@ def images_generations():
data = request.json
logger.debug(f"data: {data}")
api_key = None
# messages = data.get('messages')
model = data.get('model')
model = data.get('model', "gpt-3.5-turbo")
ori_model_name = model
accessible_model_list = get_accessible_model_list()
if model not in accessible_model_list and not 'gpt-4-gizmo-' in model:
return jsonify({"error": "model is not accessible"}), 401
model_config = find_model_config(model)
if model_config:
ori_model_name = model_config.get('ori_name', model)
if "o1-" in ori_model_name:
# 使用列表推导式过滤系统角色
messages = [message for message in messages if message["role"] in ["user", "assistant"]]
# 获取请求中的response_format参数，默认为"url"
response_format = data.get('response_format', 'url')
# 获取请求中的size参数，默认为"1024x1024"