sapiopycommons 2025.10.9a776__tar.gz → 2025.10.10a778__tar.gz

{sapiopycommons-2025.10.9a776 → sapiopycommons-2025.10.10a778}/PKG-INFO

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

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

{sapiopycommons-2025.10.9a776 → sapiopycommons-2025.10.10a778}/src/sapiopycommons/ai/agent_service_base.py

@@ -17,6 +17,8 @@ from grpc import ServicerContext
 from sapiopylib.rest.User import SapioUser, ensure_logger_initialized
 from sapiopylib.rest.pojo.datatype.FieldDefinition import AbstractVeloxFieldDefinition
 
+from sapiopycommons.ai.external_credentials import ExternalCredentials
+from sapiopycommons.ai.protoapi.externalcredentials.external_credentials_pb2 import ExternalCredentialsPbo
 from sapiopycommons.ai.protoapi.fielddefinitions.fields_pb2 import FieldValueMapPbo, FieldValuePbo
 from sapiopycommons.ai.protoapi.fielddefinitions.velox_field_def_pb2 import VeloxFieldDefPbo, FieldTypePbo, \
     SelectionPropertiesPbo, IntegerPropertiesPbo, DoublePropertiesPbo, BooleanPropertiesPbo, StringPropertiesPbo, \
@@ -278,7 +280,7 @@ class AgentServiceBase(ToolServiceServicer, ABC):
         :param info: The SapioConnectionInfo proto object.
         :param timeout_seconds: The request timeout for calls made from this user object.
         """
-        user = SapioUser(info.webservice_url, True, timeout_seconds, guid=info.app_guid)
+        user = SapioUser(info.webservice_url.rstrip("/"), True, timeout_seconds, guid=info.app_guid)
         match info.secret_type:
             case SapioUserSecretTypePbo.SESSION_TOKEN:
                 user.api_token = info.secret
@@ -876,6 +878,38 @@ class AgentBase(ABC):
         """
         pass
 
+    def get_credentials(self, category: str = None, host: str = None) -> ExternalCredentials:
+        """
+        Get credentials for the given category and host.
+
+        :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.
+        """
+        # Remove leading/trailing whitespace
+        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 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}'.")
+        if len(matching_creds) > 1:
+            raise ValueError(f"Multiple credentials found for category '{category}' and host '{host}'.")
+
+        return ExternalCredentials(matching_creds[0])
+
     def call_subprocess(self,
                         args: str | bytes | PathLike[str] | PathLike[bytes] | Sequence[str | bytes | PathLike[str] | PathLike[bytes]],
                         cwd: str | bytes | PathLike[str] | PathLike[bytes] | None = None,

sapiopycommons-2025.10.10a778/src/sapiopycommons/ai/external_credentials.py

@@ -0,0 +1,37 @@
+from sapiopycommons.ai.protoapi.externalcredentials.external_credentials_pb2 import ExternalCredentialsPbo
+
+
+class ExternalCredentials:
+    """
+    A class representing external credentials.
+    """
+    id: str
+    display_name: str
+    description: str
+    category: str
+    url: str
+    username: str
+    password: str
+    token: str
+    custom_fields: dict[str, str]
+
+    def __init__(self, pbo: ExternalCredentialsPbo):
+        self.id = pbo.id
+        self.display_name = pbo.display_name
+        self.description = pbo.description
+        self.category = pbo.category
+        self.url = pbo.url
+        self.username = pbo.username
+        self.password = pbo.password
+        self.token = pbo.token
+        self.custom_fields = dict(pbo.custom_field)
+
+    def get_custom_field(self, key: str, default: str = None) -> str | None:
+        """
+        Get a custom field by key.
+
+        :param key: The key of the custom field to retrieve.
+        :param default: The value to return if the key does not exist.
+        :return: The value of the custom field, or None if the key does not exist.
+        """
+        return self.custom_fields.get(key, default)