hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-11 20:06:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix annotation query exception (#1771)

Browse Source 
Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong
2023-12-18 16:48:34 +08:00
committed by GitHub
parent 5bb841935e
commit ebc2cdad2e
1 changed files with 59 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/completion.py
View File

@@ -341,6 +341,7 @@ class Completion:
app = conversation_message_task.app app = conversation_message_task.app
annotation_reply = app_model_config.annotation_reply_dict annotation_reply = app_model_config.annotation_reply_dict
if annotation_reply['enabled']: if annotation_reply['enabled']:
try:
score_threshold = annotation_reply.get('score_threshold', 1) score_threshold = annotation_reply.get('score_threshold', 1)
embedding_provider_name = annotation_reply['embedding_model']['embedding_provider_name'] embedding_provider_name = annotation_reply['embedding_model']['embedding_provider_name']
embedding_model_name = annotation_reply['embedding_model']['embedding_model_name'] embedding_model_name = annotation_reply['embedding_model']['embedding_model_name']
@@ -401,6 +402,9 @@ class Completion:
from_source, from_source,
score) score)
return True return True
except Exception as e:
logging.warning(f'Query annotation failed, exception: {str(e)}.')
return False
return False return False
@classmethod @classmethod