Carlos Isael Ramírez González commited on
Commit
a3373fd
·
1 Parent(s): ce79a7e

Ahora si espero se haya arreglado el error de manejo de casos especiales

Browse files
Files changed (1) hide show
  1. mojica_agent.py +1 -5
mojica_agent.py CHANGED
@@ -71,11 +71,7 @@ class MojicaAgent:
71
  def _handle_speecial_cases(self, question: str) -> Optional[str]:
72
  lower_question = question.lower()
73
  table = self.schema["table_name"]
74
- if (
75
- "clientes 0" in lower_question
76
- or "clientes cero"
77
- or "cliente cero" in lower_question
78
- ):
79
  return f"""
80
  SELECT DISTINCT "Cliente", "Razon Social"
81
  FROM "{table}"
 
71
  def _handle_speecial_cases(self, question: str) -> Optional[str]:
72
  lower_question = question.lower()
73
  table = self.schema["table_name"]
74
+ if any(kw in lower_question for kw in ["clientes 0", "clientes cero", "cliente cero"]):
 
 
 
 
75
  return f"""
76
  SELECT DISTINCT "Cliente", "Razon Social"
77
  FROM "{table}"