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

chore: skip unnecessary key checks prior to accessing a dictionary (#4497)

Browse source
This commit is contained in:
Bowen Liang 2024-05-19 18:30:45 +08:00 • committed by GitHub
commit 04ad46dd31
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
30 changed files with 45 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/docstore/dataset_docstore.py
View file

@ -121,13 +121,13 @@ class DatasetDocumentStore:
enabled=False,
created_by=self._user_id,
)
if 'answer' in doc.metadata and doc.metadata['answer']:
if doc.metadata.get('answer'):
segment_document.answer = doc.metadata.pop('answer', '')
db.session.add(segment_document)
else:
segment_document.content = doc.page_content
if 'answer' in doc.metadata and doc.metadata['answer']:
if doc.metadata.get('answer'):
segment_document.answer = doc.metadata.pop('answer', '')
segment_document.index_node_hash = doc.metadata['doc_hash']
segment_document.word_count = len(doc.page_content)