15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: 修复工作流中,函数库使用知识库检索参数错误

This commit is contained in:
shaohuzhang1 2024-09-12 14:13:09 +08:00 committed by shaohuzhang1
parent 0d762821f7
commit 989d0848c0
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py
View File

@ -79,7 +79,12 @@ class BaseSearchDatasetNode(ISearchDatasetStepNode):
**paragraph,
'similarity': find_embedding.get('similarity'),
'is_hit_handling_method': find_embedding.get('similarity') > paragraph.get(
'directly_return_similarity') and paragraph.get('hit_handling_method') == 'directly_return'
'directly_return_similarity') and paragraph.get('hit_handling_method') == 'directly_return',
'update_time': paragraph.get('update_time').strftime("%Y-%m-%d %H:%M:%S"),
'create_time': paragraph.get('create_time').strftime("%Y-%m-%d %H:%M:%S"),
'id': str(paragraph.get('id')),
'dataset_id': str(paragraph.get('dataset_id')),
'document_id': str(paragraph.get('document_id'))
}
@staticmethod

2
apps/common/constants/authentication_type.py
View File

@ -16,3 +16,5 @@ class AuthenticationType(Enum):
APPLICATION_ACCESS_TOKEN = "APPLICATION_ACCESS_TOKEN"
# key API
API_KEY = "API_KEY"
# 第三方对接
PLATFORM = 'PLATFORM'