jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: replace all dataset.Model.query to db.session.query(Model) (#19509)

Browse source
This commit is contained in:
非法操作 2025-05-12 13:52:33 +08:00 • committed by GitHub
commit b00f94df64
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
21 changed files with 430 additions and 265 deletions
Download patch file Download diff file Expand all files Collapse all files

16
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View file

@ -275,12 +275,16 @@ class KnowledgeRetrievalNode(LLMNode):
if records:
for record in records:
segment = record.segment
dataset = Dataset.query.filter_by(id=segment.dataset_id).first()
document = Document.query.filter(
Document.id == segment.document_id,
Document.enabled == True,
Document.archived == False,
).first()
dataset = db.session.query(Dataset).filter_by(id=segment.dataset_id).first() # type: ignore
document = (
db.session.query(Document)
.filter(
Document.id == segment.document_id,
Document.enabled == True,
Document.archived == False,
)
.first()
)
if dataset and document:
source = {
"metadata": {