codemie-sdk-python 0.1.227__tar.gz → 0.1.229__tar.gz

{codemie_sdk_python-0.1.227 → codemie_sdk_python-0.1.229}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.227
+Version: 0.1.229
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com
@@ -831,6 +831,50 @@ client = CodeMieClient(
 )
 ```
 
+### Assistant Versioning via SDK
+
+The Python SDK exposes full assistant versioning capabilities delivered in EPMCDME-8285.
+
+- List versions
+```python
+versions = client.assistants.list_versions("assistant-id", page=0, per_page=20)
+print([v.version_number for v in versions])
+```
+
+- Get specific version
+```python
+version = client.assistants.get_version("assistant-id", 2)
+print(version.system_prompt)
+```
+
+- Compare two versions
+```python
+from codemie_sdk.models.assistant import AssistantVersionDiff
+
+diff: AssistantVersionDiff = client.assistants.compare_versions("assistant-id", 1, 3)
+print(diff.summary)
+```
+
+- Rollback to a version
+```python
+resp = client.assistants.rollback_to_version("assistant-id", 2)
+print(resp)
+```
+
+- Chat with a specific version
+```python
+from codemie_sdk.models.assistant import AssistantChatRequest
+
+req = AssistantChatRequest(text="Hi", stream=False)
+resp = client.assistants.chat_with_version("assistant-id", 2, req)
+print(resp.generated)
+```
+
+Quick CLI example
+```bash
+python codemie-sdk/examples/assistant_versions.py <assistant_id> [version_number]
+```
+
 ## Support
 For providing credentials please contact AI/Run CodeMie Team: Vadym_Vlasenko@epam.com or Nikita_Levyankov@epam.com
 

{codemie_sdk_python-0.1.227 → codemie_sdk_python-0.1.229}/README.md

@@ -818,6 +818,50 @@ client = CodeMieClient(
 )
 ```
 
+### Assistant Versioning via SDK
+
+The Python SDK exposes full assistant versioning capabilities delivered in EPMCDME-8285.
+
+- List versions
+```python
+versions = client.assistants.list_versions("assistant-id", page=0, per_page=20)
+print([v.version_number for v in versions])
+```
+
+- Get specific version
+```python
+version = client.assistants.get_version("assistant-id", 2)
+print(version.system_prompt)
+```
+
+- Compare two versions
+```python
+from codemie_sdk.models.assistant import AssistantVersionDiff
+
+diff: AssistantVersionDiff = client.assistants.compare_versions("assistant-id", 1, 3)
+print(diff.summary)
+```
+
+- Rollback to a version
+```python
+resp = client.assistants.rollback_to_version("assistant-id", 2)
+print(resp)
+```
+
+- Chat with a specific version
+```python
+from codemie_sdk.models.assistant import AssistantChatRequest
+
+req = AssistantChatRequest(text="Hi", stream=False)
+resp = client.assistants.chat_with_version("assistant-id", 2, req)
+print(resp.generated)
+```
+
+Quick CLI example
+```bash
+python codemie-sdk/examples/assistant_versions.py <assistant_id> [version_number]
+```
+
 ## Support
 For providing credentials please contact AI/Run CodeMie Team: Vadym_Vlasenko@epam.com or Nikita_Levyankov@epam.com
 

{codemie_sdk_python-0.1.227 → codemie_sdk_python-0.1.229}/pyproject.toml

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

{codemie_sdk_python-0.1.227 → codemie_sdk_python-0.1.229}/src/codemie_sdk/__init__.py

@@ -60,9 +60,23 @@ from .models.vendor_knowledgebase import (
     VendorKnowledgeBaseInstallResponse,
     VendorKnowledgeBaseUninstallResponse,
 )
+from .models.vendor_guardrail import (
+    VendorGuardrailSetting,
+    VendorGuardrailSettingsResponse,
+    VendorGuardrail,
+    VendorGuardrailStatus,
+    VendorGuardrailsResponse,
+    VendorGuardrailVersion,
+    VendorGuardrailVersionsResponse,
+    VendorGuardrailInstallRequest,
+    VendorGuardrailInstallSummary,
+    VendorGuardrailInstallResponse,
+    VendorGuardrailUninstallResponse,
+)
 from .services.vendor_assistant import VendorAssistantService
 from .services.vendor_workflow import VendorWorkflowService
 from .services.vendor_knowledgebase import VendorKnowledgeBaseService
+from .services.vendor_guardrail import VendorGuardrailService
 
 __version__ = "0.2.11"
 __all__ = [
@@ -106,4 +120,16 @@ __all__ = [
     "VendorKnowledgeBaseInstallResponse",
     "VendorKnowledgeBaseUninstallResponse",
     "VendorKnowledgeBaseService",
+    "VendorGuardrailSetting",
+    "VendorGuardrailSettingsResponse",
+    "VendorGuardrail",
+    "VendorGuardrailStatus",
+    "VendorGuardrailsResponse",
+    "VendorGuardrailVersion",
+    "VendorGuardrailVersionsResponse",
+    "VendorGuardrailInstallRequest",
+    "VendorGuardrailInstallSummary",
+    "VendorGuardrailInstallResponse",
+    "VendorGuardrailUninstallResponse",
+    "VendorGuardrailService",
 ]

{codemie_sdk_python-0.1.227 → codemie_sdk_python-0.1.229}/src/codemie_sdk/client/client.py

@@ -16,6 +16,7 @@ from ..services.webhook import WebhookService
 from ..services.vendor_assistant import VendorAssistantService
 from ..services.vendor_workflow import VendorWorkflowService
 from ..services.vendor_knowledgebase import VendorKnowledgeBaseService
+from ..services.vendor_guardrail import VendorGuardrailService
 
 
 class CodeMieClient:
@@ -101,6 +102,9 @@ class CodeMieClient:
         self.vendor_knowledgebases = VendorKnowledgeBaseService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.vendor_guardrails = VendorGuardrailService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
 
     @property
     def token(self) -> str:
@@ -156,4 +160,7 @@ class CodeMieClient:
         self.vendor_knowledgebases = VendorKnowledgeBaseService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.vendor_guardrails = VendorGuardrailService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
         return self._token

{codemie_sdk_python-0.1.227 → codemie_sdk_python-0.1.229}/src/codemie_sdk/models/assistant.py

@@ -164,6 +164,7 @@ class Assistant(AssistantBase):
     user_abilities: Optional[List[Any]] = None
     mcp_servers: List[MCPServerDetails] = Field(default_factory=list)
     assistant_ids: List[str] = Field(default_factory=list)
+    version_count: Optional[int] = None
 
 
 class AssistantRequestBase(AssistantBase):
@@ -205,6 +206,26 @@ class AssistantUpdateRequest(AssistantRequestBase):
     pass
 
 
+class AssistantVersion(BaseModel):
+    """Immutable snapshot of assistant configuration for a specific version."""
+
+    model_config = ConfigDict(extra="ignore", use_enum_values=True)
+
+    version_number: int
+    created_date: datetime
+    created_by: Optional[User] = None
+    change_notes: Optional[str] = None
+    description: Optional[str] = None
+    system_prompt: str
+    llm_model_type: Optional[str] = None
+    temperature: Optional[float] = None
+    top_p: Optional[float] = None
+    context: List[Context] = Field(default_factory=list)
+    toolkits: List[ToolKitDetails] = Field(default_factory=list)
+    mcp_servers: List[MCPServerDetails] = Field(default_factory=list)
+    assistant_ids: List[str] = Field(default_factory=list)
+
+
 class ChatRole(str, Enum):
     """Enum for chat message roles."""
 

{codemie_sdk_python-0.1.227 → codemie_sdk_python-0.1.229}/src/codemie_sdk/models/datasource.py

@@ -25,6 +25,7 @@ class DataSourceType(str, Enum):
     CHUNK_SUMMARY = "chunk-summary"
     JSON = "knowledge_base_json"
     BEDROCK = "knowledge_base_bedrock"
+    PLATFORM = "platform_marketplace_assistant"
 
 
 class DataSourceStatus(str, Enum):

codemie_sdk_python-0.1.229/src/codemie_sdk/models/vendor_guardrail.py

@@ -0,0 +1,152 @@
+"""Models for vendor guardrail 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 VendorGuardrailSetting(BaseModel):
+    """Model representing a vendor guardrail 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 VendorGuardrailSettingsResponse(BaseModel):
+    """Response model for vendor guardrail settings list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorGuardrailSetting] = Field(
+        ..., description="List of vendor guardrail settings"
+    )
+    pagination: PaginationInfo = Field(..., description="Pagination information")
+
+
+class VendorGuardrailStatus(str, Enum):
+    """Status of vendor guardrail."""
+
+    PREPARED = "PREPARED"
+    NOT_PREPARED = "NOT_PREPARED"
+
+
+class VendorGuardrail(BaseModel):
+    """Model representing a vendor guardrail."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the guardrail")
+    name: str = Field(..., description="Name of the guardrail")
+    status: VendorGuardrailStatus = Field(..., description="Status of the guardrail")
+    description: Optional[str] = Field(None, description="Description of the guardrail")
+    version: str = Field(..., description="Version of the guardrail")
+    createdAt: datetime = Field(
+        ..., description="Creation timestamp", alias="createdAt"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+
+
+class VendorGuardrailsResponse(BaseModel):
+    """Response model for vendor guardrails list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorGuardrail] = Field(..., description="List of vendor guardrails")
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorGuardrailVersion(BaseModel):
+    """Model representing a vendor guardrail version."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the guardrail")
+    version: str = Field(..., description="Version of the guardrail")
+    name: str = Field(..., description="Name of the guardrail")
+    status: VendorGuardrailStatus = Field(..., description="Status of the version")
+    description: Optional[str] = Field(None, description="Description of the version")
+    blockedInputMessaging: Optional[str] = Field(
+        None,
+        description="Message to display when input is blocked by guardrail",
+        alias="blockedInputMessaging",
+    )
+    blockedOutputsMessaging: Optional[str] = Field(
+        None,
+        description="Message to display when output is blocked by guardrail",
+        alias="blockedOutputsMessaging",
+    )
+    createdAt: datetime = Field(
+        ..., description="Creation timestamp", alias="createdAt"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+
+
+class VendorGuardrailVersionsResponse(BaseModel):
+    """Response model for vendor guardrail versions list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorGuardrailVersion] = Field(
+        ..., description="List of vendor guardrail versions"
+    )
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorGuardrailInstallRequest(BaseModel):
+    """Model for a single guardrail installation request."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Guardrail ID to install")
+    version: str = Field(..., description="Version to use for the guardrail")
+    setting_id: str = Field(..., description="Vendor setting ID")
+
+
+class VendorGuardrailInstallSummary(BaseModel):
+    """Model for guardrail installation summary."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    guardrailId: str = Field(..., description="Installed guardrail ID")
+    version: str = Field(..., description="Version used for installation")
+    aiRunId: str = Field(..., description="AI run ID for the installation")
+
+
+class VendorGuardrailInstallResponse(BaseModel):
+    """Response model for guardrail installation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    summary: List[VendorGuardrailInstallSummary] = Field(
+        ..., description="List of installation summaries"
+    )
+
+
+class VendorGuardrailUninstallResponse(BaseModel):
+    """Response model for guardrail uninstallation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    success: bool = Field(..., description="Whether the uninstallation was successful")

{codemie_sdk_python-0.1.227 → codemie_sdk_python-0.1.229}/src/codemie_sdk/services/assistant.py

@@ -170,6 +170,141 @@ class AssistantService:
         """
         return self._api.get(f"/v1/assistants/prebuilt/{slug}", Assistant)
 
+    def list_versions(
+        self, assistant_id: str, page: int = 0, per_page: Optional[int] = None
+    ):
+        """List assistant versions.
+
+        Args:
+            assistant_id: Assistant identifier
+            page: Page number for pagination
+            per_page: Items per page (optional). If not provided, backend defaults are used.
+
+        Returns:
+            List of AssistantVersion objects
+        """
+
+        params: Dict[str, Any] = {"page": page}
+        if per_page is not None:
+            params["per_page"] = per_page
+        from ..models.assistant import AssistantVersion
+
+        raw = self._api.get(
+            f"/v1/assistants/{assistant_id}/versions",
+            dict,
+            params=params,
+            wrap_response=False,
+        )
+        items = []
+        if isinstance(raw, list):
+            items = raw
+        elif isinstance(raw, dict):
+            items = raw.get("data") or raw.get("versions") or []
+        else:
+            items = []
+        return [AssistantVersion.model_validate(it) for it in items]
+
+    def get_version(self, assistant_id: str, version_number: int):
+        """Get a specific assistant version by number.
+
+        Args:
+            assistant_id: Assistant identifier
+            version_number: Version number to retrieve
+
+        Returns:
+            AssistantVersion object
+        """
+        from ..models.assistant import AssistantVersion
+
+        raw = self._api.get(
+            f"/v1/assistants/{assistant_id}/versions/{version_number}", AssistantVersion
+        )
+        if isinstance(raw, dict):
+            return AssistantVersion.model_validate(raw)
+        return raw
+
+    def compare_versions(self, assistant_id: str, v1: int, v2: int) -> Dict[str, Any]:
+        """Compare two assistant versions and return diff summary.
+
+        Args:
+            assistant_id: Assistant identifier
+            v1: First version number
+            v2: Second version number
+
+        Returns:
+            Generic dictionary with comparison result (diff, summary, etc.)
+        """
+        return self._api.get(
+            f"/v1/assistants/{assistant_id}/versions/{v1}/compare/{v2}",
+            dict,
+        )
+
+    def rollback_to_version(
+        self, assistant_id: str, version_number: int, change_notes: Optional[str] = None
+    ) -> dict:
+        """Rollback assistant to a specific version. Creates a new version mirroring target.
+
+        Args:
+            assistant_id: Assistant identifier
+            version_number: Target version to rollback to
+            change_notes: Optional description of why rollback is performed
+
+        Returns:
+            Backend response (dict)
+        """
+        payload: Dict[str, Any] = {}
+        if change_notes:
+            payload["change_notes"] = change_notes
+        try:
+            return self._api.post(
+                f"/v1/assistants/{assistant_id}/versions/{version_number}/rollback",
+                dict,
+                json_data=payload,
+            )
+        except requests.HTTPError as err:
+            try:
+                assistant = self.get(assistant_id)
+                version = self.get_version(assistant_id, version_number)
+
+                update_req = AssistantUpdateRequest(
+                    name=assistant.name,
+                    description=assistant.description or "",
+                    system_prompt=version.system_prompt,
+                    project=assistant.project,
+                    llm_model_type=version.llm_model_type or assistant.llm_model_type,
+                    temperature=version.temperature
+                    if hasattr(version, "temperature")
+                    else assistant.temperature,
+                    top_p=version.top_p
+                    if hasattr(version, "top_p")
+                    else assistant.top_p,
+                    context=version.context
+                    if hasattr(version, "context")
+                    else assistant.context,
+                    toolkits=version.toolkits
+                    if hasattr(version, "toolkits")
+                    else assistant.toolkits,
+                    user_prompts=assistant.user_prompts,
+                    shared=assistant.shared,
+                    is_react=assistant.is_react,
+                    is_global=assistant.is_global,
+                    slug=assistant.slug,
+                    mcp_servers=version.mcp_servers
+                    if hasattr(version, "mcp_servers")
+                    else assistant.mcp_servers,
+                    assistant_ids=version.assistant_ids
+                    if hasattr(version, "assistant_ids")
+                    else assistant.assistant_ids,
+                )
+                resp = self.update(assistant_id, update_req)
+                resp["_rollback_fallback"] = True
+                resp["_target_version"] = version_number
+                if change_notes:
+                    resp["change_notes"] = change_notes
+                return resp
+            except Exception:
+                raise err
+
     def chat(
         self,
         assistant_id: str,
@@ -206,6 +341,45 @@ class AssistantService:
 
         return response
 
+    def chat_with_version(
+        self,
+        assistant_id: str,
+        version_number: int,
+        request: AssistantChatRequest,
+    ) -> Union[requests.Response, BaseModelResponse]:
+        """Send a chat request to a specific assistant version.
+
+        Uses the stable chat endpoint with an explicit `version` parameter to
+        ensure compatibility with environments that don't expose
+        /versions/{version}/model.
+
+        Args:
+            assistant_id: ID of the assistant to chat with
+            version_number: version to pin chat to
+            request: Chat request details
+
+        Returns:
+            Chat response or streaming response
+        """
+        pydantic_schema = None
+        if issubclass(request.output_schema, BaseModel):
+            pydantic_schema = deepcopy(request.output_schema)
+            request.output_schema = request.output_schema.model_json_schema()
+
+        payload = request.model_dump(exclude_none=True, by_alias=True)
+        payload["version"] = version_number
+
+        response = self._api.post(
+            f"/v1/assistants/{assistant_id}/model",
+            BaseModelResponse,
+            json_data=payload,
+            stream=request.stream,
+        )
+        if not request.stream and pydantic_schema:
+            response.generated = pydantic_schema.model_validate(response.generated)
+
+        return response
+
     def upload_file_to_chat(self, file_path: Path):
         """Upload a file to assistant chat and return the response containing file_url."""
 

codemie_sdk_python-0.1.229/src/codemie_sdk/services/vendor_guardrail.py

@@ -0,0 +1,375 @@
+"""Vendor guardrail service implementation for managing cloud vendor guardrail settings."""
+
+from typing import Union, Optional, List
+
+from ..models.vendor_assistant import VendorType
+from ..models.vendor_guardrail import (
+    VendorGuardrailSettingsResponse,
+    VendorGuardrailsResponse,
+    VendorGuardrail,
+    VendorGuardrailVersion,
+    VendorGuardrailVersionsResponse,
+    VendorGuardrailInstallRequest,
+    VendorGuardrailInstallResponse,
+    VendorGuardrailUninstallResponse,
+)
+from ..utils import ApiRequestHandler
+
+
+class VendorGuardrailService:
+    """Service for managing cloud vendor guardrail settings (AWS, Azure, GCP)."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the vendor 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 get_guardrail_settings(
+        self,
+        vendor: Union[VendorType, str],
+        page: int = 0,
+        per_page: int = 10,
+    ) -> VendorGuardrailSettingsResponse:
+        """Get guardrail 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:
+            VendorGuardrailSettingsResponse containing list of settings and pagination info
+
+        Example:
+            >>> # Using enum
+            >>> settings = client.vendor_guardrails.get_guardrail_settings(VendorType.AWS, page=0, per_page=10)
+            >>> # Using string
+            >>> settings = client.vendor_guardrails.get_guardrail_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}/guardrails/settings",
+            VendorGuardrailSettingsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_guardrails(
+        self,
+        vendor: Union[VendorType, str],
+        setting_id: str,
+        per_page: int = 10,
+        next_token: Optional[str] = None,
+    ) -> VendorGuardrailsResponse:
+        """Get guardrails 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 guardrails for
+            per_page: Number of items per page
+            next_token: Token for pagination (optional, for retrieving next page)
+
+        Returns:
+            VendorGuardrailsResponse containing list of guardrails and pagination token
+
+        Example:
+            >>> # Get first page
+            >>> guardrails = client.vendor_guardrails.get_guardrails(
+            ...     vendor=VendorType.AWS,
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...     per_page=8
+            ... )
+            >>> # Access guardrail data
+            >>> for guardrail in guardrails.data:
+            ...     print(f"Name: {guardrail.name}, Status: {guardrail.status}")
+            ...     print(f"Version: {guardrail.version}, Description: {guardrail.description}")
+            >>> # Get next page if available
+            >>> if guardrails.pagination.next_token:
+            ...     next_page = client.vendor_guardrails.get_guardrails(
+            ...         vendor=VendorType.AWS,
+            ...         setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...         per_page=8,
+            ...         next_token=guardrails.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}/guardrails",
+            VendorGuardrailsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_guardrail(
+        self,
+        vendor: Union[VendorType, str],
+        guardrail_id: str,
+        setting_id: str,
+    ) -> VendorGuardrail:
+        """Get a specific guardrail by ID for a vendor setting.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            guardrail_id: ID of the guardrail to retrieve
+            setting_id: ID of the vendor setting
+
+        Returns:
+            VendorGuardrail containing guardrail details
+
+        Example:
+            >>> guardrail = client.vendor_guardrails.get_guardrail(
+            ...     vendor=VendorType.AWS,
+            ...     guardrail_id="lss9vxro9oxg",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> print(f"Name: {guardrail.name}, Status: {guardrail.status}")
+            >>> print(f"Version: {guardrail.version}, Description: {guardrail.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}/guardrails/{guardrail_id}",
+            VendorGuardrail,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_guardrail_version(
+        self,
+        vendor: Union[VendorType, str],
+        guardrail_id: str,
+        version: str,
+        setting_id: str,
+    ) -> VendorGuardrailVersion:
+        """Get a specific version of a vendor guardrail with detailed information.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            guardrail_id: ID of the guardrail to retrieve
+            version: Version number to retrieve (e.g., "1", "2", "DRAFT")
+            setting_id: ID of the vendor setting
+
+        Returns:
+            VendorGuardrailVersion containing detailed version information including
+            blocked messaging settings and timestamps
+
+        Example:
+            >>> version_details = client.vendor_guardrails.get_guardrail_version(
+            ...     vendor=VendorType.AWS,
+            ...     guardrail_id="lss9vxro9oxg",
+            ...     version="1",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> print(f"Name: {version_details.name}")
+            >>> print(f"Version: {version_details.version}")
+            >>> print(f"Status: {version_details.status}")
+            >>> print(f"Blocked Input Message: {version_details.blockedInputMessaging}")
+            >>> print(f"Blocked Output Message: {version_details.blockedOutputsMessaging}")
+        """
+        # 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}/guardrails/{guardrail_id}/{version}",
+            VendorGuardrailVersion,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_guardrail_versions(
+        self,
+        vendor: Union[VendorType, str],
+        guardrail_id: str,
+        setting_id: str,
+        per_page: int = 10,
+        next_token: Optional[str] = None,
+    ) -> VendorGuardrailVersionsResponse:
+        """Get versions for a specific vendor guardrail.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            guardrail_id: ID of the guardrail to retrieve versions 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:
+            VendorGuardrailVersionsResponse containing list of versions and pagination token
+
+        Example:
+            >>> # Get first page of versions
+            >>> versions = client.vendor_guardrails.get_guardrail_versions(
+            ...     vendor=VendorType.AWS,
+            ...     guardrail_id="lss9vxro9oxg",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...     per_page=5
+            ... )
+            >>> for version in versions.data:
+            ...     print(f"{version.name} (v{version.version}): {version.status}")
+            >>> # Get next page if available
+            >>> if versions.pagination.next_token:
+            ...     next_page = client.vendor_guardrails.get_guardrail_versions(
+            ...         vendor=VendorType.AWS,
+            ...         guardrail_id="lss9vxro9oxg",
+            ...         setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...         per_page=5,
+            ...         next_token=versions.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}/guardrails/{guardrail_id}/versions",
+            VendorGuardrailVersionsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def install_guardrails(
+        self,
+        vendor: Union[VendorType, str],
+        guardrails: List[VendorGuardrailInstallRequest],
+    ) -> VendorGuardrailInstallResponse:
+        """Install/activate vendor guardrails.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            guardrails: List of guardrail installation requests with guardrail ID, version, and setting ID
+
+        Returns:
+            VendorGuardrailInstallResponse containing installation summary with AI run IDs
+
+        Example:
+            >>> from codemie_sdk import VendorGuardrailInstallRequest
+            >>> # Install single guardrail
+            >>> install_request = VendorGuardrailInstallRequest(
+            ...     id="lss9vxro9oxg",
+            ...     version="1",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> response = client.vendor_guardrails.install_guardrails(
+            ...     vendor=VendorType.AWS,
+            ...     guardrails=[install_request]
+            ... )
+            >>> for item in response.summary:
+            ...     print(f"Installed {item.guardrailId} version {item.version} with run ID: {item.aiRunId}")
+            >>>
+            >>> # Install multiple guardrails
+            >>> requests = [
+            ...     VendorGuardrailInstallRequest(
+            ...         id="GUARDRAIL_ID_1",
+            ...         version="1",
+            ...         setting_id="SETTING_ID"
+            ...     ),
+            ...     VendorGuardrailInstallRequest(
+            ...         id="GUARDRAIL_ID_2",
+            ...         version="2",
+            ...         setting_id="SETTING_ID"
+            ...     )
+            ... ]
+            >>> response = client.vendor_guardrails.install_guardrails(
+            ...     vendor=VendorType.AWS,
+            ...     guardrails=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 = [guardrail.model_dump(by_alias=True) for guardrail in guardrails]
+
+        return self._api.post(
+            f"/v1/vendors/{vendor_str}/guardrails",
+            VendorGuardrailInstallResponse,
+            json_data=payload,
+            wrap_response=False,
+        )
+
+    def uninstall_guardrail(
+        self,
+        vendor: Union[VendorType, str],
+        ai_run_id: str,
+    ) -> VendorGuardrailUninstallResponse:
+        """Uninstall/deactivate a vendor guardrail.
+
+        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:
+            VendorGuardrailUninstallResponse with success status
+
+        Example:
+            >>> # First, install a guardrail
+            >>> install_request = VendorGuardrailInstallRequest(
+            ...     id="lss9vxro9oxg",
+            ...     version="1",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> install_response = client.vendor_guardrails.install_guardrails(
+            ...     vendor=VendorType.AWS,
+            ...     guardrails=[install_request]
+            ... )
+            >>> ai_run_id = install_response.summary[0].aiRunId
+            >>>
+            >>> # Later, uninstall the guardrail using the AI run ID
+            >>> response = client.vendor_guardrails.uninstall_guardrail(
+            ...     vendor=VendorType.AWS,
+            ...     ai_run_id=ai_run_id
+            ... )
+            >>> if response.success:
+            ...     print("Guardrail 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}/guardrails/{ai_run_id}",
+            VendorGuardrailUninstallResponse,
+            wrap_response=False,
+        )