jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: keywords (#7357)

Browse source
This commit is contained in:
crazywoola 2024-08-16 20:43:55 +08:00 • committed by GitHub
commit 4d4af00399
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 16 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

7
api/services/dataset_service.py
View file

@ -1429,7 +1429,10 @@ class SegmentService:
segment_data_list.append(segment_document)
pre_segment_data_list.append(segment_document)
keywords_list.append(segment_item['keywords'])
if 'keywords' in segment_item:
keywords_list.append(segment_item['keywords'])
else:
keywords_list.append(None)
try:
# save vector index
@ -1482,7 +1485,7 @@ class SegmentService:
db.session.add(segment)
db.session.commit()
# update segment index task
if args['keywords']:
if 'keywords' in args:
keyword = Keyword(dataset)
keyword.delete_by_ids([segment.index_node_id])
document = RAGDocument(