kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Merge pull request #659 from supreme0597/fix/model-select-v2

Browse Source 
fix(knowledge): 修复知识库测试问题生成报错 Unknown model provider
This commit is contained in:
Wenjie Zhang 2026-04-27 09:11:28 +08:00 committed by GitHub
commit 701ce09901
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/server/routers/knowledge_router.py
View File

@ -1102,7 +1102,7 @@ async def generate_sample_questions(
logger.info(f"开始生成知识库问题，知识库: {db_name}, 文件数量: {len(files_info)}, 问题数量: {count}") 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}] messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": user_message}]
response = await model.call(messages, stream=False) response = await model.call(messages, stream=False)
@ -1523,7 +1523,7 @@ async def generate_description(
""").strip() """).strip()
try: try:
model = select_model() model = select_model(model_spec=config.default_model)
response = await model.call(prompt) response = await model.call(prompt)
description = response.content.strip() description = response.content.strip()
logger.debug(f"Generated description: {description}") logger.debug(f"Generated description: {description}")

4
backend/server/routers/mindmap_router.py
View File

@ -15,7 +15,7 @@ from fastapi import APIRouter, Body, Depends, HTTPException
from yuxi.storage.postgres.models_business import User from yuxi.storage.postgres.models_business import User
from server.utils.auth_middleware import get_admin_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.models import select_model
from yuxi.utils import logger from yuxi.utils import logger
@ -214,7 +214,7 @@ async def generate_mindmap(
logger.info(f"开始生成思维导图，知识库: {db_name}, 文件数量: {len(files_info)}") 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}] messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": user_message}]
response = await model.call(messages, stream=False) response = await model.call(messages, stream=False)