From 8f04d832ac1a8c7188c71620f20be2319c472cfd Mon Sep 17 00:00:00 2001 From: Stan Girard Date: Wed, 17 Apr 2024 17:22:40 +0200 Subject: [PATCH] feat(assistant): check if key of file is same as filename uploaded --- backend/modules/assistant/ito/summary.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/backend/modules/assistant/ito/summary.py b/backend/modules/assistant/ito/summary.py index 9cccb3ce4..3d140201c 100644 --- a/backend/modules/assistant/ito/summary.py +++ b/backend/modules/assistant/ito/summary.py @@ -57,6 +57,11 @@ class SummaryAssistant(ITO): raise ValueError("The key of the file should be doc_to_summarize") if not self.input.inputs.files[0].value: raise ValueError("No file was uploaded") + # Check if name of file is same as the key + if not self.input.inputs.files[0].value == self.files[0].filename: + raise ValueError( + "The key of the file should be the same as the name of the file" + ) if not ( self.input.outputs.brain.activated or self.input.outputs.email.activated ):