Retrieval service optimization (#13849)
This commit is contained in:
@@ -301,7 +301,6 @@ class RetrievalService:
|
|||||||
DocumentSegment.id,
|
DocumentSegment.id,
|
||||||
DocumentSegment.content,
|
DocumentSegment.content,
|
||||||
DocumentSegment.answer,
|
DocumentSegment.answer,
|
||||||
DocumentSegment.doc_metadata,
|
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
.first()
|
.first()
|
||||||
@@ -362,7 +361,6 @@ class RetrievalService:
|
|||||||
record = {
|
record = {
|
||||||
"segment": segment,
|
"segment": segment,
|
||||||
"score": document.metadata.get("score"), # type: ignore
|
"score": document.metadata.get("score"), # type: ignore
|
||||||
"segment_metadata": segment.doc_metadata,
|
|
||||||
}
|
}
|
||||||
records.append(record)
|
records.append(record)
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user