pbesa 4.0.12__tar.gz → 4.0.14__tar.gz

{pbesa-4.0.12 → pbesa-4.0.14}/PKG-INFO

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

{pbesa-4.0.12 → pbesa-4.0.14}/pbesa/celulas/celula_casos.py

@@ -42,13 +42,13 @@ Clasificación:
 """
 
 # Efectua la inferencia del modelo.
-def derive(service, text) -> any:
+def derive(service, text, max_tkns=4096) -> any:
     try:
         logging.info(f"Procesando: {text}")
         tmp_work_memory = []
         prompt  = PROMPT % text
         tmp_work_memory.append({"role": "user", "content": prompt})
-        res = service.generate(tmp_work_memory)
+        res = service.generate(tmp_work_memory, max_tokens=max_tkns)
         logging.info(f"Respuesta: {res}")
         if not res or res == "":
             res = text

{pbesa-4.0.12 → pbesa-4.0.14}/pbesa/celulas/celula_consultas.py

@@ -27,13 +27,13 @@ Clasificación:
 """
 
 # Efectua la inferencia del modelo.
-def derive(service, text) -> any:
+def derive(service, text, max_tkns=4096) -> any:
     try:
         logging.info(f"Procesando: {text}")
         tmp_work_memory = []
         prompt  = PROMPT % text
         tmp_work_memory.append({"role": "user", "content": prompt})
-        res = service.generate(tmp_work_memory)
+        res = service.generate(tmp_work_memory, max_tokens=max_tkns)
         logging.info(f"Respuesta: {res}")
         if not res or res == "":
             res = text

{pbesa-4.0.12 → pbesa-4.0.14}/pbesa/celulas/celula_datos_identificables.py

@@ -35,12 +35,12 @@ Clasificación:
 """
 
 # Efectua la inferencia del modelo.
-def derive(service, text) -> any:
+def derive(service, text, max_tkns=4096) -> any:
     try:
         tmp_work_memory = []
         prompt  = PROMPT % text
         tmp_work_memory.append({"role": "user", "content": prompt})
-        res = service.generate(tmp_work_memory)
+        res = service.generate(tmp_work_memory, max_tokens=max_tkns)
         logging.info(f"Procesando: {text}")
         logging.info(f"Respuesta: {res}")
         if not res or res == "":

{pbesa-4.0.12 → pbesa-4.0.14}/pbesa/celulas/celula_generar_documento.py

@@ -26,13 +26,13 @@ DEMANDA FORMATEADA:
 """
 
 # Efectua la inferencia del modelo.
-def derive(service, formato, text) -> any:
+def derive(service, formato, text, max_tkns=4096) -> any:
     try:
         logging.info(f"Procesando: {text}")
         tmp_work_memory = []
         prompt  = PROMPT % (formato, text)
         tmp_work_memory.append({"role": "user", "content": prompt})
-        res = service.generate(tmp_work_memory)
+        res = service.generate(tmp_work_memory, max_tokens=max_tkns)
         logging.info(f"Respuesta: {res}")
         if not res or res == "":
             res = text

pbesa-4.0.14/pbesa/celulas/celula_preguntas.py

@@ -0,0 +1,41 @@
+import logging
+
+PROMPT = """
+Eres un clasificador de preguntas. Tu tarea es analizar el siguiente texto y determinar si contiene preguntas.
+
+Responde únicamente con una de estas opciones:
+- CONTIENE
+- NO_CONTIENE
+
+**Definición de pregunta**: Una pregunta es una expresión que busca obtener información o aclaración sobre un tema específico. Esto incluye preguntas directas, indirectas y retóricas. No se consideran preguntas las afirmaciones o comentarios que no buscan respuesta.
+
+Ejemplos:
+- "Hola, ¿cómo estás?" → CONTIENE
+- "Buenos días, equipo" → NO_CONTIENE
+- "¿Cómo se hace este análisis?" → CONTIENE
+- "Ey, qué más" → NO_CONTIENE
+- "Necesito ayuda con el código" → NO_CONTIENE
+- "Saludos" → NO_CONTIENE
+
+Texto: "%s"
+
+Clasificación:
+"""
+
+# Efectua la inferencia del modelo.
+def derive(service, text, max_tkns=4096) -> any:
+    try:
+        logging.info(f"Procesando: {text}")
+        tmp_work_memory = []
+        prompt  = PROMPT % text
+        tmp_work_memory.append({"role": "user", "content": prompt})
+        res = service.generate(tmp_work_memory, max_tokens=max_tkns)
+        logging.info(f"Respuesta: {res}")
+        if not res or res == "":
+            res = text
+            logging.warning(f"No obtener una respuesta.")
+        return res
+    except Exception as e:
+        logging.error(f"Error al procesar: {text}")
+        logging.error(e)
+        return None

{pbesa-4.0.12 → pbesa-4.0.14}/pbesa/celulas/celula_saludos.py

@@ -23,13 +23,13 @@ Clasificación:
 """
 
 # Efectua la inferencia del modelo.
-def derive(service, text) -> any:
+def derive(service, text, max_tkns=4096) -> any:
     try:
         logging.info(f"Procesando: {text}")
         tmp_work_memory = []
         prompt  = PROMPT % text
         tmp_work_memory.append({"role": "user", "content": prompt})
-        res = service.generate(tmp_work_memory)
+        res = service.generate(tmp_work_memory, max_tokens=max_tkns)
         logging.info(f"Respuesta: {res}")
         if not res or res == "":
             res = text