codemie-sdk-python 0.1.456__tar.gz → 0.1.458__tar.gz

{codemie_sdk_python-0.1.456 → codemie_sdk_python-0.1.458}/PKG-INFO

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

{codemie_sdk_python-0.1.456 → codemie_sdk_python-0.1.458}/pyproject.toml

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

{codemie_sdk_python-0.1.456 → codemie_sdk_python-0.1.458}/src/codemie_sdk/__init__.py

@@ -108,6 +108,10 @@ from .models.vendor_runtime import (
     VendorRuntimeEndpoint,
     VendorRuntimeEndpointsResponse,
     VendorRuntimeEndpointDetail,
+    VendorRuntimeInstallRequest,
+    VendorRuntimeInstallSummary,
+    VendorRuntimeInstallResponse,
+    VendorRuntimeUninstallResponse,
 )
 from .services.vendor_runtime import VendorRuntimeService
 from .services.codemie_guardrails import CodemieGuardrailService
@@ -208,6 +212,10 @@ __all__ = [
     "VendorRuntimeEndpoint",
     "VendorRuntimeEndpointsResponse",
     "VendorRuntimeEndpointDetail",
+    "VendorRuntimeInstallRequest",
+    "VendorRuntimeInstallSummary",
+    "VendorRuntimeInstallResponse",
+    "VendorRuntimeUninstallResponse",
     "VendorRuntimeService",
     "GuardrailAssignmentSetting",
     "GuardrailAssignmentEntity",

{codemie_sdk_python-0.1.456 → codemie_sdk_python-0.1.458}/src/codemie_sdk/models/vendor_runtime.py

@@ -108,14 +108,65 @@ class VendorRuntimeEndpointsResponse(BaseModel):
     )
 
 
+class VendorRuntimeInstallRequest(BaseModel):
+    """Request model for installing a vendor AgentCore runtime endpoint into CodeMie."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Runtime ID")
+    agentcoreRuntimeEndpointName: str = Field(
+        ..., description="Endpoint name to install"
+    )
+    setting_id: str = Field(..., description="Vendor setting ID")
+    configuration_json: str = Field(
+        ..., description="JSON template string containing __QUERY_PLACEHOLDER__"
+    )
+
+
+class VendorRuntimeInstallSummary(BaseModel):
+    """Per-endpoint summary returned from a runtime install operation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    runtimeId: str = Field(..., description="Runtime ID")
+    endpointName: str = Field(..., description="Endpoint name")
+    aiRunId: str = Field(
+        ..., description="CodeMie assistant ID assigned to the endpoint"
+    )
+
+
+class VendorRuntimeInstallResponse(BaseModel):
+    """Response model for vendor runtime endpoint install."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    summary: list[VendorRuntimeInstallSummary] = Field(
+        ..., description="Per-endpoint installation results"
+    )
+
+
+class VendorRuntimeUninstallResponse(BaseModel):
+    """Response model for vendor runtime endpoint uninstall."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    success: bool = Field(..., description="Whether the uninstall succeeded")
+
+
 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")
+    model_config = ConfigDict(extra="ignore", populate_by_name=True)
+
+    arn: str | None = Field(
+        None, alias="agentRuntimeEndpointArn", description="Endpoint ARN"
+    )
+    runtimeArn: str | None = Field(
+        None, alias="agentRuntimeArn", description="Parent runtime ARN"
+    )
     failureReason: str | None = Field(
         None, description="Failure reason when status is NOT_PREPARED"
     )

{codemie_sdk_python-0.1.456 → codemie_sdk_python-0.1.458}/src/codemie_sdk/services/vendor_runtime.py

@@ -9,6 +9,9 @@ from ..models.vendor_runtime import (
     VendorRuntime,
     VendorRuntimeEndpointsResponse,
     VendorRuntimeEndpointDetail,
+    VendorRuntimeInstallRequest,
+    VendorRuntimeInstallResponse,
+    VendorRuntimeUninstallResponse,
 )
 from ..utils import ApiRequestHandler, TokenSource
 
@@ -38,7 +41,7 @@ class VendorRuntimeService:
         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",
+            f"/v1/vendors/{vendor_str}/agentcore-runtimes/settings",
             VendorRuntimeSettingsResponse,
             params=params,
             wrap_response=False,
@@ -67,7 +70,7 @@ class VendorRuntimeService:
         if next_token:
             params["next_token"] = next_token
         return self._api.get(
-            f"/v1/vendors/{vendor_str}/runtimes",
+            f"/v1/vendors/{vendor_str}/agentcore-runtimes",
             VendorRuntimesResponse,
             params=params,
             wrap_response=False,
@@ -92,7 +95,7 @@ class VendorRuntimeService:
         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}",
+            f"/v1/vendors/{vendor_str}/agentcore-runtimes/{runtime_id}",
             VendorRuntime,
             params=params,
             wrap_response=False,
@@ -123,12 +126,56 @@ class VendorRuntimeService:
         if next_token:
             params["next_token"] = next_token
         return self._api.get(
-            f"/v1/vendors/{vendor_str}/runtimes/{runtime_id}/endpoints",
+            f"/v1/vendors/{vendor_str}/agentcore-runtimes/{runtime_id}/endpoints",
             VendorRuntimeEndpointsResponse,
             params=params,
             wrap_response=False,
         )
 
+    def install_endpoints(
+        self,
+        vendor: Union[VendorType, str],
+        endpoints: list[VendorRuntimeInstallRequest],
+    ) -> VendorRuntimeInstallResponse:
+        """Install one or more vendor AgentCore runtime endpoints into CodeMie.
+
+        Args:
+            vendor: Cloud vendor type.
+            endpoints: List of endpoint install requests.
+
+        Returns:
+            VendorRuntimeInstallResponse with per-endpoint aiRunId summaries.
+        """
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+        payload = [e.model_dump(by_alias=True) for e in endpoints]
+        return self._api.post(
+            f"/v1/vendors/{vendor_str}/agentcore-runtimes",
+            VendorRuntimeInstallResponse,
+            json_data=payload,
+            wrap_response=False,
+        )
+
+    def uninstall_endpoint(
+        self,
+        vendor: Union[VendorType, str],
+        ai_run_id: str,
+    ) -> VendorRuntimeUninstallResponse:
+        """Uninstall a vendor AgentCore runtime endpoint from CodeMie.
+
+        Args:
+            vendor: Cloud vendor type.
+            ai_run_id: CodeMie assistant ID returned from install.
+
+        Returns:
+            VendorRuntimeUninstallResponse with success flag.
+        """
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+        return self._api.delete(
+            f"/v1/vendors/{vendor_str}/agentcore-runtimes/{ai_run_id}",
+            VendorRuntimeUninstallResponse,
+            wrap_response=False,
+        )
+
     def get_endpoint(
         self,
         vendor: Union[VendorType, str],
@@ -150,7 +197,7 @@ class VendorRuntimeService:
         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}",
+            f"/v1/vendors/{vendor_str}/agentcore-runtimes/{runtime_id}/{endpoint_name}",
             VendorRuntimeEndpointDetail,
             params=params,
             wrap_response=False,