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

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

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

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

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

{codemie_sdk_python-0.1.275 → 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.275 → codemie_sdk_python-0.1.277}/src/codemie_sdk/client/client.py

@@ -3,6 +3,7 @@
 from typing import Optional
 
 from ..auth.credentials import KeycloakCredentials
+from ..services.admin import AdminService
 from ..services.assistant import AssistantService
 from ..services.conversation import ConversationService
 from ..services.datasource import DatasourceService
@@ -17,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:
@@ -69,6 +71,7 @@ class CodeMieClient:
         self._token = "" if self._is_localhost else self.auth.get_token()
 
         # Initialize services with verify_ssl parameter and token
+        self.admin = AdminService(self._api_domain, self._token, verify_ssl=verify_ssl)
         self.assistants = AssistantService(
             self._api_domain, self._token, verify_ssl=verify_ssl
         )
@@ -105,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:
@@ -128,6 +134,9 @@ class CodeMieClient:
         """Force token refresh."""
         self._token = "" if self._is_localhost else self.auth.get_token()
         # Update token in services
+        self.admin = AdminService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
         self.assistants = AssistantService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
@@ -170,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/admin.py

@@ -0,0 +1,28 @@
+"""Admin models for managing applications/projects."""
+
+from typing import List
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class ApplicationsListResponse(BaseModel):
+    """Response model for list applications endpoint."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    applications: List[str] = Field(..., description="List of application names")
+
+
+class ApplicationCreateRequest(BaseModel):
+    """Request model for creating an application/project."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    name: str = Field(..., description="Application/project name")
+
+
+class ApplicationCreateResponse(BaseModel):
+    """Response model for create application endpoint."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    message: str = Field(..., description="Created application name")

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/admin.py

@@ -0,0 +1,60 @@
+"""Admin service implementation."""
+
+from typing import List, Optional
+
+from ..models.admin import (
+    ApplicationsListResponse,
+    ApplicationCreateRequest,
+    ApplicationCreateResponse,
+)
+from ..utils import ApiRequestHandler
+
+
+class AdminService:
+    """Service for managing CodeMie applications/projects."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the admin 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 list_applications(self, project_name: Optional[str] = None) -> List[str]:
+        """Get list of all applications/projects.
+
+        Args:
+            project_name: Optional project name to filter by
+
+        Returns:
+            List of application names
+        """
+        params = {}
+        if project_name:
+            params["search"] = project_name
+
+        response = self._api.get(
+            "/v1/admin/applications",
+            ApplicationsListResponse,
+            params=params if params else None,
+        )
+        return response.applications
+
+    def create_application(self, request: ApplicationCreateRequest) -> str:
+        """Create a new application/project.
+
+        Args:
+            request: Application creation request
+
+        Returns:
+            Created application name
+        """
+        response = self._api.post(
+            "/v1/admin/application",
+            ApplicationCreateResponse,
+            json_data=request.model_dump(exclude_none=True),
+        )
+        return response.message

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,
+        )