pbesa 4.0.36__tar.gz → 4.0.38__tar.gz

{pbesa-4.0.36 → pbesa-4.0.38}/PKG-INFO

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

{pbesa-4.0.36 → pbesa-4.0.38}/pbesa/cognitive.py

@@ -834,23 +834,30 @@ class Dialog(ABC):
     def stage_one_classification(self, session_id, messages, attemps, query):
         """ Stage one classification """
         res = ""
-        #evaluating = True
+
         dicriminador = None
         logging.info(f"------------Flujo de excepcion---------------")
         self.notify(session_id, "identificando intención...")        
         logging.info(f"Intento: {attemps}") 
-        
             
         self.__analaizer_work_memory:list = [{"role": "system", "content": ANALIZER_PROMPT}]
         
-        es_verificacion = False
-        if len(messages) >= 3:
-            last_message = messages[-2]['text']
-            es_verificacion = '¿Desea que le ayude con una consulta o una demanda?' in last_message
+        #es_verificacion = False
+        #if len(messages) >= 3:
+        #    last_message = messages[-2]['text']
+        #    es_verificacion = '¿Desea que le ayude con una consulta o una demanda?' in last_message
         
-        if attemps > 1 and not es_verificacion:
-            saludo = "NO_SALUDO"        
-        elif attemps > 2 or es_verificacion:
+        #if attemps == 2:
+        #    saludo = "NO_SALUDO"
+        #    consulta = celula_consultas.derive(self.__ai_service, query, max_tkns=10)
+
+        if attemps >= 2:# or es_verificacion:
+            rectificacion_consulta = 0
+
+            consulta = celula_consultas.derive(self.__ai_service, query, max_tkns=10)
+            es_consulta = ("PREGUNTA_O_SOLICITUD" in consulta) and not ("NO_PREGUNTA" in consulta)
+            if es_consulta:
+                rectificacion_consulta += 1
             
             saludo = "NO_SALUDO"
             self.__sintetizer_work_memory:list = [{"role": "system", "content": SINTETIZER_PROMPT}]
@@ -875,20 +882,27 @@ class Dialog(ABC):
                     self.__analaizer_work_memory.append({"role": "assistant", "content": message['text']})
                     self.__sintetizer_work_memory.append({"role": "assistant", "content": message['text']})
 
-            #self.__analaizer_work_memory.append({"role": "user", "content": query})
-            #self.__sintetizer_work_memory.append({"role": "user", "content": query})
-        
-            # Desde la seegunda iteeraccion de la
-            # conversacion, se utiliza el sintentizador
             logging.info("\n\n\n--------------SINTETIZER------------------")
             logging.info("\n%s", json.dumps(self.__sintetizer_work_memory, indent=4))
             logging.info("\n\n\n")
-            query = self.__ai_service.generate(self.__sintetizer_work_memory, max_tokens=50)
-            res = self.get_text(query)
-            query = res
-            logging.info(f"[Stage-1][Sintetizer][Thought]: {query}")
+            res = self.__ai_service.generate(self.__sintetizer_work_memory, max_tokens=50)
+            res = self.get_text(res)
+            logging.info(f"[Stage-1][Sintetizer][Thought]: {res}")
+
+            consulta = celula_consultas.derive(self.__ai_service, res, max_tkns=10)
+            es_consulta = ("PREGUNTA_O_SOLICITUD" in consulta) and not ("NO_PREGUNTA" in consulta)
+            if es_consulta:
+                rectificacion_consulta += 1
+            
+            if not rectificacion_consulta == 2:
+                query = res
+                logging.info("[Stage-1][Sintetizer][Thought]: Se sustituye el query por el pensamiento.")
+            else:
+                logging.info("[Stage-1][Sintetizer][Thought]: Se mantiene la consulta inicial.")
+                        
         else:
             saludo = celula_saludos.derive(self.__ai_service, query, max_tkns=10)
+            consulta = celula_consultas.derive(self.__ai_service, res, max_tkns=10)
 
             user_prompt = f"""
             user: {query}Texto: 
@@ -902,13 +916,9 @@ class Dialog(ABC):
         #----------------------------------
         # Verifica si es una consulta o un caso
         caso = celula_casos.derive(self.__ai_service, query, max_tkns=10)
-        consulta = celula_consultas.derive(self.__ai_service, query, max_tkns=10)
-        # Verifica si es un saludo, consulta o caso
-
-        #es_saludo = ("SALUDO" in saludo) and ("NO_PREGUNTA" in consulta) and ("NO_QUEJA_DEMANDA" in caso) and not ("NO_SALUDO" in saludo)    
-        #es_consulta = ("PREGUNTA_O_SOLICITUD" in consulta) and ("NO_QUEJA_DEMANDA" in caso) and ("NO_SALUDO" in saludo) and not ("NO_PREGUNTA" in consulta)
-        #es_caso = ("QUEJA_DEMANDA" in caso) and ("NO_PREGUNTA" in consulta) and ("NO_SALUDO" in saludo) and not ("NO_QUEJA_DEMANDA" in caso)
         
+
+        # Verifica si es un saludo, consulta o caso        
         es_saludo = ("SALUDO" in saludo) and not ("NO_SALUDO" in saludo)    
         es_consulta = ("PREGUNTA_O_SOLICITUD" in consulta) and not ("NO_PREGUNTA" in consulta)
         es_caso = ("QUEJA_DEMANDA" in caso) and not ("NO_QUEJA_DEMANDA" in caso)
@@ -988,18 +998,27 @@ class Dialog(ABC):
                     res = msg
                     ambiguedad = False
 
-        if ambiguedad:    
+        if ambiguedad:
             logging.info("[Stage-1]: Respuesta con ambiguedad")
             self.notify(session_id, "identificando ambiguedad...")
-            logging.info("\n\n\n--------------ANALIZER------------------")
-            logging.info("\n%s", json.dumps(self.__analaizer_work_memory, indent=4))
-            logging.info("------------------------------------------\n\n\n")
-            res = self.__ai_service.generate(self.__analaizer_work_memory, max_tokens=50)
-            logging.info(f"[Stage-1][Thought]: {res}")
-            self.__analaizer_work_memory.append({"role": "assistant", "content": res})                
-            self.__sintetizer_work_memory.append({"role": "assistant", "content": res})
-        #----------------------------------
-
+            
+            aclaracion_fired = False
+            for message in messages:
+                if not message['user'] and '¿Desea que le ayude con una consulta o una demanda?' in message['text']:
+                    aclaracion_fired = True
+                    break
+            
+            if aclaracion_fired:
+                logging.info("\n\n\n--------------ANALIZER------------------")
+                logging.info("\n%s", json.dumps(self.__analaizer_work_memory, indent=4))
+                logging.info("------------------------------------------\n\n\n")
+                res = self.__ai_service.generate(self.__analaizer_work_memory, max_tokens=50)
+                logging.info(f"[Stage-1][Thought]: {res}")
+                self.__analaizer_work_memory.append({"role": "assistant", "content": res})                
+                self.__sintetizer_work_memory.append({"role": "assistant", "content": res})
+            else:
+                res = "¿Desea que le ayude con una consulta o una demanda?"
+        
         return dicriminador, res
 
     def get_node(self, performative):
@@ -1413,14 +1432,14 @@ class Dialog(ABC):
                 if not node.is_terminal:
                     # Verifica recursion
                     if not self.chek_user_interaction(node.children):
-                        if self.__visited_nodes > 3:
-                            self.__visited_nodes = 0
-                            logging.info(f"[Inferencia]:[Recursion]: Deep limit")
-                            logging.info(f"------------RESET---------------")
-                            self.reset()
-                            self.notify(session['session_id'], "STOP")
-                            res = DialogState.ERROR
-                            return "Web", DialogState.START, res, "Web"
+                        #if self.__visited_nodes > 3:
+                        #    self.__visited_nodes = 0
+                        #    logging.info(f"[Inferencia]:[Recursion]: Deep limit")
+                        #    logging.info(f"------------RESET---------------")
+                        #    self.reset()
+                        #    self.notify(session['session_id'], "STOP")
+                        #    res = DialogState.ERROR
+                        #    return "Web", DialogState.START, res, "Web"
                         logging.info(f"[Inferencia]:[Recursion]:[Performativa]: {node.performative}")
                         self.notify(session['session_id'], "efectuando inferencia en profundidad")
                         #return self.do_transition(session, owner, node, query)