Dinesh310 commited on
Commit
6602c82
·
verified ·
1 Parent(s): 14182bc

Update src/rag_engine.py

Browse files
Files changed (1) hide show
  1. src/rag_engine.py +1 -1
src/rag_engine.py CHANGED
@@ -10,7 +10,7 @@ from langchain_core.runnables import RunnablePassthrough, RunnableParallel
10
  class ProjectRAGEngine:
11
  def __init__(self, api_key):
12
  self.embeddings = OpenAIEmbeddings(openai_api_key=api_key)
13
- self.llm = ChatOpenAI(model="gpt-oss-120b", openai_api_key=api_key, temperature=0)
14
  self.vector_store = None
15
 
16
  def process_documents(self, pdf_paths):
 
10
  class ProjectRAGEngine:
11
  def __init__(self, api_key):
12
  self.embeddings = OpenAIEmbeddings(openai_api_key=api_key)
13
+ self.llm = ChatOpenAI(model="openai/gpt-oss-120b:free", openai_api_key=api_key, temperature=0)
14
  self.vector_store = None
15
 
16
  def process_documents(self, pdf_paths):