Bläddra i källkod

fix document could be None (#15818)

tags/1.1.0
Jyong 7 månader sedan
förälder
incheckning
84a866028a
Inget konto är kopplat till bidragsgivarens mejladress

+ 2
- 0
api/core/rag/datasource/retrieval_service.py Visa fil

continue continue


dataset_document = dataset_documents[document_id] dataset_document = dataset_documents[document_id]
if not dataset_document:
continue


if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX: if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
# Handle parent-child documents # Handle parent-child documents

+ 23
- 20
api/core/rag/retrieval/dataset_retrieval.py Visa fil

dataset_document = DatasetDocument.query.filter( dataset_document = DatasetDocument.query.filter(
DatasetDocument.id == document.metadata["document_id"] DatasetDocument.id == document.metadata["document_id"]
).first() ).first()
if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
child_chunk = ChildChunk.query.filter(
ChildChunk.index_node_id == document.metadata["doc_id"],
ChildChunk.dataset_id == dataset_document.dataset_id,
ChildChunk.document_id == dataset_document.id,
).first()
if child_chunk:
segment = DocumentSegment.query.filter(DocumentSegment.id == child_chunk.segment_id).update(
{DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False
if dataset_document:
if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
child_chunk = ChildChunk.query.filter(
ChildChunk.index_node_id == document.metadata["doc_id"],
ChildChunk.dataset_id == dataset_document.dataset_id,
ChildChunk.document_id == dataset_document.id,
).first()
if child_chunk:
segment = DocumentSegment.query.filter(DocumentSegment.id == child_chunk.segment_id).update(
{DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False
)
db.session.commit()
else:
query = db.session.query(DocumentSegment).filter(
DocumentSegment.index_node_id == document.metadata["doc_id"]
) )
db.session.commit()
else:
query = db.session.query(DocumentSegment).filter(
DocumentSegment.index_node_id == document.metadata["doc_id"]
)


# if 'dataset_id' in document.metadata:
if "dataset_id" in document.metadata:
query = query.filter(DocumentSegment.dataset_id == document.metadata["dataset_id"])
# if 'dataset_id' in document.metadata:
if "dataset_id" in document.metadata:
query = query.filter(DocumentSegment.dataset_id == document.metadata["dataset_id"])


# add hit count to document segment
query.update({DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False)
# add hit count to document segment
query.update(
{DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False
)


db.session.commit()
db.session.commit()


# get tracing instance # get tracing instance
trace_manager: TraceQueueManager | None = ( trace_manager: TraceQueueManager | None = (

Laddar…
Avbryt
Spara