pbesa 4.0.6__tar.gz → 4.0.7__tar.gz

{pbesa-4.0.6 → pbesa-4.0.7}/PKG-INFO

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

{pbesa-4.0.6 → pbesa-4.0.7}/pbesa/cognitive.py

@@ -13,6 +13,7 @@
 # Define resources
 # --------------------------------------------------------
 
+import uuid
 import json
 import logging
 import traceback
@@ -20,13 +21,16 @@ from pydantic import BaseModel
 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.dialog import (
+    DialogState, imprimir_grafo, recorrer_interacciones, extraer_diccionario_nodos, 
+    ActionNode, DeclarativeNode, GotoNode) #, TerminalNode
 from pbesa.social.prompts import CLASSIFICATION_PROMPT, DERIVE_PROMPT, RECOVERY_PROMPT
 # --------------------------------------------------------
 # Define DTOs
 # --------------------------------------------------------
 
 class InteraccionDTO(BaseModel):
+    id: str
     tipo: str
     texto: str
     actor: str
@@ -38,6 +42,7 @@ def interaccion_serializer(interaccion):
     if isinstance(interaccion, list):
         return [interaccion_serializer(i) for i in interaccion]
     return {
+        "id": interaccion["id"] if "id" in interaccion and interaccion["id"] else str(uuid.uuid4()),
         "tipo": interaccion["tipo"],
         "texto": interaccion["texto"],
         "actor": interaccion["actor"],
@@ -477,6 +482,10 @@ class Dialog(ABC):
         }
         # Define recovery message
         self.RECOVERY_MSG = "Lo lamento, puedes darme más detalles o reformular"
+        # Define vertices list
+        self.__vertices = []
+        # Define visited nodes
+        self.__visited_nodes = 0
 
     def setup_world(self):
         """ Set up model method """
@@ -642,9 +651,10 @@ class Dialog(ABC):
 
     def get_text(self, mensaje) -> str:
         if mensaje:
-            mensaje_limpio = mensaje.replace("<|im_start|>user<|im_sep|>", "").replace("<|im_start|>system<|im_sep|>", "")
-            mensaje_limpio = mensaje_limpio.replace("<|im_start|>", "").replace("<|im_sep|>", "").replace("<|im_end|>", "")
-            mensaje_limpio = mensaje_limpio.replace("[Usuario]: ", "").replace("[Sistema]: ", "")
+            mensaje_limpio = mensaje.replace("<|im_start|>user<|im_sep|>", "").replace("<|im_start|>system<|im_sep|>", "") \
+                .replace("<|im_start|>", "").replace("<|im_sep|>", "").replace("<|im_end|>", "") \
+                .replace("[Usuario]: ", "").replace("[Sistema]: ", "") \
+                .replace("<|user|>", "").replace("<|system|>", "")
             return mensaje_limpio.strip()
         else:
             return ""
@@ -680,20 +690,25 @@ class Dialog(ABC):
             }
             logging.info(f"Recovery attemps: {self.__recovery["counter"]}")
             # Verifica si se ha alcanzado el límite de recuperación
-            if self.__recovery['counter'] <= 3:
-                self.notify("identificando la performativa...")
+            if self.__recovery['counter'] <= 3 and self.__visited_nodes < 5:
+                self.notify("identificando concepto...")
+                #--------------------------
                 # Verifica que exista la performativa
                 if not dialog_state in self.__dfa:
-                    self.notify("performativa no encontrada")
+                    self.notify("concepto no encontrado")
                     return self.recovery(query)
                 # Performativa encontrada
                 children = None
-                self.notify("performativa encontrada")
+                self.notify("concepto encontrado")
+                #--------------------------
+                # Obtiene los hijos del nodo
                 node = self.__dfa[dialog_state]
                 if not isinstance(node, list):
                     children = node.children
                 else:
+                    # Es una lista de nodos
                     children = node
+                #--------------------------
                 # Flujo de selección
                 select_node = None
                 self.notify("flujo de seleccion...")
@@ -724,20 +739,52 @@ class Dialog(ABC):
                     logging.info(f"--> Una opción.")
                     select_node = children[0]
                 else:
-                    logging.info(f"???????????????????> text: {res} es terminal")    
-                # Nuevo nodo
+                    logging.info("???> Es un nodo terminal o iniciador")
+                    return self.recovery(query)
+                #--------------------------
+                # Verifica si es un nodo
+                # que ya fue recorrido
+                if select_node.performative in self.__vertices:
+                    logging.info(f"-> nodo ya recorrido: {select_node.text}")
+                    self.__visited_nodes += 1
+                else:
+                    logging.info(f"-> nodo no recorrido: {select_node.text}")
+                    self.__visited_nodes = 0
+                    # Maraca el nodo como visitado
+                    self.__vertices.append(select_node.performative)
+                #---------------------------
+                # Verifica si el nodo es un
+                # nodo de salto
+                if isinstance(select_node, GotoNode):
+                    logging.info(f"-> node de salto: {select_node.text}")
+                    performative = select_node.text.replace("Salta a:", "").strip()
+                    select_node = self.__dfa[performative]
+                    logging.info(f"-> Salto: {select_node.text}, performativa: {select_node.performative}")
+                #---------------------------
+                # Efectua transicion
                 if select_node:
                     res = select_node.text
                     node = select_node.children[0]
-                # Nodo seleccionado
                 logging.info(f"Flujo normal: {res}")
                 self.notify(res)
+                #---------------------------
+                # Verifica si el nuevo nodo 
+                # es un nodo de salto
+                if isinstance(node, GotoNode):
+                    logging.info(f"-> node de salto: {node.text}")
+                    performative = node.text.replace("Salta a:", "").strip()
+                    node = self.__dfa[performative]
+                    logging.info(f"-> Salto: {node.text}, performativa: {node.performative}")
+                #---------------------------
+                # Actualiza la memoria de 
+                # trabajo
                 if team_source:
                     self.__work_memory.append({"role": "user", "content": res})
                 else:
                     self.__work_memory.append({"role": "user", "content": query})
                     self.__work_memory.append({"role": "user", "content": res})
-                # Efectua inferencia  
+                #---------------------------
+                # Efectua inferencia
                 new_owner, new_dialog_state, res, team = self.do_transition(owner, node, query)
                 res = self.get_text(res)
                 return new_owner, new_dialog_state, res, team

{pbesa-4.0.6 → pbesa-4.0.7}/pbesa/social/dialog.py

@@ -37,7 +37,7 @@ class DialogState:
 class Node:
     def __init__(self, actor, performative, text=None, is_terminal=False):
         self.actor = actor
-        self.performative = performative  # Se asigna el ID del objeto (en str)
+        self.performative = performative
         self.text = text
         self.is_terminal = is_terminal
         self.children = []
@@ -57,6 +57,10 @@ class ResponseNode(Node):
     def __init__(self, actor, performative, text, is_terminal=False):
         super().__init__(actor, performative=performative, text=text, is_terminal=is_terminal)
 
+class GotoNode(Node):
+    def __init__(self, actor, performative, text, is_terminal=False):
+        super().__init__(actor, performative=performative, text=text, is_terminal=is_terminal)
+
 #------------------------------------------
 # Define functions
 #------------------------------------------
@@ -125,7 +129,7 @@ def recorrer_interacciones(obj):
             # Si tiene la clave "tipo", creamos un nodo (según su valor) y procesamos sus hijos
             if "tipo" in obj:
                 # Se usa el id del diccionario convertido a cadena para 'performative'
-                current_id = str(id(obj))
+                current_id = obj["id"] if "id" in obj else str(id(obj))
                 texto = obj.get("texto")
                 
                 # Procesamos los nodos hijos buscando en las claves "interacciones" y "Interacciones"
@@ -150,6 +154,8 @@ def recorrer_interacciones(obj):
                     nuevo_nodo = ActionNode(actor=obj["actor"], performative=current_id, text=texto, action=tipo, team=obj["equipo"], tool=obj["herramienta"], is_terminal=is_terminal)
                 elif tipo == "respuesta de equipo":
                     nuevo_nodo = ResponseNode(actor=obj["actor"], performative=current_id, text=texto, is_terminal=is_terminal)
+                elif tipo == "salta a":
+                    nuevo_nodo = GotoNode(actor=obj["actor"], performative=current_id, text=texto, is_terminal=is_terminal)
                 else:
                     nuevo_nodo = Node(actor=obj["actor"], performative=current_id, text=texto, is_terminal=is_terminal)
                 
@@ -168,8 +174,10 @@ def recorrer_interacciones(obj):
                                 nodos.extend(resultado)
                             else:
                                 nodos.append(resultado)
+                # Devolvemos la lista de nodos encontrados
                 return nodos if nodos else None
         else:
+            # Si no es ni lista ni diccionario, devolvemos None
             return None
     except Exception as e:
         print(f"Error al recorrer interacciones: {e}")

{pbesa-4.0.6 → pbesa-4.0.7}/pbesa.egg-info/PKG-INFO

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

{pbesa-4.0.6 → pbesa-4.0.7}/setup.py

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