pbesa 4.0.30__tar.gz → 4.0.31__tar.gz

{pbesa-4.0.30 → pbesa-4.0.31}/PKG-INFO

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

{pbesa-4.0.30 → pbesa-4.0.31}/pbesa/cognitive.py

@@ -589,6 +589,7 @@ class Dialog(ABC):
         self.rules = None
         self.__q_attemps = 0
 
+        
     def setup_world(self):
         """ Set up model method """
         # Define role
@@ -834,10 +835,16 @@ class Dialog(ABC):
             
         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
         
-        if attemps > 1:
+        if attemps > 1 and not es_verificacion:
+            saludo = "NO_SALUDO"        
+        elif attemps > 2 or es_verificacion:
+            
             saludo = "NO_SALUDO"
-
             self.__sintetizer_work_memory:list = [{"role": "system", "content": SINTETIZER_PROMPT}]
 
             cont = 0
@@ -1751,10 +1758,15 @@ class SpecialDispatch():
                     logging.info("=> No se seleccionó ningun agente")
 
                 major_id, response, score = self.retrieval(query)
-                major_id_split = major_id.split("_")
-                major_id = major_id_split[0] if len(major_id_split) > 0 else major_id
-                major_id_split = major_id.split("-")
-                major_id = major_id_split[0] if len(major_id_split) > 0 else major_id
+                if 'glosario' in major_id.lower():
+                    major_id = 'glosario'
+                elif 'inicial' in major_id.lower():
+                    major_id = 'portal'
+                else:
+                    major_id_split = major_id.split("_")
+                    major_id = major_id_split[0] if len(major_id_split) > 0 else major_id
+                    major_id_split = major_id.split("-")
+                    major_id = major_id_split[1] if len(major_id_split) > 0 else major_id
 
                 compare_llm = 0
                 compare_index = 0
@@ -1797,4 +1809,5 @@ class SpecialDispatch():
 
             self.kusto_close()
             self.__meta_work_memory = []
+            self.major_id = None
             return select_agent, response

{pbesa-4.0.30 → pbesa-4.0.31}/pbesa.egg-info/PKG-INFO

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

{pbesa-4.0.30 → pbesa-4.0.31}/setup.py

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