pbesa 4.0.33__tar.gz → 4.0.35__tar.gz

{pbesa-4.0.33 → pbesa-4.0.35}/PKG-INFO

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

{pbesa-4.0.33 → pbesa-4.0.35}/pbesa/__init__.py

@@ -3,4 +3,5 @@ from .social import *
 from .remote import *
 from .cognitive import *
 from .models import *
-from .celulas import *
+from .celulas import *
+from .simulator import *

{pbesa-4.0.33 → pbesa-4.0.35}/pbesa/celulas/celula_generar_caso.py

@@ -19,7 +19,7 @@ Se te proporcionarán **definiciones clave** y **reglas específicas** que debes
 
 **FORMATO DE SALIDA:**
 
-Se conciso y directo. El caso debe tener menos de 100 palabras.
+Se conciso y directo. El caso debe tener menos de 50 palabras.
 
 Simula la descripción del caso desde la perspectiva de la persona afectada, incluyendo los detalles necesarios para comprender la situación y la entidad jurisdiccional que debería intervenir.
 

{pbesa-4.0.33 → pbesa-4.0.35}/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_simulador_ciudadano)
+                      celula_instruccion, celula_generar_caso)
 from pbesa.social.prompts import ANALIZER_PROMPT, CLASSIFICATION_PROMPT, DERIVE_PROMPT, RECOVERY_PROMPT, ADAPT_PROMPT, SINTETIZER_PROMPT
 
 # --------------------------------------------------------
@@ -1498,7 +1498,7 @@ class Dialog(ABC):
             logging.error(e)
             return None
 
-    def command_derive(self, command, query, max_tkns=2000) -> str | None:
+    def command_derive(self, command, query, max_tkns=2000, simulator = None) -> str | None:
         # data_extraction_cel
         if command == "IDENTIFICAR_VALORES":
             return celula_datos_identificables.derive(self.__ai_service, query, max_tkns=max_tkns)
@@ -1515,7 +1515,9 @@ class Dialog(ABC):
         if command == "GENERAR_CASO":
             return celula_generar_caso.derive(self.__ai_service, self.definitions, self.rules, max_tkns=max_tkns)
         if command == "CONTINUAR_DIALOGO":
-            return celula_simulador_ciudadano.derive(self.__ai_service, query, max_tkns=max_tkns)
+            if simulator:
+                simulator.set_service(self.__ai_service)
+                return simulator.derive(query, max_tkns=max_tkns)
         return None
     
     def parse_conversation(self) -> str:

pbesa-4.0.35/pbesa/simulator/__init__.py

@@ -0,0 +1 @@
+from .dialog_simulator import DialogSimulator

pbesa-4.0.33/pbesa/celulas/celula_simulador_ciudadano.py → pbesa-4.0.35/pbesa/simulator/dialog_simulator.py

@@ -47,69 +47,48 @@ USER_PROMPT = """
 Demanda:
 %s
 """
-class UserSimulator(object):
+
+class DialogSimulator(object):
     """
     Clase que simula un usuario demandante ante entes de control.
     """
 
-    class __UserSimulator:
-
-        def __init__(self, service):
-            self.iteracion = 1
-            self.work_memory = []
-            self.service = service
-
-        # Efectua la inferencia del modelo.
-        def evaluate(self, text, max_tkns=2000, reset=False) -> any:
-            try:
-                if reset:
-                    self.iteracion = 1
-                    self.work_memory = []
-                logging.info("\n")
-                logging.info(f"Procesando: {text} - Iteración: {self.iteracion}")
-                if self.iteracion == 1:
-                    prompt  = SYSTEM_PROMPT
-                    self.work_memory.append({"role": "system", "content": SYSTEM_PROMPT})    
-                    prompt  = USER_PROMPT % text
-                    self.work_memory.append({"role": "user", "content": prompt})
-                else:
-                    self.work_memory.append({"role": "user", "content": text})
-                res = self.service.generate(self.work_memory, max_tkns)
-                self.work_memory.append({"role": "assistant", "content": res})
-                self.iteracion += 1
-                logging.info(f"Respuesta: {res}")
-                logging.info("\n")
-                if not res or res == "":
-                    res = text
-                    logging.warning(f"No obtener una respuesta.")
-                return res
-            except Exception as e:
-                logging.error(f"Error al procesar: {text}")
-                logging.error(e)
-                return None
-    
-    # Singleton instance
-    instance = None
-
-    def __new__(cls, service) -> object:
-        """ Create new instance """
-        if not UserSimulator.instance:
-            UserSimulator.instance = UserSimulator.__UserSimulator(service)
-        return UserSimulator.instance
+    def __init__(self):
+        self.iteracion = 1
+        self.work_memory = []
+        self.service = None
+
+    # Efectua la inferencia del modelo.
+    def evaluate(self, text, max_tkns=2000, reset=False) -> any:
+        try:
+            if reset:
+                self.iteracion = 1
+                self.work_memory = []
+            logging.info("\n")
+            logging.info(f"Procesando: {text} - Iteración: {self.iteracion}")
+            if self.iteracion == 1:
+                prompt  = SYSTEM_PROMPT
+                self.work_memory.append({"role": "system", "content": SYSTEM_PROMPT})    
+                prompt  = USER_PROMPT % text
+                self.work_memory.append({"role": "user", "content": prompt})
+            else:
+                self.work_memory.append({"role": "user", "content": text})
+            res = self.service.generate(self.work_memory, max_tkns)
+            self.work_memory.append({"role": "assistant", "content": res})
+            self.iteracion += 1
+            logging.info(f"Respuesta: {res}")
+            logging.info("\n")
+            if not res or res == "":
+                res = text
+                logging.warning(f"No obtener una respuesta.")
+            return res
+        except Exception as e:
+            logging.error(f"Error al procesar: {text}")
+            logging.error(e)
+            return None
     
-    def __getattr__(self, name: str) -> object:
-        """ Get attribute
-        @param name Attribute name
-        @return object Attribute
-        """
-        return getattr(self.instance, name)
+    def derive(self, text, max_tkns=2000, reset=False) -> any:
+        return self.evaluate(text, max_tkns=max_tkns, reset=reset)
     
-    def __setattr__(self, name:str) -> object:
-        """ Set attribute
-        @param name Attribute name
-        @return object Attribute
-        """
-        return setattr(self.instance, name)
-
-def derive(service, text, max_tkns=2000, reset=False) -> any:
-    return UserSimulator(service).evaluate(text, max_tkns=max_tkns, reset=reset)
+    def set_service(self, service):
+        self.service = service

{pbesa-4.0.33 → pbesa-4.0.35}/pbesa.egg-info/PKG-INFO

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

{pbesa-4.0.33 → pbesa-4.0.35}/pbesa.egg-info/SOURCES.txt

@@ -87,7 +87,6 @@ pbesa/celulas/celula_pertinencia.py
 pbesa/celulas/celula_preguntas.py
 pbesa/celulas/celula_respuesta.py
 pbesa/celulas/celula_saludos.py
-pbesa/celulas/celula_simulador_ciudadano.py
 pbesa/celulas/data_extraction_cel.py
 pbesa/celulas/web.py
 pbesa/kernel/__init__.py
@@ -110,6 +109,8 @@ pbesa/remote/adm_listener_handler.py
 pbesa/remote/exceptions.py
 pbesa/remote/remote_adm.py
 pbesa/remote/remote_adm_handler.py
+pbesa/simulator/__init__.py
+pbesa/simulator/dialog_simulator.py
 pbesa/social/__init__.py
 pbesa/social/collaborative_team.py
 pbesa/social/delegator.py

{pbesa-4.0.33 → pbesa-4.0.35}/setup.py

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