AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-20 12:39:14 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 59 additions and 55 deletions
Show Stats 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