Seth0330 commited on
Commit
0cca271
·
verified ·
1 Parent(s): 9b356a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -154,7 +154,7 @@ class SQLiteVectorRetriever(BaseRetriever):
154
  def _get_relevant_documents(self, query, run_manager=None, **kwargs):
155
  return query_vector_db(query, self.top_k)
156
 
157
- llm = OpenAI(model="gpt-4o", openai_api_key=OPENAI_API_KEY, temperature=0)
158
  retriever = SQLiteVectorRetriever(top_k=5)
159
  qa_chain = RetrievalQA.from_chain_type(
160
  llm=llm,
 
154
  def _get_relevant_documents(self, query, run_manager=None, **kwargs):
155
  return query_vector_db(query, self.top_k)
156
 
157
+ llm = OpenAI(model="gpt-4.1", openai_api_key=OPENAI_API_KEY, temperature=0)
158
  retriever = SQLiteVectorRetriever(top_k=5)
159
  qa_chain = RetrievalQA.from_chain_type(
160
  llm=llm,