Update src/rag_engine.py

src/rag_engine.py

@@ -8,6 +8,7 @@ from langchain_core.runnables import RunnablePassthrough, RunnableParallel
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_openai import ChatOpenAI
 from langchain_community.chat_models import ChatLiteLLM
+from langchain_core.messages import HumanMessage, AIMessage
 
 class ProjectRAGEngine:
     def __init__(self):
@@ -18,11 +19,11 @@ class ProjectRAGEngine:
             encode_kwargs={"normalize_embeddings": True}
         )
         # ✅ OpenRouter LLM (Chat only)
-        self.llm = ChatLiteLLM(
-            model="openrouter/openai/gpt-oss-120b:free",
-            api_key=os.getenv("OPENROUTER_API_KEY"),
-            temperature=0
-        )
+        self.llm = ChatOpenAI(
+                model="openai/gpt-oss-120b:free",
+                base_url="https://openrouter.ai/api/v1",
+                api_key=os.getenv("OPENROUTER_API_KEY"),
+                extra_body={"reasoning": {"enabled": True}})
         self.vector_store = None
 
     def process_documents(self, pdf_paths):
@@ -56,10 +57,8 @@ class ProjectRAGEngine:
         Answer strictly using the context.
         If unknown, say you don't know.
         Cite document names and page numbers.
-
         Context:
         {context}
-
         Question:
         {question}
         """
@@ -87,4 +86,4 @@ class ProjectRAGEngine:
             for d in result["context"]
         ]
 
-        return result["answer"], sources
+        return result["answer"], sources