sapiopycommons 2025.10.9a776__py3-none-any.whl → 2025.10.9a777__py3-none-any.whl

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, \
@@ -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/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)

{sapiopycommons-2025.10.9a776.dist-info → sapiopycommons-2025.10.9a777.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.10.9a776
+Version: 2025.10.9a777
 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.dist-info → sapiopycommons-2025.10.9a777.dist-info}/RECORD

@@ -1,7 +1,8 @@
 sapiopycommons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sapiopycommons/ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sapiopycommons/ai/agent_service_base.py,sha256=52_QRjbMi6XYHC3oakX3mO5TOh6w8hieTbxzfKbFZWw,54600
+sapiopycommons/ai/agent_service_base.py,sha256=3qU3_wy9wPuhZaRrPAzXNUe63pbQAQxqLnFPdhrm-08,56280
 sapiopycommons/ai/converter_service_base.py,sha256=HiUXmwqv1STgyQeF9_eTFXzjIFXp5-NJ7sEhMpV3aAU,6351
+sapiopycommons/ai/external_credentials.py,sha256=40AI7VtHf6PzvwJLR_mZemUCrfAUvC--tGH-npaDIgo,1163
 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
@@ -105,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.9a776.dist-info/METADATA,sha256=5yZk5DAAW1IHX_9fduVcHOOjludn5fBGykcvM0i4uE4,3142
-sapiopycommons-2025.10.9a776.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-sapiopycommons-2025.10.9a776.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
-sapiopycommons-2025.10.9a776.dist-info/RECORD,,
+sapiopycommons-2025.10.9a777.dist-info/METADATA,sha256=jvXJcvHI64AzlqRPyRwn7AJ5n1ZLP6eua8j6uOKZBEs,3142
+sapiopycommons-2025.10.9a777.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+sapiopycommons-2025.10.9a777.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
+sapiopycommons-2025.10.9a777.dist-info/RECORD,,