From ccdf7ae81da4107fbd5eb2d4686b2ebd2b62fe11 Mon Sep 17 00:00:00 2001
From: Timebomb2018 <18868801967@163.com>
Date: Thu, 16 Apr 2026 17:40:30 +0800
Subject: [PATCH] refactor(model): replace VolcanoChatOpenAI with
 CompatibleChatOpenAI for unified omni model support

---
 api/app/core/models/base.py                               | 8 ++++----
 .../core/models/{volcano_chat.py => compatible_chat.py}   | 4 ++--
 2 files changed, 6 insertions(+), 6 deletions(-)
 rename api/app/core/models/{volcano_chat.py => compatible_chat.py} (92%)

diff --git a/api/app/core/models/base.py b/api/app/core/models/base.py
index 57054da1..7b570b47 100644
--- a/api/app/core/models/base.py
+++ b/api/app/core/models/base.py
@@ -14,7 +14,7 @@ from pydantic import BaseModel, Field, model_validator
 from app.core.error_codes import BizCode
 from app.core.exceptions import BusinessException
 from app.models.models_model import ModelProvider, ModelType
-from app.core.models.volcano_chat import VolcanoChatOpenAI
+from app.core.models.compatible_chat import CompatibleChatOpenAI
 
 T = TypeVar("T")
 
@@ -255,11 +255,11 @@ def get_provider_llm_class(config: RedBearModelConfig, type: ModelType = ModelTy
     """根据模型提供商获取对应的模型类"""
     provider = config.provider.lower()
 
-    # dashscope 的 omni 模型使用 OpenAI 兼容模式
+    # dashscope的omni模型 和 volcano模型使用
     if provider == ModelProvider.DASHSCOPE and config.is_omni:
-        return ChatOpenAI
+        return CompatibleChatOpenAI
     if provider == ModelProvider.VOLCANO:
-        return VolcanoChatOpenAI
+        return CompatibleChatOpenAI
     if provider in [ModelProvider.OPENAI, ModelProvider.XINFERENCE, ModelProvider.GPUSTACK]:
         if type == ModelType.LLM:
             return OpenAI
diff --git a/api/app/core/models/volcano_chat.py b/api/app/core/models/compatible_chat.py
similarity index 92%
rename from api/app/core/models/volcano_chat.py
rename to api/app/core/models/compatible_chat.py
index d9a51d13..114a3567 100644
--- a/api/app/core/models/volcano_chat.py
+++ b/api/app/core/models/compatible_chat.py
@@ -12,8 +12,8 @@ from langchain_core.outputs import ChatGenerationChunk, ChatResult
 from langchain_openai import ChatOpenAI
 
 
-class VolcanoChatOpenAI(ChatOpenAI):
-    """火山引擎 Chat 模型，支持深度思考内容（reasoning_content）的流式和非流式透传。"""
+class CompatibleChatOpenAI(ChatOpenAI):
+    """火山和千问的omni兼容模型，支持深度思考内容（reasoning_content）的流式和非流式透传。"""
 
     def _create_chat_result(self, response: Union[dict, Any], generation_info: Optional[dict] = None) -> ChatResult:
         result = super()._create_chat_result(response, generation_info)