sapiopycommons 2025.10.10a779__py3-none-any.whl → 2025.10.13a781__py3-none-any.whl

sapiopycommons/ai/test_client.py

@@ -1,5 +1,6 @@
 import base64
 import json
+import os
 from enum import Enum
 from typing import Any
 
@@ -20,6 +21,7 @@ from sapiopycommons.ai.protoapi.plan.tool.tool_pb2_grpc import ToolServiceStub
 from sapiopycommons.ai.protoapi.session.sapio_conn_info_pb2 import SapioConnectionInfoPbo, SapioUserSecretTypePbo
 from sapiopycommons.ai.protobuf_utils import ProtobufUtils
 from sapiopycommons.general.aliases import FieldValue
+from sapiopycommons.general.time_util import TimeUtil
 
 
 class ContainerType(Enum):
@@ -63,39 +65,42 @@ class AgentOutput:
         self.new_records = []
         self.logs = []
 
-    def save_outputs(self, path: str = "test_outputs", file_extensions: list[str] | None = None) -> None:
+    def save_outputs(self, path: str = "test_outputs", subfolder: str | None = None, file_extensions: list[str] | None = None) -> None:
         """
-        Save all outputs to files in the specified directory. The files will be named according to the agent name
-        and the output type.
+        Save all outputs to files in the specified output directory.
 
         :param path: The directory to save the output files to.
+        :param subfolder: An optional subfolder within the path to save the output files to. Useful for when you are
+            calling the same agent multiple times for separate test cases.
         :param file_extensions: A list of file extensions to use for binary output files. The length of this list
             should match the number of binary outputs.
         """
         if not self:
             return
-        import os
-        os.makedirs(path, exist_ok=True)
-        if self.binary_output and (not file_extensions or len(file_extensions) != len(self.binary_output)):
+        output_path: str = os.path.join(path, self.agent_name)
+        if subfolder:
+            output_path = os.path.join(output_path, subfolder)
+        os.makedirs(output_path, exist_ok=True)
+        if self.binary_output and (file_extensions is None or len(file_extensions) != len(self.binary_output)):
             raise ValueError("File extensions must be provided for each binary output.")
         for i, output in enumerate(self.binary_output):
             ext: str = "." + file_extensions[i].lstrip(".")
             for j, entry in enumerate(output):
-                with open(os.path.join(path, self.agent_name, f"binary_output_{i}_{j}{ext}"), "wb") as f:
+                with open(os.path.join(output_path, f"binary_output_{i}_{j}{ext}"), "wb") as f:
                     f.write(entry)
         for i, output in enumerate(self.csv_output):
-            with open(os.path.join(path, self.agent_name, f"csv_output_{i}.csv"), "w", encoding="utf-8") as f:
+            with open(os.path.join(output_path, f"csv_output_{i}.csv"), "w", encoding="utf-8") as f:
                 headers = output[0].keys()
                 f.write(",".join(headers) + "\n")
                 for row in output:
                     f.write(",".join(f'"{str(row[h])}"' for h in headers) + "\n")
         for i, output in enumerate(self.json_output):
             for j, entry in enumerate(output):
-                with open(os.path.join(path, self.agent_name, f"json_output_{i}_{j}.json"), "w", encoding="utf-8") as f:
+                with open(os.path.join(output_path, f"json_output_{i}_{j}.json"), "w", encoding="utf-8") as f:
                     json.dump(entry, f, indent=2)
         for i, output in enumerate(self.text_output):
             for j, entry in enumerate(output):
-                with open(os.path.join(path, self.agent_name, f"text_output_{i}_{j}.txt"), "w", encoding="utf-8") as f:
+                with open(os.path.join(output_path, f"text_output_{i}_{j}.txt"), "w", encoding="utf-8") as f:
                     f.write(entry)
 
     def __bool__(self):
@@ -314,9 +319,11 @@ class TestClient:
         :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.
         """
+        print(f"Calling agent \"{agent_name}\"...")
         with grpc.insecure_channel(self.grpc_server_url, options=self.options) as channel:
             stub = ToolServiceStub(channel)
 
+            start = TimeUtil.now_in_millis()
             response: ProcessStepResponsePbo = stub.ProcessData(
                 ProcessStepRequestPbo(
                     sapio_user=self.connection,
@@ -330,6 +337,8 @@ class TestClient:
                     ]
                 )
             )
+            end = TimeUtil.now_in_millis()
+            print(f"Agent call completed in {(end - start) / 1000.:.3f} seconds")
 
             results = AgentOutput(agent_name)
 

{sapiopycommons-2025.10.10a779.dist-info → sapiopycommons-2025.10.13a781.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.10.10a779
+Version: 2025.10.13a781
 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.10a779.dist-info → sapiopycommons-2025.10.13a781.dist-info}/RECORD

@@ -6,7 +6,7 @@ sapiopycommons/ai/external_credentials.py,sha256=40AI7VtHf6PzvwJLR_mZemUCrfAUvC-
 sapiopycommons/ai/protobuf_utils.py,sha256=cBjbxoFAwU02kNUxEce95WnMU2CMuDD-qFaeWgvQJMQ,24599
 sapiopycommons/ai/request_validation.py,sha256=TD2EUi_G5cy1OOVK1AmY2SQc3PEoAKGWs2pT8Qnp2Oo,25092
 sapiopycommons/ai/server.py,sha256=gutSskn_Fenq1uz0DDMvjx4QVFiKt2WVEP3-01a69eU,6384
-sapiopycommons/ai/test_client.py,sha256=jHIKaBy2jta2UCcci6z0dvX_BZZuiV_0z2HPr9dEI_s,18897
+sapiopycommons/ai/test_client.py,sha256=aiS58O_A3KSgRCzDT61iKyVXy9F4UY7xUyGQu5E7bTw,19432
 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
@@ -106,7 +106,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.10a779.dist-info/METADATA,sha256=5rW00M6gy8crugEHSZZCOCNdoe-Bi8gbVBwfOqoNk0U,3143
-sapiopycommons-2025.10.10a779.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-sapiopycommons-2025.10.10a779.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
-sapiopycommons-2025.10.10a779.dist-info/RECORD,,
+sapiopycommons-2025.10.13a781.dist-info/METADATA,sha256=Ex7Fc6UXveEGSs8qgYNAUSzjwKhsBOZU3jyOzIWKlyY,3143
+sapiopycommons-2025.10.13a781.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+sapiopycommons-2025.10.13a781.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
+sapiopycommons-2025.10.13a781.dist-info/RECORD,,