Update src/rag_engine.py

src/rag_engine.py

@@ -1,21 +1,24 @@
 import os
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from langchain_openai import OpenAIEmbeddings, ChatOpenAI
 from langchain_community.vectorstores import FAISS
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnablePassthrough, RunnableParallel
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_openai import ChatOpenAI
 
 
 class ProjectRAGEngine:
     def __init__(self):
-        # ✅ OpenAI embeddings (OFFICIAL)
-        self.embeddings = OpenAIEmbeddings(
-            model="text-embedding-3-small"
+        # ✅ Hugging Face Embeddings (LOCAL / FREE)
+        self.embeddings = HuggingFaceEmbeddings(
+            model_name="sentence-transformers/all-MiniLM-L6-v2",
+            model_kwargs={"device": "cpu"},   # change to "cuda" if GPU available
+            encode_kwargs={"normalize_embeddings": True}
         )
 
-        # ✅ OpenRouter LLM
+        # ✅ OpenRouter LLM (Chat only)
         self.llm = ChatOpenAI(
             model="openai/gpt-oss-120b:free",
             temperature=0,
@@ -43,6 +46,7 @@ class ProjectRAGEngine:
 
         splits = splitter.split_documents(all_docs)
 
+        # ✅ FAISS with HuggingFace embeddings
         self.vector_store = FAISS.from_documents(
             splits, self.embeddings
         )
@@ -59,10 +63,10 @@ class ProjectRAGEngine:
         Answer strictly using the context.
         If unknown, say you don't know.
         Cite document names and page numbers.
-        
+
         Context:
         {context}
-        
+
         Question:
         {question}
         """