sapiopycommons 2025.10.10a778__tar.gz → 2025.10.10a779__tar.gz

{sapiopycommons-2025.10.10a778 → sapiopycommons-2025.10.10a779}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.10.10a778
+Version: 2025.10.10a779
 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.10a778 → sapiopycommons-2025.10.10a779}/pyproject.toml

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

{sapiopycommons-2025.10.10a778 → sapiopycommons-2025.10.10a779}/src/sapiopycommons/ai/test_client.py

@@ -43,6 +43,8 @@ class AgentOutput:
     status: str
     message: str
 
+    # Outputs are lists of lists, where the outer lists are the different outputs of the tool, and the inner lists
+    # are the entries for that output.
     binary_output: list[list[bytes]]
     csv_output: list[list[dict[str, Any]]]
     json_output: list[list[dict[str, Any]]]
@@ -61,10 +63,51 @@ class AgentOutput:
         self.new_records = []
         self.logs = []
 
+    def save_outputs(self, path: str = "test_outputs", 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.
+
+        :param path: The directory to save the output files to.
+        :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)):
+            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:
+                    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:
+                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:
+                    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:
+                    f.write(entry)
+
     def __bool__(self):
+        """
+        Return True if the agent call was successful, False otherwise.
+        """
         return self.status == "Success"
 
     def __str__(self):
+        """
+        Return a string representing a summary of the agent output.
+        """
         ret_val: str = f"{self.agent_name} Output:\n"
         ret_val += f"\tStatus: {self.status}\n"
         ret_val += f"\tMessage: {self.message}\n"