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

1Panel-dev · Sep 12, 2024 · 47d36fe · 47d36fe
1 parent 2429149
commit 47d36fe
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py b/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py
@@ -79,7 +79,12 @@ def reset_paragraph(paragraph: Dict, embedding_list: List):
                 **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

diff --git a/apps/common/constants/authentication_type.py b/apps/common/constants/authentication_type.py
@@ -16,3 +16,5 @@ class AuthenticationType(Enum):
     APPLICATION_ACCESS_TOKEN = "APPLICATION_ACCESS_TOKEN"
     # key API
     API_KEY = "API_KEY"
+    # 第三方对接
+    PLATFORM = 'PLATFORM'