|
|
|
@ -189,11 +189,11 @@ class Dataset(Base):
|
|
|
|
)
|
|
|
|
)
|
|
|
|
if not external_knowledge_binding:
|
|
|
|
if not external_knowledge_binding:
|
|
|
|
return None
|
|
|
|
return None
|
|
|
|
external_knowledge_api = db.session.scalars(
|
|
|
|
external_knowledge_api = db.session.scalar(
|
|
|
|
select(ExternalKnowledgeApis)
|
|
|
|
select(ExternalKnowledgeApis).where(
|
|
|
|
.where(ExternalKnowledgeApis.id == external_knowledge_binding.external_knowledge_api_id)
|
|
|
|
ExternalKnowledgeApis.id == external_knowledge_binding.external_knowledge_api_id
|
|
|
|
.limit(1)
|
|
|
|
)
|
|
|
|
).first()
|
|
|
|
)
|
|
|
|
if not external_knowledge_api:
|
|
|
|
if not external_knowledge_api:
|
|
|
|
return None
|
|
|
|
return None
|
|
|
|
return {
|
|
|
|
return {
|
|
|
|
@ -687,27 +687,27 @@ class DocumentSegment(Base):
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
@property
|
|
|
|
def dataset(self):
|
|
|
|
def dataset(self):
|
|
|
|
return db.session.scalars(select(Dataset).where(Dataset.id == self.dataset_id).limit(1)).first()
|
|
|
|
return db.session.scalar(select(Dataset).where(Dataset.id == self.dataset_id))
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
@property
|
|
|
|
def document(self):
|
|
|
|
def document(self):
|
|
|
|
return db.session.scalars(select(Document).where(Document.id == self.document_id).limit(1)).first()
|
|
|
|
return db.session.scalar(select(Document).where(Document.id == self.document_id))
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
@property
|
|
|
|
def previous_segment(self):
|
|
|
|
def previous_segment(self):
|
|
|
|
return db.session.scalars(
|
|
|
|
return db.session.scalar(
|
|
|
|
select(DocumentSegment)
|
|
|
|
select(DocumentSegment).where(
|
|
|
|
.where(DocumentSegment.document_id == self.document_id, DocumentSegment.position == self.position - 1)
|
|
|
|
DocumentSegment.document_id == self.document_id, DocumentSegment.position == self.position - 1
|
|
|
|
.limit(1)
|
|
|
|
)
|
|
|
|
).first()
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
@property
|
|
|
|
def next_segment(self):
|
|
|
|
def next_segment(self):
|
|
|
|
return db.session.scalars(
|
|
|
|
return db.session.scalar(
|
|
|
|
select(DocumentSegment)
|
|
|
|
select(DocumentSegment).where(
|
|
|
|
.where(DocumentSegment.document_id == self.document_id, DocumentSegment.position == self.position + 1)
|
|
|
|
DocumentSegment.document_id == self.document_id, DocumentSegment.position == self.position + 1
|
|
|
|
.limit(1)
|
|
|
|
)
|
|
|
|
).first()
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
@property
|
|
|
|
def child_chunks(self):
|
|
|
|
def child_chunks(self):
|
|
|
|
|