codemie-sdk-python 0.1.452__tar.gz → 0.1.453__tar.gz

{codemie_sdk_python-0.1.452 → codemie_sdk_python-0.1.453}/PKG-INFO

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

{codemie_sdk_python-0.1.452 → codemie_sdk_python-0.1.453}/pyproject.toml

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

{codemie_sdk_python-0.1.452 → codemie_sdk_python-0.1.453}/src/codemie_sdk/__init__.py

@@ -98,6 +98,18 @@ 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 .models.vendor_runtime import (
+    VendorRuntimeSetting,
+    VendorRuntimeSettingsResponse,
+    VendorRuntime,
+    VendorRuntimeStatus,
+    VendorRuntimesResponse,
+    VendorRuntimeEndpointStatus,
+    VendorRuntimeEndpoint,
+    VendorRuntimeEndpointsResponse,
+    VendorRuntimeEndpointDetail,
+)
+from .services.vendor_runtime import VendorRuntimeService
 from .services.codemie_guardrails import CodemieGuardrailService
 from .services.analytics import AnalyticsService
 from .models.mermaid import (
@@ -187,6 +199,16 @@ __all__ = [
     "VendorGuardrailInstallResponse",
     "VendorGuardrailUninstallResponse",
     "VendorGuardrailService",
+    "VendorRuntimeSetting",
+    "VendorRuntimeSettingsResponse",
+    "VendorRuntime",
+    "VendorRuntimeStatus",
+    "VendorRuntimesResponse",
+    "VendorRuntimeEndpointStatus",
+    "VendorRuntimeEndpoint",
+    "VendorRuntimeEndpointsResponse",
+    "VendorRuntimeEndpointDetail",
+    "VendorRuntimeService",
     "GuardrailAssignmentSetting",
     "GuardrailAssignmentEntity",
     "GuardrailAssignmentRequest",

{codemie_sdk_python-0.1.452 → codemie_sdk_python-0.1.453}/src/codemie_sdk/client/client.py

@@ -23,6 +23,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.vendor_runtime import VendorRuntimeService
 from ..services.codemie_guardrails import CodemieGuardrailService
 
 
@@ -150,6 +151,9 @@ class CodeMieClient:
         self.vendor_guardrails = VendorGuardrailService(
             self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
+        self.vendor_runtimes = VendorRuntimeService(
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
+        )
         self.codemie_guardrails = CodemieGuardrailService(
             self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )

codemie_sdk_python-0.1.453/src/codemie_sdk/models/vendor_runtime.py

@@ -0,0 +1,121 @@
+"""Models for vendor AgentCore runtime settings."""
+
+from datetime import datetime
+from enum import Enum
+
+from pydantic import BaseModel, ConfigDict, Field
+
+from .vendor_assistant import PaginationInfo, TokenPagination
+
+
+class VendorRuntimeStatus(str, Enum):
+    """Status of a vendor AgentCore runtime. AWS READY maps to PREPARED."""
+
+    PREPARED = "PREPARED"
+    NOT_PREPARED = "NOT_PREPARED"
+
+
+class VendorRuntimeSetting(BaseModel):
+    """Model representing a vendor runtime setting."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    setting_id: str = Field(..., description="Unique identifier for the setting")
+    setting_name: str = Field(..., description="Name of the setting")
+    project: str = Field(..., description="Project associated with the setting")
+    entities: list[str] = Field(
+        default_factory=list, description="List of entities associated with the setting"
+    )
+    invalid: bool | None = Field(None, description="Whether the setting is invalid")
+    error: str | None = Field(
+        None, description="Error message if the setting is invalid"
+    )
+
+
+class VendorRuntimeSettingsResponse(BaseModel):
+    """Response model for vendor runtime settings list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: list[VendorRuntimeSetting] = Field(
+        ..., description="List of vendor runtime settings"
+    )
+    pagination: PaginationInfo = Field(..., description="Pagination information")
+
+
+class VendorRuntime(BaseModel):
+    """Model representing a vendor AgentCore runtime."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the runtime")
+    name: str = Field(..., description="Name of the runtime")
+    status: VendorRuntimeStatus = Field(..., description="Status of the runtime")
+    description: str | None = Field(None, description="Description of the runtime")
+    version: str | None = Field(None, description="Version of the runtime")
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+
+
+class VendorRuntimesResponse(BaseModel):
+    """Response model for vendor runtimes list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: list[VendorRuntime] = Field(..., description="List of vendor runtimes")
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorRuntimeEndpointStatus(str, Enum):
+    """Status of a vendor AgentCore runtime endpoint. AWS READY maps to PREPARED."""
+
+    PREPARED = "PREPARED"
+    NOT_PREPARED = "NOT_PREPARED"
+
+
+class VendorRuntimeEndpoint(BaseModel):
+    """Model representing a vendor AgentCore runtime endpoint."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the endpoint")
+    name: str = Field(..., description="Name of the endpoint")
+    status: VendorRuntimeEndpointStatus = Field(
+        ..., description="Status of the endpoint"
+    )
+    description: str | None = Field(None, description="Description of the endpoint")
+    liveVersion: str | None = Field(None, description="Currently live version")
+    targetVersion: str | None = Field(None, description="Target/in-progress version")
+    createdAt: datetime = Field(..., description="Creation timestamp")
+    updatedAt: datetime = Field(..., description="Last update timestamp")
+    aiRunId: str | None = Field(
+        None,
+        description="CodeMie assistant ID when endpoint is already imported; None otherwise",
+    )
+
+
+class VendorRuntimeEndpointsResponse(BaseModel):
+    """Response model for vendor runtime endpoints list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: list[VendorRuntimeEndpoint] = Field(..., description="List of endpoints")
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorRuntimeEndpointDetail(VendorRuntimeEndpoint):
+    """Detail model for a vendor AgentCore runtime endpoint (GET by name).
+
+    Extends the list view with fields only present in the detail response.
+    """
+
+    arn: str | None = Field(None, description="Endpoint ARN")
+    runtimeArn: str | None = Field(None, description="Parent runtime ARN")
+    failureReason: str | None = Field(
+        None, description="Failure reason when status is NOT_PREPARED"
+    )

codemie_sdk_python-0.1.453/src/codemie_sdk/services/vendor_runtime.py

@@ -0,0 +1,157 @@
+"""Vendor service implementation for managing cloud vendor AgentCore runtimes."""
+
+from typing import Union
+
+from ..models.vendor_assistant import VendorType
+from ..models.vendor_runtime import (
+    VendorRuntimeSettingsResponse,
+    VendorRuntimesResponse,
+    VendorRuntime,
+    VendorRuntimeEndpointsResponse,
+    VendorRuntimeEndpointDetail,
+)
+from ..utils import ApiRequestHandler, TokenSource
+
+
+class VendorRuntimeService:
+    """Service for managing cloud vendor AgentCore runtimes (AWS Bedrock)."""
+
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
+        self._api = ApiRequestHandler(api_domain, token, verify_ssl)
+
+    def get_runtime_settings(
+        self,
+        vendor: Union[VendorType, str],
+        page: int = 0,
+        per_page: int = 10,
+    ) -> VendorRuntimeSettingsResponse:
+        """Get runtime settings for a specific cloud vendor.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp).
+            page: Page number for pagination (0-based)
+            per_page: Number of items per page
+
+        Returns:
+            VendorRuntimeSettingsResponse containing list of settings and pagination info
+        """
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+        params = {"page": page, "per_page": per_page}
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/runtimes/settings",
+            VendorRuntimeSettingsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_runtimes(
+        self,
+        vendor: Union[VendorType, str],
+        setting_id: str,
+        per_page: int = 10,
+        next_token: str | None = None,
+    ) -> VendorRuntimesResponse:
+        """Get runtimes for a specific vendor setting.
+
+        Args:
+            vendor: Cloud vendor type.
+            setting_id: ID of the vendor setting to retrieve runtimes for
+            per_page: Number of items per page
+            next_token: Token for pagination (optional, for retrieving next page)
+
+        Returns:
+            VendorRuntimesResponse containing list of runtimes and pagination token
+        """
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+        params: dict = {"setting_id": setting_id, "per_page": per_page}
+        if next_token:
+            params["next_token"] = next_token
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/runtimes",
+            VendorRuntimesResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_runtime(
+        self,
+        vendor: Union[VendorType, str],
+        runtime_id: str,
+        setting_id: str,
+    ) -> VendorRuntime:
+        """Get a specific runtime by ID.
+
+        Args:
+            vendor: Cloud vendor type.
+            runtime_id: ID of the runtime to retrieve
+            setting_id: ID of the vendor setting
+
+        Returns:
+            VendorRuntime containing runtime details
+        """
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+        params = {"setting_id": setting_id}
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/runtimes/{runtime_id}",
+            VendorRuntime,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_endpoints(
+        self,
+        vendor: Union[VendorType, str],
+        runtime_id: str,
+        setting_id: str,
+        per_page: int = 10,
+        next_token: str | None = None,
+    ) -> VendorRuntimeEndpointsResponse:
+        """Get endpoints for a specific runtime.
+
+        Args:
+            vendor: Cloud vendor type.
+            runtime_id: ID of the runtime whose endpoints to retrieve
+            setting_id: ID of the vendor setting
+            per_page: Number of items per page
+            next_token: Token for pagination (optional)
+
+        Returns:
+            VendorRuntimeEndpointsResponse containing list of endpoints and pagination token
+        """
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+        params: dict = {"setting_id": setting_id, "per_page": per_page}
+        if next_token:
+            params["next_token"] = next_token
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/runtimes/{runtime_id}/endpoints",
+            VendorRuntimeEndpointsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_endpoint(
+        self,
+        vendor: Union[VendorType, str],
+        runtime_id: str,
+        endpoint_name: str,
+        setting_id: str,
+    ) -> VendorRuntimeEndpointDetail:
+        """Get detail for a specific runtime endpoint by name.
+
+        Args:
+            vendor: Cloud vendor type.
+            runtime_id: ID of the runtime that owns the endpoint.
+            endpoint_name: Name of the endpoint (route uses name, not ID).
+            setting_id: ID of the vendor setting.
+
+        Returns:
+            VendorRuntimeEndpointDetail with all list fields plus arn, runtimeArn, failureReason.
+        """
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+        params = {"setting_id": setting_id}
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/runtimes/{runtime_id}/endpoints/{endpoint_name}",
+            VendorRuntimeEndpointDetail,
+            params=params,
+            wrap_response=False,
+        )