YueHuihui
/
RAGflow
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: 修复知识库检索问题

This commit is contained in:
zstar 2025-05-16 21:42:19 +08:00
parent 73d4fe573c
commit 42cf80c082
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
management/server/services/knowledgebases/service.py
View File

@ -205,7 +205,7 @@ class KnowledgebaseService:
SELECT llm_name SELECT llm_name
FROM tenant_llm FROM tenant_llm
WHERE model_type = 'embedding' WHERE model_type = 'embedding'
ORDER BY create_time ASC ORDER BY create_time DESC
LIMIT 1 LIMIT 1
""" """
cursor.execute(query_embedding_model) cursor.execute(query_embedding_model)
@ -213,6 +213,9 @@ class KnowledgebaseService:
if embedding_model and embedding_model.get('llm_name'): if embedding_model and embedding_model.get('llm_name'):
dynamic_embd_id = embedding_model['llm_name'] dynamic_embd_id = embedding_model['llm_name']
# 对硅基流动平台进行特异性处理
if dynamic_embd_id == "netease-youdao/bce-embedding-base_v1":
dynamic_embd_id = "BAAI/bge-m3"
print(f"动态获取到的 embedding 模型 ID: {dynamic_embd_id}") print(f"动态获取到的 embedding 模型 ID: {dynamic_embd_id}")
else: else:
dynamic_embd_id = default_embd_id dynamic_embd_id = default_embd_id