Update manabUtils.py

manabUtils.py

@@ -1,4 +1,4 @@
-#to be updated
+#manabUtils.py
 from langchain_community.vectorstores import FAISS
 from langchain_huggingface import HuggingFaceEmbeddings
 from huggingface_hub import hf_hub_download
@@ -37,4 +37,23 @@ def retrieve_chunks(repo_id, embedding_model="sentence-transformers/all-MiniLM-L
     except Exception as e:
       print(f"Error in generate_qa_chain: {e}")
       return None
+    return retriever
+
+def retrieve_chunks_GPC():
+    """
+    Retreive chunks from HF dataset for GPC
+    """    
+    embedding_model="sentence-transformers/all-MiniLM-L6-v2"
+    
+    try:
+        embeddings = HuggingFaceEmbeddings(model_name=embedding_model)
+        vectorstore = FAISS.load_local(
+            folder_path="hf://manabb/NRLGPC/faiss_gpc_goods_merged",
+            embeddings=embeddings,
+            allow_dangerous_deserialization=True  # Required for .pkl
+        )      
+        retriever = vectorstore.as_retriever(search_kwargs={"k": 5})
+    except Exception as e:
+        print(f"Error in generate_qa_chain: {e}")
+        return None
     return retriever