youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-30 15:38:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: resolve basedpyright-check.

Browse Source
This commit is contained in:
FFXN
2026-01-28 13:26:40 +08:00
parent 5f71fd42bf
commit 5f611ad2ba
6 changed files with 18 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/controllers/console/datasets/datasets_document.py
View File

@ -1462,9 +1462,6 @@ class DocumentSummaryStatusApi(DocumentResource):
dataset_id = str(dataset_id)
document_id = str(document_id)
# Get document
document = self.get_document(dataset_id, document_id)
# Get dataset
dataset = DatasetService.get_dataset(dataset_id)
if not dataset:

4
api/controllers/console/datasets/datasets_segments.py
View File

@ -43,7 +43,7 @@ from tasks.batch_create_segment_to_index_task import batch_create_segment_to_ind
def _get_segment_with_summary(segment, dataset_id):
"""Helper function to marshal segment and add summary information."""
segment_dict = marshal(segment, segment_fields)
segment_dict = dict(marshal(segment, segment_fields))
# Query summary for this segment (only enabled summaries)
summary = (
db.session.query(DocumentSegmentSummary)
@ -219,7 +219,7 @@ class DatasetDocumentSegmentListApi(Resource):
# Add summary to each segment
segments_with_summary = []
for segment in segments.items:
segment_dict = marshal(segment, segment_fields)
segment_dict = dict(marshal(segment, segment_fields))
segment_dict["summary"] = summaries.get(segment.id)
segments_with_summary.append(segment_dict)