sapiopycommons 2025.5.12a519__tar.gz → 2025.5.12a521__tar.gz

{sapiopycommons-2025.5.12a519 → sapiopycommons-2025.5.12a521}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.5.12a519
+Version: 2025.5.12a521
 Summary: Official Sapio Python API Utilities Package
 Project-URL: Homepage, https://github.com/sapiosciences
 Author-email: Jonathan Steck <jsteck@sapiosciences.com>, Yechen Qiao <yqiao@sapiosciences.com>

{sapiopycommons-2025.5.12a519 → sapiopycommons-2025.5.12a521}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "sapiopycommons"
-version='2025.05.12a519'
+version='2025.05.12a521'
 authors = [
     { name="Jonathan Steck", email="jsteck@sapiosciences.com" },
     { name="Yechen Qiao", email="yqiao@sapiosciences.com" },

{sapiopycommons-2025.5.12a519 → sapiopycommons-2025.5.12a521}/src/sapiopycommons/ai/tool_service_base.py

@@ -1,12 +1,14 @@
 from __future__ import annotations
 
 import json
+import logging
 import traceback
 from abc import abstractmethod, ABC
+from logging import Logger
 from typing import Any, Iterable, Sequence
 
 from grpc import ServicerContext
-from sapiopylib.rest.User import SapioUser
+from sapiopylib.rest.User import SapioUser, ensure_logger_initialized
 from sapiopylib.rest.pojo.datatype.FieldDefinition import AbstractVeloxFieldDefinition
 
 from sapiopycommons.ai.api.fielddefinitions.proto.fields_pb2 import FieldValueMapPbo, FieldValuePbo
@@ -305,8 +307,8 @@ class ToolServiceBase(ToolServiceServicer, ABC):
             tool.setup(user, request, context)
             results: list[SapioToolResult] = tool.run(user)
             return True, results, tool.logs
-        except Exception:
-            tool.log_message(traceback.format_exc())
+        except Exception as e:
+            tool.log_exception("Exception occurred during tool execution.", e)
             return False, [], tool.logs
 
 
@@ -321,6 +323,7 @@ class ToolBase(ABC):
     outputs: list[ToolOutputDetailsPbo]
     configs: list[VeloxFieldDefPbo]
     logs: list[str]
+    logger: Logger
 
     user: SapioUser
     request: ProcessStepRequestPbo
@@ -341,6 +344,8 @@ class ToolBase(ABC):
         self.outputs = []
         self.configs = []
         self.logs = []
+        self.logger = logging.getLogger(f"ToolBase.{self.name}")
+        ensure_logger_initialized(self.logger)
 
     def setup(self, user: SapioUser, request: ProcessStepRequestPbo, context: ServicerContext) -> None:
         """
@@ -624,16 +629,50 @@ class ToolBase(ABC):
         """
         pass
 
-    def log_message(self, message: str, verbose: bool = False) -> None:
+    def log_info(self, message: str) -> None:
+        """
+        Log an info message for this tool. If verbose logging is enabled, this message will be included in the logs
+        returned to the caller.
+        .
+
+        :param message: The message to log.
+        """
+        msg = f"INFO: {self.name}: {message}"
+        if self.verbose_logging:
+            self.logs.append(msg)
+        self.logger.info(msg)
+
+    def log_warning(self, message: str) -> None:
+        """
+        Log a warning message for this tool. This message will be included in the logs returned to the caller.
+
+        :param message: The message to log.
+        """
+        msg = f"WARNING: {self.name}: {message}"
+        self.logs.append(msg)
+        self.logger.warning(msg)
+
+    def log_error(self, message: str) -> None:
+        """
+        Log an error message for this tool. This message will be included in the logs returned to the caller.
+
+        :param message: The message to log.
+        """
+        msg = f"ERROR: {self.name}: {message}"
+        self.logs.append(msg)
+        self.logger.error(msg)
+
+    def log_exception(self, message: str, e: Exception) -> None:
         """
-        Log a message for this tool. This message will be included in the logs returned to the caller.
+        Log an exception for this tool. This message will be included in the logs returned to the caller.
 
         :param message: The message to log.
-        :param verbose: If true, the message will only be logged if verbose logging is enabled. If false, the message
-            will be logged regardless of the verbose logging setting.
+        :param e: The exception to log.
         """
-        if not verbose or self.verbose_logging:
-            self.logs.append(f"{self.name}: {message}")
+        msg = f"EXCEPTION: {self.name}: {message} - {e}"
+        self.logs.append(msg)
+        msg += "\n" + traceback.format_exc()
+        self.logger.error(msg)
 
     def get_input_binary(self, index: int = 0) -> list[bytes]:
         """