From f71065bdbfad32709af5fd140c7a31e36d028f15 Mon Sep 17 00:00:00 2001
From: sreyakumar <121137643+sreyakumar@users.noreply.github.com>
Date: Thu, 21 Nov 2024 10:15:24 -0800
Subject: [PATCH] added cache dir

---
 .gitignore                                     | 1 +
 src/metadata_chatbot/agents/docdb_retriever.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index cabac1d..fa161a2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -14,6 +14,7 @@ umap_visualization.py
 
 # Byte-compiled / optimized / DLL files
 __pycache__/
+huggingface_cache/
 *.py[cod]
 *$py.class
 
diff --git a/src/metadata_chatbot/agents/docdb_retriever.py b/src/metadata_chatbot/agents/docdb_retriever.py
index 283378e..a1f75f7 100644
--- a/src/metadata_chatbot/agents/docdb_retriever.py
+++ b/src/metadata_chatbot/agents/docdb_retriever.py
@@ -17,7 +17,8 @@
 hf = HuggingFaceEmbeddings(
     model_name=model_name,
     model_kwargs=model_kwargs,
-    encode_kwargs=encode_kwargs
+    encode_kwargs=encode_kwargs,
+    cache_folder="/scratch/huggingface_cache"
 )
 
 BEDROCK_CLIENT = boto3.client(