Add Prompt Guard config APIs

Author: kenany

CHANGELOG.md

@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - AI Guard: config APIs.
+- Prompt Guard: config APIs.
 
 ## 6.1.1 - 2025-05-12
 

packages/pangea-sdk/pangea/asyncio/services/prompt_guard.py

@@ -1,10 +1,18 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Literal
 
 from pangea.asyncio.services.base import ServiceBaseAsync
 from pangea.config import PangeaConfig
-from pangea.services.prompt_guard import GuardResult, Message
+from pangea.response import PangeaResponseResult
+from pangea.services.prompt_guard import (
+    AuditDataActivityConfig,
+    GuardResult,
+    Message,
+    ServiceConfigFilter,
+    ServiceConfigsPage,
+)
 
 if TYPE_CHECKING:
     from collections.abc import Iterable
@@ -82,3 +90,105 @@ async def guard(
             GuardResult,
             data={"messages": messages, "analyzers": analyzers, "classify": classify},
         )
+
+    async def get_service_config(
+        self,
+        *,
+        id: str | None = None,
+        version: str | None = None,
+        analyzers: Mapping[str, bool] | None = None,
+        malicious_detection_threshold: float | None = None,
+        benign_detection_threshold: float | None = None,
+        audit_data_activity: AuditDataActivityConfig | None = None,
+    ) -> PangeaResponse[PangeaResponseResult]:
+        """
+        OperationId: prompt_guard_post_v1beta_config
+        """
+        return await self.request.post(
+            "v1beta/config",
+            data={
+                "id": id,
+                "version": version,
+                "analyzers": analyzers,
+                "malicious_detection_threshold": malicious_detection_threshold,
+                "benign_detection_threshold": benign_detection_threshold,
+                "audit_data_activity": audit_data_activity,
+            },
+            result_class=PangeaResponseResult,
+        )
+
+    async def create_service_config(
+        self,
+        *,
+        id: str | None = None,
+        version: str | None = None,
+        analyzers: Mapping[str, bool] | None = None,
+        malicious_detection_threshold: float | None = None,
+        benign_detection_threshold: float | None = None,
+        audit_data_activity: AuditDataActivityConfig | None = None,
+    ) -> PangeaResponse[PangeaResponseResult]:
+        """
+        OperationId: prompt_guard_post_v1beta_config_create
+        """
+        return await self.request.post(
+            "v1beta/config/create",
+            data={
+                "id": id,
+                "version": version,
+                "analyzers": analyzers,
+                "malicious_detection_threshold": malicious_detection_threshold,
+                "benign_detection_threshold": benign_detection_threshold,
+                "audit_data_activity": audit_data_activity,
+            },
+            result_class=PangeaResponseResult,
+        )
+
+    async def update_service_config(
+        self,
+        *,
+        id: str | None = None,
+        version: str | None = None,
+        analyzers: Mapping[str, bool] | None = None,
+        malicious_detection_threshold: float | None = None,
+        benign_detection_threshold: float | None = None,
+        audit_data_activity: AuditDataActivityConfig | None = None,
+    ) -> PangeaResponse[PangeaResponseResult]:
+        """
+        OperationId: prompt_guard_post_v1beta_config_update
+        """
+        return await self.request.post(
+            "v1beta/config/update",
+            data={
+                "id": id,
+                "version": version,
+                "analyzers": analyzers,
+                "malicious_detection_threshold": malicious_detection_threshold,
+                "benign_detection_threshold": benign_detection_threshold,
+                "audit_data_activity": audit_data_activity,
+            },
+            result_class=PangeaResponseResult,
+        )
+
+    async def delete_service_config(self, id: str) -> PangeaResponse[PangeaResponseResult]:
+        """
+        OperationId: prompt_guard_post_v1beta_config_delete
+        """
+        return await self.request.post("v1beta/config/delete", data={"id": id}, result_class=PangeaResponseResult)
+
+    async def list_service_configs(
+        self,
+        *,
+        filter: ServiceConfigFilter | None = None,
+        last: str | None = None,
+        order: Literal["asc", "desc"] | None = None,
+        order_by: Literal["id", "created_at", "updated_at"] | None = None,
+        size: int | None = None,
+    ) -> PangeaResponse[ServiceConfigsPage]:
+        """
+        OperationId: prompt_guard_post_v1beta_config_list
+        """
+        return await self.request.post(
+            "v1beta/config/list",
+            data={"filter": filter, "last": last, "order": order, "order_by": order_by, "size": size},
+            result_class=ServiceConfigsPage,
+        )

packages/pangea-sdk/pangea/services/ai_guard.py

@@ -1,14 +1,13 @@
 from __future__ import annotations
 
 from collections.abc import Mapping
-from datetime import datetime
 from typing import Annotated, Generic, Literal, Optional, Union, overload
 
 from pydantic import BaseModel, ConfigDict, Field, RootModel
 from typing_extensions import TypeVar
 
 from pangea.config import PangeaConfig
-from pangea.response import APIRequestModel, APIResponseModel, PangeaResponse, PangeaResponseResult
+from pangea.response import APIRequestModel, APIResponseModel, PangeaDateTime, PangeaResponse, PangeaResponseResult
 from pangea.services.base import ServiceBase
 
 # This is named "prompt injection" in the API spec even though it is also used
@@ -648,43 +647,43 @@ class ServiceConfigFilter(BaseModel):
     """
     Only records where id equals one of the provided substrings.
     """
-    created_at: Optional[datetime] = None
+    created_at: Optional[PangeaDateTime] = None
     """
     Only records where created_at equals this value.
     """
-    created_at__gt: Optional[datetime] = None
+    created_at__gt: Optional[PangeaDateTime] = None
     """
     Only records where created_at is greater than this value.
     """
-    created_at__gte: Optional[datetime] = None
+    created_at__gte: Optional[PangeaDateTime] = None
     """
     Only records where created_at is greater than or equal to this value.
     """
-    created_at__lt: Optional[datetime] = None
+    created_at__lt: Optional[PangeaDateTime] = None
     """
     Only records where created_at is less than this value.
     """
-    created_at__lte: Optional[datetime] = None
+    created_at__lte: Optional[PangeaDateTime] = None
     """
     Only records where created_at is less than or equal to this value.
     """
-    updated_at: Optional[datetime] = None
+    updated_at: Optional[PangeaDateTime] = None
     """
     Only records where updated_at equals this value.
     """
-    updated_at__gt: Optional[datetime] = None
+    updated_at__gt: Optional[PangeaDateTime] = None
     """
     Only records where updated_at is greater than this value.
     """
-    updated_at__gte: Optional[datetime] = None
+    updated_at__gte: Optional[PangeaDateTime] = None
     """
     Only records where updated_at is greater than or equal to this value.
     """
-    updated_at__lt: Optional[datetime] = None
+    updated_at__lt: Optional[PangeaDateTime] = None
     """
     Only records where updated_at is less than this value.
     """
-    updated_at__lte: Optional[datetime] = None
+    updated_at__lte: Optional[PangeaDateTime] = None
     """
     Only records where updated_at is less than or equal to this value.
     """