Spaces:
Sleeping
Sleeping
Update rag_api.py
Browse files- rag_api.py +10 -3
rag_api.py
CHANGED
|
@@ -34,8 +34,15 @@ INTENT_PROMPT = PromptTemplate(
|
|
| 34 |
|
| 35 |
Analiza el mensaje del usuario y clasifícalo en UNA de estas categorías:
|
| 36 |
- SALUDO: saludos, despedidas, conversación casual ("hola", "gracias", "adiós", "¿cómo estás?")
|
| 37 |
-
- NUTRICION: preguntas sobre nutrición, dieta, salud,
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
|
| 40 |
Responde SOLO con la categoría, sin explicación.
|
| 41 |
|
|
@@ -176,7 +183,7 @@ async def process_query(request: QueryRequest):
|
|
| 176 |
}
|
| 177 |
|
| 178 |
else:
|
| 179 |
-
# NUTRICION → RAG
|
| 180 |
respuesta = qa_chain.invoke(request.query)
|
| 181 |
docs = retriever.invoke(request.query)
|
| 182 |
sources = [doc.metadata.get("source", "N/A") for doc in docs]
|
|
|
|
| 34 |
|
| 35 |
Analiza el mensaje del usuario y clasifícalo en UNA de estas categorías:
|
| 36 |
- SALUDO: saludos, despedidas, conversación casual ("hola", "gracias", "adiós", "¿cómo estás?")
|
| 37 |
+
- NUTRICION: preguntas sobre nutrición, dieta, salud, alimentos, calorías, macros, IMC,
|
| 38 |
+
planes alimenticios, recetas, suplementos, hábitos saludables, Y TAMBIÉN cualquier
|
| 39 |
+
pregunta relacionada con NutriActive como empresa: sus cursos, servicios, productos,
|
| 40 |
+
planes, precios, programas, etc.
|
| 41 |
+
- OTRO: preguntas claramente NO relacionadas con nutrición, salud ni NutriActive
|
| 42 |
+
(matemáticas, historia, tecnología general, etc.)
|
| 43 |
+
|
| 44 |
+
IMPORTANTE: Ante la duda, clasifica como NUTRICION. Solo usa OTRO cuando estés
|
| 45 |
+
completamente seguro de que no tiene relación con nutrición ni con NutriActive.
|
| 46 |
|
| 47 |
Responde SOLO con la categoría, sin explicación.
|
| 48 |
|
|
|
|
| 183 |
}
|
| 184 |
|
| 185 |
else:
|
| 186 |
+
# NUTRICION o cualquier categoría no reconocida → RAG
|
| 187 |
respuesta = qa_chain.invoke(request.query)
|
| 188 |
docs = retriever.invoke(request.query)
|
| 189 |
sources = [doc.metadata.get("source", "N/A") for doc in docs]
|