feat: 支持图像生成流式响应并优化配置

- 为 OpenAI 兼容路由的图像生成聊天添加流式支持。
- 重构 `gemini-2.0-flash-exp` 安全设置,使用常量统一管理。
- 更改图像生成默认响应格式为 `url`。
- 启用 `.env.example` 中的 `AUTH_TOKEN`。
- 清理部分代码注释。
This commit is contained in:
snaily
2025-05-03 20:37:09 +08:00
parent 929592bbc4
commit 56f6f5e198
6 changed files with 16 additions and 24 deletions
+5 -3
View File
@@ -90,7 +90,11 @@ async def chat_completion(
if is_image_chat:
# 图像生成聊天
response = await chat_service.create_image_chat_completion(request, current_api_key)
return response # 直接返回,不处理流式
# 处理流式响应
if request.stream:
return StreamingResponse(response, media_type="text/event-stream")
# 非流式直接返回结果
return response
else:
# 普通聊天补全
response = await chat_service.create_chat_completion(request, current_api_key)
@@ -111,8 +115,6 @@ async def generate_image(
operation_name = "generate_image"
async with handle_route_errors(logger, operation_name):
logger.info(f"Handling image generation request for prompt: {request.prompt}")
# 注意:这里假设 image_create_service.generate_images 是同步函数
# 如果它是异步的,需要 await
response = image_create_service.generate_images(request)
return response