sapiopycommons 2025.9.8a728__tar.gz → 2025.9.8a730__tar.gz

{sapiopycommons-2025.9.8a728 → sapiopycommons-2025.9.8a730}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.9.8a728
+Version: 2025.9.8a730
 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.9.8a728 → sapiopycommons-2025.9.8a730}/pyproject.toml

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

{sapiopycommons-2025.9.8a728 → sapiopycommons-2025.9.8a730}/src/sapiopycommons/ai/server.py

@@ -1,4 +1,7 @@
+from __future__ import annotations
+
 import asyncio
+from argparse import ArgumentParser
 from typing import Any
 
 import grpc
@@ -16,12 +19,30 @@ class SapioGrpcServer:
     """
     port: int
     options: list[tuple[str, Any]]
+    debug_mode: bool
     _converter_services: list[ConverterServiceServicer]
     _script_services: list[ScriptServiceServicer]
     _tool_services: list[ToolServiceServicer]
 
-    def __init__(self, port: int = 50051, message_mb_size: int = 1024, options: list[tuple[str, Any]] | None = None) \
-            -> None:
+    @staticmethod
+    def args_parser() -> ArgumentParser:
+        """
+        Create an argument parser for the gRPC server.
+
+        :return: The argument parser.
+        """
+        parser = ArgumentParser()
+        parser.add_argument("--debug_mode", "-d", action="store_true", type=bool)
+        parser.add_argument("--port", "-p", default=50051, type=int)
+        parser.add_argument("--message_mb_size", "-s", default=1024, type=int)
+        return parser
+
+    @staticmethod
+    def from_args(options: list[tuple[str, Any]] | None = None) -> SapioGrpcServer:
+        return SapioGrpcServer(options=options, **vars(SapioGrpcServer.args_parser().parse_args()))
+
+    def __init__(self, port: int = 50051, message_mb_size: int = 1024, debug_mode: bool = False,
+                 options: list[tuple[str, Any]] | None = None) -> None:
         """
         Initialize the gRPC server with the specified port and message size.
 
@@ -29,6 +50,7 @@ class SapioGrpcServer:
         :param message_mb_size: The maximum size of a message in megabytes.
         :param options: Additional gRPC server options to set. This should be a list of tuples where the first item is
             the option name and the second item is the option value.
+        :param debug_mode: Sets the debug mode for services.
         """
         if isinstance(port, str):
             port = int(port)
@@ -39,6 +61,9 @@ class SapioGrpcServer:
         ]
         if options:
             self.options.extend(options)
+        self.debug_mode = debug_mode
+        if debug_mode:
+            print("Debug mode is enabled.")
         self._converter_services = []
         self._script_services = []
         self._tool_services = []
@@ -65,6 +90,7 @@ class SapioGrpcServer:
 
         :param service: The tool service to register with the server.
         """
+        service.debug_mode = self.debug_mode
         self._tool_services.append(service)
 
     def start(self) -> None:

{sapiopycommons-2025.9.8a728 → sapiopycommons-2025.9.8a730}/src/sapiopycommons/ai/tool_service_base.py

@@ -217,6 +217,8 @@ class ToolServiceBase(ToolServiceServicer, ABC):
     A base class for implementing a tool service. Subclasses should implement the register_tools method to register
     their tools with the service.
     """
+    debug_mode: bool = False
+
     def GetToolDetails(self, request: ToolDetailsRequestPbo, context: ServicerContext) -> ToolDetailsResponsePbo:
         try:
             # Get the tool details from the registered tools.
@@ -327,7 +329,7 @@ class ToolServiceBase(ToolServiceServicer, ABC):
         tool: ToolBase = registered_tools[find_tool]()
         try:
             # Setup the tool with details from the request.
-            tool.setup(user, request, context)
+            tool.setup(user, request, context, self.debug_mode)
             # Validate that the provided inputs match the tool's expected inputs.
             if len(request.input) != len(tool.input_configs):
                 msg: str = (f"Expected {len(tool.input_configs)} inputs for this tool, but got {len(request.input)} "
@@ -350,8 +352,16 @@ class ToolServiceBase(ToolServiceServicer, ABC):
             tool.log_exception("Exception occurred during tool execution.", e)
             return False, str(e), [], tool.logs
         finally:
-            # Clean up any temporary files created by the tool.
-            tool.temp_data.cleanup()
+            # Clean up any temporary files created by the tool. If in debug mode, then log the files instead
+            # so that they can be manually inspected.
+            if self.debug_mode:
+                print("Temporary files/directories created during tool execution:")
+                for directory in tool.temp_data.directories:
+                    print(f"\tDirectory: {directory}")
+                for file in tool.temp_data.files:
+                    print(f"\tFile: {file}")
+            else:
+                tool.temp_data.cleanup()
 
 
 class ToolBase(ABC):
@@ -370,6 +380,7 @@ class ToolBase(ABC):
     logs: list[str]
     logger: Logger
     verbose_logging: bool
+    debug_mode: bool
 
     temp_data: TempFileHandler
 
@@ -415,7 +426,7 @@ class ToolBase(ABC):
         self.logger = logging.getLogger(f"ToolBase.{self._name}")
         ensure_logger_initialized(self.logger)
 
-    def setup(self, user: SapioUser, request: ProcessStepRequestPbo, context: ServicerContext) -> None:
+    def setup(self, user: SapioUser, request: ProcessStepRequestPbo, context: ServicerContext, debug_mode: bool) -> None:
         """
         Setup the tool with the user, request, and context. This method can be overridden by subclasses to perform
         additional setup.
@@ -424,11 +435,14 @@ class ToolBase(ABC):
             system.
         :param request: The request object containing the input data.
         :param context: The gRPC context.
+        :param debug_mode: If true, the tool should run in debug mode, providing additional logging and not cleaning
+            up temporary files.
         """
         self.user = user
         self.request = request
         self.context = context
         self.verbose_logging = request.verbose_logging
+        self.debug_mode = debug_mode
 
     def add_input(self, container_type: ContainerType, content_type: str, display_name: str, description: str,
                   structure_example: str | bytes | None = None, validation: str | None = None,