pbesa 4.0.21__tar.gz → 4.0.22__tar.gz

{pbesa-4.0.21 → pbesa-4.0.22}/PKG-INFO

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

{pbesa-4.0.21 → pbesa-4.0.22}/pbesa/cognitive.py

@@ -1457,53 +1457,172 @@ class SpecialDispatch():
             return mensaje_limpio.strip()
         else:
             return ""
+    
+    @abstractmethod
+    def retrieval(self, query) -> str:
+        """ Set retrieval method
+        :param query: query
+        :return: str
+        """
+        pass
+    
+    @abstractmethod
+    def kusto_connection(self):
+        pass
+
+    @abstractmethod
+    def kusto_open(self) -> bool:
+        pass
 
     def special_dispatch(self, data: any) -> None:
         """ 
         Response.
         @param data Event data 
         """
-        logging.info("Despachando por descripcion...")
-        options = ""
-        cont = 1
-        agent_options = {}
-        for agent, item in self.__options_dict.items():
-            agent_options[cont] = agent
-            options += f"{cont}) {item}\n"
-            cont += 1
-        query = data['dto']['text'] if data and 'dto' in data and data['dto']['text'] is not None else ''
-        prompt = CLASSIFICATION_PROMPT % options
-        logging.info(f"Query: {query},\n Options:\n{options}")
-        self.__meta_work_memory.append({"role": "system", "content": prompt})
-        
+        if data and not data['dto']['session']['team'] == "Funcionalidades":
+            logging.info("Despachando por descripcion...")
+            options = ""
+            cont = 1
+            agent_options = {}
+            for agent, item in self.__options_dict.items():
+                agent_options[cont] = agent
+                options += f"{cont}) {item}\n"
+                cont += 1
+            query = data['dto']['text'] if data and 'dto' in data and data['dto']['text'] is not None else ''
+            prompt = CLASSIFICATION_PROMPT % options
+            logging.info(f"Query: {query},\n Options:\n{options}")
+            self.__meta_work_memory.append({"role": "system", "content": prompt})
+            
 
-        user_prompt = f"""
-        Texto: "%s"
+            user_prompt = f"""
+            Texto: "%s"
 
-        Respuesta:
-        """ % query
-        self.__meta_work_memory.append({"role": "user", "content": user_prompt})
-        
-        res = self.__ai_service.generate(self.__meta_work_memory, max_tokens=10)
-        logging.info(f"Thought: {res}")
-        self.__meta_work_memory = []
-        res = self.get_text(res)
-        select_agent = None
-        compare = re.findall(r'\d+', res)
-        print(f"Compare: {compare}")
-        if len(compare) > 0:
-            compare = compare[0]
+            Respuesta:
+            """ % query
+            self.__meta_work_memory.append({"role": "user", "content": user_prompt})
+            
+            res = self.__ai_service.generate(self.__meta_work_memory, max_tokens=10)
+            logging.info(f"Thought: {res}")
+            self.__meta_work_memory = []
+            res = self.get_text(res)
+            select_agent = None
+            compare = re.findall(r'\d+', res)
+            print(f"Compare: {compare}")
+            if len(compare) > 0:
+                compare = compare[0]
+            else:
+                compare = res
+            compare = compare.strip()
+            print(f"Compare: {compare}")
+            for option in range(1, cont+1):
+                print(f"Option: {option} - Compare: {compare}")
+                if str(option) == compare:                
+                    select_agent = agent_options[option]
+                    logging.info(f"Descripcion del agente seleccionado: {select_agent}")
+                    break
+            if not select_agent:
+                logging.info("=> No se seleccionó ningun agente")
+            return select_agent
         else:
-            compare = res
-        compare = compare.strip()
-        print(f"Compare: {compare}")
-        for option in range(1, cont+1):
-            print(f"Option: {option} - Compare: {compare}")
-            if str(option) == compare:                
-                select_agent = agent_options[option]
-                logging.info(f"Descripcion del agente seleccionado: {select_agent}")
-                break
-        if not select_agent:
-            logging.info("=> No se seleccionó ningun agente")
-        return select_agent
- 
+            logging.info("Despachando por descripcion...")
+            options = ""
+            cont = 1
+
+            selection_options = {}
+
+            agent_options = {}
+            for agent, item in self.__options_dict.items():
+                agent_options[cont] = agent
+                options += f"{cont}) {item}\n"
+                cont += 1
+                selection_options[agent] = 0
+
+            query = data['dto']['text'] if data and 'dto' in data and data['dto']['text'] is not None else ''
+            prompt = CLASSIFICATION_PROMPT % options
+            logging.info(f"Query: {query},\n Options:\n{options}")
+            self.__meta_work_memory.append({"role": "system", "content": prompt})
+            
+
+            user_prompt = f"""
+            Texto: "%s"
+
+            Respuesta:
+            """ % query
+            self.__meta_work_memory.append({"role": "user", "content": user_prompt})
+
+            self.kusto_open()
+
+            attemps = 1
+            while attemps <= 3:
+                logging.info(f"Intento de selección de agente: {attemps}")
+
+                res = self.__ai_service.generate(self.__meta_work_memory, max_tokens=10)
+                logging.info(f"Thought: {res}")
+                
+                res = self.get_text(res)
+                select_agent = None
+                compare = re.findall(r'\d+', res)
+                print(f"Compare: {compare}")
+                if len(compare) > 0:
+                    compare = compare[0]
+                else:
+                    compare = res
+                compare = compare.strip()
+                print(f"Compare: {compare}")
+                for option in range(1, cont+1):
+                    print(f"Option: {option} - Compare: {compare}")
+                    if str(option) == compare:                
+                        select_agent = agent_options[option]
+                        logging.info(f"Descripcion del agente seleccionado: {select_agent}")
+                        break
+                if not select_agent:
+                    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
+
+                compare_llm = 0
+                compare_index = 0
+                idex_result = None
+
+                # Check if the selected agent is in the retrieval response
+                if major_id and major_id in select_agent:
+                    logging.info(f"Se seleccionó el agente: {select_agent} con score: {score}")
+                    exit = True
+                    continue
+                else:
+                    for ag in selection_options:
+                        if select_agent and select_agent in ag:
+                            selection_options[ag] += 1
+                            compare_llm += 1 
+                        if major_id and major_id in ag:
+                            selection_options[ag] += 1
+                            compare_index += 1
+                            idex_result = ag
+
+                attemps += 1
+
+            if attemps > 3:
+                logging.info("No se pudo seleccionar un agente, identificando el agente con mayor score...")
+                
+                if compare_llm == compare_index:
+                    logging.warning("Dando prioridad al resultado del indice")
+                    select_agent = idex_result
+                else:
+                    logging.info("Evaluando los scores de los agentes...")
+                    major_id = ""
+                    max_score = 0
+                    for ag, score in selection_options.items():
+                        if score > max_score:
+                            max_score = score
+                            major_id = ag
+                    if major_id:
+                        select_agent = major_id
+                        logging.info(f"Se seleccionó el agente: {select_agent} con score: {max_score}") 
+
+            self.kusto_close()
+            self.__meta_work_memory = []
+            return select_agent

{pbesa-4.0.21 → pbesa-4.0.22}/pbesa.egg-info/PKG-INFO

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

{pbesa-4.0.21 → pbesa-4.0.22}/setup.py

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