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

[feat] 调整项目结构以及支持机器人模式更多的自定义以及支持自定义会话是否自动删除

Browse Source
This commit is contained in:
Wizerd
2023-12-23 23:12:44 +08:00
parent fcdd99a1fc
commit 944ff3fa63
5 changed files with 134 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
Readme.md
View File

@@ -83,29 +83,39 @@
仓库内已包含相关文件和目录，拉到本地后修改 docker-compose.yml 文件里的环境变量后运行`docker-compose up -d`即可。 仓库内已包含相关文件和目录，拉到本地后修改 docker-compose.yml 文件里的环境变量后运行`docker-compose up -d`即可。
## 环境变量说明： ## config.json 变量说明：
- `BASE_URL`：Pandora-Next 的部署地址，如：`https://pandoranext.com`，注意：不要以 `/` 结尾。可以填写为本项目可以访问到的 PandoraNext 的内网地址。 - `log_level`: 用于设置日志等级，可选值为：`DEBUG`、`INFO`、`WARNING`、`ERROR`，默认为 `DEBUG`
- `PROXY_API_PREFIX`: PandoraNext Proxy 模式下的 API 前缀 - `need_log_to_file`: 用于设置是否需要将日志输出到文件，可选值为：`true`、`false`，默认为 `true`，日志文件路径为：`./log/access.log`，默认每天会自动分割日志文件。
- `UPLOAD_BASE_URL`：用于dalle模型生成图片的时候展示所用，需要设置为使用如 [ChatGPT-Next-Web](https://github.com/ChatGPTNextWebTeam/ChatGPT-Next-Web) 的用户可以访问到的本项目地址，如：`http://1.2.3.4:50011` - `pandora_base_url`: Pandora-Next 的部署地址，如：`https://pandoranext.com`，注意：不要以 `/` 结尾。可以填写为本项目可以访问到的 PandoraNext 的内网地址。
- `KEY_FOR_GPTS_INFO`：仅获取 GPTS 信息的 key，需要该 key 能够访问所有配置的 GPTS。后续发送消息仍需要在请求头携带请求所用的 key，如果未配置该项，请将 `gpts.json` 文件修改为： - `pandora_api_prefix`: PandoraNext Proxy 模式下的 API 前缀
- `backend_container_url`: 用于dalle模型生成图片的时候展示所用，需要设置为使用如 [ChatGPT-Next-Web](https://github.com/ChatGPTNextWebTeam/ChatGPT-Next-Web) 的用户可以访问到的本项目地址，如：`http://1.2.3.4:50011`，同原环境变量中的 `UPLOAD_BASE_URL`
- `backend_container_api_prefix`: 用于设置本项目 `/v1/xxx` 接口的前缀，如果留空则与官方api调用接口一致。设置示例：`666 `
- `key_for_gpts_info`: 仅获取 GPTS 信息的 key，需要该 key 能够访问所有配置的 GPTS。后续发送消息仍需要在请求头携带请求所用的 key，如果未配置该项，请将 `gpts.json` 文件修改为：
```json ```json
{} {}
``` ```
- `GPT_4_S_New_Name`、`GPT_4_MOBILE_NEW_NAME`、`GPT_3_5_NEW_NAME`: 用于设置 gpt-4-s、gpt-4-mobile、gpt-3.5-turbo 的模型名称与别名，如果不需要修改，可以保持不变。如果需要修改，每个模型均支持设置多个别名，多个别名之间以英文逗号隔开，例如：`gpt-4-s` 的别名可以设置为 `gpt-4-s,dall-e-3`，这样在调用的时候就可以使用 `gpt-4-s` 或者 `dall-e-3` 来调用该模型。 - `gpt_4_s_new_name`、`gpt_4_mobile_new_name`、`gpt_3_5_new_name`: 用于设置 gpt-4-s、gpt-4-mobile、gpt-3.5-turbo 的模型名称与别名，如果不需要修改，可以保持不变。如果需要修改，每个模型均支持设置多个别名，多个别名之间以英文逗号隔开，例如：`gpt-4-s` 的别名可以设置为 `gpt-4-s,dall-e-3`，这样在调用的时候就可以使用 `gpt-4-s` 或者 `dall-e-3` 来调用该模型。
- `API_PREFIX`: 用于设置本项目 `/v1/xxx` 接口的前缀，如果留空则与官方api调用接口一致。设置示例：`666` - `need_delete_conversation_after_response`: 用于设置是否在响应后删除对话，可选值为：`true`、`false`，默认为 `false`，如果设置为 `true`，则会在响应后删除对话，这样可以保证在页面上不会留下通过本项目调用的对话记录.
- `LOG_LEVEL`: 用于设置日志等级，可选值为：`DEBUG`、`INFO`、`WARNING`、`ERROR`，默认为 `DEBUG` - `bot_mode`
- `NEED_LOG_TO_FILE`: 用于设置是否需要将日志输出到文件，可选值为：`true`、`false`，默认为 `true`，日志文件路径为：`./log/access.log`，默认每天会自动分割日志文件。 - `enabled`: 用于设置是否开启 Bot 模式，可选值为：`true`、`false`，默认为 `false`，开启 Bot 模式后，将可以自定义联网插件的引引用、绘图插件的markdown格式的图片以及插件执行过程的输出，仅建议在 QQ、微信机器人等 Bot 项目中开启，网页应用请不要开启。
- `BOT_MODE`: 用于设置是否开启 Bot 模式，可选值为：`true`、`false`，默认为 `false`，开启 Bot 模式后，将不会输出联网插件的引用以及绘图插件的markdown格式的图片，仅建议在 QQ、微信机器人等 Bot 项目中开启，网页应用请不要开启。 - `enabled_markdown_image_output`: 用于设置是否开启 Bot 模式下绘图插件的markdown格式的图片输出，可选值为：`true`、`false`，默认为 `false`，开启后，将会输出markdown格式的图片输出，仅在 `bot_mode.enabled` 为 `true` 时生效。
- `enabled_bing_reference_output`: 用于设置是否开启 Bot 模式下联网插件的引用输出，可选值为：`true`、`false`，默认为 `false`，开启后，将会输出联网插件的引用，仅在 `bot_mode.enabled` 为 `true` 时生效。
- `enabled_plugin_output`: 用于设置是否开启 Bot 模式下插件执行过程的输出，可选值为：`true`、`false`，默认为 `false`，开启后，将会输出插件执行过程的输出，仅在 `bot_mode.enabled` 为 `true` 时生效。
## GPTS配置说明 ## GPTS配置说明

19
data/config.json Normal file
View File

@@ -0,0 +1,19 @@
{
"log_level": "DEBUG",
"need_log_to_file": "true",
"pandora_base_url": "",
"pandora_api_prefix": "",
"backend_container_url": "",
"backend_container_api_prefix": "",
"key_for_gpts_info": "",
"gpt_4_s_new_name": "gpt-4-s",
"gpt_4_mobile_new_name": "gpt-4-mobile,dall-e-3",
"gpt_3_5_new_name": "gpt-3.5-turbo",
"need_delete_conversation_after_response": "true",
"bot_mode": {
"enabled": "false",
"enabled_markdown_image_output": "false",
"enabled_bing_reference_output": "false",
"enabled_plugin_output": "false"
}
}

0
gpts.json → data/gpts.json
View File

14
docker-compose.yml
View File

@@ -6,20 +6,8 @@ services:
restart: always restart: always
ports: ports:
- "50011:33333" - "50011:33333"
environment:
- BASE_URL=<Pandora-Next的部署地址，如：https://pandoranext.com>
- PROXY_API_PREFIX=<Pandora-Next的PROXY_API_PREFIX>
- UPLOAD_BASE_URL=<50011端口在公网可以访问到的地址，如：http://1.2.3.4:50011，如果使用了反代需填入反代后的域名如：https://pandora-backend-api.com>
- KEY_FOR_GPTS_INFO=<一个仅用于获取GPTs信息的fk> # 如果不需要额外使用gpts，可以不填
- GPT_4_S_New_Name=gpt-4-s # gpt-4-s模型的自定义模型名称，支持同时设置多个，用英文逗号分隔
- GPT_4_MOBILE_NEW_NAME=gpt-4-mobile # gpt-4-mobile模型的自定义模型名称，支持同时设置多个，用英文逗号分隔
- GPT_3_5_NEW_NAME=gpt-3.5-turbo # gpt-3.5-turbo模型的自定义模型名称，支持同时设置多个，用英文逗号分隔
- API_PREFIX=<本项目接口前缀> # 本项目/v1接口的前缀，示例：666，如果留空默认为原版本一致
- LOG_LEVEL=<日志等级> # 日志输出等级，可选值：DEBUG,INFO,WARNING,ERROR,CRITICAL
- NEED_LOG_TO_FILE=<是否需要输出日志到文件> # 是否需要输出日志到文件，可选值：true,false
- BOT_MODE=false
volumes: volumes:
- ./log:/app/log - ./log:/app/log
- ./images:/app/images - ./images:/app/images
- ./gpts.json:/app/gpts.json - ./data:/app/data

140
main.py
View File

@@ -24,11 +24,35 @@ from io import BytesIO
from urllib.parse import urlparse, urlunparse from urllib.parse import urlparse, urlunparse
import base64 import base64
# 读取配置文件
def load_config(file_path):
with open(file_path, 'r', encoding='utf-8') as file:
return json.load(file)
NEED_LOG_TO_FILE = os.getenv('NEED_LOG_TO_FILE', 'true').lower() == 'true' CONFIG = load_config('./data/config.json')
# 获取环境变量中的日志级别，如果没有设置，则默认为DEBUG LOG_LEVEL = CONFIG.get('log_level', 'DEBUG').upper()
LOG_LEVEL = os.getenv('LOG_LEVEL', 'DEBUG').upper() NEED_LOG_TO_FILE = CONFIG.get('need_log_to_file', 'true').lower() == 'true'
# 使用 get 方法获取配置项，同时提供默认值
BASE_URL = CONFIG.get('pandora_base_url', '')
PROXY_API_PREFIX = CONFIG.get('pandora_api_prefix', '')
if PROXY_API_PREFIX != '':
PROXY_API_PREFIX = "/" + PROXY_API_PREFIX
UPLOAD_BASE_URL = CONFIG.get('backend_container_url', '')
KEY_FOR_GPTS_INFO = CONFIG.get('key_for_gpts_info', '')
API_PREFIX = CONFIG.get('backend_container_api_prefix', '')
GPT_4_S_New_Names = CONFIG.get('gpt_4_s_new_name', 'gpt-4-s').split(',')
GPT_4_MOBILE_NEW_NAMES = CONFIG.get('gpt_4_mobile_new_name', 'gpt-4-mobile').split(',')
GPT_3_5_NEW_NAMES = CONFIG.get('gpt_3_5_new_name', 'gpt-3.5-turbo').split(',')
BOT_MODE = CONFIG.get('bot_mode', {})
BOT_MODE_ENABLED = BOT_MODE.get('enabled', 'false').lower() == 'true'
BOT_MODE_ENABLED_MARKDOWN_IMAGE_OUTPUT = BOT_MODE.get('enabled_markdown_image_output', 'false').lower() == 'true'
BOT_MODE_ENABLED_BING_REFERENCE_OUTPUT = BOT_MODE.get('enabled_bing_reference_output', 'false').lower() == 'true'
BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT = BOT_MODE.get('enabled_plugin_output', 'false').lower() == 'true'
NEED_DELETE_CONVERSATION_AFTER_RESPONSE = CONFIG.get('need_delete_conversation_after_response', 'true').lower() == 'true'
# 设置日志级别 # 设置日志级别
log_level_dict = { log_level_dict = {
@@ -83,7 +107,7 @@ def load_gpts_config(file_path):
# 根据 ID 发送请求并获取配置信息 # 根据 ID 发送请求并获取配置信息
def fetch_gizmo_info(base_url, proxy_api_prefix, model_id): def fetch_gizmo_info(base_url, proxy_api_prefix, model_id):
url = f"{base_url}/{proxy_api_prefix}/backend-api/gizmos/{model_id}" url = f"{base_url}{proxy_api_prefix}/backend-api/gizmos/{model_id}"
headers = { headers = {
"Authorization": f"Bearer {KEY_FOR_GPTS_INFO}" "Authorization": f"Bearer {KEY_FOR_GPTS_INFO}"
} }
@@ -142,21 +166,14 @@ def generate_gpts_payload(model, messages):
app = Flask(__name__) app = Flask(__name__)
CORS(app, resources={r"/images/*": {"origins": "*"}}) CORS(app, resources={r"/images/*": {"origins": "*"}})
# 添加环境变量配置
BASE_URL = os.getenv('BASE_URL', '')
PROXY_API_PREFIX = os.getenv('PROXY_API_PREFIX', '')
UPLOAD_BASE_URL = os.getenv('UPLOAD_BASE_URL', '')
KEY_FOR_GPTS_INFO = os.getenv('KEY_FOR_GPTS_INFO', '')
# 添加环境变量配置
API_PREFIX = os.getenv('API_PREFIX', '')
BOT_MODE = os.getenv('BOT_MODE', 'false').lower() == 'true'
PANDORA_UPLOAD_URL = 'files.pandoranext.com' PANDORA_UPLOAD_URL = 'files.pandoranext.com'
VERSION = '0.2.1' VERSION = '0.3.0'
# VERSION = 'test' # VERSION = 'test'
UPDATE_INFO = '支持 BOT 模式' UPDATE_INFO = '调整项目目录结构以适配更多的配置项'
# UPDATE_INFO = '【仅供临时测试使用】 ' # UPDATE_INFO = '【仅供临时测试使用】 '
with app.app_context(): with app.app_context():
@@ -170,44 +187,49 @@ with app.app_context():
logger.info(f"LOG_LEVEL: {LOG_LEVEL}") logger.info(f"LOG_LEVEL: {LOG_LEVEL}")
logger.info(f"NEED_LOG_TO_FILE: {NEED_LOG_TO_FILE}") logger.info(f"NEED_LOG_TO_FILE: {NEED_LOG_TO_FILE}")
logger.info(f"BOT_MODE: {BOT_MODE}") logger.info(f"BOT_MODE_ENABLED: {BOT_MODE_ENABLED}")
if BOT_MODE_ENABLED:
logger.info(f"enabled_markdown_image_output: {BOT_MODE_ENABLED_MARKDOWN_IMAGE_OUTPUT}")
logger.info(f"enabled_bing_reference_output: {BOT_MODE_ENABLED_BING_REFERENCE_OUTPUT}")
logger.info(f"enabled_plugin_output: {BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT}")
if not BASE_URL: if not BASE_URL:
raise Exception('BASE_URL is not set') raise Exception('pandora_base_url is not set')
else: else:
logger.info(f"BASE_URL: {BASE_URL}") logger.info(f"pandora_base_url: {BASE_URL}")
if not PROXY_API_PREFIX: if not PROXY_API_PREFIX:
raise Exception('PROXY_API_PREFIX is not set') logger.warning('pandora_api_prefix is not set')
else: else:
logger.info(f"PROXY_API_PREFIX: {PROXY_API_PREFIX}") logger.info(f"pandora_api_prefix: {PROXY_API_PREFIX}")
if not UPLOAD_BASE_URL: if not UPLOAD_BASE_URL:
logger.info("UPLOAD_BASE_URL 未设置，绘图功能将无法正常使用") logger.info("backend_container_url 未设置，绘图功能将无法正常使用")
else: else:
logger.info(f"UPLOAD_BASE_URL: {UPLOAD_BASE_URL}") logger.info(f"backend_container_url: {UPLOAD_BASE_URL}")
if not KEY_FOR_GPTS_INFO: if not KEY_FOR_GPTS_INFO:
logger.warning("KEY_FOR_GPTS_INFO 未设置，请将 gpts.json 中仅保留 “{}” 作为内容") logger.warning("key_for_gpts_info 未设置，请将 gpts.json 中仅保留 “{}” 作为内容")
else: else:
logger.info(f"KEY_FOR_GPTS_INFO: {KEY_FOR_GPTS_INFO}") logger.info(f"key_for_gpts_info: {KEY_FOR_GPTS_INFO}")
if not API_PREFIX: if not API_PREFIX:
logger.warning("API_PREFIX 未设置，安全性会有所下降") logger.warning("backend_container_api_prefix 未设置，安全性会有所下降")
logger.info(f'Chat 接口 URI: /v1/chat/completions') logger.info(f'Chat 接口 URI: /v1/chat/completions')
logger.info(f'绘图接口 URI: /v1/images/generations') logger.info(f'绘图接口 URI: /v1/images/generations')
else: else:
logger.info(f"API_PREFIX: {API_PREFIX}") logger.info(f"backend_container_api_prefix: {API_PREFIX}")
logger.info(f'Chat 接口 URI: /{API_PREFIX}/v1/chat/completions') logger.info(f'Chat 接口 URI: /{API_PREFIX}/v1/chat/completions')
logger.info(f'绘图接口 URI: /{API_PREFIX}/v1/images/generations') logger.info(f'绘图接口 URI: /{API_PREFIX}/v1/images/generations')
logger.info(f"need_delete_conversation_after_response: {NEED_DELETE_CONVERSATION_AFTER_RESPONSE}")
logger.info(f"==========================================") logger.info(f"==========================================")
# 从环境变量中读取模型名称，支持用逗号分隔的多个名称
GPT_4_S_New_Names = os.getenv('GPT_4_S_New_Name', 'gpt-4-s').split(',')
GPT_4_MOBILE_NEW_NAMES = os.getenv('GPT_4_MOBILE_NEW_NAME', 'gpt-4-mobile').split(',')
GPT_3_5_NEW_NAMES = os.getenv('GPT_3_5_NEW_NAME', 'gpt-3.5-turbo').split(',')
# 更新 gpts_configurations 列表，支持多个映射 # 更新 gpts_configurations 列表，支持多个映射
gpts_configurations = [] gpts_configurations = []
@@ -231,7 +253,7 @@ with app.app_context():
logger.info(f"GPTS 配置信息") logger.info(f"GPTS 配置信息")
# 加载配置并添加到全局列表 # 加载配置并添加到全局列表
gpts_data = load_gpts_config("./gpts.json") gpts_data = load_gpts_config("./data/gpts.json")
add_config_to_global_list(BASE_URL, PROXY_API_PREFIX, gpts_data) add_config_to_global_list(BASE_URL, PROXY_API_PREFIX, gpts_data)
# print("当前可用GPTS：" + get_accessible_model_list()) # print("当前可用GPTS：" + get_accessible_model_list())
# 输出当前可用 GPTS name # 输出当前可用 GPTS name
@@ -252,7 +274,7 @@ with app.app_context():
# 定义获取 token 的函数 # 定义获取 token 的函数
def get_token(): def get_token():
url = f"{BASE_URL}/{PROXY_API_PREFIX}/api/arkose/token" url = f"{BASE_URL}{PROXY_API_PREFIX}/api/arkose/token"
payload = {'type': 'gpt-4'} payload = {'type': 'gpt-4'}
response = requests.post(url, data=payload) response = requests.post(url, data=payload)
if response.status_code == 200: if response.status_code == 200:
@@ -314,7 +336,7 @@ def upload_file(file_content, mime_type, api_key):
logger.debug(f"非已知文件类型，MINE置空") logger.debug(f"非已知文件类型，MINE置空")
# 第1步：调用/backend-api/files接口获取上传URL # 第1步：调用/backend-api/files接口获取上传URL
upload_api_url = f"{BASE_URL}/{PROXY_API_PREFIX}/backend-api/files" upload_api_url = f"{BASE_URL}{PROXY_API_PREFIX}/backend-api/files"
upload_request_payload = { upload_request_payload = {
"file_name": file_name, "file_name": file_name,
"file_size": file_size, "file_size": file_size,
@@ -350,7 +372,7 @@ def upload_file(file_content, mime_type, api_key):
raise Exception("Failed to upload file") raise Exception("Failed to upload file")
# 第3步：检测上传是否成功并检查响应 # 第3步：检测上传是否成功并检查响应
check_url = f"{BASE_URL}/{PROXY_API_PREFIX}/backend-api/files/{file_id}/uploaded" check_url = f"{BASE_URL}{PROXY_API_PREFIX}/backend-api/files/{file_id}/uploaded"
check_response = requests.post(check_url, json={}, headers=headers) check_response = requests.post(check_url, json={}, headers=headers)
logger.debug(f"check_response: {check_response.text}") logger.debug(f"check_response: {check_response.text}")
if check_response.status_code != 200: if check_response.status_code != 200:
@@ -428,7 +450,7 @@ my_files_types = [
# 定义发送请求的函数 # 定义发送请求的函数
def send_text_prompt_and_get_response(messages, api_key, stream, model): def send_text_prompt_and_get_response(messages, api_key, stream, model):
url = f"{BASE_URL}/{PROXY_API_PREFIX}/backend-api/conversation" url = f"{BASE_URL}{PROXY_API_PREFIX}/backend-api/conversation"
headers = { headers = {
"Authorization": f"Bearer {api_key}" "Authorization": f"Bearer {api_key}"
} }
@@ -603,8 +625,11 @@ def send_text_prompt_and_get_response(messages, api_key, stream, model):
def delete_conversation(conversation_id, api_key): def delete_conversation(conversation_id, api_key):
logger.info(f"准备删除的会话id： {conversation_id}") logger.info(f"准备删除的会话id： {conversation_id}")
if conversation_id: if not NEED_DELETE_CONVERSATION_AFTER_RESPONSE:
patch_url = f"{BASE_URL}/{PROXY_API_PREFIX}/backend-api/conversation/{conversation_id}" logger.info(f"自动删除会话功能已禁用")
return
if conversation_id and NEED_DELETE_CONVERSATION_AFTER_RESPONSE:
patch_url = f"{BASE_URL}{PROXY_API_PREFIX}/backend-api/conversation/{conversation_id}"
patch_headers = { patch_headers = {
"Authorization": f"Bearer {api_key}", "Authorization": f"Bearer {api_key}",
} }
@@ -685,7 +710,7 @@ def replace_complete_citation(text, citations):
logger.debug(f"citation: {citation}") logger.debug(f"citation: {citation}")
if cited_message_idx == int(citation_number): if cited_message_idx == int(citation_number):
url = citation.get("metadata", {}).get("url", "") url = citation.get("metadata", {}).get("url", "")
if BOT_MODE == False: if ((BOT_MODE_ENABLED == False) or (BOT_MODE_ENABLED == True and BOT_MODE_ENABLED_BING_REFERENCE_OUTPUT == True)):
return f"[[{citation_number}]({url})]" return f"[[{citation_number}]({url})]"
else: else:
return "" return ""
@@ -761,8 +786,8 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
content = message.get("content", {}) content = message.get("content", {})
role = message.get("author", {}).get("role") role = message.get("author", {}).get("role")
content_type = content.get("content_type") content_type = content.get("content_type")
# print(f"content_type: {content_type}") print(f"content_type: {content_type}")
# print(f"last_content_type: {last_content_type}") print(f"last_content_type: {last_content_type}")
metadata = {} metadata = {}
citations = [] citations = []
@@ -795,7 +820,7 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
is_img_message = True is_img_message = True
asset_pointer = part.get('asset_pointer').replace('file-service://', '') asset_pointer = part.get('asset_pointer').replace('file-service://', '')
logger.debug(f"asset_pointer: {asset_pointer}") logger.debug(f"asset_pointer: {asset_pointer}")
image_url = f"{BASE_URL}/{PROXY_API_PREFIX}/backend-api/files/{asset_pointer}/download" image_url = f"{BASE_URL}{PROXY_API_PREFIX}/backend-api/files/{asset_pointer}/download"
headers = { headers = {
"Authorization": f"Bearer {api_key}" "Authorization": f"Bearer {api_key}"
@@ -813,12 +838,16 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
logger.debug(f"下载图片成功") logger.debug(f"下载图片成功")
image_data = image_download_response.content image_data = image_download_response.content
today_image_url = save_image(image_data) # 保存图片，并获取文件名 today_image_url = save_image(image_data) # 保存图片，并获取文件名
if BOT_MODE == False: if ((BOT_MODE_ENABLED == False) or (BOT_MODE_ENABLED == True and BOT_MODE_ENABLED_MARKDOWN_IMAGE_OUTPUT == True)):
new_text = f"\n![image]({UPLOAD_BASE_URL}/{today_image_url})\n[下载链接]({UPLOAD_BASE_URL}/{today_image_url})\n" new_text = f"\n![image]({UPLOAD_BASE_URL}/{today_image_url})\n[下载链接]({UPLOAD_BASE_URL}/{today_image_url})\n"
else: else:
logger.error(f"下载图片失败: {image_download_response.text}") logger.error(f"下载图片失败: {image_download_response.text}")
if last_content_type == "code": if last_content_type == "code":
new_text = "\n```\n" + new_text if BOT_MODE_ENABLED and BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT == False:
new_text = content.get("text", "")
else:
new_text = "\n```\n" + new_text
logger.debug(f"new_text: {new_text}") logger.debug(f"new_text: {new_text}")
is_img_message = True is_img_message = True
else: else:
@@ -831,6 +860,8 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
# print(f"data_json: {data_json}") # print(f"data_json: {data_json}")
if content_type == "multimodal_text" and last_content_type == "code": if content_type == "multimodal_text" and last_content_type == "code":
new_text = "\n```\n" + content.get("text", "") new_text = "\n```\n" + content.get("text", "")
if BOT_MODE_ENABLED and BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT == False:
new_text = content.get("text", "")
elif role == "tool" and name == "dalle.text2im": elif role == "tool" and name == "dalle.text2im":
logger.debug(f"无视消息: {content.get('text', '')}") logger.debug(f"无视消息: {content.get('text', '')}")
continue continue
@@ -842,6 +873,8 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
# print(f"last_full_code: {last_full_code}") # print(f"last_full_code: {last_full_code}")
# print(f"new_text: {new_text}") # print(f"new_text: {new_text}")
last_full_code = full_code # 更新完整代码以备下次比较 last_full_code = full_code # 更新完整代码以备下次比较
if BOT_MODE_ENABLED and BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT == False:
new_text = ""
elif last_content_type == "code" and content_type != "code" and content_type != None: elif last_content_type == "code" and content_type != "code" and content_type != None:
full_code = ''.join(content.get("text", "")) full_code = ''.join(content.get("text", ""))
@@ -850,6 +883,8 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
# print(f"last_full_code: {last_full_code}") # print(f"last_full_code: {last_full_code}")
# print(f"new_text: {new_text}") # print(f"new_text: {new_text}")
last_full_code = "" # 更新完整代码以备下次比较 last_full_code = "" # 更新完整代码以备下次比较
if BOT_MODE_ENABLED and BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT == False:
new_text = ""
elif content_type == "code" and last_content_type == "code" and content_type != None: elif content_type == "code" and last_content_type == "code" and content_type != None:
full_code = ''.join(content.get("text", "")) full_code = ''.join(content.get("text", ""))
@@ -858,6 +893,8 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
# print(f"last_full_code: {last_full_code}") # print(f"last_full_code: {last_full_code}")
# print(f"new_text: {new_text}") # print(f"new_text: {new_text}")
last_full_code = full_code # 更新完整代码以备下次比较 last_full_code = full_code # 更新完整代码以备下次比较
if BOT_MODE_ENABLED and BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT == False:
new_text = ""
else: else:
# 只获取新的 parts # 只获取新的 parts
@@ -906,7 +943,10 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
full_code_result = ''.join(content.get("text", "")) full_code_result = ''.join(content.get("text", ""))
new_text = "`Result:` \n```\n" + full_code_result[len(last_full_code_result):] new_text = "`Result:` \n```\n" + full_code_result[len(last_full_code_result):]
if last_content_type == "code": if last_content_type == "code":
new_text = "\n```\n" + new_text if BOT_MODE_ENABLED and BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT == False:
new_text = ""
else:
new_text = "\n```\n" + new_text
# print(f"full_code_result: {full_code_result}") # print(f"full_code_result: {full_code_result}")
# print(f"last_full_code_result: {last_full_code_result}") # print(f"last_full_code_result: {last_full_code_result}")
# print(f"new_text: {new_text}") # print(f"new_text: {new_text}")
@@ -915,15 +955,23 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
# new_text = content.get("text", "") + "\n```" # new_text = content.get("text", "") + "\n```"
full_code_result = ''.join(content.get("text", "")) full_code_result = ''.join(content.get("text", ""))
new_text = full_code_result[len(last_full_code_result):] + "\n```\n" new_text = full_code_result[len(last_full_code_result):] + "\n```\n"
if BOT_MODE_ENABLED and BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT == False:
new_text = ""
if content_type == "code": if content_type == "code":
new_text = new_text + "\n```\n" if BOT_MODE_ENABLED and BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT == False:
new_text = ""
else:
new_text = new_text + "\n```\n"
# print(f"full_code_result: {full_code_result}") # print(f"full_code_result: {full_code_result}")
# print(f"last_full_code_result: {last_full_code_result}") # print(f"last_full_code_result: {last_full_code_result}")
# print(f"new_text: {new_text}") # print(f"new_text: {new_text}")
last_full_code_result = "" # 更新完整代码以备下次比较 last_full_code_result = "" # 更新完整代码以备下次比较
elif last_content_type == "execution_output" and role == "tool" and name == "python" and content_type != None: elif last_content_type == "execution_output" and role == "tool" and name == "python" and content_type != None:
full_code_result = ''.join(content.get("text", "")) full_code_result = ''.join(content.get("text", ""))
new_text = full_code_result[len(last_full_code_result):] if BOT_MODE_ENABLED and BOT_MODE_ENABLED_CODE_BLOCK_OUTPUT == False:
new_text = ""
else:
new_text = full_code_result[len(last_full_code_result):]
# print(f"full_code_result: {full_code_result}") # print(f"full_code_result: {full_code_result}")
# print(f"last_full_code_result: {last_full_code_result}") # print(f"last_full_code_result: {last_full_code_result}")
# print(f"new_text: {new_text}") # print(f"new_text: {new_text}")
@@ -1299,7 +1347,7 @@ def images_generations():
is_img_message = True is_img_message = True
asset_pointer = part.get('asset_pointer').replace('file-service://', '') asset_pointer = part.get('asset_pointer').replace('file-service://', '')
logger.debug(f"asset_pointer: {asset_pointer}") logger.debug(f"asset_pointer: {asset_pointer}")
image_url = f"{BASE_URL}/{PROXY_API_PREFIX}/backend-api/files/{asset_pointer}/download" image_url = f"{BASE_URL}{PROXY_API_PREFIX}/backend-api/files/{asset_pointer}/download"
headers = { headers = {
"Authorization": f"Bearer {api_key}" "Authorization": f"Bearer {api_key}"