From 775ac129dce34af4c72714df0ade4b33a8fe2b7c Mon Sep 17 00:00:00 2001 From: supreme0597 Date: Mon, 27 Apr 2026 06:36:12 +0800 Subject: [PATCH] =?UTF-8?q?fix(knowledge):=20=E4=BF=AE=E5=A4=8D=E7=9F=A5?= =?UTF-8?q?=E8=AF=86=E5=BA=93=E6=B5=8B=E8=AF=95=E9=97=AE=E9=A2=98=E7=94=9F?= =?UTF-8?q?=E6=88=90=E6=8A=A5=E9=94=99=20Unknown=20model=20provider?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit select_model() 无参调用走了 V1 旧逻辑，无法正确解析 V2 格式的 default_model 配置（如 supreme_newapi:openai）。传入 config.default_model 作为 model_spec 参数，使模型选择走 V2 路径。 影响范围：知识库测试问题生成、知识库描述生成、思维导图生成 --- backend/server/routers/knowledge_router.py | 4 ++-- backend/server/routers/mindmap_router.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/backend/server/routers/knowledge_router.py b/backend/server/routers/knowledge_router.py index c3ade1c4..8b734297 100644 --- a/backend/server/routers/knowledge_router.py +++ b/backend/server/routers/knowledge_router.py @@ -1102,7 +1102,7 @@ async def generate_sample_questions( logger.info(f"开始生成知识库问题，知识库: {db_name}, 文件数量: {len(files_info)}, 问题数量: {count}") # 选择模型并调用 - model = select_model() + model = select_model(model_spec=config.default_model) messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": user_message}] response = await model.call(messages, stream=False) @@ -1523,7 +1523,7 @@ async def generate_description( """).strip() try: - model = select_model() + model = select_model(model_spec=config.default_model) response = await model.call(prompt) description = response.content.strip() logger.debug(f"Generated description: {description}") diff --git a/backend/server/routers/mindmap_router.py b/backend/server/routers/mindmap_router.py index e9b90b96..57f2d2c5 100644 --- a/backend/server/routers/mindmap_router.py +++ b/backend/server/routers/mindmap_router.py @@ -15,7 +15,7 @@ from fastapi import APIRouter, Body, Depends, HTTPException from yuxi.storage.postgres.models_business import User from server.utils.auth_middleware import get_admin_user -from yuxi import knowledge_base +from yuxi import config, knowledge_base from yuxi.models import select_model from yuxi.utils import logger @@ -214,7 +214,7 @@ async def generate_mindmap( logger.info(f"开始生成思维导图，知识库: {db_name}, 文件数量: {len(files_info)}") # 选择模型并调用 - model = select_model() + model = select_model(model_spec=config.default_model) messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": user_message}] response = await model.call(messages, stream=False)