temp fix pdf upload sigkill

weni-ai · Nov 26, 2024 · c70bb97 · c70bb97
1 parent 5abb53e
commit c70bb97
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/nexus/task_managers/file_manager/celery_file_manager.py b/nexus/task_managers/file_manager/celery_file_manager.py
@@ -46,6 +46,16 @@ def upload_file(
 
         if indexer_database == Project.BEDROCK:
             print("[+ 🦑 Using BEDROCK 🦑 +]")
+
+            if "pdf" in extension_file:
+                tasks_bedrock.bedrock_upload_file(
+                    pickled_file,
+                    content_base_uuid,
+                    user_email,
+                    str(content_base_file.uuid),
+                )
+                return {"uuid": str(content_base_file.uuid), "extension_file": extension_file}
+
             tasks_bedrock.bedrock_upload_file.delay(
                 pickled_file,
                 content_base_uuid,
@@ -54,6 +64,16 @@ def upload_file(
             )
             return {"uuid": str(content_base_file.uuid), "extension_file": extension_file}
 
+        if "pdf" in extension_file:
+            tasks.upload_file(
+                pickled_file,
+                content_base_uuid,
+                extension_file,
+                user_email, str(content_base_file.uuid),
+                load_type
+            )
+            return {"uuid": str(content_base_file.uuid), "extension_file": extension_file}
+
         tasks.upload_file.delay(
             pickled_file,
             content_base_uuid,