Changed num context retrieved

src/vectorstore.py

@@ -1,5 +1,6 @@
 from langchain_community.vectorstores import FAISS
-from langchain_community.embeddings import HuggingFaceEmbeddings
+# from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_huggingface import HuggingFaceEmbeddings
 from huggingface_hub import snapshot_download
 import logging
 

tools.py

@@ -85,7 +85,9 @@ vector_store = VectorStore(
     # embeddings_model="BAAI/bge-m3", 
     embeddings_model=os.environ.get("EMBEDDINGS_MODEL"),
     vs_local_path=os.environ.get("VS_LOCAL_PATH"), 
-    vs_hf_path=os.environ.get("VS_HF_PATH"))
+    vs_hf_path=os.environ.get("VS_HF_PATH"),
+    number_of_contexts=int(os.environ.get("RETRIEVE_NUM_CONTEXTS", 3))
+)
 
 
 def tool_register(cls: BaseModel):