codemie-sdk-python 0.1.204__py3-none-any.whl → 0.1.230__py3-none-any.whl

codemie_sdk/__init__.py

@@ -18,6 +18,118 @@ Basic usage:
 """
 
 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 .models.vendor_knowledgebase import (
+    VendorKnowledgeBaseSetting,
+    VendorKnowledgeBaseSettingsResponse,
+    VendorKnowledgeBase,
+    VendorKnowledgeBaseStatus,
+    VendorKnowledgeBasesResponse,
+    VendorKnowledgeBaseDetail,
+    VendorKnowledgeBaseInstallRequest,
+    VendorKnowledgeBaseInstallSummary,
+    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.7"
-__all__ = ["CodeMieClient"]
+__version__ = "0.2.12"
+__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",
+    "VendorKnowledgeBaseSetting",
+    "VendorKnowledgeBaseSettingsResponse",
+    "VendorKnowledgeBase",
+    "VendorKnowledgeBaseStatus",
+    "VendorKnowledgeBasesResponse",
+    "VendorKnowledgeBaseDetail",
+    "VendorKnowledgeBaseInstallRequest",
+    "VendorKnowledgeBaseInstallSummary",
+    "VendorKnowledgeBaseInstallResponse",
+    "VendorKnowledgeBaseUninstallResponse",
+    "VendorKnowledgeBaseService",
+    "VendorGuardrailSetting",
+    "VendorGuardrailSettingsResponse",
+    "VendorGuardrail",
+    "VendorGuardrailStatus",
+    "VendorGuardrailsResponse",
+    "VendorGuardrailVersion",
+    "VendorGuardrailVersionsResponse",
+    "VendorGuardrailInstallRequest",
+    "VendorGuardrailInstallSummary",
+    "VendorGuardrailInstallResponse",
+    "VendorGuardrailUninstallResponse",
+    "VendorGuardrailService",
+]

codemie_sdk/client/client.py

@@ -12,6 +12,11 @@ from ..services.task import TaskService
 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
+from ..services.vendor_knowledgebase import VendorKnowledgeBaseService
+from ..services.vendor_guardrail import VendorGuardrailService
 
 
 class CodeMieClient:
@@ -85,6 +90,21 @@ class CodeMieClient:
         self.files = FileOperationService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        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
+        )
+        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:
@@ -131,4 +151,16 @@ 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
+        )
+        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/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."""
 
@@ -263,6 +284,14 @@ class AssistantChatRequest(BaseModel):
         default=None, description="DataSource in conversation history"
     )
     stream: bool = Field(default=False, description="Enable streaming response")
+    propagate_headers: bool = Field(
+        default=False,
+        description="Enable propagation of X-* HTTP headers to MCP servers during tool execution",
+    )
+    custom_metadata: Optional[dict[str, Any]] = Field(
+        default=None,
+        description="Custom metadata for the AI Assistant",
+    )
     top_k: int = Field(default=10, description="Top K results to consider")
     system_prompt: str = Field(default="", description="Override system prompt")
     background_task: bool = Field(default=False, description="Run as background task")

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/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/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")