codemie-sdk-python 0.1.221__tar.gz → 0.1.223__tar.gz

{codemie_sdk_python-0.1.221 → codemie_sdk_python-0.1.223}/PKG-INFO

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

{codemie_sdk_python-0.1.221 → codemie_sdk_python-0.1.223}/pyproject.toml

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

codemie_sdk_python-0.1.223/src/codemie_sdk/__init__.py

@@ -0,0 +1,85 @@
+"""
+CodeMie SDK for Python
+~~~~~~~~~~~~~~~~~~~~~
+
+A Python SDK for interacting with CodeMie API.
+
+Basic usage:
+
+    >>> from codemie_sdk import CodeMieClient
+    >>> client = CodeMieClient(
+    ...     auth_server_url="https://auth.example.com",
+    ...     auth_client_id="client_id",
+    ...     auth_client_secret="secret",
+    ...     auth_realm_name="realm",
+    ...     codemie_api_domain="api.codemie.com"
+    ... )
+    >>> assistants = client.assistants.list()
+"""
+
+from .client.client import CodeMieClient
+from .models.vendor_assistant import (
+    VendorType,
+    VendorAssistantSetting,
+    VendorAssistantSettingsResponse,
+    VendorAssistant,
+    VendorAssistantVersion,
+    VendorAssistantStatus,
+    VendorAssistantsResponse,
+    VendorAssistantAlias,
+    VendorAssistantAliasesResponse,
+    VendorAssistantInstallRequest,
+    VendorAssistantInstallSummary,
+    VendorAssistantInstallResponse,
+    VendorAssistantUninstallResponse,
+    PaginationInfo,
+    TokenPagination,
+)
+from .models.vendor_workflow import (
+    VendorWorkflowSetting,
+    VendorWorkflowSettingsResponse,
+    VendorWorkflow,
+    VendorWorkflowStatus,
+    VendorWorkflowsResponse,
+    VendorWorkflowAlias,
+    VendorWorkflowAliasesResponse,
+    VendorWorkflowInstallRequest,
+    VendorWorkflowInstallSummary,
+    VendorWorkflowInstallResponse,
+    VendorWorkflowUninstallResponse,
+)
+from .services.vendor_assistant import VendorAssistantService
+from .services.vendor_workflow import VendorWorkflowService
+
+__version__ = "0.2.10"
+__all__ = [
+    "CodeMieClient",
+    "VendorType",
+    "VendorAssistantSetting",
+    "VendorAssistantSettingsResponse",
+    "VendorAssistant",
+    "VendorAssistantVersion",
+    "VendorAssistantStatus",
+    "VendorAssistantsResponse",
+    "VendorAssistantAlias",
+    "VendorAssistantAliasesResponse",
+    "VendorAssistantInstallRequest",
+    "VendorAssistantInstallSummary",
+    "VendorAssistantInstallResponse",
+    "VendorAssistantUninstallResponse",
+    "PaginationInfo",
+    "TokenPagination",
+    "VendorAssistantService",
+    "VendorWorkflowSetting",
+    "VendorWorkflowSettingsResponse",
+    "VendorWorkflow",
+    "VendorWorkflowStatus",
+    "VendorWorkflowsResponse",
+    "VendorWorkflowAlias",
+    "VendorWorkflowAliasesResponse",
+    "VendorWorkflowInstallRequest",
+    "VendorWorkflowInstallSummary",
+    "VendorWorkflowInstallResponse",
+    "VendorWorkflowUninstallResponse",
+    "VendorWorkflowService",
+]

{codemie_sdk_python-0.1.221 → codemie_sdk_python-0.1.223}/src/codemie_sdk/client/client.py

@@ -13,6 +13,8 @@ from ..services.user import UserService
 from ..services.workflow import WorkflowService
 from ..services.files import FileOperationService
 from ..services.webhook import WebhookService
+from ..services.vendor_assistant import VendorAssistantService
+from ..services.vendor_workflow import VendorWorkflowService
 
 
 class CodeMieClient:
@@ -89,6 +91,12 @@ class CodeMieClient:
         self.webhook = WebhookService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.vendor_assistants = VendorAssistantService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_workflows = VendorWorkflowService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
 
     @property
     def token(self) -> str:
@@ -135,4 +143,10 @@ class CodeMieClient:
         self.conversations = ConversationService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.vendor_assistants = VendorAssistantService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_workflows = VendorWorkflowService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
         return self._token

codemie_sdk_python-0.1.223/src/codemie_sdk/models/vendor_assistant.py

@@ -0,0 +1,187 @@
+"""Models for vendor assistant settings."""
+
+from datetime import datetime
+from enum import Enum
+from typing import Optional, List
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class VendorType(str, Enum):
+    """Supported cloud vendor types."""
+
+    AWS = "aws"
+    AZURE = "azure"
+    GCP = "gcp"
+
+
+class VendorAssistantStatus(str, Enum):
+    """Status of vendor assistant."""
+
+    PREPARED = "PREPARED"
+    NOT_PREPARED = "NOT_PREPARED"
+
+
+class VendorAssistantSetting(BaseModel):
+    """Model representing a vendor assistant 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: Optional[bool] = Field(None, description="Whether the setting is invalid")
+    error: Optional[str] = Field(
+        None, description="Error message if the setting is invalid"
+    )
+
+
+class PaginationInfo(BaseModel):
+    """Pagination information for list responses."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    total: int = Field(..., description="Total number of items")
+    pages: int = Field(..., description="Total number of pages")
+    page: int = Field(..., description="Current page number (0-based)")
+    per_page: int = Field(..., description="Number of items per page")
+
+
+class VendorAssistantSettingsResponse(BaseModel):
+    """Response model for vendor assistant settings list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorAssistantSetting] = Field(
+        ..., description="List of vendor assistant settings"
+    )
+    pagination: PaginationInfo = Field(..., description="Pagination information")
+
+
+class VendorAssistant(BaseModel):
+    """Model representing a vendor assistant."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the assistant")
+    name: str = Field(..., description="Name of the assistant")
+    status: VendorAssistantStatus = Field(..., description="Status of the assistant")
+    description: Optional[str] = Field(None, description="Description of the assistant")
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+
+
+class VendorAssistantVersion(BaseModel):
+    """Model representing a specific version of a vendor assistant with detailed information."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the assistant")
+    name: str = Field(..., description="Name of the assistant")
+    status: VendorAssistantStatus = Field(..., description="Status of the assistant")
+    version: str = Field(..., description="Version of the assistant")
+    instruction: str = Field(..., description="Instructions for the assistant")
+    foundationModel: str = Field(
+        ...,
+        description="ARN or identifier of the foundation model",
+        alias="foundationModel",
+    )
+    description: Optional[str] = Field(None, description="Description of the assistant")
+    createdAt: datetime = Field(
+        ..., description="Creation timestamp", alias="createdAt"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+
+
+class TokenPagination(BaseModel):
+    """Token-based pagination information."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    next_token: Optional[str] = Field(None, description="Token for the next page")
+
+
+class VendorAssistantsResponse(BaseModel):
+    """Response model for vendor assistants list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorAssistant] = Field(..., description="List of vendor assistants")
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorAssistantAlias(BaseModel):
+    """Model representing a vendor assistant alias."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the alias")
+    name: str = Field(..., description="Name of the alias")
+    status: VendorAssistantStatus = Field(..., description="Status of the alias")
+    description: Optional[str] = Field(None, description="Description of the alias")
+    version: str = Field(..., description="Version of the alias")
+    createdAt: datetime = Field(
+        ..., description="Creation timestamp", alias="createdAt"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+
+
+class VendorAssistantAliasesResponse(BaseModel):
+    """Response model for vendor assistant aliases list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorAssistantAlias] = Field(
+        ..., description="List of vendor assistant aliases"
+    )
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorAssistantInstallRequest(BaseModel):
+    """Model for a single assistant installation request."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Assistant ID to install")
+    agentAliasId: str = Field(..., description="Alias ID to use for the assistant")
+    setting_id: str = Field(..., description="Vendor setting ID")
+
+
+class VendorAssistantInstallSummary(BaseModel):
+    """Model for assistant installation summary."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    agentId: str = Field(..., description="Installed assistant ID")
+    agentAliasId: str = Field(..., description="Alias ID used for installation")
+    aiRunId: str = Field(..., description="AI run ID for the installation")
+
+
+class VendorAssistantInstallResponse(BaseModel):
+    """Response model for assistant installation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    summary: List[VendorAssistantInstallSummary] = Field(
+        ..., description="List of installation summaries"
+    )
+
+
+class VendorAssistantUninstallResponse(BaseModel):
+    """Response model for assistant uninstallation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    success: bool = Field(..., description="Whether the uninstallation was successful")

codemie_sdk_python-0.1.223/src/codemie_sdk/models/vendor_workflow.py

@@ -0,0 +1,145 @@
+"""Models for vendor workflow settings."""
+
+from datetime import datetime
+from enum import Enum
+from typing import Optional, List
+
+from pydantic import BaseModel, ConfigDict, Field
+
+from .vendor_assistant import PaginationInfo, TokenPagination
+
+
+class VendorWorkflowSetting(BaseModel):
+    """Model representing a vendor workflow 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: Optional[bool] = Field(None, description="Whether the setting is invalid")
+    error: Optional[str] = Field(
+        None, description="Error message if the setting is invalid"
+    )
+
+
+class VendorWorkflowSettingsResponse(BaseModel):
+    """Response model for vendor workflow settings list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorWorkflowSetting] = Field(
+        ..., description="List of vendor workflow settings"
+    )
+    pagination: PaginationInfo = Field(..., description="Pagination information")
+
+
+class VendorWorkflowStatus(str, Enum):
+    """Status of vendor workflow."""
+
+    PREPARED = "PREPARED"
+    NOT_PREPARED = "NOT_PREPARED"
+
+
+class VendorWorkflow(BaseModel):
+    """Model representing a vendor workflow."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the workflow")
+    name: str = Field(..., description="Name of the workflow")
+    status: VendorWorkflowStatus = Field(..., description="Status of the workflow")
+    description: Optional[str] = Field(None, description="Description of the workflow")
+    version: str = Field(..., description="Version of the workflow")
+    createdAt: datetime = Field(
+        ..., description="Creation timestamp", alias="createdAt"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+
+
+class VendorWorkflowsResponse(BaseModel):
+    """Response model for vendor workflows list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorWorkflow] = Field(..., description="List of vendor workflows")
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorWorkflowAlias(BaseModel):
+    """Model representing a vendor workflow alias."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the alias")
+    name: str = Field(..., description="Name of the alias")
+    status: VendorWorkflowStatus = Field(..., description="Status of the alias")
+    description: Optional[str] = Field(None, description="Description of the alias")
+    version: str = Field(..., description="Version of the alias")
+    createdAt: datetime = Field(
+        ..., description="Creation timestamp", alias="createdAt"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+    aiRunId: Optional[str] = Field(
+        None, description="AI run ID if the alias is installed", alias="aiRunId"
+    )
+
+
+class VendorWorkflowAliasesResponse(BaseModel):
+    """Response model for vendor workflow aliases list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorWorkflowAlias] = Field(
+        ..., description="List of vendor workflow aliases"
+    )
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorWorkflowInstallRequest(BaseModel):
+    """Model for a single workflow installation request."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Workflow ID to install")
+    flowAliasId: str = Field(..., description="Flow alias ID to use for the workflow")
+    setting_id: str = Field(..., description="Vendor setting ID")
+
+
+class VendorWorkflowInstallSummary(BaseModel):
+    """Model for workflow installation summary."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    flowId: str = Field(..., description="Installed workflow ID")
+    flowAliasId: str = Field(..., description="Flow alias ID used for installation")
+    aiRunId: str = Field(..., description="AI run ID for the installation")
+
+
+class VendorWorkflowInstallResponse(BaseModel):
+    """Response model for workflow installation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    summary: List[VendorWorkflowInstallSummary] = Field(
+        ..., description="List of installation summaries"
+    )
+
+
+class VendorWorkflowUninstallResponse(BaseModel):
+    """Response model for workflow uninstallation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    success: bool = Field(..., description="Whether the uninstallation was successful")

codemie_sdk_python-0.1.223/src/codemie_sdk/services/vendor_assistant.py

@@ -0,0 +1,364 @@
+"""Vendor service implementation for managing cloud vendor assistant settings."""
+
+from typing import Union, Optional, List
+
+from ..models.vendor_assistant import (
+    VendorType,
+    VendorAssistantSettingsResponse,
+    VendorAssistantsResponse,
+    VendorAssistant,
+    VendorAssistantVersion,
+    VendorAssistantAliasesResponse,
+    VendorAssistantInstallRequest,
+    VendorAssistantInstallResponse,
+    VendorAssistantUninstallResponse,
+)
+from ..utils import ApiRequestHandler
+
+
+class VendorAssistantService:
+    """Service for managing cloud vendor assistant settings (AWS, Azure, GCP)."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the vendor 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 get_assistant_settings(
+        self,
+        vendor: Union[VendorType, str],
+        page: int = 0,
+        per_page: int = 10,
+    ) -> VendorAssistantSettingsResponse:
+        """Get assistant settings for a specific cloud vendor.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            page: Page number for pagination (0-based)
+            per_page: Number of items per page
+
+        Returns:
+            VendorAssistantSettingsResponse containing list of settings and pagination info
+
+        Example:
+            >>> # Using enum
+            >>> settings = client.vendor_assistants.get_assistant_settings(VendorType.AWS, page=0, per_page=10)
+            >>> # Using string
+            >>> settings = client.vendor_assistants.get_assistant_settings("aws", page=0, per_page=10)
+        """
+        # Convert enum to string value if needed
+        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}/assistants/settings",
+            VendorAssistantSettingsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_assistants(
+        self,
+        vendor: Union[VendorType, str],
+        setting_id: str,
+        per_page: int = 10,
+        next_token: Optional[str] = None,
+    ) -> VendorAssistantsResponse:
+        """Get assistants for a specific vendor setting.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            setting_id: ID of the vendor setting to retrieve assistants for
+            per_page: Number of items per page
+            next_token: Token for pagination (optional, for retrieving next page)
+
+        Returns:
+            VendorAssistantsResponse containing list of assistants and pagination token
+
+        Example:
+            >>> # Get first page
+            >>> assistants = client.vendor_assistants.get_assistants(
+            ...     vendor=VendorType.AWS,
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...     per_page=8
+            ... )
+            >>> # Get next page if available
+            >>> if assistants.pagination.next_token:
+            ...     next_page = client.vendor_assistants.get_assistants(
+            ...         vendor=VendorType.AWS,
+            ...         setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...         per_page=8,
+            ...         next_token=assistants.pagination.next_token
+            ...     )
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "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}/assistants",
+            VendorAssistantsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_assistant(
+        self,
+        vendor: Union[VendorType, str],
+        assistant_id: str,
+        setting_id: str,
+    ) -> VendorAssistant:
+        """Get a specific assistant by ID for a vendor setting.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            assistant_id: ID of the assistant to retrieve
+            setting_id: ID of the vendor setting
+
+        Returns:
+            VendorAssistant containing assistant details
+
+        Example:
+            >>> assistant = client.vendor_assistants.get_assistant(
+            ...     vendor=VendorType.AWS,
+            ...     assistant_id="TJBKR0DGWT",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> print(f"Name: {assistant.name}, Status: {assistant.status}")
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "setting_id": setting_id,
+        }
+
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/assistants/{assistant_id}",
+            VendorAssistant,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_assistant_version(
+        self,
+        vendor: Union[VendorType, str],
+        assistant_id: str,
+        version: str,
+        setting_id: str,
+    ) -> VendorAssistantVersion:
+        """Get a specific version of a vendor assistant with detailed information.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            assistant_id: ID of the assistant to retrieve
+            version: Version number to retrieve (e.g., "1", "DRAFT")
+            setting_id: ID of the vendor setting
+
+        Returns:
+            VendorAssistantVersion containing detailed version information including
+            instruction, foundation model, and timestamps
+
+        Example:
+            >>> version_details = client.vendor_assistants.get_assistant_version(
+            ...     vendor=VendorType.AWS,
+            ...     assistant_id="TJBKR0DGWT",
+            ...     version="1",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> print(f"Name: {version_details.name}")
+            >>> print(f"Version: {version_details.version}")
+            >>> print(f"Instruction: {version_details.instruction}")
+            >>> print(f"Model: {version_details.foundationModel}")
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "setting_id": setting_id,
+        }
+
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/assistants/{assistant_id}/{version}",
+            VendorAssistantVersion,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_assistant_aliases(
+        self,
+        vendor: Union[VendorType, str],
+        assistant_id: str,
+        setting_id: str,
+        per_page: int = 10,
+        next_token: Optional[str] = None,
+    ) -> VendorAssistantAliasesResponse:
+        """Get aliases for a specific vendor assistant.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            assistant_id: ID of the assistant to retrieve aliases for
+            setting_id: ID of the vendor setting
+            per_page: Number of items per page
+            next_token: Token for pagination (optional, for retrieving next page)
+
+        Returns:
+            VendorAssistantAliasesResponse containing list of aliases and pagination token
+
+        Example:
+            >>> # Get first page of aliases
+            >>> aliases = client.vendor_assistants.get_assistant_aliases(
+            ...     vendor=VendorType.AWS,
+            ...     assistant_id="TJBKR0DGWT",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...     per_page=5
+            ... )
+            >>> for alias in aliases.data:
+            ...     print(f"{alias.name} (v{alias.version}): {alias.status}")
+            >>> # Get next page if available
+            >>> if aliases.pagination.next_token:
+            ...     next_page = client.vendor_assistants.get_assistant_aliases(
+            ...         vendor=VendorType.AWS,
+            ...         assistant_id="TJBKR0DGWT",
+            ...         setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...         per_page=5,
+            ...         next_token=aliases.pagination.next_token
+            ...     )
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "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}/assistants/{assistant_id}/aliases",
+            VendorAssistantAliasesResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def install_assistants(
+        self,
+        vendor: Union[VendorType, str],
+        assistants: List[VendorAssistantInstallRequest],
+    ) -> VendorAssistantInstallResponse:
+        """Install/activate vendor assistants.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            assistants: List of assistant installation requests with assistant ID, alias ID, and setting ID
+
+        Returns:
+            VendorAssistantInstallResponse containing installation summary with AI run IDs
+
+        Example:
+            >>> from codemie_sdk import VendorAssistantInstallRequest
+            >>> # Install single assistant
+            >>> install_request = VendorAssistantInstallRequest(
+            ...     id="TJBKR0DGWT",
+            ...     agentAliasId="MNULODIW4N",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> response = client.vendor_assistants.install_assistants(
+            ...     vendor=VendorType.AWS,
+            ...     assistants=[install_request]
+            ... )
+            >>> for item in response.summary:
+            ...     print(f"Installed {item.agentId} with run ID: {item.aiRunId}")
+            >>>
+            >>> # Install multiple assistants
+            >>> requests = [
+            ...     VendorAssistantInstallRequest(
+            ...         id="ASSISTANT_ID_1",
+            ...         agentAliasId="ALIAS_ID_1",
+            ...         setting_id="SETTING_ID"
+            ...     ),
+            ...     VendorAssistantInstallRequest(
+            ...         id="ASSISTANT_ID_2",
+            ...         agentAliasId="ALIAS_ID_2",
+            ...         setting_id="SETTING_ID"
+            ...     )
+            ... ]
+            >>> response = client.vendor_assistants.install_assistants(
+            ...     vendor=VendorType.AWS,
+            ...     assistants=requests
+            ... )
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        # Convert list of Pydantic models to list of dicts
+        payload = [assistant.model_dump(by_alias=True) for assistant in assistants]
+
+        return self._api.post(
+            f"/v1/vendors/{vendor_str}/assistants",
+            VendorAssistantInstallResponse,
+            json_data=payload,
+            wrap_response=False,
+        )
+
+    def uninstall_assistant(
+        self,
+        vendor: Union[VendorType, str],
+        ai_run_id: str,
+    ) -> VendorAssistantUninstallResponse:
+        """Uninstall/deactivate a vendor assistant.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            ai_run_id: AI run ID returned from the install operation
+
+        Returns:
+            VendorAssistantUninstallResponse with success status
+
+        Example:
+            >>> # First, install an assistant
+            >>> install_request = VendorAssistantInstallRequest(
+            ...     id="TJBKR0DGWT",
+            ...     agentAliasId="MNULODIW4N",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> install_response = client.vendor_assistants.install_assistants(
+            ...     vendor=VendorType.AWS,
+            ...     assistants=[install_request]
+            ... )
+            >>> ai_run_id = install_response.summary[0].aiRunId
+            >>>
+            >>> # Later, uninstall the assistant using the AI run ID
+            >>> response = client.vendor_assistants.uninstall_assistant(
+            ...     vendor=VendorType.AWS,
+            ...     ai_run_id=ai_run_id
+            ... )
+            >>> if response.success:
+            ...     print("Assistant successfully uninstalled!")
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        return self._api.delete(
+            f"/v1/vendors/{vendor_str}/assistants/{ai_run_id}",
+            VendorAssistantUninstallResponse,
+            wrap_response=False,
+        )

codemie_sdk_python-0.1.223/src/codemie_sdk/services/vendor_workflow.py

@@ -0,0 +1,330 @@
+"""Vendor workflow service implementation for managing cloud vendor workflow settings."""
+
+from typing import Union, Optional, List
+
+from ..models.vendor_assistant import VendorType
+from ..models.vendor_workflow import (
+    VendorWorkflowSettingsResponse,
+    VendorWorkflowsResponse,
+    VendorWorkflow,
+    VendorWorkflowAliasesResponse,
+    VendorWorkflowInstallRequest,
+    VendorWorkflowInstallResponse,
+    VendorWorkflowUninstallResponse,
+)
+from ..utils import ApiRequestHandler
+
+
+class VendorWorkflowService:
+    """Service for managing cloud vendor workflow settings (AWS, Azure, GCP)."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the vendor workflow 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 get_workflow_settings(
+        self,
+        vendor: Union[VendorType, str],
+        page: int = 0,
+        per_page: int = 10,
+    ) -> VendorWorkflowSettingsResponse:
+        """Get workflow settings for a specific cloud vendor.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            page: Page number for pagination (0-based)
+            per_page: Number of items per page
+
+        Returns:
+            VendorWorkflowSettingsResponse containing list of settings and pagination info
+
+        Example:
+            >>> # Using enum
+            >>> settings = client.vendor_workflows.get_workflow_settings(VendorType.AWS, page=0, per_page=10)
+            >>> # Using string
+            >>> settings = client.vendor_workflows.get_workflow_settings("aws", page=0, per_page=10)
+            >>> # Access settings data
+            >>> for setting in settings.data:
+            ...     print(f"Setting: {setting.setting_name}, Project: {setting.project}")
+            ...     if setting.invalid:
+            ...         print(f"Error: {setting.error}")
+        """
+        # Convert enum to string value if needed
+        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}/workflows/settings",
+            VendorWorkflowSettingsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_workflows(
+        self,
+        vendor: Union[VendorType, str],
+        setting_id: str,
+        per_page: int = 10,
+        next_token: Optional[str] = None,
+    ) -> VendorWorkflowsResponse:
+        """Get workflows for a specific vendor setting.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            setting_id: ID of the vendor setting to retrieve workflows for
+            per_page: Number of items per page
+            next_token: Token for pagination (optional, for retrieving next page)
+
+        Returns:
+            VendorWorkflowsResponse containing list of workflows and pagination token
+
+        Example:
+            >>> # Get first page
+            >>> workflows = client.vendor_workflows.get_workflows(
+            ...     vendor=VendorType.AWS,
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...     per_page=8
+            ... )
+            >>> # Access workflow data
+            >>> for workflow in workflows.data:
+            ...     print(f"Name: {workflow.name}, Status: {workflow.status}")
+            ...     print(f"Version: {workflow.version}, Description: {workflow.description}")
+            >>> # Get next page if available
+            >>> if workflows.pagination.next_token:
+            ...     next_page = client.vendor_workflows.get_workflows(
+            ...         vendor=VendorType.AWS,
+            ...         setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...         per_page=8,
+            ...         next_token=workflows.pagination.next_token
+            ...     )
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "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}/workflows",
+            VendorWorkflowsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_workflow(
+        self,
+        vendor: Union[VendorType, str],
+        workflow_id: str,
+        setting_id: str,
+    ) -> VendorWorkflow:
+        """Get a specific workflow by ID for a vendor setting.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            workflow_id: ID of the workflow to retrieve
+            setting_id: ID of the vendor setting
+
+        Returns:
+            VendorWorkflow containing workflow details
+
+        Example:
+            >>> workflow = client.vendor_workflows.get_workflow(
+            ...     vendor=VendorType.AWS,
+            ...     workflow_id="9HXLQ7J9YP",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> print(f"Name: {workflow.name}, Status: {workflow.status}")
+            >>> print(f"Version: {workflow.version}, Description: {workflow.description}")
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "setting_id": setting_id,
+        }
+
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/workflows/{workflow_id}",
+            VendorWorkflow,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_workflow_aliases(
+        self,
+        vendor: Union[VendorType, str],
+        workflow_id: str,
+        setting_id: str,
+        per_page: int = 10,
+        next_token: Optional[str] = None,
+    ) -> VendorWorkflowAliasesResponse:
+        """Get aliases for a specific vendor workflow.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            workflow_id: ID of the workflow to retrieve aliases for
+            setting_id: ID of the vendor setting
+            per_page: Number of items per page
+            next_token: Token for pagination (optional, for retrieving next page)
+
+        Returns:
+            VendorWorkflowAliasesResponse containing list of aliases and pagination token
+
+        Example:
+            >>> # Get first page of aliases
+            >>> aliases = client.vendor_workflows.get_workflow_aliases(
+            ...     vendor=VendorType.AWS,
+            ...     workflow_id="9HXLQ7J9YP",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...     per_page=5
+            ... )
+            >>> for alias in aliases.data:
+            ...     print(f"{alias.name} (v{alias.version}): {alias.status}")
+            ...     if alias.aiRunId:
+            ...         print(f"  AI Run ID: {alias.aiRunId}")
+            >>> # Get next page if available
+            >>> if aliases.pagination.next_token:
+            ...     next_page = client.vendor_workflows.get_workflow_aliases(
+            ...         vendor=VendorType.AWS,
+            ...         workflow_id="9HXLQ7J9YP",
+            ...         setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...         per_page=5,
+            ...         next_token=aliases.pagination.next_token
+            ...     )
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "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}/workflows/{workflow_id}/aliases",
+            VendorWorkflowAliasesResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def install_workflows(
+        self,
+        vendor: Union[VendorType, str],
+        workflows: List[VendorWorkflowInstallRequest],
+    ) -> VendorWorkflowInstallResponse:
+        """Install/activate vendor workflows.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            workflows: List of workflow installation requests with workflow ID, flow alias ID, and setting ID
+
+        Returns:
+            VendorWorkflowInstallResponse containing installation summary with AI run IDs
+
+        Example:
+            >>> from codemie_sdk import VendorWorkflowInstallRequest
+            >>> # Install single workflow
+            >>> install_request = VendorWorkflowInstallRequest(
+            ...     id="9HXLQ7J9YP",
+            ...     flowAliasId="9RUV0BI2L7",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> response = client.vendor_workflows.install_workflows(
+            ...     vendor=VendorType.AWS,
+            ...     workflows=[install_request]
+            ... )
+            >>> for item in response.summary:
+            ...     print(f"Installed workflow {item.flowId} with run ID: {item.aiRunId}")
+            >>>
+            >>> # Install multiple workflows
+            >>> requests = [
+            ...     VendorWorkflowInstallRequest(
+            ...         id="WORKFLOW_ID_1",
+            ...         flowAliasId="ALIAS_ID_1",
+            ...         setting_id="SETTING_ID"
+            ...     ),
+            ...     VendorWorkflowInstallRequest(
+            ...         id="WORKFLOW_ID_2",
+            ...         flowAliasId="ALIAS_ID_2",
+            ...         setting_id="SETTING_ID"
+            ...     )
+            ... ]
+            >>> response = client.vendor_workflows.install_workflows(
+            ...     vendor=VendorType.AWS,
+            ...     workflows=requests
+            ... )
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        # Convert list of Pydantic models to list of dicts
+        payload = [workflow.model_dump(by_alias=True) for workflow in workflows]
+
+        return self._api.post(
+            f"/v1/vendors/{vendor_str}/workflows",
+            VendorWorkflowInstallResponse,
+            json_data=payload,
+            wrap_response=False,
+        )
+
+    def uninstall_workflow(
+        self,
+        vendor: Union[VendorType, str],
+        ai_run_id: str,
+    ) -> VendorWorkflowUninstallResponse:
+        """Uninstall/deactivate a vendor workflow.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            ai_run_id: AI run ID returned from the workflow alias (aiRunId field)
+
+        Returns:
+            VendorWorkflowUninstallResponse with success status
+
+        Example:
+            >>> # Get workflow aliases to find the aiRunId
+            >>> aliases = client.vendor_workflows.get_workflow_aliases(
+            ...     vendor=VendorType.AWS,
+            ...     workflow_id="9HXLQ7J9YP",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> # Find an installed alias with aiRunId
+            >>> for alias in aliases.data:
+            ...     if alias.aiRunId:
+            ...         ai_run_id = alias.aiRunId
+            ...         break
+            >>>
+            >>> # Uninstall the workflow using the AI run ID
+            >>> response = client.vendor_workflows.uninstall_workflow(
+            ...     vendor=VendorType.AWS,
+            ...     ai_run_id="56fed66d-f66e-46e3-b420-bb3a8d93eed4"
+            ... )
+            >>> if response.success:
+            ...     print("Workflow successfully uninstalled!")
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        return self._api.delete(
+            f"/v1/vendors/{vendor_str}/workflows/{ai_run_id}",
+            VendorWorkflowUninstallResponse,
+            wrap_response=False,
+        )

codemie_sdk_python-0.1.221/src/codemie_sdk/__init__.py

@@ -1,23 +0,0 @@
-"""
-CodeMie SDK for Python
-~~~~~~~~~~~~~~~~~~~~~
-
-A Python SDK for interacting with CodeMie API.
-
-Basic usage:
-
-    >>> from codemie_sdk import CodeMieClient
-    >>> client = CodeMieClient(
-    ...     auth_server_url="https://auth.example.com",
-    ...     auth_client_id="client_id",
-    ...     auth_client_secret="secret",
-    ...     auth_realm_name="realm",
-    ...     codemie_api_domain="api.codemie.com"
-    ... )
-    >>> assistants = client.assistants.list()
-"""
-
-from .client.client import CodeMieClient
-
-__version__ = "0.2.7"
-__all__ = ["CodeMieClient"]