浏览代码

Post fix of #23224 (#25007)

tags/1.8.1
Yongtao Huang 2 个月前
父节点
当前提交
8fcc864fb7
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 1 次插入2 次删除
  1. 1
    2
      api/core/rag/retrieval/dataset_retrieval.py

+ 1
- 2
api/core/rag/retrieval/dataset_retrieval.py 查看文件

@@ -534,7 +534,6 @@ class DatasetRetrieval:
synchronize_session=False,
)
)
db.session.commit()
else:
query = db.session.query(DocumentSegment).where(
DocumentSegment.index_node_id == document.metadata["doc_id"]
@@ -596,7 +595,7 @@ class DatasetRetrieval:
with flask_app.app_context():
with Session(db.engine) as session:
dataset_stmt = select(Dataset).where(Dataset.id == dataset_id)
dataset = db.session.scalar(dataset_stmt)
dataset = session.scalar(dataset_stmt)

if not dataset:
return []

正在加载...
取消
保存