pangea-sdk 5.4.0b1__py3-none-any.whl → 5.4.0b2__py3-none-any.whl

pangea/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "5.4.0beta1"
+__version__ = "5.4.0beta2"
 
 from pangea.asyncio.request import PangeaRequestAsync
 from pangea.config import PangeaConfig

pangea/asyncio/request.py

@@ -53,6 +53,10 @@ class PangeaRequestAsync(PangeaRequestBase):
         if isinstance(data, BaseModel):
             data = data.model_dump(exclude_none=True)
 
+        if isinstance(data, dict):
+            # Remove `None` values.
+            data = {k: v for k, v in data.items() if v is not None}
+
         if data is None:
             data = {}
 

pangea/asyncio/services/prompt_guard.py

@@ -51,7 +51,9 @@ class PromptGuardAsync(ServiceBaseAsync):
 
         super().__init__(token, config, logger_name, config_id)
 
-    async def guard(self, messages: Iterable[Message]) -> PangeaResponse[GuardResult]:
+    async def guard(
+        self, messages: Iterable[Message], *, analyzers: Iterable[str] | None = None
+    ) -> PangeaResponse[GuardResult]:
         """
         Guard (Beta)
 
@@ -63,6 +65,7 @@ class PromptGuardAsync(ServiceBaseAsync):
 
         Args:
             messages: Messages.
+            analyzers: Specific analyzers to be used in the call.
 
         Examples:
             from pangea.asyncio.services.prompt_guard import Message
@@ -70,4 +73,4 @@ class PromptGuardAsync(ServiceBaseAsync):
             response = await prompt_guard.guard([Message(role="user", content="hello world")])
         """
 
-        return await self.request.post("v1beta/guard", GuardResult, data={"messages": messages})
+        return await self.request.post("v1beta/guard", GuardResult, data={"messages": messages, "analyzers": analyzers})

pangea/request.py

@@ -230,6 +230,10 @@ class PangeaRequest(PangeaRequestBase):
         if isinstance(data, BaseModel):
             data = data.model_dump(exclude_none=True)
 
+        if isinstance(data, dict):
+            # Remove `None` values.
+            data = {k: v for k, v in data.items() if v is not None}
+
         if data is None:
             data = {}
 

pangea/services/prompt_guard.py

@@ -1,9 +1,9 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Literal, Optional
 
 from pangea.config import PangeaConfig
-from pangea.response import APIRequestModel, PangeaResponse, PangeaResponseResult
+from pangea.response import APIRequestModel, APIResponseModel, PangeaResponse, PangeaResponseResult
 from pangea.services.base import ServiceBase
 
 if TYPE_CHECKING:
@@ -15,11 +15,35 @@ class Message(APIRequestModel):
     content: str
 
 
+class Classification(APIResponseModel):
+    category: str
+    """Classification category"""
+
+    label: str
+    """Classification label"""
+
+    confidence: float
+    """Confidence score for the classification"""
+
+
 class GuardResult(PangeaResponseResult):
     detected: bool
-    type: Optional[str] = None
-    detector: Optional[str] = None
+    """Boolean response for if the prompt was considered malicious or not"""
+
+    type: Optional[Literal["direct", "indirect"]] = None
+    """Type of analysis, either direct or indirect"""
+
+    analyzer: Optional[str] = None
+    """Prompt Analyzers for identifying and rejecting properties of prompts"""
+
     confidence: int
+    """Percent of confidence in the detection result, ranging from 0 to 100"""
+
+    info: Optional[str] = None
+    """Extra information about the detection result"""
+
+    classifications: list[Classification]
+    """List of classification results with labels and confidence scores"""
 
 
 class PromptGuard(ServiceBase):
@@ -61,7 +85,9 @@ class PromptGuard(ServiceBase):
 
         super().__init__(token, config, logger_name, config_id)
 
-    def guard(self, messages: Iterable[Message]) -> PangeaResponse[GuardResult]:
+    def guard(
+        self, messages: Iterable[Message], *, analyzers: Iterable[str] | None = None
+    ) -> PangeaResponse[GuardResult]:
         """
         Guard (Beta)
 
@@ -72,7 +98,8 @@ class PromptGuard(ServiceBase):
         OperationId: prompt_guard_post_v1beta_guard
 
         Args:
-            messages: Messages.
+            messages: Prompt content and role array.
+            analyzers: Specific analyzers to be used in the call.
 
         Examples:
             from pangea.services.prompt_guard import Message
@@ -80,4 +107,4 @@ class PromptGuard(ServiceBase):
             response = prompt_guard.guard([Message(role="user", content="hello world")])
         """
 
-        return self.request.post("v1beta/guard", GuardResult, data={"messages": messages})
+        return self.request.post("v1beta/guard", GuardResult, data={"messages": messages, "analyzers": analyzers})

{pangea_sdk-5.4.0b1.dist-info → pangea_sdk-5.4.0b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pangea-sdk
-Version: 5.4.0b1
+Version: 5.4.0b2
 Summary: Pangea API SDK
 License: MIT
 Keywords: Pangea,SDK,Audit
@@ -63,13 +63,13 @@ the same compatibility guarantees as stable releases.
 Via pip:
 
 ```bash
-$ pip3 install pangea-sdk==5.4.0b1
+$ pip3 install pangea-sdk==5.4.0b2
 ```
 
 Via poetry:
 
 ```bash
-$ poetry add pangea-sdk==5.4.0b1
+$ poetry add pangea-sdk==5.4.0b2
 ```
 
 ## Usage

{pangea_sdk-5.4.0b1.dist-info → pangea_sdk-5.4.0b2.dist-info}/RECORD

@@ -1,7 +1,7 @@
-pangea/__init__.py,sha256=5GVWxTAjdvD4yfQsWwkX4JBF9NJnedj3Xos107reJy0,251
+pangea/__init__.py,sha256=D6T0FmtXtWk1W0BXPe9snPKzeo-dOtTEQHD3z5tFfxk,251
 pangea/asyncio/__init__.py,sha256=kjEMkqMQ521LlMSu5jn3_WgweyArwVZ2C-s3x7mR6Pk,45
 pangea/asyncio/file_uploader.py,sha256=wI7epib7Rc5jtZw4eJ1L1SlmutDG6CPv59C8N2UPhtY,1436
-pangea/asyncio/request.py,sha256=e-_wpOGsOLGlC4BXI6WBa_MFzkUoaMQEqYgXWaqoXQk,17900
+pangea/asyncio/request.py,sha256=lpLY-o405r3-VUfrAE5uxYxI8UjM4hjPqUzAUtOGE5o,18040
 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=rPaCx4cMzj-g9WFMRIysFCJAz6Btp6YrhcKe_exky8k,26283
@@ -11,7 +11,7 @@ pangea/asyncio/services/base.py,sha256=vRFVcO_uEAGJte3OUUBLD43RoiiFB1vC7SPyN6yEM
 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=BcxGKSoZ1nJiEHyZM9yOwKSSPJUrB6ibJ19KR27VlgQ,40261
-pangea/asyncio/services/prompt_guard.py,sha256=gRh7Z-ZS21EN8vXXnK1OJ5kYvmOWxKppia8tZYeuhmM,2202
+pangea/asyncio/services/prompt_guard.py,sha256=KMScsVybwQEwipN7CaC4AX4KRqlIkC8xZ2j7m-BDL9E,2349
 pangea/asyncio/services/redact.py,sha256=JPJcmeKFloMZRpkjAHAZbpZJpO993WsTfEwA-S5ov18,7951
 pangea/asyncio/services/sanitize.py,sha256=EbSdq_v9yZWce9xEYWvZharE9bJcxw8cg5Pv8LVxdxc,8627
 pangea/asyncio/services/share.py,sha256=AXXtFtmbXud0dAAom7qqHVOK9zBfX5S10MFS_1DQvio,30767
@@ -25,7 +25,7 @@ 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=Xv2aWAmDBZ3oasIEMCHulsnXjAGzqubYTf3vRCcX2es,24942
+pangea/request.py,sha256=vGB8owXUiNQoeiiACFvfXvg44JJo_L6WfcHlF6ug8co,25082
 pangea/response.py,sha256=lPAcYsF9Xg166CiyhCofVmQA-W4jevh0MQXxUa8Re68,7737
 pangea/services/__init__.py,sha256=h36HzyIGaI5kO6l3UCwKHx_Kd-m_9mYVwn5MLRVzblI,408
 pangea/services/ai_guard.py,sha256=-Ztyf9WUo1EB-gRzVH--OU62xzPoeVQ7BiFMTGUK5fI,3303
@@ -41,7 +41,7 @@ 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=y1EX2ctYIxQc52lmHp6-Q_UIDM--t3fOpXDssWiRPfo,56474
-pangea/services/prompt_guard.py,sha256=baaYbYW_HoZJNB8NCIxxFjvJelqrl9UOtWa10-N1jgg,2312
+pangea/services/prompt_guard.py,sha256=k-b8ukyXaHruqC4S2NjFKhDLcvY5Wp8Ga0g_Ep3WFKY,3244
 pangea/services/redact.py,sha256=ovIcT0jkXe57O7keGzSClWNCic8y-4NZoemXoSKjjww,12913
 pangea/services/sanitize.py,sha256=eAN1HhObiKqygy6HHcfl0NmxYfPMvqSKepwEAVVIIEE,12936
 pangea/services/share/file_format.py,sha256=1svO1ee_aenA9zoO_AaU-Rk5Ulp7kcPOc_KwNoluyQE,2797
@@ -55,6 +55,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.4.0b1.dist-info/METADATA,sha256=7kHpv2AVDGKa_KQJSTv5dZa1TlvM4VC1_bpddpL5o1M,6843
-pangea_sdk-5.4.0b1.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-pangea_sdk-5.4.0b1.dist-info/RECORD,,
+pangea_sdk-5.4.0b2.dist-info/METADATA,sha256=7_vZ3tIafJ9-TdGTZphBcqHo0MQPqp-bNb4L870CwgU,6843
+pangea_sdk-5.4.0b2.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+pangea_sdk-5.4.0b2.dist-info/RECORD,,