pangea-sdk 5.4.0b3__tar.gz → 5.5.0__tar.gz

{pangea_sdk-5.4.0b3 → pangea_sdk-5.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pangea-sdk
-Version: 5.4.0b3
+Version: 5.5.0
 Summary: Pangea API SDK
 License: MIT
 Keywords: Pangea,SDK,Audit
@@ -9,13 +9,13 @@ Author-email: glenn.gallien@pangea.cloud
 Requires-Python: >=3.9,<4.0.0
 Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries
-Requires-Dist: aiohttp (>=3.11.11,<4.0.0)
+Requires-Dist: aiohttp (>=3.11.12,<4.0.0)
 Requires-Dist: cryptography (>=43.0.3,<44.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.10.5,<3.0.0)
-Requires-Dist: python-dateutil (>=2.9.0,<3.0.0)
-Requires-Dist: requests (>=2.31.0,<3.0.0)
+Requires-Dist: deprecated (>=1.2.18,<2.0.0)
+Requires-Dist: google-crc32c (>=1.6.0,<2.0.0)
+Requires-Dist: pydantic (>=2.10.6,<3.0.0)
+Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
+Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: requests-toolbelt (>=1.0.0,<2.0.0)
 Requires-Dist: typing-extensions (>=4.12.2,<5.0.0)
 Description-Content-Type: text/markdown
@@ -63,13 +63,13 @@ the same compatibility guarantees as stable releases.
 Via pip:
 
 ```bash
-$ pip3 install pangea-sdk==5.4.0b2
+$ pip3 install pangea-sdk==5.5.0b2
 ```
 
 Via poetry:
 
 ```bash
-$ poetry add pangea-sdk==5.4.0b2
+$ poetry add pangea-sdk==5.5.0b2
 ```
 
 ## Usage

{pangea_sdk-5.4.0b3 → pangea_sdk-5.5.0}/README.md

@@ -41,13 +41,13 @@ the same compatibility guarantees as stable releases.
 Via pip:
 
 ```bash
-$ pip3 install pangea-sdk==5.4.0b2
+$ pip3 install pangea-sdk==5.5.0b2
 ```
 
 Via poetry:
 
 ```bash
-$ poetry add pangea-sdk==5.4.0b2
+$ poetry add pangea-sdk==5.5.0b2
 ```
 
 ## Usage

{pangea_sdk-5.4.0b3 → pangea_sdk-5.5.0}/pangea/__init__.py

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

pangea_sdk-5.5.0/pangea/asyncio/services/ai_guard.py

@@ -0,0 +1,221 @@
+from __future__ import annotations
+
+from typing import overload
+
+from typing_extensions import TypeVar
+
+from pangea.asyncio.services.base import ServiceBaseAsync
+from pangea.config import PangeaConfig
+from pangea.response import PangeaResponse
+from pangea.services.ai_guard import LogFields, TextGuardResult
+
+_T = TypeVar("_T")
+
+
+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)
+
+    @overload
+    async def guard_text(
+        self,
+        text: str,
+        *,
+        recipe: str | None = None,
+        debug: bool | None = None,
+        llm_info: str | None = None,
+        log_fields: LogFields | None = None,
+    ) -> PangeaResponse[TextGuardResult[None]]:
+        """
+        Text Guard for scanning LLM inputs and outputs
+
+        Analyze and redact text to avoid manipulation of the model, addition of
+        malicious content, and other undesirable data transfers.
+
+        OperationId: ai_guard_post_v1_text_guard
+
+        Args:
+            text: Text to be scanned by AI Guard for PII, sensitive data,
+                malicious content, and other data types defined by the
+                configuration. Supports processing up to 10KB of text.
+            recipe: Recipe key of a configuration of data types and settings
+                defined in the Pangea User Console. It specifies the rules that
+                are to be applied to the text, such as defang malicious URLs.
+            debug: Setting this value to true will provide a detailed analysis
+                of the text data
+            llm_info: Short string hint for the LLM Provider information
+            log_field: Additional fields to include in activity log
+
+        Examples:
+            response = await ai_guard.guard_text("text")
+        """
+
+    @overload
+    async def guard_text(
+        self,
+        *,
+        messages: _T,
+        recipe: str | None = None,
+        debug: bool | None = None,
+        llm_info: str | None = None,
+        log_fields: LogFields | None = None,
+    ) -> PangeaResponse[TextGuardResult[_T]]:
+        """
+        Text Guard for scanning LLM inputs and outputs
+
+        Analyze and redact text to avoid manipulation of the model, addition of
+        malicious content, and other undesirable data transfers.
+
+        OperationId: ai_guard_post_v1_text_guard
+
+        Args:
+            messages: Structured messages data to be scanned by AI Guard for
+                PII, sensitive data, malicious content, and other data types
+                defined by the configuration. Supports processing up to 10KB of
+                JSON text
+            recipe: Recipe key of a configuration of data types and settings
+                defined in the Pangea User Console. It specifies the rules that
+                are to be applied to the text, such as defang malicious URLs.
+            debug: Setting this value to true will provide a detailed analysis
+                of the text data
+            llm_info: Short string hint for the LLM Provider information
+            log_field: Additional fields to include in activity log
+
+        Examples:
+            response = await ai_guard.guard_text(messages=[{"role": "user", "content": "hello world"}])
+        """
+
+    @overload
+    async def guard_text(
+        self,
+        *,
+        llm_input: _T,
+        recipe: str | None = None,
+        debug: bool | None = None,
+        llm_info: str | None = None,
+        log_fields: LogFields | None = None,
+    ) -> PangeaResponse[TextGuardResult[_T]]:
+        """
+        Text Guard for scanning LLM inputs and outputs
+
+        Analyze and redact text to avoid manipulation of the model, addition of
+        malicious content, and other undesirable data transfers.
+
+        OperationId: ai_guard_post_v1_text_guard
+
+        Args:
+            llm_input: Structured full llm payload data to be scanned by AI
+                Guard for PII, sensitive data, malicious content, and other data
+                types defined by the configuration. Supports processing up to
+                10KB of JSON text
+            recipe: Recipe key of a configuration of data types and settings
+                defined in the Pangea User Console. It specifies the rules that
+                are to be applied to the text, such as defang malicious URLs.
+            debug: Setting this value to true will provide a detailed analysis
+                of the text data
+            llm_info: Short string hint for the LLM Provider information
+            log_field: Additional fields to include in activity log
+
+        Examples:
+            response = await ai_guard.guard_text(
+                llm_input={"model": "gpt-4o", "messages": [{"role": "user", "content": "hello world"}]}
+            )
+        """
+
+    async def guard_text(  # type: ignore[misc]
+        self,
+        text: str | None = None,
+        *,
+        messages: _T | None = None,
+        llm_input: _T | None = None,
+        recipe: str | None = None,
+        debug: bool | None = None,
+        llm_info: str | None = None,
+        log_fields: LogFields | None = None,
+    ) -> PangeaResponse[TextGuardResult[None]]:
+        """
+        Text Guard for scanning LLM inputs and outputs
+
+        Analyze and redact text to avoid manipulation of the model, addition of
+        malicious content, and other undesirable data transfers.
+
+        OperationId: ai_guard_post_v1_text_guard
+
+        Args:
+            text: Text to be scanned by AI Guard for PII, sensitive data,
+                malicious content, and other data types defined by the
+                configuration. Supports processing up to 10KB of text.
+            messages: Structured messages data to be scanned by AI Guard for
+                PII, sensitive data, malicious content, and other data types
+                defined by the configuration. Supports processing up to 10KB of
+                JSON text
+            llm_input: Structured full llm payload data to be scanned by AI
+                Guard for PII, sensitive data, malicious content, and other data
+                types defined by the configuration. Supports processing up to
+                10KB of JSON text
+            recipe: Recipe key of a configuration of data types and settings
+                defined in the Pangea User Console. It specifies the rules that
+                are to be applied to the text, such as defang malicious URLs.
+            debug: Setting this value to true will provide a detailed analysis
+                of the text data
+            llm_info: Short string hint for the LLM Provider information
+            log_field: Additional fields to include in activity log
+
+        Examples:
+            response = await ai_guard.guard_text("text")
+        """
+
+        if not any((text, messages, llm_input)):
+            raise ValueError("Exactly one of `text`, `messages`, or `llm_input` must be given")
+
+        if sum((text is not None, messages is not None, llm_input is not None)) > 1:
+            raise ValueError("Only one of `text`, `messages`, or `llm_input` can be given at once")
+
+        return await self.request.post(
+            "v1/text/guard",
+            TextGuardResult,
+            data={
+                "text": text,
+                "messages": messages,
+                "llm_input": llm_input,
+                "recipe": recipe,
+                "debug": debug,
+                "llm_info": llm_info,
+                "log_fields": log_fields,
+            },
+        )

{pangea_sdk-5.4.0b3 → pangea_sdk-5.5.0}/pangea/asyncio/services/prompt_guard.py

@@ -52,20 +52,24 @@ class PromptGuardAsync(ServiceBaseAsync):
         super().__init__(token, config, logger_name, config_id)
 
     async def guard(
-        self, messages: Iterable[Message], *, analyzers: Iterable[str] | None = None
+        self,
+        messages: Iterable[Message],
+        *,
+        analyzers: Iterable[str] | None = None,
+        classify: bool | None = None,
     ) -> PangeaResponse[GuardResult]:
         """
-        Guard (Beta)
+        Guard
 
         Guard messages.
 
-        How to install a [Beta release](https://pangea.cloud/docs/sdk/python/#beta-releases).
-
-        OperationId: prompt_guard_post_v1beta_guard
+        OperationId: prompt_guard_post_v1_guard
 
         Args:
-            messages: Messages.
-            analyzers: Specific analyzers to be used in the call.
+            messages: Prompt content and role array in JSON format. The
+              `content` is the text that will be analyzed for redaction.
+            analyzers: Specific analyzers to be used in the call
+            classify: Boolean to enable classification of the content
 
         Examples:
             from pangea.asyncio.services.prompt_guard import Message
@@ -73,4 +77,8 @@ 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, "analyzers": analyzers})
+        return await self.request.post(
+            "v1/guard",
+            GuardResult,
+            data={"messages": messages, "analyzers": analyzers, "classify": classify},
+        )

{pangea_sdk-5.4.0b3 → pangea_sdk-5.5.0}/pangea/asyncio/services/share.py

@@ -34,7 +34,7 @@ class ShareAsync(ServiceBaseAsync):
 
         Examples:
              config = PangeaConfig(domain="aws.us.pangea.cloud")
-             authz = ShareAsync(token="pangea_token", config=config)
+             share = ShareAsync(token="pangea_token", config=config)
         """
 
         super().__init__(token, config, logger_name, config_id=config_id)
@@ -51,7 +51,7 @@ class ShareAsync(ServiceBaseAsync):
             A PangeaResponse. Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/share).
 
         Examples:
-            response = share.buckets()
+            response = await share.buckets()
         """
 
         return await self.request.post("v1/buckets", m.BucketsResult)

{pangea_sdk-5.4.0b3 → pangea_sdk-5.5.0}/pangea/config.py

@@ -2,7 +2,7 @@
 # Author: Pangea Cyber Corporation
 
 from dataclasses import dataclass
-from typing import Optional
+from typing import Literal, Optional
 
 
 @dataclass
@@ -16,10 +16,12 @@ class PangeaConfig:
     scheme (http:// or https://), subdomain, domain and port.
     """
 
-    environment: str = "production"
+    environment: Literal["production", "local"] = "production"
     """
-    Used to generate service url.
-    It should be only 'production' or 'local' in cases of particular services that can run locally as Redact.
+    Pangea environment, used to construct service URLs.
+
+    If set to "local", then `domain` must be the full host (i.e., hostname and
+    port) for the Pangea service that this `PangeaConfig` will be used for.
     """
 
     config_id: Optional[str] = None

pangea_sdk-5.5.0/pangea/services/ai_guard.py

@@ -0,0 +1,323 @@
+from __future__ import annotations
+
+from typing import Any, Dict, Generic, List, Optional, TypeVar, overload
+
+from pangea.config import PangeaConfig
+from pangea.response import APIRequestModel, APIResponseModel, PangeaResponse, PangeaResponseResult
+from pangea.services.base import ServiceBase
+
+
+class LogFields(APIRequestModel):
+    """Additional fields to include in activity log"""
+
+    citations: Optional[str] = None
+    """Origin or source application of the event"""
+
+    extra_info: Optional[str] = None
+    """Stores supplementary details related to the event"""
+
+    model: Optional[str] = None
+    """Model used to perform the event"""
+
+    source: Optional[str] = None
+    """IP address of user or app or agent"""
+
+    tools: Optional[str] = None
+    """Tools used to perform the event"""
+
+
+class AnalyzerResponse(APIResponseModel):
+    analyzer: str
+    confidence: float
+
+
+class PromptInjectionResult(APIResponseModel):
+    action: str
+    analyzer_responses: List[AnalyzerResponse]
+    """Triggered prompt injection analyzers."""
+
+
+class PiiEntity(APIResponseModel):
+    type: str
+    value: str
+    action: str
+    start_pos: Optional[int] = None
+
+
+class PiiEntityResult(APIResponseModel):
+    entities: List[PiiEntity]
+
+
+class MaliciousEntity(APIResponseModel):
+    type: str
+    value: str
+    action: str
+    start_pos: Optional[int] = None
+    raw: Optional[Dict[str, Any]] = None
+
+
+class MaliciousEntityResult(APIResponseModel):
+    entities: List[MaliciousEntity]
+
+
+class SecretsEntity(APIResponseModel):
+    type: str
+    value: str
+    action: str
+    start_pos: Optional[int] = None
+    redacted_value: Optional[str] = None
+
+
+class SecretsEntityResult(APIResponseModel):
+    entities: List[SecretsEntity]
+
+
+class LanguageDetectionResult(APIResponseModel):
+    language: str
+    action: str
+
+
+class CodeDetectionResult(APIResponseModel):
+    language: str
+    action: str
+
+
+_T = TypeVar("_T")
+
+
+class TextGuardDetector(APIResponseModel, Generic[_T]):
+    detected: bool
+    data: Optional[_T] = None
+
+
+class TextGuardDetectors(APIResponseModel):
+    prompt_injection: Optional[TextGuardDetector[PromptInjectionResult]] = None
+    pii_entity: Optional[TextGuardDetector[PiiEntityResult]] = None
+    malicious_entity: Optional[TextGuardDetector[MaliciousEntityResult]] = None
+    secrets_detection: Optional[TextGuardDetector[SecretsEntityResult]] = None
+    profanity_and_toxicity: Optional[TextGuardDetector[Any]] = None
+    custom_entity: Optional[TextGuardDetector[Any]] = None
+    language_detection: Optional[TextGuardDetector[LanguageDetectionResult]] = None
+    code_detection: Optional[TextGuardDetector[CodeDetectionResult]] = None
+
+
+class TextGuardResult(PangeaResponseResult, Generic[_T]):
+    detectors: TextGuardDetectors
+    """Result of the recipe analyzing and input prompt."""
+
+    prompt_text: Optional[str] = None
+    """Updated prompt text, if applicable."""
+
+    prompt_messages: Optional[_T] = None
+    """Updated structured prompt, if applicable."""
+
+    blocked: bool
+
+
+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)
+
+    @overload
+    def guard_text(
+        self,
+        text: str,
+        *,
+        recipe: str | None = None,
+        debug: bool | None = None,
+        llm_info: str | None = None,
+        log_fields: LogFields | None = None,
+    ) -> PangeaResponse[TextGuardResult[None]]:
+        """
+        Text Guard for scanning LLM inputs and outputs
+
+        Analyze and redact text to avoid manipulation of the model, addition of
+        malicious content, and other undesirable data transfers.
+
+        OperationId: ai_guard_post_v1_text_guard
+
+        Args:
+            text: Text to be scanned by AI Guard for PII, sensitive data,
+                malicious content, and other data types defined by the
+                configuration. Supports processing up to 10KB of text.
+            recipe: Recipe key of a configuration of data types and settings
+                defined in the Pangea User Console. It specifies the rules that
+                are to be applied to the text, such as defang malicious URLs.
+            debug: Setting this value to true will provide a detailed analysis
+                of the text data
+            llm_info: Short string hint for the LLM Provider information
+            log_field: Additional fields to include in activity log
+
+        Examples:
+            response = ai_guard.guard_text("text")
+        """
+
+    @overload
+    def guard_text(
+        self,
+        *,
+        messages: _T,
+        recipe: str | None = None,
+        debug: bool | None = None,
+        llm_info: str | None = None,
+        log_fields: LogFields | None = None,
+    ) -> PangeaResponse[TextGuardResult[_T]]:
+        """
+        Text Guard for scanning LLM inputs and outputs
+
+        Analyze and redact text to avoid manipulation of the model, addition of
+        malicious content, and other undesirable data transfers.
+
+        OperationId: ai_guard_post_v1_text_guard
+
+        Args:
+            messages: Structured messages data to be scanned by AI Guard for
+                PII, sensitive data, malicious content, and other data types
+                defined by the configuration. Supports processing up to 10KB of
+                JSON text
+            recipe: Recipe key of a configuration of data types and settings
+                defined in the Pangea User Console. It specifies the rules that
+                are to be applied to the text, such as defang malicious URLs.
+            debug: Setting this value to true will provide a detailed analysis
+                of the text data
+            llm_info: Short string hint for the LLM Provider information
+            log_field: Additional fields to include in activity log
+
+        Examples:
+            response = ai_guard.guard_text(messages=[{"role": "user", "content": "hello world"}])
+        """
+
+    @overload
+    def guard_text(
+        self,
+        *,
+        llm_input: _T,
+        recipe: str | None = None,
+        debug: bool | None = None,
+        llm_info: str | None = None,
+        log_fields: LogFields | None = None,
+    ) -> PangeaResponse[TextGuardResult[_T]]:
+        """
+        Text Guard for scanning LLM inputs and outputs
+
+        Analyze and redact text to avoid manipulation of the model, addition of
+        malicious content, and other undesirable data transfers.
+
+        OperationId: ai_guard_post_v1_text_guard
+
+        Args:
+            llm_input: Structured full llm payload data to be scanned by AI
+                Guard for PII, sensitive data, malicious content, and other data
+                types defined by the configuration. Supports processing up to
+                10KB of JSON text
+            recipe: Recipe key of a configuration of data types and settings
+                defined in the Pangea User Console. It specifies the rules that
+                are to be applied to the text, such as defang malicious URLs.
+            debug: Setting this value to true will provide a detailed analysis
+                of the text data
+            llm_info: Short string hint for the LLM Provider information
+            log_field: Additional fields to include in activity log
+
+        Examples:
+            response = ai_guard.guard_text(
+                llm_input={"model": "gpt-4o", "messages": [{"role": "user", "content": "hello world"}]}
+            )
+        """
+
+    def guard_text(  # type: ignore[misc]
+        self,
+        text: str | None = None,
+        *,
+        messages: _T | None = None,
+        llm_input: _T | None = None,
+        recipe: str | None = None,
+        debug: bool | None = None,
+        llm_info: str | None = None,
+        log_fields: LogFields | None = None,
+    ) -> PangeaResponse[TextGuardResult[None]]:
+        """
+        Text Guard for scanning LLM inputs and outputs
+
+        Analyze and redact text to avoid manipulation of the model, addition of
+        malicious content, and other undesirable data transfers.
+
+        OperationId: ai_guard_post_v1_text_guard
+
+        Args:
+            text: Text to be scanned by AI Guard for PII, sensitive data,
+                malicious content, and other data types defined by the
+                configuration. Supports processing up to 10KB of text.
+            messages: Structured messages data to be scanned by AI Guard for
+                PII, sensitive data, malicious content, and other data types
+                defined by the configuration. Supports processing up to 10KB of
+                JSON text
+            llm_input: Structured full llm payload data to be scanned by AI
+                Guard for PII, sensitive data, malicious content, and other data
+                types defined by the configuration. Supports processing up to
+                10KB of JSON text
+            recipe: Recipe key of a configuration of data types and settings
+                defined in the Pangea User Console. It specifies the rules that
+                are to be applied to the text, such as defang malicious URLs.
+            debug: Setting this value to true will provide a detailed analysis
+                of the text data
+            llm_info: Short string hint for the LLM Provider information
+            log_field: Additional fields to include in activity log
+
+        Examples:
+            response = ai_guard.guard_text("text")
+        """
+
+        if not any((text, messages, llm_input)):
+            raise ValueError("At least one of `text`, `messages`, or `llm_input` must be given")
+
+        if sum((text is not None, messages is not None, llm_input is not None)) > 1:
+            raise ValueError("Only one of `text`, `messages`, or `llm_input` can be given at once")
+
+        return self.request.post(
+            "v1/text/guard",
+            TextGuardResult,
+            data={
+                "text": text,
+                "messages": messages,
+                "llm_input": llm_input,
+                "recipe": recipe,
+                "debug": debug,
+                "llm_info": llm_info,
+                "log_fields": log_fields,
+            },
+        )

{pangea_sdk-5.4.0b3 → pangea_sdk-5.5.0}/pangea/services/prompt_guard.py

@@ -19,8 +19,8 @@ class Classification(APIResponseModel):
     category: str
     """Classification category"""
 
-    label: str
-    """Classification label"""
+    detected: bool
+    """Classification detection result"""
 
     confidence: float
     """Confidence score for the classification"""
@@ -30,7 +30,7 @@ class GuardResult(PangeaResponseResult):
     detected: bool
     """Boolean response for if the prompt was considered malicious or not"""
 
-    type: Optional[Literal["direct", "indirect"]] = None
+    type: Optional[Literal["direct", "indirect", ""]] = None
     """Type of analysis, either direct or indirect"""
 
     analyzer: Optional[str] = None
@@ -86,20 +86,24 @@ class PromptGuard(ServiceBase):
         super().__init__(token, config, logger_name, config_id)
 
     def guard(
-        self, messages: Iterable[Message], *, analyzers: Iterable[str] | None = None
+        self,
+        messages: Iterable[Message],
+        *,
+        analyzers: Iterable[str] | None = None,
+        classify: bool | None = None,
     ) -> PangeaResponse[GuardResult]:
         """
-        Guard (Beta)
+        Guard
 
         Guard messages.
 
-        How to install a [Beta release](https://pangea.cloud/docs/sdk/python/#beta-releases).
-
-        OperationId: prompt_guard_post_v1beta_guard
+        OperationId: prompt_guard_post_v1_guard
 
         Args:
-            messages: Prompt content and role array.
-            analyzers: Specific analyzers to be used in the call.
+            messages: Prompt content and role array in JSON format. The
+              `content` is the text that will be analyzed for redaction.
+            analyzers: Specific analyzers to be used in the call
+            classify: Boolean to enable classification of the content
 
         Examples:
             from pangea.services.prompt_guard import Message
@@ -107,4 +111,8 @@ class PromptGuard(ServiceBase):
             response = prompt_guard.guard([Message(role="user", content="hello world")])
         """
 
-        return self.request.post("v1beta/guard", GuardResult, data={"messages": messages, "analyzers": analyzers})
+        return self.request.post(
+            "v1/guard",
+            GuardResult,
+            data={"messages": messages, "analyzers": analyzers, "classify": classify},
+        )

{pangea_sdk-5.4.0b3 → pangea_sdk-5.5.0}/pangea/services/share/share.py

@@ -762,7 +762,7 @@ class Share(ServiceBase):
 
         Examples:
              config = PangeaConfig(domain="aws.us.pangea.cloud")
-             authz = Share(token="pangea_token", config=config)
+             share = Share(token="pangea_token", config=config)
         """
 
         super().__init__(token, config, logger_name, config_id=config_id)

{pangea_sdk-5.4.0b3 → pangea_sdk-5.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pangea-sdk"
-version = "5.4.0beta3"
+version = "5.5.0"
 description = "Pangea API SDK"
 authors = [
     {name = "Glenn Gallien", email = "glenn.gallien@pangea.cloud"}
@@ -16,13 +16,13 @@ classifiers = [
 ]
 requires-python = ">=3.9,<4.0.0"
 dependencies = [
-  "aiohttp (>=3.11.11,<4.0.0)",
+  "aiohttp (>=3.11.12,<4.0.0)",
   "cryptography (>=43.0.3,<44.0.0)",
-  "deprecated (>=1.2.15,<2.0.0)",
-  "google-crc32c (>=1.5.0,<2.0.0)",
-  "pydantic (>=2.10.5,<3.0.0)",
-  "python-dateutil (>=2.9.0,<3.0.0)",
-  "requests (>=2.31.0,<3.0.0)",
+  "deprecated (>=1.2.18,<2.0.0)",
+  "google-crc32c (>=1.6.0,<2.0.0)",
+  "pydantic (>=2.10.6,<3.0.0)",
+  "python-dateutil (>=2.9.0.post0,<3.0.0)",
+  "requests (>=2.32.3,<3.0.0)",
   "requests-toolbelt (>=1.0.0,<2.0.0)",
   "typing-extensions (>=4.12.2,<5.0.0)"
 ]
@@ -33,10 +33,10 @@ packages = [
 ]
 
 [tool.poetry.group.dev.dependencies]
-black = "^24.10.0"
+black = "^25.1.0"
 docstring-parser = "^0.15"
-isort = "^5.13.2"
-mypy = "1.14.1"
+isort = "^6.0.0"
+mypy = "1.15.0"
 types-Deprecated = "^1.2.9.3"
 types-python-dateutil = "^2.8.19.14"
 types-requests = "^2.31.0.10"

pangea_sdk-5.4.0b3/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_sdk-5.4.0b3/pangea/services/ai_guard.py

@@ -1,128 +0,0 @@
-from __future__ import annotations
-
-from typing import Any, Dict, Generic, List, Optional, TypeVar
-
-from pangea.config import PangeaConfig
-from pangea.response import APIResponseModel, PangeaResponse, PangeaResponseResult
-from pangea.services.base import ServiceBase
-
-
-class AnalyzerResponse(APIResponseModel):
-    analyzer: str
-    confidence: float
-
-
-class PromptInjectionResult(APIResponseModel):
-    analyzer_responses: List[AnalyzerResponse]
-    """Triggered prompt injection analyzers."""
-
-
-class PiiEntity(APIResponseModel):
-    type: str
-    value: str
-    redacted: bool
-    start_pos: Optional[int] = None
-
-
-class PiiEntityResult(APIResponseModel):
-    entities: List[PiiEntity]
-
-
-class MaliciousEntity(APIResponseModel):
-    type: str
-    value: str
-    redacted: Optional[bool] = None
-    start_pos: Optional[int] = None
-    raw: Optional[Dict[str, Any]] = None
-
-
-class MaliciousEntityResult(APIResponseModel):
-    entities: List[MaliciousEntity]
-
-
-T = TypeVar("T")
-
-
-class TextGuardDetector(APIResponseModel, Generic[T]):
-    detected: bool
-    data: Optional[T] = None
-
-
-class TextGuardDetectors(APIResponseModel):
-    prompt_injection: Optional[TextGuardDetector[PromptInjectionResult]] = None
-    pii_entity: Optional[TextGuardDetector[PiiEntityResult]] = None
-    malicious_entity: Optional[TextGuardDetector[MaliciousEntityResult]] = None
-
-
-class TextGuardResult(PangeaResponseResult):
-    detectors: TextGuardDetectors
-    prompt: str
-
-
-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}
-        )