pbesa 4.0.3__tar.gz → 4.0.5__tar.gz

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

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.1
 Name: pbesa
-Version: 4.0.3
+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.3 → 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
 # --------------------------------------------------------
@@ -466,6 +466,8 @@ class Dialog(ABC):
         # Define deep
         self.__deep_count = 0
         self.__deep_limit = 3
+        # Define knowledge
+        self.knowledge = None
 
     def setup_world(self):
         """ Set up model method """
@@ -473,10 +475,14 @@ class Dialog(ABC):
         instrucciones = f"Instrucciones:\n{self.__role.description}\n Tu tarea: {self.__role.objective}\n"
         requisitos = f"Requisitos:\n{self.__role.arquetype}\n"
         ejemplo = f"Ejemplo:\n{self.__role.example}\n"
+        if self.knowledge:
+            conocimiento = f"Conocimiento:\n{self.knowledge}\n"
         continuar = "Ahora, evalúa el siguiente caso:\n"
         self.__work_memory.append({"role": "user", "content": instrucciones})
         self.__work_memory.append({"role": "user", "content": requisitos})
         self.__work_memory.append({"role": "user", "content": ejemplo})
+        if self.knowledge:
+            self.__work_memory.append({"role": "user", "content": conocimiento})
         self.__work_memory.append({"role": "user", "content": continuar})
         
     def get_model(self) -> any:
@@ -512,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
 
@@ -602,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):
@@ -721,7 +731,10 @@ class Dialog(ABC):
                     self.__work_memory.append({"role": "user", "content": node.text})
                     logging.info(f"-> node team -> envia: {query}")
                     text = self.team_inquiry(node.team, query, node.tool, False)
-            logging.info(f"-> node team -> text: {text}")
+                logging.info(f"-> node team -> text: {text}")
+                if text:
+                    logging.info(f"-> Adicion WM node team -> text: {text}")
+                    self.__work_memory.append({"role": "system", "content": text})
             self.__deep_count += 1
             if self.__deep_count < self.__deep_limit:
                 return self.transition(owner, node.performative, text, True)
@@ -765,3 +778,10 @@ class Dialog(ABC):
 
         logging.info(f"$$$> new_owner: {owner} new_dialog_state: {new_dialog_state}")
         return owner, new_dialog_state, res, owner
+
+    def set_knowledge(self, knowledge) -> str:
+        """ Set knowledge method
+        :param query: query
+        :return: str
+        """
+        self.knowledge = knowledge

{pbesa-4.0.3 → pbesa-4.0.5}/pbesa/social/collaborative_team.py

@@ -33,6 +33,7 @@ clear and unified results.
 # Define resources
 # --------------------------------------------------------
 
+import logging
 from threading import Timer
 from abc import abstractmethod
 from ..kernel.agent import Agent
@@ -94,7 +95,7 @@ class DelegateAction(Action):
         self.agent.reset()
         self.delegate(data['dto'])
         
-    def activeTimeout(self, time:int) -> None:
+    def active_timeout(self, time:int) -> None:
         """ Active timeout
         @param time: Time
         """
@@ -109,6 +110,27 @@ class DelegateAction(Action):
             self.adm.send_event(ag, 'task', data)
         else:
             raise CollaborativeException('[Warn, toAssign]: The number of data packets exceeds the number of agents')
+        
+    def get_list_agents(self) -> list:
+        """ Get list of agents
+        @return: List of agents
+        """
+        return self.agent.get_free_list()
+    
+    def to_individual_assign(self, ag_id, data:any) -> bool:
+        """ Assign data to an agent
+        @param data: Data
+        """
+        if len(self.agent.get_free_list()) > 0:
+            index = self.agent.get_free_list().index(ag_id)
+            if index >= 0:
+                self.agent.get_free_list().pop(index)
+                self.adm.send_event(ag_id, 'task', data)
+            else:
+                return False
+        else:
+            raise CollaborativeException('[Warn, toAssign]: The number of data packets exceeds the number of agents')
+        return True
 
     @abstractmethod
     def delegate(self, data:any) -> None:
@@ -129,17 +151,24 @@ class ResponseAction(Action):
         @param data: Event data
         """
         if 'timeout' == data:
+            logging.info('Stopping timeout')
             results = {}
             for key, res in self.agent.get_check_dict().items():
                 results[key] = res
             self.end_of_process(results, True)
         else:
+            logging.info("Response received")
+            logging.info(data)
             result = 'None'
             agentID = data['source']
-            if data['result']:
+            if 'result' in data and data['result']:
                 result =  data['result']
-            self.agent.get_check_dict()[agentID] = result
+                self.agent.get_check_dict()[agentID] = result
+            else:
+                logging.warning(f"No result received for agent {agentID}")
+                self.agent.get_check_dict()[agentID] = "None"
             if self.check():
+                logging.info("All agents have responded")
                 results = {}
                 for key, res in self.agent.get_check_dict().items():
                     results[key] = res

{pbesa-4.0.3 → 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.3 → 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.3 → 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.3 → 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.3 → 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.3
+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.3 → pbesa-4.0.5}/setup.py

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