From 97b9b9923558901825be8fce22d629f997f2f14c Mon Sep 17 00:00:00 2001
From: yinpeng <2291314224@qq.com>
Date: Tue, 11 Feb 2025 04:20:28 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=A0=B9=E6=8D=AE=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E7=B1=BB=E5=9E=8B=E9=80=89=E6=8B=A9=E4=B8=8D=E5=90=8C=E7=9A=84?=
 =?UTF-8?q?API=E5=AF=86=E9=92=A5=E5=A4=84=E7=90=86=E8=81=8A=E5=A4=A9?=
 =?UTF-8?q?=E8=AF=B7=E6=B1=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/openai_routes.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/app/api/openai_routes.py b/app/api/openai_routes.py
index 9812c16..9982ea8 100644
--- a/app/api/openai_routes.py
+++ b/app/api/openai_routes.py
@@ -54,7 +54,11 @@ async def chat_completion(
     logger.info(f"Request: \n{request.model_dump_json(indent=2)}")
     logger.info(f"Using API key: {api_key}")
     try:
-        response = await chat_service.create_image_chat_completion(request=request)
+        # 如果model是imagen3,使用paid_key
+        if request.model == f"{settings.CREATE_IMAGE_MODEL}-chat":
+            response = await chat_service.create_image_chat_completion(request=request)
+        else:
+            response = await chat_service.create_chat_completion(request,api_key)
         # 处理流式响应
         if request.stream:
             return StreamingResponse(response, media_type="text/event-stream")