pbesa 4.0.4__tar.gz → 4.0.5__tar.gz

{pbesa-4.0.4 → pbesa-4.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.1
 Name: pbesa
-Version: 4.0.4
+Version: 4.0.5
 License-File: LICENSE
 License-File: LICENSE.txt
 Requires-Dist: pymongo==4.6.3
@@ -8,4 +8,3 @@ Requires-Dist: requests==2.32.3
 Requires-Dist: azure-ai-projects==1.0.0b6
 Requires-Dist: azure-ai-inference==1.0.0b9
 Requires-Dist: azure-identity==1.20.0
-Dynamic: requires-dist

{pbesa-4.0.4 → pbesa-4.0.5}/pbesa/cognitive.py

@@ -21,7 +21,7 @@ from typing import List, Optional
 from abc import ABC, abstractmethod
 from pbesa.models import AIFoundry, AzureInference, GPTService, ServiceProvider
 from pbesa.social.dialog import DialogState, imprimir_grafo, recorrer_interacciones, extraer_diccionario_nodos, ActionNode, DeclarativeNode#, TerminalNode
-from pbesa.social.prompts import CLASSIFICATION_PROMPT, DERIVE_PROMPT
+from pbesa.social.prompts import CLASSIFICATION_PROMPT, DERIVE_PROMPT, RECOVERY_PROMPT
 # --------------------------------------------------------
 # Define DTOs
 # --------------------------------------------------------
@@ -518,29 +518,29 @@ class Dialog(ABC):
         self.setup_world()
         interations = agent_metadata.role.interactions
         grafo = recorrer_interacciones(interations)
-        logging.info("")
+        logging.debug("")
         # Si el grafo es una lista de nodos, lo imprimimos cada uno
         if isinstance(grafo, list):
             for nodo in grafo:
                 imprimir_grafo(nodo)
         else:
             imprimir_grafo(grafo)
-        logging.info("")
+        logging.debug("")
         # Ejemplo de uso:
         self.__dfa = extraer_diccionario_nodos(grafo)
         # Mostrar el diccionario
         for clave, valor in self.__dfa.items():
-            logging.info(f"{clave}: {valor.text}")
-        logging.info("")
+            logging.debug(f"{clave}: {valor.text}")
+        logging.debug("")
         iniciadores = []
         for item in interations:
             for clave, valor in self.__dfa.items():
                 if valor.text == item['texto']:
                     iniciadores.append(valor)
-        logging.info("Iniciadores:")
+        logging.debug("Iniciadores:")
         for iniciador in iniciadores:
-            logging.info(iniciador.text)
-        logging.info("")
+            logging.debug(iniciador.text)
+        logging.debug("")
         # Set dialog state
         self.__dfa['start'] = iniciadores
 
@@ -608,7 +608,11 @@ class Dialog(ABC):
         if not dialog_state in self.__dfa:
             logging.warning(f"------------Performativa no existe---------------")
             self.reset()
-            return "Web", DialogState.START, "Lo lamento, no puedo responder en este momento", "Web"
+            prompt = RECOVERY_PROMPT % query
+            temp_work_memory = [{"role": "user", "content": query}]
+            res = self.__ai_service.generate(temp_work_memory)
+            res = self.get_text(res)
+            return "Web", DialogState.START, res, "Web"
 
         node = self.__dfa[dialog_state]
         if not isinstance(node, list):

{pbesa-4.0.4 → pbesa-4.0.5}/pbesa/social/dialog.py

@@ -13,6 +13,7 @@
 # Define resources
 # --------------------------------------------------------
 
+import logging
 import datetime
 import traceback
 from .templates import AFIRMATIVE_RESPONSE, GET_DATE
@@ -199,6 +200,6 @@ def imprimir_grafo(nodo, nivel=0):
     indent = "  " * nivel
     # Se muestra la clase del nodo y algunos atributos
     clase = nodo.__class__.__name__
-    print(f"{indent}{clase} (performative: {nodo.performative}, text/action: {nodo.text if hasattr(nodo, 'text') and nodo.text is not None else getattr(nodo, 'action', None)}, terminal: {nodo.is_terminal})")
+    logging.debug(f"{indent}{clase} (performative: {nodo.performative}, text/action: {nodo.text if hasattr(nodo, 'text') and nodo.text is not None else getattr(nodo, 'action', None)}, terminal: {nodo.is_terminal})")
     for child in nodo.children:
         imprimir_grafo(child, nivel + 1)

{pbesa-4.0.4 → pbesa-4.0.5}/pbesa/social/dispatcher_team.py

@@ -130,7 +130,7 @@ class ResponseAction(Action):
 class DispatcherController(Agent):
     """ Represents the agent that delegates the execution of actions to other agents """
     
-    def __init__(self, agent_id:str, buffer_size:int, pool_size:int) -> None:
+    def __init__(self, agent_id:str, buffer_size:int, pool_size:int, delegate=None) -> None:
         """ Constructor
         @param agent_id: Agent ID
         @param type: Pool type
@@ -139,15 +139,20 @@ class DispatcherController(Agent):
         """
         self.__buffer_size = buffer_size
         self.__request_dict = {}
+        self.__pool_size = pool_size
         self.__free_queue = Queue(pool_size)
         self.__agent_list = []
+        self.__delegate = delegate
         super().__init__(agent_id)
 
     def setup(self) -> None:
         """ Set up method """
         self._social = True
         self.add_behavior('Delegate')
-        self.bind_action('Delegate', 'delegate', Delegate())
+        if self.__delegate:
+            self.bind_action('Delegate', 'delegate', self.__delegate())
+        else:
+            self.bind_action('Delegate', 'delegate', Delegate())
         self.add_behavior('Notify')
         self.bind_action('Notify', 'notify', NotifyFreeAction())
         self.add_behavior('Response')
@@ -161,6 +166,8 @@ class DispatcherController(Agent):
         self.__agent_list.append(agent.id)
         agent.set_controller(self.id)
         agent.set_controller_type('POOL')
+        if len(self.__agent_list) > self.__pool_size:
+            raise DispatcherException('[Warn, suscribeAgent]: The pool is full')
         self.__free_queue.put(agent.id)
         actions = agent.get_actions()
         for action in actions:
@@ -190,6 +197,12 @@ class DispatcherController(Agent):
         """ Builds the agent """
         pass
 
+    def get_agent_list(self) -> list:
+        """ Gets the agent list
+        @return: Agent list
+        """
+        return self.__agent_list
+
     def get_free_queue(self) -> Queue:
         """ Gets the free queue
         @return: Free queue

{pbesa-4.0.4 → pbesa-4.0.5}/pbesa/social/prompts.py

@@ -36,5 +36,29 @@ Ahora, evalúa el siguiente caso:
 
 Texto: "%s"
 
+Respuesta:
+"""
+
+RECOVERY_PROMPT = """
+Instrucciones:
+
+Eres un agente especializado en parafrasear. Tu tarea es analizar un texto principal y parafrasearlo generando opciones al usuario.
+
+Requisitos:
+
+- No incluyas explicaciones, razonamientos ni texto adicional.
+- Genera opciones en lenguaje natural sin involucrar números o listas.
+- Si no logras parafrasear el texto principal, responde solicitando más información.
+
+Ejemplo:
+
+Texto: "Hola mi caso esta relacionado con la Superfinanciera"
+
+Respuesta: Perdona, tu solicitud ¿Es sobre la Superintendencia Financiera de Colombia o sobre la Superintendencia de Sociedades?
+
+Ahora, evalúa el siguiente caso:
+
+Texto: "%s"
+
 Respuesta:
 """

{pbesa-4.0.4 → pbesa-4.0.5}/pbesa/social/selected_dispatcher_team.py

@@ -104,7 +104,7 @@ class DelegateAction(Action):
                         self.__rewier[ag] = 0
                         exit = True
                     else:
-                        logging.info('The agent is not available')
+                        logging.info('The agent is not operational')
                         self.adm.send_event(agent_obj.get_controller(), 'notify', ag)
                         if ag in self.__rewier:
                             self.__rewier[ag] = self.__rewier[ag] + 1

{pbesa-4.0.4 → pbesa-4.0.5}/pbesa.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.1
 Name: pbesa
-Version: 4.0.4
+Version: 4.0.5
 License-File: LICENSE
 License-File: LICENSE.txt
 Requires-Dist: pymongo==4.6.3
@@ -8,4 +8,3 @@ Requires-Dist: requests==2.32.3
 Requires-Dist: azure-ai-projects==1.0.0b6
 Requires-Dist: azure-ai-inference==1.0.0b9
 Requires-Dist: azure-identity==1.20.0
-Dynamic: requires-dist

{pbesa-4.0.4 → pbesa-4.0.5}/setup.py

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