pangea-sdk 5.2.0b2__py3-none-any.whl → 5.2.1__py3-none-any.whl

pangea/services/redact.py

@@ -66,6 +66,17 @@ class RedactRequest(APIRequestModel):
     rulesets: Optional[List[str]] = None
     return_result: Optional[bool] = None
     redaction_method_overrides: Optional[RedactionMethodOverrides] = None
+    vault_parameters: Optional[VaultParameters] = None
+    llm_request: Optional[bool] = None
+    """Is this redact call going to be used in an LLM request?"""
+
+
+class VaultParameters(APIRequestModel):
+    fpe_key_id: Optional[str] = None
+    """A vault key ID of an exportable key used to redact with FPE instead of using the service config default."""
+
+    salt_secret_id: Optional[str] = None
+    """A vault secret ID of a secret used to salt a hash instead of using the service config default."""
 
 
 class RecognizerResult(APIResponseModel):
@@ -108,11 +119,13 @@ class RedactResult(PangeaResponseResult):
     redact_text: Redacted text result
     count: Number of redactions present in the text
     report: Describes the decision process for redactions
+    fpe_context: FPE context used to encrypt and redact data
     """
 
     redacted_text: Optional[str] = None
     count: int
     report: Optional[DebugReport] = None
+    fpe_context: Optional[str] = None
 
 
 class StructuredRequest(APIRequestModel):
@@ -134,6 +147,9 @@ class StructuredRequest(APIRequestModel):
     rulesets: Optional[List[str]] = None
     return_result: Optional[bool] = None
     redaction_method_overrides: Optional[RedactionMethodOverrides] = None
+    vault_parameters: Optional[VaultParameters] = None
+    llm_request: Optional[bool] = None
+    """Is this redact call going to be used in an LLM request?"""
 
 
 class StructuredResult(PangeaResponseResult):
@@ -228,6 +244,8 @@ class Redact(ServiceBase):
         rulesets: Optional[List[str]] = None,
         return_result: Optional[bool] = None,
         redaction_method_overrides: Optional[RedactionMethodOverrides] = None,
+        llm_request: Optional[bool] = None,
+        vault_parameters: Optional[VaultParameters] = None,
     ) -> PangeaResponse[RedactResult]:
         """
         Redact
@@ -244,6 +262,8 @@ class Redact(ServiceBase):
             rulesets (list[str], optional): An array of redact rulesets short names
             return_result(bool, optional): Setting this value to false will omit the redacted result only returning count
             redaction_method_overrides: A set of redaction method overrides for any enabled rule. These methods override the config declared methods
+            llm_request: Boolean flag to enable FPE redaction for LLM requests
+            vault_parameters: A set of vault parameters to use for redaction
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -264,6 +284,8 @@ class Redact(ServiceBase):
             rulesets=rulesets,
             return_result=return_result,
             redaction_method_overrides=redaction_method_overrides,
+            llm_request=llm_request,
+            vault_parameters=vault_parameters,
         )
         return self.request.post("v1/redact", RedactResult, data=input.model_dump(exclude_none=True))
 
@@ -277,6 +299,8 @@ class Redact(ServiceBase):
         rulesets: Optional[List[str]] = None,
         return_result: Optional[bool] = None,
         redaction_method_overrides: Optional[RedactionMethodOverrides] = None,
+        llm_request: Optional[bool] = None,
+        vault_parameters: Optional[VaultParameters] = None,
     ) -> PangeaResponse[StructuredResult]:
         """
         Redact structured
@@ -297,6 +321,8 @@ class Redact(ServiceBase):
             rulesets (list[str], optional): An array of redact rulesets short names
             return_result(bool, optional): Setting this value to false will omit the redacted result only returning count
             redaction_method_overrides: A set of redaction method overrides for any enabled rule. These methods override the config declared methods
+            llm_request: Boolean flag to enable FPE redaction for LLM requests
+            vault_parameters: A set of vault parameters to use for redaction
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -324,6 +350,8 @@ class Redact(ServiceBase):
             rulesets=rulesets,
             return_result=return_result,
             redaction_method_overrides=redaction_method_overrides,
+            llm_request=llm_request,
+            vault_parameters=vault_parameters,
         )
         return self.request.post("v1/redact_structured", StructuredResult, data=input.model_dump(exclude_none=True))
 

pangea/services/sanitize.py

@@ -7,6 +7,7 @@ from typing import Dict, List, Optional, Tuple
 
 from pydantic import Field
 
+from pangea.config import PangeaConfig
 from pangea.response import APIRequestModel, PangeaResponse, PangeaResponseResult, TransferMethod
 from pangea.services.base import ServiceBase
 from pangea.utils import FileUploadParams, get_file_upload_params
@@ -197,6 +198,27 @@ class Sanitize(ServiceBase):
 
     service_name = "sanitize"
 
+    def __init__(
+        self, token: str, config: PangeaConfig | None = None, logger_name: str = "pangea", config_id: str | None = None
+    ) -> None:
+        """
+        Sanitize client
+
+        Initializes a new Sanitize client.
+
+        Args:
+            token: Pangea API token.
+            config: Configuration.
+            logger_name: Logger name.
+            config_id: Configuration ID.
+
+        Examples:
+             config = PangeaConfig(domain="aws.us.pangea.cloud")
+             authz = Sanitize(token="pangea_token", config=config)
+        """
+
+        super().__init__(token, config, logger_name, config_id=config_id)
+
     def sanitize(
         self,
         transfer_method: TransferMethod = TransferMethod.POST_URL,

pangea/services/share/share.py

@@ -6,6 +6,7 @@ import enum
 import io
 from typing import Dict, List, NewType, Optional, Tuple, Union
 
+from pangea.config import PangeaConfig
 from pangea.response import APIRequestModel, PangeaResponse, PangeaResponseResult, TransferMethod
 from pangea.services.base import ServiceBase
 from pangea.services.share.file_format import FileFormat
@@ -669,10 +670,31 @@ class BucketsResult(PangeaResponseResult):
 
 
 class Share(ServiceBase):
-    """Share service client."""
+    """Secure Share service client."""
 
     service_name = "share"
 
+    def __init__(
+        self, token: str, config: PangeaConfig | None = None, logger_name: str = "pangea", config_id: str | None = None
+    ) -> None:
+        """
+        Secure Share client
+
+        Initializes a new Secure Share client.
+
+        Args:
+            token: Pangea API token.
+            config: Configuration.
+            logger_name: Logger name.
+            config_id: Configuration ID.
+
+        Examples:
+             config = PangeaConfig(domain="aws.us.pangea.cloud")
+             authz = Share(token="pangea_token", config=config)
+        """
+
+        super().__init__(token, config, logger_name, config_id=config_id)
+
     def buckets(self) -> PangeaResponse[BucketsResult]:
         """
         Buckets

{pangea_sdk-5.2.0b2.dist-info → pangea_sdk-5.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pangea-sdk
-Version: 5.2.0b2
+Version: 5.2.1
 Summary: Pangea API SDK
 Home-page: https://pangea.cloud/docs/sdk/python/
 License: MIT
@@ -17,11 +17,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries
-Requires-Dist: aiohttp (>=3.10.10,<4.0.0)
+Requires-Dist: aiohttp (>=3.11.10,<4.0.0)
 Requires-Dist: cryptography (>=43.0.3,<44.0.0)
-Requires-Dist: deprecated (>=1.2.14,<2.0.0)
+Requires-Dist: deprecated (>=1.2.15,<2.0.0)
 Requires-Dist: google-crc32c (>=1.5.0,<2.0.0)
-Requires-Dist: pydantic (>=2.9.2,<3.0.0)
+Requires-Dist: pydantic (>=2.10.3,<3.0.0)
 Requires-Dist: python-dateutil (>=2.9.0,<3.0.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)
 Requires-Dist: requests-toolbelt (>=1.0.0,<2.0.0)

{pangea_sdk-5.2.0b2.dist-info → pangea_sdk-5.2.1.dist-info}/RECORD

@@ -1,51 +1,47 @@
-pangea/__init__.py,sha256=YW7Y-Lb2D13OAwVGQH3AL6xmqtVbLjqDaQFROwi_Yd4,251
+pangea/__init__.py,sha256=YdVCSgR-OBewhxnne1SnOliwM7KXiu_EV88j_zSLRmM,246
 pangea/asyncio/__init__.py,sha256=kjEMkqMQ521LlMSu5jn3_WgweyArwVZ2C-s3x7mR6Pk,45
 pangea/asyncio/file_uploader.py,sha256=wI7epib7Rc5jtZw4eJ1L1SlmutDG6CPv59C8N2UPhtY,1436
-pangea/asyncio/request.py,sha256=Pwk3dAzO_A7I1NyrIqK5V80xvtAL80KiNSjbrXjzQzA,17519
-pangea/asyncio/services/__init__.py,sha256=L6Tdhjfx_ZECHskhLMPaCcOefi-r-imw6q_zlU4j-FY,464
-pangea/asyncio/services/ai_guard.py,sha256=FxO8GRBztfOhpd7Kzj4-jWAiG0kb89-jVWL_AnbEou8,2142
-pangea/asyncio/services/audit.py,sha256=bZ7gdkVWkzqLqUVc1Wnf3oDAaCLg97-zTWhY8UdX0_Y,26549
+pangea/asyncio/request.py,sha256=BREsLY8_MCxGSplHbdKFZTzr4TX5ya0-BsXLffZhsco,17849
+pangea/asyncio/services/__init__.py,sha256=3IkiTqY_RtFndI7aoDTrb1yLv8xos_cKhmGS1TULcmw,386
+pangea/asyncio/services/audit.py,sha256=rPaCx4cMzj-g9WFMRIysFCJAz6Btp6YrhcKe_exky8k,26283
 pangea/asyncio/services/authn.py,sha256=rPeLJweL8mYH_t4ebcQn4n_Wglr3kClKNnCXNCimZU4,46622
-pangea/asyncio/services/authz.py,sha256=HgW9R8DeW19wS7fpgq0NWOx41wZWcn6NYS4NMbi8p1A,9482
-pangea/asyncio/services/base.py,sha256=1_W3ImEdH5C3rGynC_JWeKx1E2X9HHvUWBa5uRiZFNM,2763
+pangea/asyncio/services/authz.py,sha256=B_0_nhDMJcjNpjpCx3Vi2LDRhlmfV9325GKbUZ8reos,10025
+pangea/asyncio/services/base.py,sha256=vRFVcO_uEAGJte3OUUBLD43RoiiFB1vC7SPyN6yEMoA,3158
 pangea/asyncio/services/embargo.py,sha256=ctzj3kip6xos-Eu3JuOskrCGYC8T3JlsgAopZHiPSXM,3068
 pangea/asyncio/services/file_scan.py,sha256=PLG1O-PL4Yk9uY9D6NbMrZ5LHg70Z311s7bFe46UMZA,7108
-pangea/asyncio/services/intel.py,sha256=cCm3VwWxUzEUCNhuPCeejJvr4uOeLXuYDbDwTzNG6Aw,38121
-pangea/asyncio/services/prompt_guard.py,sha256=gRh7Z-ZS21EN8vXXnK1OJ5kYvmOWxKppia8tZYeuhmM,2202
-pangea/asyncio/services/redact.py,sha256=jRNtXr_DZ_cY7guhut-eZmOEhy2uN_VCXrjGH6bkh74,7265
-pangea/asyncio/services/sanitize.py,sha256=bf98J-s-P51oSKqNBgR0wj5mlHOCBwpjWz7k0NdXCKQ,7899
-pangea/asyncio/services/share.py,sha256=UYJeUKA3NLSFA8R0X7B6yBi2U1g4q04O4ftrp9SMCzA,26097
+pangea/asyncio/services/intel.py,sha256=Iwz_DleAPtmd1taekT4W8lVo65uHjIvA4TQ7WUQGrRk,38306
+pangea/asyncio/services/redact.py,sha256=JPJcmeKFloMZRpkjAHAZbpZJpO993WsTfEwA-S5ov18,7951
+pangea/asyncio/services/sanitize.py,sha256=4pRWBH595kFUXBkmu8euyk6H7F1M_-xXB2Qxnz6St6c,8627
+pangea/asyncio/services/share.py,sha256=YPJm_Gc4tfcx2cX6P_vLWIsHOR6M3RYy9LdU2UzEZbk,26791
 pangea/asyncio/services/vault.py,sha256=VqrJGSEdq6MlZRI6cJpkthhIsqLClSQdgVxwYCbIwEk,77079
 pangea/audit_logger.py,sha256=gRkCfUUT5LDNaycwxkhZUySgY47jDfn1ZeKOul4XCQI,3842
 pangea/config.py,sha256=mQUu8GX_6weIuv3vjNdG5plppXskXYASmxMWtFQh-hc,1662
 pangea/crypto/rsa.py,sha256=mwSiNy571KAGr3F6oEM0CXWkl9D023ch8ldbZZeLj_4,4747
-pangea/deep_verify.py,sha256=mocaGbC6XLbMTVWxTpMv4oJtXGPWpT-SbFqT3obpiZs,8443
+pangea/deep_verify.py,sha256=ZGraaL7TCxwRBIDqjBFR0clKlhAC-Yce6kD-1LClhG8,8616
 pangea/deprecated.py,sha256=IjFYEVvY1E0ld0SMkEYC1o62MAleX3nnT1If2dFVbHo,608
-pangea/dump_audit.py,sha256=1Je8D2fXwU4PWcZ-ZD4icfO3DNFvWqJkwsac4qFEhOo,7025
+pangea/dump_audit.py,sha256=IevqaUUh7GDepdIW7slSxeZbkPrWIVbcX3sr4DgpJXI,7090
 pangea/exceptions.py,sha256=OBtzUECpNa6vNp8ySkHC-tm4QjFRCOAHBkMHqzAlOu8,5656
 pangea/file_uploader.py,sha256=4RQ44xt-faApC61nn2PlwHT7XYrJ4GeQA8Ug4tySEAg,1227
 pangea/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pangea/request.py,sha256=ZRH4NCnbzjpzkcNnOomelK5uB49T_20PfHP_j4Swgmw,24545
+pangea/request.py,sha256=zxFvqbeGYtxN29nS8SPPZKlcGCqdUtenxaq84pQ4cxs,24807
 pangea/response.py,sha256=lPAcYsF9Xg166CiyhCofVmQA-W4jevh0MQXxUa8Re68,7737
-pangea/services/__init__.py,sha256=h36HzyIGaI5kO6l3UCwKHx_Kd-m_9mYVwn5MLRVzblI,408
-pangea/services/ai_guard.py,sha256=W9OxmqQiC2ymDWAoE8FK3_8prDTQhHR8lz4xE81tlMM,4157
-pangea/services/audit/audit.py,sha256=IFv7jANA8S2SypQVS47x94_Cr5Z9zSsL9Dp9eXw9RHk,39593
+pangea/services/__init__.py,sha256=-QsZxRzRq_V1x1lmS_mu4310MNm0DkM4r6g6rfVGnOc,340
+pangea/services/audit/audit.py,sha256=7-c9l7jyGtpG7SqRUMpqsAzcUDhMZ5izgPalxHXsUvM,39320
 pangea/services/audit/exceptions.py,sha256=bhVuYe4ammacOVxwg98CChxvwZf5FKgR2DcgqILOcwc,471
 pangea/services/audit/models.py,sha256=1h1B9eSYQMYG3f8WNi1UcDX2-impRrET_ErjJYUnj7M,14678
 pangea/services/audit/signing.py,sha256=5A4hvPtpfP2kMz8bsiiKUACriXbh5dv9gb_rbqiUtuI,5583
 pangea/services/audit/util.py,sha256=Zq1qvfeplYfhCP_ud5YMvntSB0UvnCdsuYbOzZkHbjg,7620
 pangea/services/authn/authn.py,sha256=cZKl2Ixc6HwHnkRecpSaAGTQUgaZUtxfLa0T3S03HMs,45478
 pangea/services/authn/models.py,sha256=HH5su6jx3O9AwVGzASXZ99-eIWjgXEP5LhIVdewM13s,22394
-pangea/services/authz.py,sha256=HfDnovAokzAHvnjYdOCwceM-1sCmzODnjNEbQBUSfo8,12222
-pangea/services/base.py,sha256=nnVVzw6Y_hej5Vcy0WvN_CR0BG_U5F5k_XTNiBX1we0,3464
+pangea/services/authz.py,sha256=bB0ZEUuXLT7Xjs5kZef1hZK6Du6VUusHe5aekNaxamw,12746
+pangea/services/base.py,sha256=43pWQcR9CeT4sGzgctF3Sy4M_h7DaUzkuZD2Z7CcDUU,3845
 pangea/services/embargo.py,sha256=9Wfku4td5ORaIENKmnGmS5jxJJIRfWp6Q51L36Jsy0I,3897
 pangea/services/file_scan.py,sha256=QiO80uKqB_BnAOiYQKznXfxpa5j40qqETE3-zBRT_QE,7813
-pangea/services/intel.py,sha256=CziBhC5K6O_kBXpD8zgJLpDtLHzBRgATGW4gHHFJT48,52039
-pangea/services/prompt_guard.py,sha256=baaYbYW_HoZJNB8NCIxxFjvJelqrl9UOtWa10-N1jgg,2312
-pangea/services/redact.py,sha256=ZYXkzEoriLJyCqaj5dqmgsC56mIz4T3pPToZ7TcNfhg,11465
-pangea/services/sanitize.py,sha256=XP5D4CcbCZfzgU567X6H5eFBWwZuYSsHdvsdrQAZekY,12767
+pangea/services/intel.py,sha256=flVdK4WhllPutCkWh7H-MuBxMMz0f4Bl-fz2f-hPuWM,52679
+pangea/services/redact.py,sha256=ovIcT0jkXe57O7keGzSClWNCic8y-4NZoemXoSKjjww,12913
+pangea/services/sanitize.py,sha256=D_R_XIe9FvRCX40b8b_3gouhAGduDdxI9bT2w-kiVHU,13444
 pangea/services/share/file_format.py,sha256=1svO1ee_aenA9zoO_AaU-Rk5Ulp7kcPOc_KwNoluyQE,2797
-pangea/services/share/share.py,sha256=iyP32UNWoT2F9C_65FiXWrVoNoO7dBjf0tVX1mF2Fz0,45644
+pangea/services/share/share.py,sha256=8N4zXQJjkVDf2iR62pz6xpmxlbXRPNfHgYfOuEdNaZ8,46333
 pangea/services/vault/models/asymmetric.py,sha256=vspijmEvHm5WXri_fjOWfQc4maYyZfhDkLuaTM8-PZo,4991
 pangea/services/vault/models/common.py,sha256=PSZRFqHTUtEMJJGwywEFM2AU3aV8S-sbcoo3LLQ6uTc,17981
 pangea/services/vault/models/keys.py,sha256=duAuTiOby_D7MloRvN4gNj0P-b-jx9sdtplAWFxsShw,2786
@@ -55,6 +51,6 @@ pangea/services/vault/vault.py,sha256=ow-Zm7PYzfWIfUcA4UNnpeL2DHfZM4C7inRDmNR3zQ
 pangea/tools.py,sha256=2-Y4SAHWFv6Ocj42J_bWrVy27M5G3wi7a8LJn0dabHc,6427
 pangea/utils.py,sha256=dZ6MwFVEWXUgXvvDg-k6JnvVfsgslvtaBd7ez7afrqk,4983
 pangea/verify_audit.py,sha256=nSP17OzoSPdvezRExwfcf45H8ZPZnxZu-CbEp3qFJO0,17354
-pangea_sdk-5.2.0b2.dist-info/METADATA,sha256=j_pSOR8g6RfcYWO7D9lQhISqYHPzfUlc5VSFFYW1elo,7514
-pangea_sdk-5.2.0b2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-pangea_sdk-5.2.0b2.dist-info/RECORD,,
+pangea_sdk-5.2.1.dist-info/METADATA,sha256=YjlZ9hj9YmyyMIeS6Z_eSbpBeHsh_QNfzwxfbexlZVs,7513
+pangea_sdk-5.2.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pangea_sdk-5.2.1.dist-info/RECORD,,

pangea/asyncio/services/ai_guard.py

@@ -1,75 +0,0 @@
-from __future__ import annotations
-
-from pangea.asyncio.services.base import ServiceBaseAsync
-from pangea.config import PangeaConfig
-from pangea.response import PangeaResponse
-from pangea.services.ai_guard import TextGuardResult
-
-
-class AIGuardAsync(ServiceBaseAsync):
-    """AI Guard service client.
-
-    Provides methods to interact with Pangea's AI Guard service.
-
-    Examples:
-        from pangea import PangeaConfig
-        from pangea.asyncio.services import AIGuardAsync
-
-        config = PangeaConfig(domain="aws.us.pangea.cloud")
-        ai_guard = AIGuardAsync(token="pangea_token", config=config)
-    """
-
-    service_name = "ai-guard"
-
-    def __init__(
-        self, token: str, config: PangeaConfig | None = None, logger_name: str = "pangea", config_id: str | None = None
-    ) -> None:
-        """
-        AI Guard service client.
-
-        Initializes a new AI Guard client.
-
-        Args:
-            token: Pangea API token.
-            config: Pangea service configuration.
-            logger_name: Logger name.
-            config_id: Configuration ID.
-
-        Examples:
-            from pangea import PangeaConfig
-            from pangea.asyncio.services import AIGuardAsync
-
-            config = PangeaConfig(domain="aws.us.pangea.cloud")
-            ai_guard = AIGuardAsync(token="pangea_token", config=config)
-        """
-
-        super().__init__(token, config, logger_name, config_id)
-
-    async def guard_text(
-        self,
-        text: str,
-        *,
-        recipe: str = "pangea_prompt_guard",
-        debug: bool = False,
-    ) -> PangeaResponse[TextGuardResult]:
-        """
-        Text guard (Beta)
-
-        Guard text.
-
-        How to install a [Beta release](https://pangea.cloud/docs/sdk/python/#beta-releases).
-
-        OperationId: ai_guard_post_v1beta_text_guard
-
-        Args:
-            text: Text.
-            recipe: Recipe.
-            debug: Debug.
-
-        Examples:
-            response = await ai_guard.guard_text("text")
-        """
-
-        return await self.request.post(
-            "v1beta/text/guard", TextGuardResult, data={"text": text, "recipe": recipe, "debug": debug}
-        )

pangea/asyncio/services/prompt_guard.py

@@ -1,73 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from pangea.asyncio.services.base import ServiceBaseAsync
-from pangea.config import PangeaConfig
-from pangea.services.prompt_guard import GuardResult, Message
-
-if TYPE_CHECKING:
-    from collections.abc import Iterable
-
-    from pangea.response import PangeaResponse
-
-
-class PromptGuardAsync(ServiceBaseAsync):
-    """Prompt Guard service client.
-
-    Provides methods to interact with Pangea's Prompt Guard service.
-
-    Examples:
-        from pangea import PangeaConfig
-        from pangea.asyncio.services import PromptGuardAsync
-
-        config = PangeaConfig(domain="aws.us.pangea.cloud")
-        prompt_guard = PromptGuardAsync(token="pangea_token", config=config)
-    """
-
-    service_name = "prompt-guard"
-
-    def __init__(
-        self, token: str, config: PangeaConfig | None = None, logger_name: str = "pangea", config_id: str | None = None
-    ) -> None:
-        """
-        Prompt Guard service client.
-
-        Initializes a new Prompt Guard client.
-
-        Args:
-            token: Pangea API token.
-            config: Pangea service configuration.
-            logger_name: Logger name.
-            config_id: Configuration ID.
-
-        Examples:
-            from pangea import PangeaConfig
-            from pangea.asyncio.services import PromptGuardAsync
-
-            config = PangeaConfig(domain="aws.us.pangea.cloud")
-            prompt_guard = PromptGuardAsync(token="pangea_token", config=config)
-        """
-
-        super().__init__(token, config, logger_name, config_id)
-
-    async def guard(self, messages: Iterable[Message]) -> PangeaResponse[GuardResult]:
-        """
-        Guard (Beta)
-
-        Guard messages.
-
-        How to install a [Beta release](https://pangea.cloud/docs/sdk/python/#beta-releases).
-
-        OperationId: prompt_guard_post_v1beta_guard
-
-        Args:
-            messages: Messages.
-
-        Examples:
-            from pangea.asyncio.services.prompt_guard import Message
-
-            response = await prompt_guard.guard([Message(role="user", content="hello world")])
-        """
-
-        return await self.request.post("v1beta/guard", GuardResult, data={"messages": messages})

pangea/services/ai_guard.py

@@ -1,157 +0,0 @@
-from __future__ import annotations
-
-from typing import List, Optional
-
-from typing_extensions import Literal
-
-from pangea.config import PangeaConfig
-from pangea.response import PangeaResponse, PangeaResponseResult
-from pangea.services.base import ServiceBase
-from pangea.services.intel import UserBreachedData
-
-
-class TextGuardSecurityIssues(PangeaResponseResult):
-    compromised_email_addresses: int
-    malicious_domain_count: int
-    malicious_ip_count: int
-    malicious_url_count: int
-    redact_rule_match_count: int
-
-
-class TextGuardFindings(PangeaResponseResult):
-    artifact_count: Optional[int] = None
-    malicious_count: Optional[int] = None
-    security_issues: TextGuardSecurityIssues
-
-
-class RedactRecognizerResult(PangeaResponseResult):
-    field_type: str
-    """The entity name."""
-
-    score: float
-    """The certainty score that the entity matches this specific snippet."""
-
-    text: str
-    """The text snippet that matched."""
-
-    start: int
-    """The starting index of a snippet."""
-
-    end: int
-    """The ending index of a snippet."""
-
-    redacted: bool
-    """Indicates if this rule was used to anonymize a text snippet."""
-
-
-class RedactReport(PangeaResponseResult):
-    count: int
-    recognizer_results: List[RedactRecognizerResult]
-
-
-class IntelResults(PangeaResponseResult):
-    category: List[str]
-    """
-    The categories that apply to this indicator as determined by the provider.
-    """
-
-    score: int
-    """The score, given by the Pangea service, for the indicator."""
-
-    verdict: Literal["malicious", "suspicious", "unknown", "benign"]
-
-
-class TextGuardReport(PangeaResponseResult):
-    domain_intel: Optional[IntelResults] = None
-    ip_intel: Optional[IntelResults] = None
-    redact: RedactReport
-    url_intel: Optional[IntelResults] = None
-    user_intel: Optional[UserBreachedData] = None
-
-
-class TextGuardArtifact(PangeaResponseResult):
-    defanged: bool
-    end: int
-    start: int
-    type: str
-    value: str
-    verdict: Optional[str] = None
-    """The verdict, given by the Pangea service, for the indicator."""
-
-
-class TextGuardResult(PangeaResponseResult):
-    artifacts: Optional[List[TextGuardArtifact]] = None
-    findings: TextGuardFindings
-    redacted_prompt: str
-
-    # `debug=True` only.
-    report: Optional[TextGuardReport] = None
-
-
-class AIGuard(ServiceBase):
-    """AI Guard service client.
-
-    Provides methods to interact with Pangea's AI Guard service.
-
-    Examples:
-        from pangea import PangeaConfig
-        from pangea.services import AIGuard
-
-        config = PangeaConfig(domain="aws.us.pangea.cloud")
-        ai_guard = AIGuard(token="pangea_token", config=config)
-    """
-
-    service_name = "ai-guard"
-
-    def __init__(
-        self, token: str, config: PangeaConfig | None = None, logger_name: str = "pangea", config_id: str | None = None
-    ) -> None:
-        """
-        AI Guard service client.
-
-        Initializes a new AI Guard client.
-
-        Args:
-            token: Pangea API token.
-            config: Pangea service configuration.
-            logger_name: Logger name.
-            config_id: Configuration ID.
-
-        Examples:
-            from pangea import PangeaConfig
-            from pangea.services import AIGuard
-
-            config = PangeaConfig(domain="aws.us.pangea.cloud")
-            ai_guard = AIGuard(token="pangea_token", config=config)
-        """
-
-        super().__init__(token, config, logger_name, config_id)
-
-    def guard_text(
-        self,
-        text: str,
-        *,
-        recipe: str = "pangea_prompt_guard",
-        debug: bool = False,
-    ) -> PangeaResponse[TextGuardResult]:
-        """
-        Text guard (Beta)
-
-        Guard text.
-
-        How to install a [Beta release](https://pangea.cloud/docs/sdk/python/#beta-releases).
-
-        OperationId: ai_guard_post_v1beta_text_guard
-
-        Args:
-            text: Text.
-            recipe: Recipe.
-            debug: Debug.
-
-        Examples:
-            response = ai_guard.guard_text("text")
-        """
-
-        return self.request.post(
-            "v1beta/text/guard", TextGuardResult, data={"text": text, "recipe": recipe, "debug": debug}
-        )

pangea/services/prompt_guard.py

@@ -1,83 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING, Optional
-
-from pangea.config import PangeaConfig
-from pangea.response import APIRequestModel, PangeaResponse, PangeaResponseResult
-from pangea.services.base import ServiceBase
-
-if TYPE_CHECKING:
-    from collections.abc import Iterable
-
-
-class Message(APIRequestModel):
-    role: str
-    content: str
-
-
-class GuardResult(PangeaResponseResult):
-    detected: bool
-    type: Optional[str] = None
-    detector: Optional[str] = None
-    confidence: int
-
-
-class PromptGuard(ServiceBase):
-    """Prompt Guard service client.
-
-    Provides methods to interact with Pangea's Prompt Guard service.
-
-    Examples:
-        from pangea import PangeaConfig
-        from pangea.services import PromptGuard
-
-        config = PangeaConfig(domain="aws.us.pangea.cloud")
-        prompt_guard = PromptGuard(token="pangea_token", config=config)
-    """
-
-    service_name = "prompt-guard"
-
-    def __init__(
-        self, token: str, config: PangeaConfig | None = None, logger_name: str = "pangea", config_id: str | None = None
-    ) -> None:
-        """
-        Prompt Guard service client.
-
-        Initializes a new Prompt Guard client.
-
-        Args:
-            token: Pangea API token.
-            config: Pangea service configuration.
-            logger_name: Logger name.
-            config_id: Configuration ID.
-
-        Examples:
-            from pangea import PangeaConfig
-            from pangea.services import PromptGuard
-
-            config = PangeaConfig(domain="aws.us.pangea.cloud")
-            prompt_guard = PromptGuard(token="pangea_token", config=config)
-        """
-
-        super().__init__(token, config, logger_name, config_id)
-
-    def guard(self, messages: Iterable[Message]) -> PangeaResponse[GuardResult]:
-        """
-        Guard (Beta)
-
-        Guard messages.
-
-        How to install a [Beta release](https://pangea.cloud/docs/sdk/python/#beta-releases).
-
-        OperationId: prompt_guard_post_v1beta_guard
-
-        Args:
-            messages: Messages.
-
-        Examples:
-            from pangea.services.prompt_guard import Message
-
-            response = prompt_guard.guard([Message(role="user", content="hello world")])
-        """
-
-        return self.request.post("v1beta/guard", GuardResult, data={"messages": messages})