diff --git a/Part 1/main.py b/Part 1/main.py index f12dafd00a2b2862f5829481c4d21e8d8518654e..39094228bffbe26e7ce9faef8d208a00d9fcc66f 100644 --- a/Part 1/main.py +++ b/Part 1/main.py @@ -69,7 +69,7 @@ def create_index(index_name): ## Load docs from S3 -def load_docs(bucket_name,local_dir): +def download_documents(bucket_name,local_dir): response = s3_client.list_objects_v2(Bucket=bucket_name) for item in response['Contents']: key = item['Key'] @@ -117,7 +117,7 @@ def generate_store_embeddings(bedrock_client, chunks,awsauth,index_name): ## main def main(): - docs= load_docs(BUCKET_NAME,LOCAL_DIR) + docs= download_documents(BUCKET_NAME,LOCAL_DIR) chunks=split_text(docs, 1000, 100, LOCAL_DIR) print("Sample chunk:", chunks[0]) create_index(index_name)