Yongtao Huang пре 2 месеци
родитељ
комит
8fcc864fb7
No account linked to committer's email address
1 измењених фајлова са 1 додато и 2 уклоњено
  1. 1
    2
      api/core/rag/retrieval/dataset_retrieval.py

+ 1
- 2
api/core/rag/retrieval/dataset_retrieval.py Прегледај датотеку

synchronize_session=False, synchronize_session=False,
) )
) )
db.session.commit()
else: else:
query = db.session.query(DocumentSegment).where( query = db.session.query(DocumentSegment).where(
DocumentSegment.index_node_id == document.metadata["doc_id"] DocumentSegment.index_node_id == document.metadata["doc_id"]
with flask_app.app_context(): with flask_app.app_context():
with Session(db.engine) as session: with Session(db.engine) as session:
dataset_stmt = select(Dataset).where(Dataset.id == dataset_id) dataset_stmt = select(Dataset).where(Dataset.id == dataset_id)
dataset = db.session.scalar(dataset_stmt)
dataset = session.scalar(dataset_stmt)


if not dataset: if not dataset:
return [] return []

Loading…
Откажи
Сачувај