feat: 高级编排支持文件上传(WIP)

1Panel-dev · Nov 9, 2024 · 28d67a7 · 28d67a7
1 parent 4fb2523
commit 28d67a7
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/apps/application/flow/step_node/image_understand_step_node/i_image_understand_node.py b/apps/application/flow/step_node/image_understand_step_node/i_image_understand_node.py
@@ -30,8 +30,7 @@ def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
     def _run(self):
         res = self.workflow_manage.get_reference_field(self.node_params_serializer.data.get('image_list')[0],
                                                  self.node_params_serializer.data.get('image_list')[1:])
-        self.node_params_serializer.data['image'] = res
-        return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
+        return self.execute(image = res, **self.node_params_serializer.data, **self.flow_params_serializer.data)
 
     def execute(self, model_id, system, prompt, dialogue_number, history_chat_record, stream, chat_id,
                 chat_record_id,

diff --git a/.../application/flow/step_node/image_understand_step_node/impl/base_image_understand_node.py b/.../application/flow/step_node/image_understand_step_node/impl/base_image_understand_node.py
@@ -95,7 +95,7 @@ def generate_prompt_question(self, prompt):
 
     def generate_message_list(self, image_model, system: str, prompt: str, history_message, image):
         if image is not None and len(image) > 0:
-            file_id = image[0].url.split('/')[-1]
+            file_id = image[0]['file_id']
             file = QuerySet(File).filter(id=file_id).first()
 
         if system is not None and len(system) > 0:

diff --git a/apps/application/views/chat_views.py b/apps/application/views/chat_views.py
@@ -415,6 +415,6 @@ def post(self, request: Request, application_id: str, chat_id: str):
             meta = {'application_id': application_id, 'chat_id': chat_id}
             for file in files:
                 file_url = FileSerializer(data={'file': file, 'meta': meta}).upload()
-                file_ids.append({'name': file.name, 'url': file_url})
+                file_ids.append({'name': file.name, 'url': file_url, 'file_id': file_url.split('/')[-1]})
             return result.success(file_ids)
 
diff --git a/ui/src/components/ai-chat/index.vue b/ui/src/components/ai-chat/index.vue
@@ -959,6 +959,7 @@ const uploadFile = async (file: any, fileList: any) => {
       const f = response.data.filter((f: any) => f.name === file.name)
       if (f.length > 0) {
         file.url = f[0].url
+        file.file_id = f[0].file_id
       }
     })
     console.log(uploadFileList.value)