From 561267ae1621ccfaa3e4ec72472f43bffd607586 Mon Sep 17 00:00:00 2001
From: "abir.chebbi" <abir.chebbi@hes-so.ch>
Date: Mon, 2 Sep 2024 21:53:33 +0200
Subject: [PATCH] correct function name
---
Part 1/main.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/Part 1/main.py b/Part 1/main.py
index f12dafd..3909422 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)
--
GitLab