sapiopycommons 2025.10.21a791__tar.gz → 2025.10.30a798__tar.gz

{sapiopycommons-2025.10.21a791 → sapiopycommons-2025.10.30a798}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.10.21a791
+Version: 2025.10.30a798
 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.21a791 → sapiopycommons-2025.10.30a798}/pyproject.toml

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

{sapiopycommons-2025.10.21a791 → sapiopycommons-2025.10.30a798}/src/sapiopycommons/ai/agent_service_base.py

@@ -979,37 +979,32 @@ class AgentBase(ABC):
         """
         pass
 
-    def get_credentials(self, category: str = None, host: str = None) -> ExternalCredentials:
+    def get_credentials(self, name: str | None = None, category: str | None = None) -> ExternalCredentials:
         """
         Get credentials for the given category and host.
 
+        :param name: The host name of the credentials to retrieve.
         :param category: The category of the credentials to retrieve.
-        :param host: The host for which to retrieve the credentials.
         :return: An ExternalCredentials object containing the credentials for the given category and host.
         """
         if not self.__is_setup:
             raise Exception("Cannot call this function before the agent has been set up to respond to a request.")
         # Remove leading/trailing whitespace
+        name = name.strip() if name else None
         category = category.strip() if category else None
-        host = host.strip() if host else None
 
         matching_creds: list[ExternalCredentialsPbo] = []
         for cred in self.request.external_credential:
             # Do case insensitive comparison
+            if name and cred.display_name.lower != name.lower():
+                continue
             if category and cred.category.lower() != category.lower():
                 continue
-            if host:
-                # Parse the URL to get the host and compare
-                from urllib.parse import urlparse
-                parsed_url = urlparse(cred.url)
-                if parsed_url.hostname is None or parsed_url.hostname.lower() != host.lower():
-                    continue
-
             matching_creds.append(cred)
         if len(matching_creds) == 0:
-            raise ValueError(f"No credentials found for category '{category}' and host '{host}'.")
+            raise ValueError(f"No credentials found for name '{name}' and category '{category}'.")
         if len(matching_creds) > 1:
-            raise ValueError(f"Multiple credentials found for category '{category}' and host '{host}'.")
+            raise ValueError(f"Multiple credentials found for name '{name}' and category '{category}'.")
 
         return ExternalCredentials.from_pbo(matching_creds[0])
 

{sapiopycommons-2025.10.21a791 → sapiopycommons-2025.10.30a798}/src/sapiopycommons/ai/test_client.py

@@ -126,7 +126,10 @@ class AgentOutput:
             for i, output in enumerate(self.binary_output, start=1):
                 output: list[bytes]
                 ret_val += f"\tBinary Output {i}:\n"
-                for binary in output:
+                for j, binary in enumerate(output):
+                    if j > 5:
+                        ret_val += f"\t\tAnd {len(output) - 5} more binary items...\n"
+                        break
                     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)} output(s)\n"
@@ -134,14 +137,20 @@ class AgentOutput:
                 output: list[dict[str, Any]]
                 ret_val += f"\tCSV Output {i}:\n"
                 ret_val += f"\t\tHeaders: {', '.join(output[0].keys())}\n"
-                for j, csv_row in enumerate(output):
+                for j, csv_row in enumerate(output, start=1):
+                    if j > 5:
+                        ret_val += f"\t\tAnd {len(output) - 5} more CSV rows...\n"
+                        break
                     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)} output(s)\n"
             for i, output in enumerate(self.json_output, start=1):
                 output: list[Any]
                 ret_val += f"\tJSON Output {i}:\n"
-                for json_obj in output:
+                for j, json_obj in enumerate(output, start=1):
+                    if j > 5:
+                        ret_val += f"\t\tAnd {len(output) - 5} more JSON items...\n"
+                        break
                     ret_val += f"\t\t"
                     ret_val += json.dumps(json_obj, indent=2).replace("\n", "\n\t\t") + "\n"
 
@@ -149,11 +158,17 @@ class AgentOutput:
             for i, output in enumerate(self.text_output, start=1):
                 output: list[str]
                 ret_val += f"\tText Output {i}:\n"
-                for text in output:
+                for j, text in enumerate(output, start=1):
+                    if j > 5:
+                        ret_val += f"\t\tAnd {len(output) - 5} more text items...\n"
+                        break
                     ret_val += f"\t\t{text}\n"
 
             ret_val += f"New Records: {len(self.new_records)} item(s)\n"
-            for record in self.new_records:
+            for i, record in enumerate(self.new_records, start=1):
+                if i > 5:
+                    ret_val += f"\tAnd {len(self.new_records) - 5} more new records...\n"
+                    break
                 ret_val += f"{json.dumps(record, indent=2)}\n"
 
         ret_val += f"Logs: {len(self.logs)} item(s)\n"