Trabis commited on
Commit
b496d0b
1 Parent(s): d2d22a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -207,7 +207,7 @@ class RAGLoader:
207
  """Vérifie si l'index et les documents associés existent"""
208
  return self.index_path.exists() and self.documents_path.exists()
209
 
210
- def get_retriever(self, k: int = 5):
211
  """
212
  Crée un retriever pour l'utilisation avec LangChain
213
 
@@ -248,7 +248,7 @@ class RAGLoader:
248
  # Initialize the RAG system
249
  llm = ChatMistralAI(model="mistral-large-latest", mistral_api_key="QK0ZZpSxQbCEVgOLtI6FARQVmBYc6WGP")
250
  rag_loader = RAGLoader()
251
- retriever = rag_loader.get_retriever(k=5)
252
 
253
  prompt_template = ChatPromptTemplate.from_messages([
254
  ("system", """أنت مساعد مفيد يجيب على الأسئلة باللغة العربية باستخدام المعلومات المقدمة.
 
207
  """Vérifie si l'index et les documents associés existent"""
208
  return self.index_path.exists() and self.documents_path.exists()
209
 
210
+ def get_retriever(self, k: int = 10):
211
  """
212
  Crée un retriever pour l'utilisation avec LangChain
213
 
 
248
  # Initialize the RAG system
249
  llm = ChatMistralAI(model="mistral-large-latest", mistral_api_key="QK0ZZpSxQbCEVgOLtI6FARQVmBYc6WGP")
250
  rag_loader = RAGLoader()
251
+ retriever = rag_loader.get_retriever(k=10)
252
 
253
  prompt_template = ChatPromptTemplate.from_messages([
254
  ("system", """أنت مساعد مفيد يجيب على الأسئلة باللغة العربية باستخدام المعلومات المقدمة.