Skip to content

Commit 44314c2

Browse files
committed
fix: workflow input file handle extra metadata field
1 parent fbda130 commit 44314c2

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

  • src/backend/bisheng/workflow/nodes/input

src/backend/bisheng/workflow/nodes/input/input.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -315,7 +315,7 @@ def parse_upload_file(self, key: str, key_info: dict, value: List[str]) -> dict
315315
})
316316

317317
file_ext = file_name.split('.')[-1].lower()
318-
logger.debug(f"{self.id}.{key} file_parse_mode is keep_raw")
318+
logger.debug(f"{self.id}.{key} file_parse_mode is {file_parse_mode}")
319319
original_file_path.append(one_file_url)
320320
if file_ext in self._image_ext:
321321
image_files_path.append(one_file_url)
@@ -340,6 +340,9 @@ def parse_upload_file(self, key: str, key_info: dict, value: List[str]) -> dict
340340
# A file corresponding to the same variable, placed in a file_id mile
341341
for one in metadatas:
342342
one.update(all_metadata[-1])
343+
input_file_metadata_keys = {one.field_name for one in InputFileMetadata}
344+
for k in one.keys() - input_file_metadata_keys:
345+
del one[k]
343346
new_metadata.append(one)
344347

345348
# Uploaded to milvus And es

0 commit comments

Comments
 (0)