From 30ff0afd5d76a3a5aa48058210ae411253574ada Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 13 三月 2025 14:55:30 +0800
Subject: [PATCH] 增加文件多轮问答
---
app/service/knowledge.py | 5 ++++-
1 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/app/service/knowledge.py b/app/service/knowledge.py
index bb3a682..4fb834f 100644
--- a/app/service/knowledge.py
+++ b/app/service/knowledge.py
@@ -15,9 +15,11 @@
query = db.query(KnowledgeModel).filter(KnowledgeModel.knowledge_type==knowledge)
if user.permission != "admin":
klg_list = [j.id for i in user.groups for j in i.knowledges]
- query = query.filter(or_(KnowledgeModel.tenant_id == user_id, KnowledgeModel.id.in_(klg_list)))
+ query = query.filter(or_(KnowledgeModel.id.in_(klg_list), KnowledgeModel.tenant_id == str(user_id)))
+
if location:
query = query.filter(or_(KnowledgeModel.permission == "team", KnowledgeModel.tenant_id == str(user_id)))
+
if keyword:
query = query.filter(KnowledgeModel.name.like('%{}%'.format(keyword)))
total = query.count()
@@ -50,6 +52,7 @@
knowledge.permission = app_knowledge["permission"]
knowledge.documents = app_knowledge["doc_num"]
knowledge.status = app_knowledge["status"]
+ knowledge.embd_id = app_knowledge["embd_id"]
knowledge.update_date = datetime.now()
db.add(knowledge)
db.commit()
--
Gitblit v1.8.0