Merge pull request #7 from FloRul/dev

Dev
FloRul · Jan 27, 2024 · 9c9a62b · 9c9a62b
2 parents 5c7fdf6 + 1afe7ed
commit 9c9a62b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ingestion/src/index.py b/ingestion/src/index.py
@@ -56,7 +56,7 @@ def delete_documents(filename: str):
 
 def fetch_file(bucket, key):
     s3 = boto3.client("s3")
-    local_filename = f"/tmp/{key}"
+    local_filename = f"/tmp/{key.split('/')[-1]}"
 
     try:
         s3.download_file(bucket, key, local_filename)
@@ -145,7 +145,7 @@ def lambda_handler(event, context):
                     return len(docs)
 
             elif eventName.startswith(OBJECT_REMOVED):
-                return delete_documents(filename=key)
+                return delete_documents(filename=key.split("/")[-1])
 
         except Exception as e:
             print(e)