NightmareFox12 commited on
Commit
0ff702f
·
verified ·
1 Parent(s): d8caf2b

Update rag_api.py

Browse files
Files changed (1) hide show
  1. 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, cálculos como IMC, calorías, macros, alimentos, etc.
38
- - OTRO: preguntas no relacionadas con nutrición ni saludos
 
 
 
 
 
 
 
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 completo
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]