Trabis commited on
Commit
5e88b6d
·
verified ·
1 Parent(s): 461cb4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -6
app.py CHANGED
@@ -205,12 +205,19 @@ class OptimizedRAGLoader:
205
  return retriever_function
206
 
207
  # Initialize components
208
- mistral_api_key = os.getenv("mistral_api_key")
209
- llm = ChatMistralAI(
210
- model="mistral-large-latest",
211
- mistral_api_key=mistral_api_key,
212
- temperature=0.01,
213
- streaming=True,
 
 
 
 
 
 
 
214
  )
215
 
216
  rag_loader = OptimizedRAGLoader()
 
205
  return retriever_function
206
 
207
  # Initialize components
208
+ # mistral_api_key = os.getenv("mistral_api_key")
209
+ # llm = ChatMistralAI(
210
+ # model="mistral-large-latest",
211
+ # mistral_api_key=mistral_api_key,
212
+ # temperature=0.01,
213
+ # streaming=True,
214
+ # )
215
+
216
+ from langchain_openai import ChatOpenAI
217
+ llm = ChatOpenAI(
218
+ api_key="sk-bahOSQLfPZb62d-q3aZ0JGcN8raIl12mhUj38DkdpeT3BlbkFJ650KTnBNL0rsIvUcdBA1KJw8H7dCCy7-Kl02GO-l4A",
219
+ model_name="GPT-4 Turbo",
220
+ temperature=0.1,
221
  )
222
 
223
  rag_loader = OptimizedRAGLoader()