codemie-sdk-python 0.1.280__tar.gz → 0.1.282__tar.gz

{codemie_sdk_python-0.1.280 → codemie_sdk_python-0.1.282}/PKG-INFO

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

{codemie_sdk_python-0.1.280 → codemie_sdk_python-0.1.282}/pyproject.toml

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

{codemie_sdk_python-0.1.280 → codemie_sdk_python-0.1.282}/src/codemie_sdk/models/assistant.py

@@ -216,6 +216,7 @@ class AssistantRequestBase(AssistantBase):
     mcp_servers: List[MCPServerDetails] = Field(default_factory=list)
     assistant_ids: List[str] = Field(default_factory=list)
     prompt_variables: List[PromptVariable] = Field(default_factory=list)
+    skip_integration_validation: Optional[bool] = Field(default=False)
 
 
 class AssistantCreateRequest(AssistantRequestBase):
@@ -230,6 +231,92 @@ class AssistantUpdateRequest(AssistantRequestBase):
     pass
 
 
+class MissingIntegration(BaseModel):
+    """Model representing a single missing tool credential."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    toolkit: str = Field(..., description="Toolkit name (e.g., 'Data Management')")
+    tool: str = Field(..., description="Tool name (e.g., 'sql')")
+    label: str = Field(..., description="Display label for the tool (e.g., 'SQL')")
+    credential_type: Optional[str] = Field(
+        None, description="Credential type required (e.g., 'AWS', 'Jira')"
+    )
+
+
+class MissingIntegrationsByCredentialType(BaseModel):
+    """Model representing missing tools grouped by credential type."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    credential_type: str = Field(
+        ..., description="Credential type (e.g., 'AWS', 'Jira', 'Confluence')"
+    )
+    missing_tools: List[MissingIntegration] = Field(
+        ..., description="List of missing tools requiring this credential type"
+    )
+    # Optional sub-assistant context
+    assistant_id: Optional[str] = Field(
+        None, description="Sub-assistant ID (if from sub-assistant)"
+    )
+    assistant_name: Optional[str] = Field(
+        None, description="Sub-assistant name (if from sub-assistant)"
+    )
+    icon_url: Optional[str] = Field(
+        None, description="Sub-assistant icon URL (if from sub-assistant)"
+    )
+
+
+class IntegrationValidationResult(BaseModel):
+    """Complete validation result for assistant integrations."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    has_missing_integrations: bool = Field(
+        ..., description="Whether any integrations are missing"
+    )
+    missing_by_credential_type: List[MissingIntegrationsByCredentialType] = Field(
+        default_factory=list,
+        description="Missing tools in main assistant grouped by credential type",
+    )
+    sub_assistants_missing: List[MissingIntegrationsByCredentialType] = Field(
+        default_factory=list,
+        description="Missing tools in sub-assistants grouped by credential type",
+    )
+    message: Optional[str] = Field(
+        None, description="User-friendly message about missing integrations"
+    )
+
+
+class AssistantCreateResponse(BaseModel):
+    """Response model for assistant creation with validation."""
+
+    model_config = ConfigDict(extra="ignore", populate_by_name=True)
+
+    message: str = Field(..., description="Response message")
+    assistant_id: Optional[str] = Field(
+        None,
+        alias="assistantId",
+        description="Created assistant ID (None if validation failed)",
+    )
+    validation: Optional[IntegrationValidationResult] = Field(
+        None,
+        description="Validation result (populated if validation found missing integrations)",
+    )
+
+
+class AssistantUpdateResponse(BaseModel):
+    """Response model for assistant update with validation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    message: str = Field(..., description="Response message")
+    validation: Optional[IntegrationValidationResult] = Field(
+        None,
+        description="Validation result (populated if validation found missing integrations)",
+    )
+
+
 class AssistantVersion(BaseModel):
     """Immutable snapshot of assistant configuration for a specific version."""
 

{codemie_sdk_python-0.1.280 → codemie_sdk_python-0.1.282}/src/codemie_sdk/services/assistant.py

@@ -14,6 +14,8 @@ from ..models.assistant import (
     Assistant,
     AssistantCreateRequest,
     AssistantUpdateRequest,
+    AssistantCreateResponse,
+    AssistantUpdateResponse,
     ToolKitDetails,
     AssistantChatRequest,
     BaseModelResponse,
@@ -90,20 +92,24 @@ class AssistantService:
         """
         return self._api.get(f"/v1/assistants/slug/{slug}", Assistant)
 
-    def create(self, request: AssistantCreateRequest) -> dict:
+    def create(self, request: AssistantCreateRequest) -> AssistantCreateResponse:
         """Create a new assistant.
 
         Args:
             request: Assistant creation request
 
         Returns:
-            Created assistant details
+            AssistantCreateResponse with assistant_id and optional validation results
         """
         return self._api.post(
-            "/v1/assistants", dict, json_data=request.model_dump(exclude_none=True)
+            "/v1/assistants",
+            AssistantCreateResponse,
+            json_data=request.model_dump(exclude_none=True),
         )
 
-    def update(self, assistant_id: str, request: AssistantUpdateRequest) -> dict:
+    def update(
+        self, assistant_id: str, request: AssistantUpdateRequest
+    ) -> AssistantUpdateResponse:
         """Update an existing assistant.
 
         Args:
@@ -111,11 +117,11 @@ class AssistantService:
             request: Assistant update request
 
         Returns:
-            Updated assistant details
+            AssistantUpdateResponse with optional validation results
         """
         return self._api.put(
             f"/v1/assistants/{assistant_id}",
-            dict,
+            AssistantUpdateResponse,
             json_data=request.model_dump(exclude_none=True),
         )
 
@@ -296,7 +302,8 @@ class AssistantService:
                     if hasattr(version, "assistant_ids")
                     else assistant.assistant_ids,
                 )
-                resp = self.update(assistant_id, update_req)
+                update_resp = self.update(assistant_id, update_req)
+                resp = update_resp.model_dump()
                 resp["_rollback_fallback"] = True
                 resp["_target_version"] = version_number
                 if change_notes: