codemie-sdk-python 0.1.276__tar.gz → 0.1.277__tar.gz

{codemie_sdk_python-0.1.276 → codemie_sdk_python-0.1.277}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.276
+Version: 0.1.277
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com

{codemie_sdk_python-0.1.276 → codemie_sdk_python-0.1.277}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.276"
+version = "0.1.277"
 description = "CodeMie SDK for Python"
 authors = [
     "Vadym Vlasenko <vadym_vlasenko@epam.com>",

{codemie_sdk_python-0.1.276 → codemie_sdk_python-0.1.277}/src/codemie_sdk/__init__.py

@@ -73,10 +73,17 @@ from .models.vendor_guardrail import (
     VendorGuardrailInstallResponse,
     VendorGuardrailUninstallResponse,
 )
+from .models.guardrails import (
+    GuardrailAssignmentSetting,
+    GuardrailAssignmentEntity,
+    GuardrailAssignmentRequest,
+    GuardrailAssignmentResponse,
+)
 from .services.vendor_assistant import VendorAssistantService
 from .services.vendor_workflow import VendorWorkflowService
 from .services.vendor_knowledgebase import VendorKnowledgeBaseService
 from .services.vendor_guardrail import VendorGuardrailService
+from .services.codemie_guardrails import CodemieGuardrailService
 
 __version__ = "0.2.12"
 __all__ = [
@@ -132,4 +139,9 @@ __all__ = [
     "VendorGuardrailInstallResponse",
     "VendorGuardrailUninstallResponse",
     "VendorGuardrailService",
+    "GuardrailAssignmentSetting",
+    "GuardrailAssignmentEntity",
+    "GuardrailAssignmentRequest",
+    "GuardrailAssignmentResponse",
+    "CodemieGuardrailService",
 ]

{codemie_sdk_python-0.1.276 → codemie_sdk_python-0.1.277}/src/codemie_sdk/client/client.py

@@ -18,6 +18,7 @@ from ..services.vendor_assistant import VendorAssistantService
 from ..services.vendor_workflow import VendorWorkflowService
 from ..services.vendor_knowledgebase import VendorKnowledgeBaseService
 from ..services.vendor_guardrail import VendorGuardrailService
+from ..services.codemie_guardrails import CodemieGuardrailService
 
 
 class CodeMieClient:
@@ -107,6 +108,9 @@ class CodeMieClient:
         self.vendor_guardrails = VendorGuardrailService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.codemie_guardrails = CodemieGuardrailService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
 
     @property
     def token(self) -> str:
@@ -175,4 +179,7 @@ class CodeMieClient:
         self.vendor_guardrails = VendorGuardrailService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.codemie_guardrails = CodemieGuardrailService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
         return self._token

codemie_sdk_python-0.1.277/src/codemie_sdk/models/guardrails.py

@@ -0,0 +1,62 @@
+"""Models for guardrail assignments."""
+
+from typing import List
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class GuardrailAssignmentSetting(BaseModel):
+    """Model for guardrail assignment settings."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    mode: str = Field(..., description="Assignment mode (e.g., 'all', 'specific')")
+    source: str = Field(..., description="Source type (e.g., 'input', 'output')")
+
+
+class GuardrailAssignmentEntity(BaseModel):
+    """Model for guardrail assignment to specific entities."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    settings: List[GuardrailAssignmentSetting] = Field(
+        default_factory=list, description="List of assignment settings"
+    )
+    items: List[str] = Field(
+        default_factory=list, description="List of specific entity IDs"
+    )
+
+
+class GuardrailAssignmentRequest(BaseModel):
+    """Request model for assigning guardrails to entities."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    project: GuardrailAssignmentEntity = Field(
+        default_factory=lambda: GuardrailAssignmentEntity(settings=[], items=[]),
+        description="Project-level assignments",
+    )
+    assistants: GuardrailAssignmentEntity = Field(
+        default_factory=lambda: GuardrailAssignmentEntity(settings=[], items=[]),
+        description="Assistant-level assignments",
+    )
+    workflows: GuardrailAssignmentEntity = Field(
+        default_factory=lambda: GuardrailAssignmentEntity(settings=[], items=[]),
+        description="Workflow-level assignments",
+    )
+    datasources: GuardrailAssignmentEntity = Field(
+        default_factory=lambda: GuardrailAssignmentEntity(settings=[], items=[]),
+        description="Datasource-level assignments",
+    )
+
+
+class GuardrailAssignmentResponse(BaseModel):
+    """Response model for guardrail assignment."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    success: int = Field(..., description="Count of successful assignments")
+    failed: int = Field(..., description="Count of failed assignments")
+    errors: List[str] = Field(
+        default_factory=list, description="List of error messages"
+    )

codemie_sdk_python-0.1.277/src/codemie_sdk/services/codemie_guardrails.py

@@ -0,0 +1,76 @@
+"""CodeMie guardrail service implementation for managing guardrail assignments."""
+
+from ..models.guardrails import (
+    GuardrailAssignmentRequest,
+    GuardrailAssignmentResponse,
+)
+from ..utils import ApiRequestHandler
+
+
+class CodemieGuardrailService:
+    """Service for managing CodeMie guardrail assignments."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the CodeMie guardrail service.
+
+        Args:
+            api_domain: Base URL for the CodeMie API
+            token: Authentication token
+            verify_ssl: Whether to verify SSL certificates
+        """
+        self._api = ApiRequestHandler(api_domain, token, verify_ssl)
+
+    def assign_guardrails(
+        self,
+        guardrail_id: str,
+        assignment: GuardrailAssignmentRequest,
+    ) -> GuardrailAssignmentResponse:
+        """Assign guardrails to project entities (assistants, workflows, datasources).
+
+        Args:
+            guardrail_id: ID of the guardrail to assign (AI run ID from installation)
+            assignment: Assignment configuration for different entity types
+
+        Returns:
+            GuardrailAssignmentResponse with success status
+
+        Example:
+            >>> from codemie_sdk.models.guardrails import (
+            ...     GuardrailAssignmentRequest,
+            ...     GuardrailAssignmentEntity,
+            ...     GuardrailAssignmentSetting
+            ... )
+            >>> # Assign guardrail to all project assistants
+            >>> assignment = GuardrailAssignmentRequest(
+            ...     assistants=GuardrailAssignmentEntity(
+            ...         settings=[GuardrailAssignmentSetting(mode="all", source="input")],
+            ...         items=[]
+            ...     )
+            ... )
+            >>> response = client.codemie_guardrails.assign_guardrails(
+            ...     guardrail_id="ai_run_123",
+            ...     assignment=assignment
+            ... )
+            >>> if response.success:
+            ...     print("Guardrail assigned successfully!")
+            >>>
+            >>> # Assign to specific assistants
+            >>> assignment = GuardrailAssignmentRequest(
+            ...     assistants=GuardrailAssignmentEntity(
+            ...         settings=[],
+            ...         items=["asst_123", "asst_456"]
+            ...     )
+            ... )
+            >>> response = client.codemie_guardrails.assign_guardrails(
+            ...     guardrail_id="ai_run_123",
+            ...     assignment=assignment
+            ... )
+        """
+        payload = assignment.model_dump(by_alias=True)
+
+        return self._api.put(
+            f"/v1/guardrails/{guardrail_id}/assignments",
+            GuardrailAssignmentResponse,
+            json_data=payload,
+            wrap_response=False,
+        )