diff --git a/ingestion/src/index.py b/ingestion/src/index.py index 59877d7..2be0c04 100644 --- 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)