Spaces:
Sleeping
Sleeping
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- 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}"
|