Update app.py
Browse files
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 =
|
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=
|
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", """أنت مساعد مفيد يجيب على الأسئلة باللغة العربية باستخدام المعلومات المقدمة.
|