address last bandit issue to not have temp dir hardcoded

awslabs · Nov 16, 2023 · a2cffd8 · a2cffd8
1 parent a27dd6c
commit a2cffd8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lambda/aws-rag-appsync-stepfn-opensearch/embeddings_job/src/lambda.py b/lambda/aws-rag-appsync-stepfn-opensearch/embeddings_job/src/lambda.py
@@ -16,6 +16,7 @@
 import json
 import boto3, json
 import numpy as np
+import tempfile
 from helpers.credentials_helper import get_credentials
 from helpers.opensearch_helper import check_if_index_exists, process_shard
 from helpers.update_ingestion_status import updateIngestionJobStatus
@@ -75,7 +76,7 @@ def get_bedrock_client(service_name="bedrock-runtime"):
 opensearch_index = os.environ['OPENSEARCH_INDEX']
 opensearch_domain = os.environ['OPENSEARCH_DOMAIN_ENDPOINT']
 
-DATA_DIR = "/tmp"
+DATA_DIR = tempfile.gettempdir()
 CHUNCK_SIZE_DOC_SPLIT=500
 OVERLAP_FOR_DOC_SPLIT=20
 MAX_OS_DOCS_PER_PUT = 500