fix: 修改LLMProvider 类型

This commit is contained in:
XIE7654
2025-08-11 10:26:42 +08:00
parent e21a1ac716
commit a88f272c19
7 changed files with 54 additions and 12 deletions

View File

@@ -11,6 +11,7 @@ from langchain.chains import ConversationChain
from api.v1.chat.vo import MessageVO
from deps.auth import get_current_user
from llm.factory import get_adapter
from llm.enums import LLMProvider
from services.chat_service import ChatDBService
from db.session import get_db
from models.ai import ChatConversation, ChatMessage
@@ -37,11 +38,13 @@ async def chat_stream(request: Request, user=Depends(get_current_user), db: Sess
if platform == 'tongyi':
model = 'qwen-plus'
api_key = os.getenv("DASHSCOPE_API_KEY")
provider = LLMProvider.TONGYI
else:
# 默认使用 DeepSeek
model = 'deepseek-chat'
api_key = os.getenv("DEEPSEEK_API_KEY")
llm = get_adapter(platform, api_key=api_key, model=model)
provider = LLMProvider.DEEPSEEK
llm = get_adapter(provider, api_key=api_key, model=model)
if not content or not isinstance(content, str):
from fastapi.responses import JSONResponse
@@ -94,9 +97,11 @@ async def create_conversation(request: Request, db: Session = Depends(get_db), u
platform = body.get('platform')
if platform == 'tongyi':
model = 'qwen-plus'
# provider = LLMProvider.TONGYI
else:
# 默认使用 DeepSeek
model = 'deepseek-chat'
# provider = LLMProvider.DEEPSEEK
user_id = user["user_id"]
conversation = ChatDBService.get_or_create_conversation(db, None, user_id, model, '新对话')
return resp_success(data=conversation.id)