Dinesh310 commited on
Commit
349188f
·
verified ·
1 Parent(s): 7a2a450

Update src/vectorstore/vectorstore.py

Browse files
Files changed (1) hide show
  1. src/vectorstore/vectorstore.py +8 -1
src/vectorstore/vectorstore.py CHANGED
@@ -3,6 +3,7 @@
3
  from typing import List
4
  from langchain_community.vectorstores import FAISS
5
  from langchain_openai import OpenAIEmbeddings
 
6
  # from langchain.schema import Document
7
  from langchain_core.documents import Document
8
 
@@ -12,7 +13,13 @@ class VectorStore:
12
 
13
  def __init__(self):
14
  """Initialize vector store with OpenAI embeddings"""
15
- self.embedding = OpenAIEmbeddings()
 
 
 
 
 
 
16
  self.vectorstore = None
17
  self.retriever = None
18
 
 
3
  from typing import List
4
  from langchain_community.vectorstores import FAISS
5
  from langchain_openai import OpenAIEmbeddings
6
+ from langchain_community.embeddings import HuggingFaceEmbeddings
7
  # from langchain.schema import Document
8
  from langchain_core.documents import Document
9
 
 
13
 
14
  def __init__(self):
15
  """Initialize vector store with OpenAI embeddings"""
16
+ # self.embedding = OpenAIEmbeddings()
17
+ # ✅ Hugging Face Embeddings (LOCAL / FREE)
18
+ self.embeddings = HuggingFaceEmbeddings(
19
+ model_name="sentence-transformers/all-MiniLM-L6-v2",
20
+ model_kwargs={"device": "cpu"}, # change to "cuda" if GPU available
21
+ encode_kwargs={"normalize_embeddings": True}
22
+ )
23
  self.vectorstore = None
24
  self.retriever = None
25