codemie-sdk-python 0.1.52__py3-none-any.whl → 0.1.258__py3-none-any.whl

codemie_sdk/models/datasource.py

@@ -24,6 +24,9 @@ class DataSourceType(str, Enum):
     SUMMARY = "summary"
     CHUNK_SUMMARY = "chunk-summary"
     JSON = "knowledge_base_json"
+    BEDROCK = "knowledge_base_bedrock"
+    PLATFORM = "platform_marketplace_assistant"
+    AZURE_DEVOPS_WIKI = "knowledge_base_azure_devops_wiki"
 
 
 class DataSourceStatus(str, Enum):
@@ -45,6 +48,15 @@ class DataSourceProcessingInfo(BaseModel):
     processed_documents_count: Optional[int] = Field(None, alias="documents_count_key")
 
 
+class ElasticsearchStatsResponse(BaseModel):
+    """Response model for Elasticsearch index statistics."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    index_name: str = Field(..., description="Name of the index in Elasticsearch")
+    size_in_bytes: int = Field(..., ge=0, description="Size of the index in bytes")
+
+
 # Base request models
 class Confluence(BaseModel):
     """Model for Confluence-specific response fields"""
@@ -88,6 +100,17 @@ class File(BaseModel):
     model_config = ConfigDict(extra="ignore")
 
 
+class AzureDevOpsWiki(BaseModel):
+    """Model for Azure DevOps Wiki-specific response fields"""
+
+    wiki_query: Optional[str] = None
+    organization: Optional[str] = None
+    project: Optional[str] = None
+    wiki_name: Optional[str] = None
+
+    model_config = ConfigDict(extra="ignore")
+
+
 class Code(BaseModel):
     """Model for code repository datasource creation"""
 
@@ -107,7 +130,7 @@ class BaseDataSourceRequest(BaseModel):
     name: str = Field(
         ...,
         description="Name must contain only lowercase letters and underscores.",
-        max_length=25,
+        max_length=50,
     )
     project_name: str
     description: str = Field(..., max_length=100)
@@ -188,6 +211,17 @@ class FileDataSourceRequest(BaseDataSourceRequest):
         super().__init__(type=DataSourceType.FILE, **data)
 
 
+class AzureDevOpsWikiDataSourceRequest(BaseDataSourceRequest, AzureDevOpsWiki):
+    """Model for Azure DevOps Wiki datasource creation requests"""
+
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.AZURE_DEVOPS_WIKI, **data)
+
+    @classmethod
+    def required_fields(cls) -> List[str]:
+        return ["wiki_query"]
+
+
 class BaseUpdateDataSourceRequest(BaseDataSourceRequest):
     """Mixin update-specific reindex fields"""
 
@@ -258,6 +292,49 @@ class UpdateFileDataSourceRequest(BaseUpdateDataSourceRequest):
         super().__init__(type=DataSourceType.FILE, **data)
 
 
+class UpdateAzureDevOpsWikiDataSourceRequest(
+    BaseUpdateDataSourceRequest, AzureDevOpsWiki
+):
+    """Model for Azure DevOps Wiki datasource updates"""
+
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.AZURE_DEVOPS_WIKI, **data)
+
+
+class CodeAnalysisDataSourceRequest(BaseModel):
+    """Model for provider-based datasource creation requests"""
+
+    model_config = ConfigDict(extra="ignore", populate_by_name=True)
+
+    name: str = Field(..., description="Datasource name")
+    description: Optional[str] = Field(None, description="Datasource description")
+    project_name: str = Field(..., description="Project name")
+    project_space_visible: bool = Field(False, alias="projectSpaceVisible")
+    branch: Optional[str] = Field(None, description="Git branch")
+    api_url: str = Field(..., description="Repository URL")
+    access_token: str = Field(..., description="Access token for repository")
+    analyzer: Optional[str] = Field(
+        None, description="Code analyzer type (e.g., Java, Python)"
+    )
+    datasource_root: str = Field("/", description="Root directory to analyze")
+
+
+class CodeExplorationDataSourceRequest(BaseModel):
+    """Model for CodeExplorationToolkit datasource creation requests"""
+
+    model_config = ConfigDict(extra="ignore", populate_by_name=True)
+
+    name: str = Field(..., description="Datasource name")
+    description: Optional[str] = Field(None, description="Datasource description")
+    project_name: str = Field(..., description="Project name")
+    project_space_visible: bool = Field(False, alias="projectSpaceVisible")
+    code_analysis_datasource_ids: List[str] = Field(
+        ...,
+        alias="code_analysis_datasource_ids",
+        description="List of CodeAnalysisToolkit datasource IDs",
+    )
+
+
 class DataSource(BaseModel):
     model_config = ConfigDict(
         extra="ignore",
@@ -292,6 +369,8 @@ class DataSource(BaseModel):
     confluence: Optional[Confluence] = None
     # Google doc specific fields
     google_doc_link: Optional[str] = None
+    # Azure DevOps Wiki specific fields
+    azure_devops_wiki: Optional[AzureDevOpsWiki] = None
 
     @model_validator(mode="before")
     def before_init(cls, values):
@@ -308,6 +387,7 @@ class DataSource(BaseModel):
             DataSourceType.CONFLUENCE,
             DataSourceType.JIRA,
             DataSourceType.GOOGLE,
+            DataSourceType.AZURE_DEVOPS_WIKI,
         ]:
             complete_state = values.get("complete_state", 0)
             if complete_state is not None:

codemie_sdk/models/file_operation.py

@@ -0,0 +1,25 @@
+"""File operation models."""
+
+from typing import Optional, List, Any, Dict
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class FileResponse(BaseModel):
+    """Individual file response model."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    file_url: str = Field(..., description="URL or identifier for the uploaded file")
+
+
+class FileBulkCreateResponse(BaseModel):
+    """Response model for bulk file creation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    files: List[FileResponse] = Field(
+        ..., description="List of successfully uploaded files with their URLs"
+    )
+    failed_files: Optional[List[Dict[str, Any]]] = Field(
+        None, description="List of files that failed to upload, null if no failures"
+    )

codemie_sdk/models/integration.py

@@ -1,8 +1,8 @@
 """Models for assistant-related data structures."""
 
+from datetime import datetime
 from enum import Enum
 from typing import List, Optional, Any
-from datetime import datetime
 
 from pydantic import BaseModel, Field, ConfigDict, field_serializer
 
@@ -29,12 +29,14 @@ class CredentialTypes(str, Enum):
     SONAR = "Sonar"
     SQL = "SQL"
     TELEGRAM = "Telegram"
-    ZEPHYR_CLOUD = "ZephyrCloud"
+    ZEPHYR_SCALE = "ZephyrScale"
     ZEPHYR_SQUAD = "ZephyrSquad"
     SERVICE_NOW = "ServiceNow"
     DIAL = "DIAL"
     A2A = "A2A"
     MCP = "MCP"
+    LITE_LLM = "LiteLLM"
+    REPORT_PORTAL = "ReportPortal"
 
 
 class IntegrationType(str, Enum):
@@ -68,6 +70,7 @@ class Integration(BaseModel):
     project_name: str
     alias: Optional[str] = None
     default: bool = False
+    is_global: bool = False
     credential_type: CredentialTypes
     credential_values: List[CredentialValues]
     setting_type: IntegrationType = Field(default=IntegrationType.USER)
@@ -75,3 +78,21 @@ class Integration(BaseModel):
     @field_serializer("date", "update_date")
     def serialize_dt(self, dt: datetime, _info):
         return dt.isoformat()
+
+
+class IntegrationTestRequest(BaseModel):
+    """Model for integration test request."""
+
+    credential_type: str
+    credential_values: Optional[List[CredentialValues]] = None
+    setting_id: Optional[str] = None
+
+
+class IntegrationTestResponse(BaseModel):
+    """Model for integration test response."""
+
+    def __getitem__(self, key):
+        return getattr(self, key)
+
+    message: str
+    success: bool

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