pbesa 4.0.41__tar.gz → 4.0.43__tar.gz

{pbesa-4.0.41 → pbesa-4.0.43}/PKG-INFO

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

pbesa-4.0.43/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.41 → pbesa-4.0.43}/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
 
 # --------------------------------------------------------
@@ -717,13 +717,13 @@ class Dialog(ABC):
             logging.error(f"Error al enviar notificación: {text}")
             logging.error(f"Error: {e}")
     
-    def update_conversation(self, session_manager, id_conversacion, res):
-        conversation = session_manager.get_conversation(id_conversacion)
+    def update_conversation(self, session_manager, conversation_id, res):
+        conversation = session_manager.get_conversation(conversation_id)
         if conversation:
             conversation['team-response'] = res
-            session_manager.update_conversation(id_conversacion, conversation)
+            session_manager.update_conversation(conversation_id, conversation)
         else:
-            logging.error(f"Error al actualizar la conversación: {id_conversacion}")
+            logging.error(f"Error al actualizar la conversación: {conversation_id}")
 
     def team_inquiry(self, session, team, data, operation, session_flag) -> str:
         try:
@@ -750,11 +750,11 @@ class Dialog(ABC):
                 response = canal.post(team.lower(), dto, timeout=300)
                 if response and not response['status']:
                     logging.error(f'No se pudo establecer la comunicación con el agente remoto')
-                    self.update_conversation(session_manager, session['id_conversacion'], None)
+                    self.update_conversation(session_manager, session['conversation_id'], None)
                     return None
                 logging.info(f'>>>> Response: {response}')
                 res = response['message']['response']
-                self.update_conversation(session_manager, session['id_conversacion'], res)
+                self.update_conversation(session_manager, session['conversation_id'], res)
                 return res
             else:
                 if operation and not operation == "Ninguno":
@@ -762,7 +762,7 @@ class Dialog(ABC):
                         "data": {
                             'text': data,
                             'operation': operation,
-                            'id_conversacion': session['id_conversacion'],
+                            'conversation_id': session['conversation_id'],
                             'session_id': session['session_id'],
                         },
                     }
@@ -776,10 +776,10 @@ class Dialog(ABC):
                 if response['status']:
                     logging.info(f'>>>> Response: {response}')
                     res = response['message']['response']
-                    self.update_conversation(session_manager, session['id_conversacion'], res)
+                    self.update_conversation(session_manager, session['conversation_id'], res)
                     return res
                 else:
-                    self.update_conversation(session_manager, session['id_conversacion'], None)
+                    self.update_conversation(session_manager, session['conversation_id'], None)
                     logging.error(f'No se pudo establecer la comunicación con el agente remoto')
                     return None
             logging.info("END: team_inquiry")
@@ -932,6 +932,9 @@ class Dialog(ABC):
         
         msg = "No puedo ayudarle con su consulta. Dado que no está relacionada con los servicios que presta Justifacil."
 
+        if es_consulta:
+            query = f"En el contexto jurídico. {query}" 
+
         # Verifica los casos
         if eva > 0 and eva < 3:
             logging.info("[Stage-1]: Discriminador encontrado")
@@ -942,8 +945,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 +1023,22 @@ 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']
+                        query = f"En el contexto jurídico. {query}"
+                        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)
 
@@ -1091,11 +1105,11 @@ class Dialog(ABC):
                 if owner == "Web" and dialog_state == DialogState.START:
                     logging.info(f"TOPTQ: {owner} - {dialog_state} - {team_source} - {query}")
                     session_manager = self.state['session_manager']        
-                    conversation = session_manager.get_conversation(session['id_conversacion'])
+                    conversation = session_manager.get_conversation(session['conversation_id'])
                     attemps = conversation.get('attemps', 1)            
                     dicriminador, res = self.stage_one_classification(session['session_id'], conversation['messages'], attemps, query)
                     conversation['attemps'] = attemps + 1
-                    session_manager.update_conversation(session['id_conversacion'], conversation)
+                    session_manager.update_conversation(session['conversation_id'], conversation)
                     logging.info(f"Discriminador: {dicriminador}")
                     # Limpia la memoria de trabajo
                     # Para que el agente la utilice en
@@ -1348,7 +1362,7 @@ class Dialog(ABC):
                             if "especialistas" in node.text.lower():
                                 self.notify(session['session_id'], "consultando especialistas...")
                                 session_manager = self.state['session_manager']
-                                conversation = session_manager.get_conversation(session['id_conversacion'])
+                                conversation = session_manager.get_conversation(session['conversation_id'])
                                 if 'team-response' in conversation:
                                     res = conversation['team-response']
                                 else:
@@ -1391,9 +1405,9 @@ class Dialog(ABC):
                             if "expertos" in node.text.lower():
                                 self.notify(session['session_id'], "consultando expertos...")
                                 session_manager = self.state['session_manager']
-                                conversation = session_manager.get_conversation(session['id_conversacion'])
+                                conversation = session_manager.get_conversation(session['conversation_id'])
                                 conversation['case'] = query
-                                session_manager.update_conversation(session['id_conversacion'], conversation)
+                                session_manager.update_conversation(session['conversation_id'], conversation)
                                 logging.info("[Node]:[Action]:[Call]:[Continua]:[Consulta]: Caso almacenado")
                         else:
                             logging.info("[Node]:[Action]:[Call]:[Continua]:[Inferencia]:")

{pbesa-4.0.41 → pbesa-4.0.43}/pbesa/social/sel_dispatcher_team.py

@@ -188,7 +188,10 @@ class SelectedDispatcher(Action):
         try:
             logging.info('Assign to agent...')
             session_id = data['dto']['session']['session_id'] if 'session' in data['dto'] else None
-            assistant = data['dto']['session']['assistant'] if 'session' in data['dto'] and 'assistant' in data['dto']['session'] else None            
+            #assistant = data['dto']['session']['assistant'] if 'session' in data['dto'] and 'assistant' in data['dto']['session'] else None            
+            assistant = None
+            if data and 'conversation' in data and data['dto']['conversation']:
+                assistant = data['dto']['conversation']['assistant'] if data['dto']['conversation'] else None
             if assistant:
                 logging.info('The session is already assigned')
                 logging.info(f'The agent type: {assistant} will be assigned')

{pbesa-4.0.41 → pbesa-4.0.43}/pbesa.egg-info/PKG-INFO

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

{pbesa-4.0.41 → pbesa-4.0.43}/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.41 → pbesa-4.0.43}/setup.py

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