sapiopycommons 2025.10.2a773__py3-none-any.whl → 2025.10.9a776__py3-none-any.whl

sapiopycommons/ai/server.py

@@ -12,7 +12,7 @@ from sapiopycommons.ai.protoapi.plan.converter.converter_pb2_grpc import add_Con
 from sapiopycommons.ai.protoapi.plan.script.script_pb2_grpc import add_ScriptServiceServicer_to_server, \
     ScriptServiceServicer
 from sapiopycommons.ai.protoapi.plan.tool.tool_pb2_grpc import add_ToolServiceServicer_to_server, ToolServiceServicer
-from sapiopycommons.ai.tool_service_base import ToolServiceBase
+from sapiopycommons.ai.agent_service_base import AgentServiceBase
 
 
 class SapioGrpcServer:
@@ -24,7 +24,7 @@ class SapioGrpcServer:
     debug_mode: bool
     _converter_services: list[ConverterServiceServicer]
     _script_services: list[ScriptServiceServicer]
-    _tool_services: list[ToolServiceServicer]
+    _agent_services: list[ToolServiceServicer]
 
     @staticmethod
     def args_parser() -> ArgumentParser:
@@ -68,7 +68,7 @@ class SapioGrpcServer:
             print("Debug mode is enabled.")
         self._converter_services = []
         self._script_services = []
-        self._tool_services = []
+        self._agent_services = []
 
     def update_message_size(self, message_mb_size: int) -> None:
         """
@@ -97,22 +97,22 @@ class SapioGrpcServer:
         """
         self._script_services.append(service)
 
-    def add_tool_service(self, service: ToolServiceBase) -> None:
+    def add_agent_service(self, service: AgentServiceBase) -> None:
         """
-        Add a tool service to the gRPC server.
+        Add an agent service to the gRPC server.
 
-        :param service: The tool service to register with the server.
+        :param service: The agent service to register with the server.
         """
         service.debug_mode = self.debug_mode
-        self._tool_services.append(service)
+        self._agent_services.append(service)
 
     def start(self) -> None:
         """
         Start the gRPC server for the provided servicers.
         """
-        if not (self._converter_services or self._script_services or self._tool_services):
+        if not (self._converter_services or self._script_services or self._agent_services):
             raise ValueError("No services have been added to the server. Use add_converter_service, add_script_service,"
-                             "or add_tool_service to register a service before starting the server.")
+                             "or add_agent_service to register a service before starting the server.")
 
         async def serve():
             server = grpc.aio.server(options=self.options)
@@ -123,8 +123,8 @@ class SapioGrpcServer:
             for service in self._script_services:
                 print(f"Registering Script service: {service.__class__.__name__}")
                 add_ScriptServiceServicer_to_server(service, server)
-            for service in self._tool_services:
-                print(f"Registering Tool service: {service.__class__.__name__}")
+            for service in self._agent_services:
+                print(f"Registering Agent service: {service.__class__.__name__}")
                 add_ToolServiceServicer_to_server(service, server)
 
             from grpc_health.v1 import health_pb2, health_pb2_grpc

sapiopycommons/ai/test_client.py

@@ -33,27 +33,27 @@ class ContainerType(Enum):
 
 
 # FR-47422: Created class.
-class ToolOutput:
+class AgentOutput:
     """
-    A class for holding the output of a TestClient that calls a ToolService. ToolOutput objects an be
-    printed to show the output of the tool in a human-readable format.
+    A class for holding the output of a TestClient that calls an AgentService. AgentOutput objects an be
+    printed to show the output of the agent in a human-readable format.
     """
-    tool_name: str
+    agent_name: str
 
     status: str
     message: str
 
     binary_output: list[list[bytes]]
     csv_output: list[list[dict[str, Any]]]
-    json_output: list[list[Any]]
+    json_output: list[list[dict[str, Any]]]
     text_output: list[list[str]]
 
     new_records: list[dict[str, FieldValue]]
 
     logs: list[str]
 
-    def __init__(self, tool_name: str):
-        self.tool_name = tool_name
+    def __init__(self, agent_name: str):
+        self.agent_name = agent_name
         self.binary_output = []
         self.csv_output = []
         self.json_output = []
@@ -65,20 +65,20 @@ class ToolOutput:
         return self.status == "Success"
 
     def __str__(self):
-        ret_val: str = f"{self.tool_name} Output:\n"
+        ret_val: str = f"{self.agent_name} Output:\n"
         ret_val += f"\tStatus: {self.status}\n"
         ret_val += f"\tMessage: {self.message}\n"
         ret_val += "-" * 25 + "\n"
 
         if self.status == "Success":
-            ret_val += f"Binary Output: {sum(len(x) for x in self.binary_output)} item(s) across {len(self.binary_output)} outputs\n"
+            ret_val += f"Binary Output: {sum(len(x) for x in self.binary_output)} item(s) across {len(self.binary_output)} output(s)\n"
             for i, output in enumerate(self.binary_output, start=1):
                 output: list[bytes]
                 ret_val += f"\tBinary Output {i}:\n"
                 for binary in output:
                     ret_val += f"\t\t{len(binary)} byte(s): {binary[:50]}...\n"
 
-            ret_val += f"CSV Output: {sum(len(x) for x in self.csv_output)} item(s) across {len(self.csv_output)} outputs\n"
+            ret_val += f"CSV Output: {sum(len(x) for x in self.csv_output)} item(s) across {len(self.csv_output)} output(s)\n"
             for i, output in enumerate(self.csv_output, start=1):
                 output: list[dict[str, Any]]
                 ret_val += f"\tCSV Output {i}:\n"
@@ -86,7 +86,7 @@ class ToolOutput:
                 for j, csv_row in enumerate(output):
                     ret_val += f"\t\t{j}: {', '.join(f'{v}' for k, v in csv_row.items())}\n"
 
-            ret_val += f"JSON Output: {sum(len(x) for x in self.json_output)} item(s) across {len(self.json_output)} outputs\n"
+            ret_val += f"JSON Output: {sum(len(x) for x in self.json_output)} item(s) across {len(self.json_output)} output(s)\n"
             for i, output in enumerate(self.json_output, start=1):
                 output: list[Any]
                 ret_val += f"\tJSON Output {i}:\n"
@@ -94,7 +94,7 @@ class ToolOutput:
                     ret_val += f"\t\t"
                     ret_val += json.dumps(json_obj, indent=2).replace("\n", "\n\t\t") + "\n"
 
-            ret_val += f"Text Output: {sum(len(x) for x in self.text_output)} item(s) across {len(self.text_output)} outputs\n"
+            ret_val += f"Text Output: {sum(len(x) for x in self.text_output)} item(s) across {len(self.text_output)} output(s)\n"
             for i, output in enumerate(self.text_output, start=1):
                 output: list[str]
                 ret_val += f"\tText Output {i}:\n"
@@ -113,7 +113,7 @@ class ToolOutput:
 
 class TestClient:
     """
-    A client for testing a ToolService.
+    A client for testing an AgentService.
     """
     grpc_server_url: str
     options: list[tuple[str, Any]] | None
@@ -236,6 +236,7 @@ class TestClient:
         """
         Helper method for adding inputs to the next request.
         """
+        container: StepItemContainerPbo | None = None
         match container_type:
             # The content type doesn't matter when we're just testing.
             case ContainerType.BINARY:
@@ -246,27 +247,29 @@ class TestClient:
                 container = StepItemContainerPbo(content_type=ContentTypePbo(), json_container=items)
             case ContainerType.TEXT:
                 container = StepItemContainerPbo(content_type=ContentTypePbo(), text_container=items)
+            case _:
+                raise ValueError(f"Unsupported container type: {container_type}")
         self._request_inputs.append(container)
 
     def get_service_details(self) -> ToolDetailsResponsePbo:
         """
-        Get the details of the tools from the server.
+        Get the details of the agents from the server.
 
-        :return: A ToolDetailsResponsePbo object containing the details of the tool service.
+        :return: A ToolDetailsResponsePbo object containing the details of the agent service.
         """
         with grpc.insecure_channel(self.grpc_server_url, options=self.options) as channel:
             stub = ToolServiceStub(channel)
             return stub.GetToolDetails(ToolDetailsRequestPbo(sapio_conn_info=self.connection))
 
-    def call_tool(self, tool_name: str, is_verbose: bool = True, is_dry_run: bool = False) -> ToolOutput:
+    def call_agent(self, agent_name: str, is_verbose: bool = True, is_dry_run: bool = False) -> AgentOutput:
         """
-        Send the request to the tool service for a particular tool name. This will send all the inputs that have been
+        Send the request to the agent service for a particular agent name. This will send all the inputs that have been
         added using the add_X_input functions.
 
-        :param tool_name: The name of the tool to call on the server.
-        :param is_verbose: If True, the tool will log verbosely.
-        :param is_dry_run: If True, the tool will not be executed, but the request will be validated.
-        :return: A ToolOutput object containing the results of the tool service call.
+        :param agent_name: The name of the agent to call on the server.
+        :param is_verbose: If True, the agent will log verbosely.
+        :param is_dry_run: If True, the agent will not be executed, but the request will be validated.
+        :return: An AgentOutput object containing the results of the agent service call.
         """
         with grpc.insecure_channel(self.grpc_server_url, options=self.options) as channel:
             stub = ToolServiceStub(channel)
@@ -274,7 +277,7 @@ class TestClient:
             response: ProcessStepResponsePbo = stub.ProcessData(
                 ProcessStepRequestPbo(
                     sapio_user=self.connection,
-                    tool_name=tool_name,
+                    tool_name=agent_name,
                     config_field_values=self._config_fields,
                     dry_run=is_dry_run,
                     verbose_logging=is_verbose,
@@ -285,7 +288,7 @@ class TestClient:
                 )
             )
 
-            results = ToolOutput(tool_name)
+            results = AgentOutput(agent_name)
 
             match response.status:
                 case ProcessStepResponseStatusPbo.SUCCESS:

sapiopycommons/files/temp_files.py

@@ -12,7 +12,7 @@ class TempFileHandler:
     directories: list[str]
     files: list[str]
 
-    def __init__(self) -> None:
+    def __init__(self):
         self.directories = []
         self.files = []
 

{sapiopycommons-2025.10.2a773.dist-info → sapiopycommons-2025.10.9a776.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.10.2a773
+Version: 2025.10.9a776
 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.10.2a773.dist-info → sapiopycommons-2025.10.9a776.dist-info}/RECORD

@@ -1,11 +1,14 @@
 sapiopycommons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sapiopycommons/ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sapiopycommons/ai/agent_service_base.py,sha256=52_QRjbMi6XYHC3oakX3mO5TOh6w8hieTbxzfKbFZWw,54600
 sapiopycommons/ai/converter_service_base.py,sha256=HiUXmwqv1STgyQeF9_eTFXzjIFXp5-NJ7sEhMpV3aAU,6351
 sapiopycommons/ai/protobuf_utils.py,sha256=cBjbxoFAwU02kNUxEce95WnMU2CMuDD-qFaeWgvQJMQ,24599
-sapiopycommons/ai/request_validation.py,sha256=AOpLoXYRnKf0h7MRNj6ev_rfcRxXREzQgBcbpBp5loY,24014
-sapiopycommons/ai/server.py,sha256=1ksOax2cab9MH7ESXQ5dwVNZcBKiWpNVhkzO4KkFTSg,6370
-sapiopycommons/ai/test_client.py,sha256=Um93jXmIx0YaHf-YbV5NSamPTHveJ0kU_UaAfOApnZg,16342
-sapiopycommons/ai/tool_service_base.py,sha256=_KHd9Y2C8gg3Ad-4ubP9fMf5_kMC41k5LZh8EfAW6MU,54435
+sapiopycommons/ai/request_validation.py,sha256=TD2EUi_G5cy1OOVK1AmY2SQc3PEoAKGWs2pT8Qnp2Oo,25092
+sapiopycommons/ai/server.py,sha256=gutSskn_Fenq1uz0DDMvjx4QVFiKt2WVEP3-01a69eU,6384
+sapiopycommons/ai/test_client.py,sha256=C35Z9m4wCQhWPsX_p59VD8eAHP1OkZoJjnUVIdxSmZA,16546
+sapiopycommons/ai/protoapi/externalcredentials/external_credentials_pb2.py,sha256=mEonoj6Iq-AyvO4m3YsPYu85aZfD1E0a0cL8B9yPfEo,2481
+sapiopycommons/ai/protoapi/externalcredentials/external_credentials_pb2.pyi,sha256=sfExq8fFwIwFxCpV50ytdxW5ePNBjJBr_80_trq_JQw,1658
+sapiopycommons/ai/protoapi/externalcredentials/external_credentials_pb2_grpc.py,sha256=TNS1CB_QGBSa1YU9sYR_hF-pmBwv2GpxjaNQoM_r9iU,948
 sapiopycommons/ai/protoapi/fielddefinitions/fields_pb2.py,sha256=8tKXwLXcqFGdQHHSEBSi6Fd7dcaCFoOqmhjzqhenb_M,2372
 sapiopycommons/ai/protoapi/fielddefinitions/fields_pb2.pyi,sha256=FwtXmNAf7iYGEFm4kbqb04v77jNHbZg18ZmEDhle_bU,1444
 sapiopycommons/ai/protoapi/fielddefinitions/fields_pb2_grpc.py,sha256=uO25bcnfGqXpP4ggUur54Nr73Wj-DGWftExzLNcxdHI,931
@@ -24,14 +27,14 @@ sapiopycommons/ai/protoapi/plan/converter/converter_pb2_grpc.py,sha256=6T5FCmT_v
 sapiopycommons/ai/protoapi/plan/item/item_container_pb2.py,sha256=VIXmIw8-9jtH7peJZ16BEmGDfaVjjxTKhmlfcHWT82M,3863
 sapiopycommons/ai/protoapi/plan/item/item_container_pb2.pyi,sha256=bbPNQDwfFDd7_S7yU99Co1O7sRhxjle-RM0_nK8jgjc,4246
 sapiopycommons/ai/protoapi/plan/item/item_container_pb2_grpc.py,sha256=1NzBWBUINC0Rk-NGYZ-97BVKvVUxcn_I44IjQO2h-nQ,932
-sapiopycommons/ai/protoapi/plan/script/script_pb2.py,sha256=swyahlxM7fKm-RqtgMd1Czqd1JrbXRw3sYqFTkjbbyM,5144
-sapiopycommons/ai/protoapi/plan/script/script_pb2.pyi,sha256=GqlqLf9UX_B5vCR9cm-VoYLHFJNmZjJafcPJ6EDa8Qw,6192
+sapiopycommons/ai/protoapi/plan/script/script_pb2.py,sha256=4TN7NlJEENAVqwfymoEDj5hhHYxHLdrHdpRUVgzVG8A,5621
+sapiopycommons/ai/protoapi/plan/script/script_pb2.pyi,sha256=l250ak1sOfm7Ue-PyBQC-7j7-p4DeBxltnOzwDsQIYc,6870
 sapiopycommons/ai/protoapi/plan/script/script_pb2_grpc.py,sha256=ginPYpsiI6U6dB6gfWHJM8LWjSHUgcqz_gR7Dmdsyck,6864
 sapiopycommons/ai/protoapi/plan/tool/entry_pb2.py,sha256=yNyyyVvz94ewjGaHw3t0pUP-KH7ACg5hLC0QzrsFPis,2130
 sapiopycommons/ai/protoapi/plan/tool/entry_pb2.pyi,sha256=2vI0WSy0KGFHDewMSRyX5rUkmmmSaMBLvFO7txqA6Ug,2354
 sapiopycommons/ai/protoapi/plan/tool/entry_pb2_grpc.py,sha256=i24BfJEt7LtyROxHVEyS9RLYqLZKPvyJMKRFZj6NUTg,923
-sapiopycommons/ai/protoapi/plan/tool/tool_pb2.py,sha256=4SqqRZAyxTnR86irNuuOVRvo3r8MFqgVkkqpHwu3uSY,8563
-sapiopycommons/ai/protoapi/plan/tool/tool_pb2.pyi,sha256=fVx33ImMoZWVDSFLZWs6uukjRzLfY7OI9mUhDUS2-8g,17518
+sapiopycommons/ai/protoapi/plan/tool/tool_pb2.py,sha256=aLD5j60lWUxhHW8b1AoUOClLOjFPjEMmM1KrY337-gI,9095
+sapiopycommons/ai/protoapi/plan/tool/tool_pb2.pyi,sha256=HGNou2THKPk8o9G_ZPvbGXDb6cBnv8L1R_l10i9ceHI,18441
 sapiopycommons/ai/protoapi/plan/tool/tool_pb2_grpc.py,sha256=CLSGEgSpN4D6wcE-P_5lzb2Nttjr4XjumxGUrZQSip0,6915
 sapiopycommons/ai/protoapi/session/sapio_conn_info_pb2.py,sha256=MhWzTyJz3xZgpdW8_LCVKuzKXT0cv6iHMRB-UNCUNzM,2094
 sapiopycommons/ai/protoapi/session/sapio_conn_info_pb2.pyi,sha256=vLYA8Tkzq2AwgVadoUp5vAg4HgGlgga0kzeS3e_XkCQ,1621
@@ -69,7 +72,7 @@ sapiopycommons/files/file_text_converter.py,sha256=Gaj_divTiKXWd6flDOgrxNXpcn9fD
 sapiopycommons/files/file_util.py,sha256=WBA3FYG8R2HtfxjWSzQhZKW6_1s6JSxTo9lk3SeNDu8,37140
 sapiopycommons/files/file_validator.py,sha256=ryg22-93csmRO_Pv0ZpWphNkB74xWZnHyJ23K56qLj0,28761
 sapiopycommons/files/file_writer.py,sha256=hACVl0duCjP28gJ1NPljkjagNCLod0ygUlPbvUmRDNM,17605
-sapiopycommons/files/temp_files.py,sha256=i-xgh4puYLtREI1ZtKq8y5q1lRWQX_UrVyJd2yQO0hw,3183
+sapiopycommons/files/temp_files.py,sha256=s2sGvn9uh2dTI8AVAQJWOf6RAZ0xZs7DSccCi4AGmlw,3175
 sapiopycommons/flowcyto/flow_cyto.py,sha256=B6DFquLi-gcWfJWyP4vYfwTXXJKl6O9W5-k8FzkM0Oo,2610
 sapiopycommons/flowcyto/flowcyto_data.py,sha256=mYKFuLbtpJ-EsQxLGtu4tNHVlygTxKixgJxJqD68F58,2596
 sapiopycommons/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -102,7 +105,7 @@ sapiopycommons/webhook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 sapiopycommons/webhook/webhook_context.py,sha256=D793uLsb1691SalaPnBUk3rOSxn_hYLhdvkaIxjNXss,1909
 sapiopycommons/webhook/webhook_handlers.py,sha256=7o_wXOruhT9auNh8OfhJAh4WhhiPKij67FMBSpGPICc,39939
 sapiopycommons/webhook/webservice_handlers.py,sha256=cvW6Mk_110BzYqkbk63Kg7jWrltBCDALOlkJRu8h4VQ,14300
-sapiopycommons-2025.10.2a773.dist-info/METADATA,sha256=OSVEDDTYHDZMjMAbq5ptDU5Nsu4uZ774wYZqnFIZ7Ww,3142
-sapiopycommons-2025.10.2a773.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-sapiopycommons-2025.10.2a773.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
-sapiopycommons-2025.10.2a773.dist-info/RECORD,,
+sapiopycommons-2025.10.9a776.dist-info/METADATA,sha256=5yZk5DAAW1IHX_9fduVcHOOjludn5fBGykcvM0i4uE4,3142
+sapiopycommons-2025.10.9a776.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+sapiopycommons-2025.10.9a776.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
+sapiopycommons-2025.10.9a776.dist-info/RECORD,,