pbesa 4.0.40__tar.gz → 4.0.42__tar.gz

{pbesa-4.0.40 → pbesa-4.0.42}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pbesa
-Version: 4.0.40
+Version: 4.0.42
 License-File: LICENSE
 License-File: LICENSE.txt
 Requires-Dist: pymongo>=4.6.3

pbesa-4.0.42/pbesa/celulas/celula_verificar_calculos.py

@@ -0,0 +1,59 @@
+import logging
+
+SYSTEM_PROMPT = """
+
+Tu única tarea es clasificar la siguiente consulta de un usuario en una de dos categorías: PERTINENTE u OTRA.
+
+- PERTINENTE: Asigna esta etiqueta si la consulta solicita información estadística, datos agregados o cálculos que pueden extraerse de una base de datos de trámites y demandas. Busca preguntas sobre totales, promedios, porcentajes, conteos, distribuciones o tendencias a lo largo del tiempo.
+
+- OTRA: Asigna esta etiqueta si la consulta es de cualquier otra índole. Esto incluye preguntas sobre un caso personal, cómo realizar un trámite, problemas técnicos, opiniones, o cualquier tema que no sea una solicitud de datos estadísticos del observatorio.
+
+Regla clave para evitar la inversión: La clasificación depende de si el usuario pide datos agregados, no del tema.
+
+- Una consulta sobre cómo poner una demanda -> OTRA.
+
+- Una consulta sobre cuántas demandas se pusieron el mes pasado -> PERTINENTE.
+
+Ejemplos:
+
+- Consulta: "¿Cuál es el número total de trámites recibidos por la entidad en 2023?" -> Clasificación: PERTINENTE
+
+- Consulta: "Necesito ayuda para radicar una demanda contra mi operador de telefonía." -> Clasificación: OTRA
+
+- Consulta: "Dame un porcentaje de las demandas clasificadas por sector económico y por región." -> Clasificación: PERTINENTE
+
+- Consulta: "No recuerdo mi contraseña para acceder al portal, ¿cómo la recupero?" -> Clasificación: OTRA
+
+- Consulta: "¿Cuál es el tiempo promedio de resolución para un caso de protección al consumidor?" -> Clasificación: PERTINENTE
+
+- Consulta: "¿Qué opina de la nueva reforma a la justicia?" -> Clasificación: OTRA
+
+Responde únicamente con PERTINENTE u OTRA.
+"""
+
+USER_PROMPT = """
+
+Consulta: "%s"
+
+Clasificación:
+"""
+
+# Efectua la inferencia del modelo.
+def derive(service, text, max_tkns=2000) -> any:
+    try:
+        tmp_work_memory = []
+        prompt  = SYSTEM_PROMPT
+        tmp_work_memory.append({"role": "system", "content": prompt})
+        USER_PROMPT % text
+        tmp_work_memory.append({"role": "user", "content": text})
+        res = service.generate(tmp_work_memory, max_tokens=max_tkns)
+        logging.info(f"[Celula][Odj][Procesando]: {text}")        
+        logging.info(f"[Celula][Odj][Respuesta]: {res}")
+        if not res or res == "":
+            res = text
+            logging.warning(f"[Celula][Odj]: No obtener una respuesta.")
+        return res.replace("*", "").strip()
+    except Exception as e:
+        logging.error(f"[Celula][Odj]: Error al procesar: {text}")
+        logging.error(e)
+        return None

{pbesa-4.0.40 → pbesa-4.0.42}/pbesa/cognitive.py

@@ -29,7 +29,7 @@ from pbesa.social.dialog import (
 from .celulas import (celula_casos, celula_consultas, celula_saludos, celula_datos_identificables,
                       celula_generar_documento, celula_expertos, celula_pertinencia, celula_extraccion,
                       celula_evaluador, celula_respuesta, celula_conversador, celula_parafraseo, celula_cuestionador,
-                      celula_instruccion, celula_generar_caso, celula_demanda)
+                      celula_instruccion, celula_generar_caso, celula_demanda, celula_verificar_calculos)
 from pbesa.social.prompts import ANALIZER_PROMPT, CLASSIFICATION_PROMPT, DERIVE_PROMPT, RECOVERY_PROMPT, ADAPT_PROMPT, SINTETIZER_PROMPT
 
 # --------------------------------------------------------
@@ -942,8 +942,9 @@ class Dialog(ABC):
             elif es_consulta and not es_saludo and not es_caso:
                 dicriminador = "consulta"
                 per_res = celula_pertinencia.derive(self.__ai_service, query, max_tkns=10)
+                obj_check = celula_verificar_calculos.derive(self.__ai_service, query, max_tkns=10)
                 if per_res and not per_res == "":
-                    if "ABSURDO" in per_res:
+                    if "ABSURDO" in per_res and not "PERTINENTE" in obj_check:
                         dicriminador = None
                         res = msg
             elif es_caso and not es_saludo and not es_consulta:
@@ -1019,12 +1020,21 @@ class Dialog(ABC):
             if len(messages) > 1:
                 message = messages[-2]
                 if not message['user'] and '¿Desea que le ayude con una consulta o una demanda?' in message['text']:
-                    demanda = celula_demanda.derive(self.__ai_service, query, max_tkns=10)
-                    es_demanda = ("DEMANDA" in demanda) and not ("INDEFINIDO" in demanda)
-                    if es_demanda:
-                        res = self.stage_one_sintetizer(session_id, messages, attemps, query, es_demanda)
-                        dicriminador = "caso"
-                        return dicriminador, res
+                    if 'consulta' in query:
+                        message = messages[-3]
+                        query = message['text']
+                        per_res = celula_pertinencia.derive(self.__ai_service, query, max_tkns=10)
+                        obj_check = celula_verificar_calculos.derive(self.__ai_service, query, max_tkns=10)
+                        if per_res and "ABSURDO" in per_res and not "PERTINENTE" in obj_check:
+                            return None, "No puedo ayudarle con su consulta. Dado que no está relacionada con los servicios que presta Justifacil."
+                        return 'consulta', query
+                    else:
+                        demanda = celula_demanda.derive(self.__ai_service, query, max_tkns=10)
+                        es_demanda = ("DEMANDA" in demanda) and not ("INDEFINIDO" in demanda)
+                        if es_demanda:
+                            res = self.stage_one_sintetizer(session_id, messages, attemps, query, es_demanda)
+                            dicriminador = "caso"
+                            return dicriminador, res
 
             dicriminador, ambiguedad, res = self.stage_one_evaluate(session_id, messages, attemps, query, es_demanda)
 

{pbesa-4.0.40 → pbesa-4.0.42}/pbesa/social/sel_dispatcher_team.py

@@ -188,22 +188,21 @@ class SelectedDispatcher(Action):
         try:
             logging.info('Assign to agent...')
             session_id = data['dto']['session']['session_id'] if 'session' in data['dto'] else None
-            agent_list = self.agent.get_agent_list()
-            logging.debug('List of agents: ' + str(agent_list))
-            if session_id in self.agent.planilla:
-                mayor_ag_id = self.agent.planilla[session_id]
+            assistant = data['dto']['session']['assistant'] if 'session' in data['dto'] and 'assistant' in data['dto']['session'] else None            
+            if assistant:
                 logging.info('The session is already assigned')
-                logging.info(f'The agent {mayor_ag_id} will be assigned')
-                self.agent.add_to_hold(data, mayor_ag_id, session_id)
+                logging.info(f'The agent type: {assistant} will be assigned')
+                self.agent.add_to_hold(data, assistant, session_id)
             else:
                 score_mayor = 0 
                 mayor_ag = self.manual_selection(data['dto'])
                 if mayor_ag:
                     logging.info(f'The agent {mayor_ag.id} will be assigned')
-                    self.agent.add_to_hold(data, mayor_ag_id, session_id)
+                    self.agent.add_to_hold(data, mayor_ag.id, session_id)
                 else:
                     roles = []
                     # Get the agent asocciated with the data.
+                    agent_list = self.agent.get_agent_list()            
                     for agent_id in agent_list:
                         agent = self.adm.get_agent(agent_id)
                         # Chec if the agent is instance of AugmentedGeneration

{pbesa-4.0.40 → pbesa-4.0.42}/pbesa.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pbesa
-Version: 4.0.40
+Version: 4.0.42
 License-File: LICENSE
 License-File: LICENSE.txt
 Requires-Dist: pymongo>=4.6.3

{pbesa-4.0.40 → pbesa-4.0.42}/pbesa.egg-info/SOURCES.txt

@@ -88,6 +88,7 @@ pbesa/celulas/celula_pertinencia.py
 pbesa/celulas/celula_preguntas.py
 pbesa/celulas/celula_respuesta.py
 pbesa/celulas/celula_saludos.py
+pbesa/celulas/celula_verificar_calculos.py
 pbesa/celulas/data_extraction_cel.py
 pbesa/celulas/web.py
 pbesa/kernel/__init__.py

{pbesa-4.0.40 → pbesa-4.0.42}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='pbesa',
-    version='4.0.40',
+    version='4.0.42',
     packages=find_packages(),
     install_requires=[
         'pymongo>=4.6.3',