sapiopycommons 2025.5.12a519__py3-none-any.whl → 2025.5.12a521__py3-none-any.whl

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]:
         """

{sapiopycommons-2025.5.12a519.dist-info → sapiopycommons-2025.5.12a521.dist-info}/METADATA

@@ -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.dist-info → sapiopycommons-2025.5.12a521.dist-info}/RECORD

@@ -1,7 +1,7 @@
 sapiopycommons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sapiopycommons/ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sapiopycommons/ai/protobuf_utils.py,sha256=aKmZtNLrrzUe_LFsdW0zYpG4VI-ZGgukyXVDFtIqfes,22656
-sapiopycommons/ai/tool_service_base.py,sha256=lJRR4jX9oRs56f9ysRbIbWe5tFOYrV6DXQlS0dbGa3M,32496
+sapiopycommons/ai/tool_service_base.py,sha256=h9XtvPKaDBCVE9a6oRVU7ZkvqJhUPO9U0VaHX76JhXM,33728
 sapiopycommons/ai/api/fielddefinitions/proto/fields_pb2.py,sha256=YcZjb_YM-XeLErM8hEC_S7vGMVGvcXAMGs2b-u5zvOE,2377
 sapiopycommons/ai/api/fielddefinitions/proto/fields_pb2.pyi,sha256=FwtXmNAf7iYGEFm4kbqb04v77jNHbZg18ZmEDhle_bU,1444
 sapiopycommons/ai/api/fielddefinitions/proto/fields_pb2_grpc.py,sha256=wPImJPdCUZNVEVoUWzsba9kGIXjEKPdUkawP5SnVyiU,932
@@ -85,7 +85,7 @@ sapiopycommons/webhook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 sapiopycommons/webhook/webhook_context.py,sha256=D793uLsb1691SalaPnBUk3rOSxn_hYLhdvkaIxjNXss,1909
 sapiopycommons/webhook/webhook_handlers.py,sha256=L0HetSm43NvA5KyW3xbLpGFh2DbAaeZJVtXIEl2fvV8,39689
 sapiopycommons/webhook/webservice_handlers.py,sha256=Y5dHx_UFWFuSqaoPL6Re-fsKYRuxvCWZ8bj6KSZ3jfM,14285
-sapiopycommons-2025.5.12a519.dist-info/METADATA,sha256=ydyu7WKWgeim8QXQl90mu2FpgluAJUDW1TnmolC5Z9U,3143
-sapiopycommons-2025.5.12a519.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-sapiopycommons-2025.5.12a519.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
-sapiopycommons-2025.5.12a519.dist-info/RECORD,,
+sapiopycommons-2025.5.12a521.dist-info/METADATA,sha256=NDAG-QmrZ4B1l_8ZVl6iaEHGA296uwg7PL5sHF6uEoA,3143
+sapiopycommons-2025.5.12a521.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+sapiopycommons-2025.5.12a521.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
+sapiopycommons-2025.5.12a521.dist-info/RECORD,,