codemie-sdk-python 0.1.314__tar.gz → 0.1.316__tar.gz

{codemie_sdk_python-0.1.314 → codemie_sdk_python-0.1.316}/PKG-INFO

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

{codemie_sdk_python-0.1.314 → codemie_sdk_python-0.1.316}/pyproject.toml

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

{codemie_sdk_python-0.1.314 → codemie_sdk_python-0.1.316}/src/codemie_sdk/__init__.py

@@ -107,6 +107,15 @@ from .models.mermaid import (
     ResponseType,
 )
 from .services.mermaid import MermaidService
+from .models.categories import (
+    Category,
+    CategoryCreateRequest,
+    CategoryUpdateRequest,
+    CategoryResponse,
+    CategoryListResponse,
+    CategoryListMetadata,
+)
+from .services.categories import CategoryService
 
 __version__ = "0.2.12"
 __all__ = [
@@ -186,4 +195,11 @@ __all__ = [
     "ContentType",
     "ResponseType",
     "MermaidService",
+    "Category",
+    "CategoryCreateRequest",
+    "CategoryUpdateRequest",
+    "CategoryResponse",
+    "CategoryListResponse",
+    "CategoryListMetadata",
+    "CategoryService",
 ]

{codemie_sdk_python-0.1.314 → codemie_sdk_python-0.1.316}/src/codemie_sdk/client/client.py

@@ -6,6 +6,7 @@ from ..auth.credentials import KeycloakCredentials
 from ..services.admin import AdminService
 from ..services.analytics import AnalyticsService
 from ..services.assistant import AssistantService
+from ..services.categories import CategoryService
 from ..services.conversation import ConversationService
 from ..services.datasource import DatasourceService
 from ..services.llm import LLMService
@@ -80,6 +81,9 @@ class CodeMieClient:
         self.assistants = AssistantService(
             self._api_domain, self._token, verify_ssl=verify_ssl
         )
+        self.categories = CategoryService(
+            self._api_domain, self._token, verify_ssl=verify_ssl
+        )
         self.llms = LLMService(self._api_domain, self._token, verify_ssl=verify_ssl)
         self.mermaid = MermaidService(
             self._api_domain, self._token, verify_ssl=verify_ssl
@@ -151,6 +155,9 @@ class CodeMieClient:
         self.assistants = AssistantService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.categories = CategoryService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
         self.llms = LLMService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )

{codemie_sdk_python-0.1.314 → codemie_sdk_python-0.1.316}/src/codemie_sdk/models/assistant.py

@@ -105,6 +105,13 @@ class MCPServerConfig(BaseModel):
     auth_token: Optional[str] = Field(
         None, description="Authentication token for the MCP-Connect server"
     )
+    single_usage: bool = Field(
+        False,
+        description="Whether this MCP server configuration is for single use only",
+    )
+    tools: Optional[list[str]] = Field(
+        None, description="List of tool names available in this MCP server"
+    )
 
 
 class MCPServerDetails(BaseModel):
@@ -113,14 +120,17 @@ class MCPServerDetails(BaseModel):
     name: str
     description: Optional[str]
     enabled: bool
+    mcp_config_id: Optional[str] = None
     config: Optional[MCPServerConfig] = None
     mcp_connect_url: Optional[str] = None
     tools_tokens_size_limit: Optional[int] = None
     command: Optional[str] = None
     arguments: Optional[str] = None
     settings: Optional[Integration] = None
-    mcp_connect_auth_token: Optional[Integration] = None
     integration_alias: Optional[str] = None
+    mcp_connect_auth_token: Optional[Integration] = None
+    resolve_dynamic_values_in_arguments: bool = False
+    tools: Optional[list[str]] = None
 
 
 class SystemPromptHistory(BaseModel):

codemie_sdk_python-0.1.316/src/codemie_sdk/models/categories.py

@@ -0,0 +1,71 @@
+"""Models for category-related data structures."""
+
+from typing import List, Optional
+
+from pydantic import BaseModel, Field, ConfigDict
+
+
+class Category(BaseModel):
+    """Model for assistant category."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Category ID")
+    name: str = Field(..., description="Category display name")
+    description: Optional[str] = Field(None, description="Category description")
+
+
+class CategoryCreateRequest(BaseModel):
+    """Model for creating a new category."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    name: str = Field(..., description="Category name")
+    description: Optional[str] = Field(None, description="Category description")
+
+
+class CategoryUpdateRequest(BaseModel):
+    """Model for updating an existing category."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    name: str = Field(..., description="Category name")
+    description: Optional[str] = Field(None, description="Category description")
+
+
+class CategoryResponse(BaseModel):
+    """Model for category response with assistant counts."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Category ID")
+    name: str = Field(..., description="Category name")
+    description: Optional[str] = Field(None, description="Category description")
+    marketplace_count: int = Field(
+        default=0, description="Number of marketplace assistants in this category"
+    )
+    project_count: int = Field(
+        default=0, description="Number of project assistants in this category"
+    )
+
+
+class CategoryListMetadata(BaseModel):
+    """Metadata for paginated category list response."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    page: int = Field(..., description="Current page number (0-indexed)")
+    per_page: int = Field(..., description="Items per page")
+    total: int = Field(..., description="Total number of categories")
+    total_pages: int = Field(..., description="Total number of pages")
+
+
+class CategoryListResponse(BaseModel):
+    """Model for paginated category list response."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    items: List[CategoryResponse] = Field(
+        default_factory=list, description="List of categories with counts"
+    )
+    metadata: CategoryListMetadata = Field(..., description="Pagination metadata")

{codemie_sdk_python-0.1.314 → codemie_sdk_python-0.1.316}/src/codemie_sdk/models/conversation.py

@@ -83,7 +83,7 @@ class HistoryItem(BaseModel):
     fileNames: List[str]
     assistantId: Optional[str] = None
     thoughts: Optional[List[Thought]] = Field(default_factory=list)
-    workflowExecutionRef: Optional[str] = None
+    workflowExecutionRef: Optional[Union[str, bool]] = None
     executionId: Optional[str] = None
 
 
@@ -144,7 +144,9 @@ class ConversationCreateRequest(BaseModel):
     initial_assistant_id: Optional[str] = None
     folder: Optional[str] = None
     mcp_server_single_usage: Optional[bool] = False
-    is_workflow: Optional[bool] = None
+    is_workflow_conversation: Optional[bool] = Field(
+        default=None, serialization_alias="is_workflow"
+    )
 
 
 class ConversationDetails(BaseModel):

codemie_sdk_python-0.1.316/src/codemie_sdk/services/categories.py

@@ -0,0 +1,146 @@
+"""Category service implementation."""
+
+from typing import List
+
+from ..models.categories import (
+    Category,
+    CategoryCreateRequest,
+    CategoryUpdateRequest,
+    CategoryResponse,
+    CategoryListResponse,
+)
+from ..utils import ApiRequestHandler
+
+
+class CategoryService:
+    """Service for managing CodeMie assistant categories."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the category 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_categories(self) -> List[Category]:
+        """Get all available assistant categories (legacy, non-paginated).
+
+        This endpoint maintains backward compatibility with existing clients.
+        For paginated access with assistant counts, use list_categories().
+
+        Returns:
+            List of all categories
+        """
+        return self._api.get("/v1/assistants/categories", List[Category])
+
+    def list_categories(
+        self, page: int = 0, per_page: int = 10
+    ) -> CategoryListResponse:
+        """Get paginated list of categories with assistant counts.
+
+        Returns counts separated by marketplace vs project assistants.
+        This is the preferred endpoint for admin UI with pagination support.
+
+        Args:
+            page: Page number (0-indexed, default: 0)
+            per_page: Number of items per page (1-100, default: 10)
+
+        Returns:
+            Paginated response with categories and metadata
+        """
+        params = {"page": page, "per_page": per_page}
+        return self._api.get(
+            "/v1/assistants/categories/list", CategoryListResponse, params=params
+        )
+
+    def get_category(self, category_id: str) -> CategoryResponse:
+        """Get a specific category by ID with assistant counts.
+
+        Args:
+            category_id: Category ID (humanized, e.g., "migration_modernization")
+
+        Returns:
+            Category details with assistant counts
+
+        Raises:
+            NotFoundError: Category not found
+        """
+        return self._api.get(
+            f"/v1/assistants/categories/{category_id}", CategoryResponse
+        )
+
+    def create_category(self, request: CategoryCreateRequest) -> CategoryResponse:
+        """Create a new category. Admin access required.
+
+        The category ID will be auto-generated from the name using humanization logic:
+        - Special characters are removed
+        - Spaces are replaced with underscores
+        - All text is lowercased
+
+        Example: "Migration & Modernization" → "migration_modernization"
+
+        Args:
+            request: Category creation request with name and description
+
+        Returns:
+            Created category with assistant counts (will be 0 for new categories)
+
+        Raises:
+            ApiError: Invalid request data or duplicate category ID (400)
+            ApiError: User is not an admin (403)
+        """
+        return self._api.post(
+            "/v1/assistants/categories",
+            CategoryResponse,
+            json_data=request.model_dump(exclude_none=True),
+        )
+
+    def update_category(
+        self, category_id: str, request: CategoryUpdateRequest
+    ) -> CategoryResponse:
+        """Update an existing category. Admin access required.
+
+        If assistants are assigned to this category, the operation will succeed
+        but the response will include the count of affected assistants.
+        The frontend should display a warning to the admin when editing categories
+        with assigned assistants.
+
+        Args:
+            category_id: Category ID to update
+            request: Category update request with new name and description
+
+        Returns:
+            Updated category with current assistant counts
+
+        Raises:
+            ApiError: User is not an admin (403)
+            NotFoundError: Category not found (404)
+        """
+        return self._api.put(
+            f"/v1/assistants/categories/{category_id}",
+            CategoryResponse,
+            json_data=request.model_dump(exclude_none=True),
+        )
+
+    def delete_category(self, category_id: str) -> dict:
+        """Delete a category. Admin access required.
+
+        This operation will fail with a 409 Conflict error if any assistants
+        (marketplace or project) are assigned to this category.
+        The admin must first remove all assistants from the category before deletion.
+
+        Args:
+            category_id: Category ID to delete
+
+        Returns:
+            Empty dict on successful deletion (204 No Content)
+
+        Raises:
+            ApiError: User is not an admin (403)
+            NotFoundError: Category not found (404)
+            ApiError: Category has assigned assistants and cannot be deleted (409)
+        """
+        return self._api.delete(f"/v1/assistants/categories/{category_id}", dict)

{codemie_sdk_python-0.1.314 → codemie_sdk_python-0.1.316}/src/codemie_sdk/services/conversation.py

@@ -84,7 +84,7 @@ class ConversationService:
         return self._api.post(
             "/v1/conversations",
             dict,
-            json_data=request.model_dump(exclude_none=True),
+            json_data=request.model_dump(exclude_none=True, by_alias=True),
         )
 
     def chat(