sapiopycommons 2025.10.21a791__tar.gz → 2025.10.21a792__tar.gz

{sapiopycommons-2025.10.21a791 → sapiopycommons-2025.10.21a792}/PKG-INFO

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

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

{sapiopycommons-2025.10.21a791 → sapiopycommons-2025.10.21a792}/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])