codemie-sdk-python 0.1.595__tar.gz → 0.1.613__tar.gz

codemie_sdk_python-0.1.595/README.md → codemie_sdk_python-0.1.613/PKG-INFO

@@ -1,3 +1,16 @@
+Metadata-Version: 2.1
+Name: codemie-sdk-python
+Version: 0.1.613
+Summary: CodeMie SDK for Python
+Author: Vadym Vlasenko
+Author-email: vadym_vlasenko@epam.com
+Requires-Python: >=3.12,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: pydantic (>=2.12.2,<3.0.0)
+Requires-Dist: requests (>=2.31.0,<3.0.0)
+Description-Content-Type: text/markdown
+
 # CodeMie Python SDK
 
 Python SDK for CodeMie services. This SDK provides a comprehensive interface to interact with CodeMie services, including LLM (Large Language Models), assistants, workflows, and tools.
@@ -340,48 +353,6 @@ response = client.assistants.chat_with_version("assistant-id", version_number=2,
 print(response.generated)
 ```
 
-#### Project Feature Mapping
-
-Control which assistants have a feature enabled per project. Currently the only supported feature is `teams`.
-
-1. **List assistants with a feature enabled for a project**
-```python
-from codemie_sdk import AssistantProjectFeature
-
-assistants = client.assistants.list_project_mapping(
-    feature=AssistantProjectFeature.teams,
-    project="my-project",
-    page=0,
-    per_page=12,
-)
-for a in assistants:
-    print(a.name)
-```
-
-2. **Enable a feature for an assistant on a project**
-```python
-from codemie_sdk import AssistantProjectFeature, AssistantProjectMappingRequest
-
-request = AssistantProjectMappingRequest(
-    project_name="my-project",
-    feature=AssistantProjectFeature.teams,
-)
-response = client.assistants.enable_project_mapping("assistant-id", request)
-print(response.message)
-```
-
-3. **Disable a feature for an assistant on a project**
-```python
-from codemie_sdk import AssistantProjectFeature
-
-response = client.assistants.disable_project_mapping(
-    "assistant-id",
-    project="my-project",
-    feature=AssistantProjectFeature.teams,
-)
-print(response.message)
-```
-
 ### Datasource Service
 
 The Datasource service enables managing various types of data sources in CodeMie, including code repositories, Confluence spaces, Jira projects, files, and Google documents.
@@ -1430,4 +1401,4 @@ poetry build
 
 # Publish to PyPI
 make publish
-```
+```

codemie_sdk_python-0.1.595/PKG-INFO → codemie_sdk_python-0.1.613/README.md

@@ -1,16 +1,3 @@
-Metadata-Version: 2.1
-Name: codemie-sdk-python
-Version: 0.1.595
-Summary: CodeMie SDK for Python
-Author: Vadym Vlasenko
-Author-email: vadym_vlasenko@epam.com
-Requires-Python: >=3.12,<4.0
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: pydantic (>=2.12.2,<3.0.0)
-Requires-Dist: requests (>=2.31.0,<3.0.0)
-Description-Content-Type: text/markdown
-
 # CodeMie Python SDK
 
 Python SDK for CodeMie services. This SDK provides a comprehensive interface to interact with CodeMie services, including LLM (Large Language Models), assistants, workflows, and tools.
@@ -353,48 +340,6 @@ response = client.assistants.chat_with_version("assistant-id", version_number=2,
 print(response.generated)
 ```
 
-#### Project Feature Mapping
-
-Control which assistants have a feature enabled per project. Currently the only supported feature is `teams`.
-
-1. **List assistants with a feature enabled for a project**
-```python
-from codemie_sdk import AssistantProjectFeature
-
-assistants = client.assistants.list_project_mapping(
-    feature=AssistantProjectFeature.teams,
-    project="my-project",
-    page=0,
-    per_page=12,
-)
-for a in assistants:
-    print(a.name)
-```
-
-2. **Enable a feature for an assistant on a project**
-```python
-from codemie_sdk import AssistantProjectFeature, AssistantProjectMappingRequest
-
-request = AssistantProjectMappingRequest(
-    project_name="my-project",
-    feature=AssistantProjectFeature.teams,
-)
-response = client.assistants.enable_project_mapping("assistant-id", request)
-print(response.message)
-```
-
-3. **Disable a feature for an assistant on a project**
-```python
-from codemie_sdk import AssistantProjectFeature
-
-response = client.assistants.disable_project_mapping(
-    "assistant-id",
-    project="my-project",
-    feature=AssistantProjectFeature.teams,
-)
-print(response.message)
-```
-
 ### Datasource Service
 
 The Datasource service enables managing various types of data sources in CodeMie, including code repositories, Confluence spaces, Jira projects, files, and Google documents.
@@ -1443,4 +1388,4 @@ poetry build
 
 # Publish to PyPI
 make publish
-```
+```

{codemie_sdk_python-0.1.595 → codemie_sdk_python-0.1.613}/pyproject.toml

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

{codemie_sdk_python-0.1.595 → codemie_sdk_python-0.1.613}/src/codemie_sdk/__init__.py

@@ -151,10 +151,6 @@ from .models.skill import (
 from .services.skill import SkillService
 from .models.assistant import (
     BuiltinSubagent,
-    AssistantProjectFeature,
-    AssistantProjectMappingRequest,
-    AssistantProjectMappingPagination,
-    AssistantProjectMappingListResponse,
 )
 from .models.integration import (
     TransferMode,
@@ -165,10 +161,6 @@ from .models.integration import (
 __version__ = "0.2.12"
 __all__ = [
     "CodeMieClient",
-    "AssistantProjectFeature",
-    "AssistantProjectMappingRequest",
-    "AssistantProjectMappingPagination",
-    "AssistantProjectMappingListResponse",
     "VendorType",
     "VendorAssistantSetting",
     "VendorAssistantSettingsResponse",

{codemie_sdk_python-0.1.595 → codemie_sdk_python-0.1.613}/src/codemie_sdk/client/client.py

@@ -28,7 +28,9 @@ from ..services.activity_event import ActivityEventService
 from ..services.codemie_guardrails import CodemieGuardrailService
 from ..services.dynamic_config import DynamicConfigService
 from ..services.mcp_configs import MCPConfigsService
+from ..services.cost_center import CostCenterService
 from ..services.customer_config_service import CustomerConfigService
+from ..services.project_budget_groups import ProjectBudgetGroupsService
 
 
 class CodeMieClient:
@@ -173,6 +175,12 @@ class CodeMieClient:
         self.customer_config = CustomerConfigService(
             self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
+        self.cost_centers = CostCenterService(
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
+        )
+        self.project_budget_groups = ProjectBudgetGroupsService(
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
+        )
 
     @property
     def token(self) -> str:

{codemie_sdk_python-0.1.595 → codemie_sdk_python-0.1.613}/src/codemie_sdk/models/assistant.py

@@ -181,6 +181,44 @@ class AssistantListResponse(BaseModel):
     created_by: User | None = None
 
 
+class ToolCallPolicy(str, Enum):
+    """Policy governing when tool calls require user approval."""
+
+    AUTO_APPROVE = "auto_approve"
+    APPROVE_FOR_ME = "approve_for_me"
+    ASK_FOR_APPROVAL = "ask_for_approval"
+
+
+class ToolCallAction(str, Enum):
+    """User action on a pending tool-call confirmation."""
+
+    ALLOW = "allow"
+    DENY = "deny"
+
+
+class ToolPermissionsConfig(BaseModel):
+    """Configuration for tool-call approval policy on an assistant."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    tool_call_policy: ToolCallPolicy = ToolCallPolicy.AUTO_APPROVE
+    allow_override: bool = True
+
+
+class ToolCallResumeRequest(BaseModel):
+    """Request body for POST /v1/assistants/{id}/model/tool-call/resume."""
+
+    model_config = ConfigDict(extra="ignore", use_enum_values=True)
+
+    conversation_id: str = Field(
+        description="Conversation ID that has a pending tool-call checkpoint"
+    )
+    action: ToolCallAction = Field(
+        description="User decision: 'allow' executes the tool, 'deny' rejects it"
+    )
+    stream: bool = Field(default=True, description="Enable streaming NDJSON response")
+
+
 class Assistant(AssistantBase):
     """Full assistant model with additional fields."""
 
@@ -233,6 +271,7 @@ class Assistant(AssistantBase):
         default=None,
         description="Number of times this assistant has been cloned in the Marketplace",
     )
+    tool_permissions: ToolPermissionsConfig | None = None
 
 
 class AssistantRequestBase(AssistantBase):
@@ -268,6 +307,7 @@ class AssistantRequestBase(AssistantBase):
     enabled_builtin_subagents: list[str] | None = None
     prompt_variables: list[PromptVariable] = Field(default_factory=list)
     skip_integration_validation: bool | None = Field(default=False)
+    tool_permissions: ToolPermissionsConfig | None = None
 
 
 class AssistantCreateRequest(AssistantRequestBase):
@@ -612,53 +652,3 @@ class ReactionRequest(BaseModel):
     model_config = ConfigDict(extra="ignore", use_enum_values=True)
 
     reaction: ReactionType = Field(description="Reaction type: 'like' or 'dislike'")
-
-
-class AssistantProjectFeature(str, Enum):
-    """Feature discriminator for assistant/project mappings (e.g. Teams bot).
-
-    This enum is expected to grow over time on the backend; treat unrecognized
-    wire values defensively rather than assuming this is the complete set.
-    """
-
-    TEAMS = "teams"
-
-
-class AssistantProjectMappingRequest(BaseModel):
-    """Request body for POST /v1/assistants/{assistant_id}/projects/mapping."""
-
-    model_config = ConfigDict(extra="ignore", use_enum_values=True)
-
-    project_name: str = Field(description="Project (application) name")
-    feature: AssistantProjectFeature = Field(
-        description="Feature discriminator this mapping is for, e.g. 'teams'"
-    )
-
-
-class AssistantProjectMappingPagination(BaseModel):
-    """Pagination metadata returned by GET /v1/assistants/projects/mapping."""
-
-    model_config = ConfigDict(extra="ignore")
-
-    page: int = Field(description="Current page, 0-indexed")
-    per_page: int = Field(description="Items per page")
-    total: int = Field(description="Total matching assistants across all pages")
-
-
-class AssistantProjectMappingListResponse(BaseModel):
-    """Response for GET /v1/assistants/projects/mapping.
-
-    List items are intentionally typed as ``dict[str, Any]`` rather than
-    ``Assistant``/``AssistantBase``. The backend contract documents this
-    endpoint as returning "the same shape as GET /v1/assistants", but that
-    endpoint's own shape is parameterized by a ``minimal_response`` flag that
-    this new endpoint does not expose, and the exact per-item fields returned
-    here are not otherwise confirmed. Rather than guess the wire format,
-    callers get the raw dict and can assert on whichever fields they need
-    (e.g. ``item["id"]``).
-    """
-
-    model_config = ConfigDict(extra="ignore")
-
-    data: list[dict[str, Any]] = Field(default_factory=list)
-    pagination: AssistantProjectMappingPagination

{codemie_sdk_python-0.1.595 → codemie_sdk_python-0.1.613}/src/codemie_sdk/models/conversation.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from datetime import datetime
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 from codemie_sdk.models.assistant import ContextType
 
@@ -14,6 +14,8 @@ class Conversation(BaseModel):
     Model for conversation summary data as returned from the list endpoint.
     """
 
+    model_config = ConfigDict(extra="ignore")
+
     id: str
     name: str
     folder: str | None
@@ -21,6 +23,7 @@ class Conversation(BaseModel):
     date: str
     assistant_ids: list[str]
     initial_assistant_id: str | None
+    finished_at: datetime | None = None
 
 
 class Mark(BaseModel):
@@ -177,6 +180,7 @@ class ConversationDetails(BaseModel):
     is_workflow_conversation: bool | None = None
     category: str | None
     mcp_server_single_usage: bool | None = False
+    finished_at: datetime | None = None
 
 
 class ConversationShareRequest(BaseModel):
@@ -256,3 +260,58 @@ class UpsertHistoryResponse(BaseModel):
     new_messages: int
     total_messages: int
     created: bool
+
+
+class ConversationFinishResponse(BaseModel):
+    """Response for POST /conversations/{conversation_id}/finish (owner-only, not idempotent).
+
+    Already-finished conversations cause the server to raise HTTP 409 rather
+    than return this response again.
+    """
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    conversation_id: str = Field(alias="conversationId")
+    finished_at: datetime = Field(alias="finishedAt")
+
+
+class ConversationFinishResult(BaseModel):
+    """Per-conversation-ID result entry within a bulk-finish response."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    conversation_id: str = Field(alias="conversationId")
+    already_finished: bool = Field(default=False, alias="alreadyFinished")
+    finished_at: datetime | None = Field(default=None, alias="finishedAt")
+    error: str | None = None
+
+
+class ConversationFinishBulkRequest(BaseModel):
+    """Request body for POST /admin/conversations/finish-bulk."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    conversation_ids: list[str] = Field(alias="conversationIds")
+
+
+class ConversationFinishBulkResponse(BaseModel):
+    """Response for POST /admin/conversations/finish-bulk (admin-only, idempotent per ID)."""
+
+    total: int
+    results: list[ConversationFinishResult]
+
+
+class ConversationPagination(BaseModel):
+    """Pagination metadata for the admin conversation list endpoint."""
+
+    page: int
+    per_page: int
+    total: int
+    pages: int
+
+
+class AdminConversationListResponse(BaseModel):
+    """Paginated response for GET /admin/conversations (admin-only, cross-user)."""
+
+    data: list[Conversation]
+    pagination: ConversationPagination

codemie_sdk_python-0.1.613/src/codemie_sdk/models/cost_center.py

@@ -0,0 +1,42 @@
+# Copyright 2026 EPAM Systems, Inc. ("EPAM")
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Cost center management models for the CodeMie SDK."""
+
+from datetime import datetime
+from uuid import UUID
+
+from pydantic import BaseModel, ConfigDict
+
+
+class CostCenterResponse(BaseModel):
+    """Response model for a cost center."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: UUID
+    name: str
+    description: str | None = None
+    created_by: str
+    created_at: datetime
+    project_count: int = 0
+
+
+class CreateCostCenterRequest(BaseModel):
+    """Request model for creating a cost center."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    name: str
+    description: str | None = None

{codemie_sdk_python-0.1.595 → codemie_sdk_python-0.1.613}/src/codemie_sdk/models/datasource.py

@@ -123,6 +123,15 @@ class AzureDevOpsWorkItem(BaseModel):
     model_config = ConfigDict(extra="ignore")
 
 
+class XWiki(BaseModel):
+    """Model for xWiki-specific response fields"""
+
+    space: str | None = None
+    wiki: str | None = None
+
+    model_config = ConfigDict(extra="ignore")
+
+
 class Code(BaseModel):
     """Model for code repository datasource creation"""
 
@@ -239,6 +248,17 @@ class AzureDevOpsWorkItemDataSourceRequest(BaseDataSourceRequest, AzureDevOpsWor
         super().__init__(type=DataSourceType.AZURE_DEVOPS_WORK_ITEM, **data)
 
 
+class XWikiDataSourceRequest(BaseDataSourceRequest, XWiki):
+    """Model for xWiki datasource creation requests"""
+
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.XWIKI, **data)
+
+    @classmethod
+    def required_fields(cls) -> list[str]:
+        return ["space"]
+
+
 class BaseUpdateDataSourceRequest(BaseDataSourceRequest):
     """Mixin update-specific reindex fields"""
 
@@ -327,6 +347,13 @@ class UpdateAzureDevOpsWorkItemDataSourceRequest(
         super().__init__(type=DataSourceType.AZURE_DEVOPS_WORK_ITEM, **data)
 
 
+class UpdateXWikiDataSourceRequest(BaseUpdateDataSourceRequest, XWiki):
+    """Model for xWiki datasource updates"""
+
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.XWIKI, **data)
+
+
 class CodeAnalysisDataSourceRequest(BaseModel):
     """Model for provider-based datasource creation requests"""
 
@@ -399,6 +426,8 @@ class DataSource(BaseModel):
     azure_devops_wiki: AzureDevOpsWiki | None = None
     # Azure DevOps Work Item specific fields
     azure_devops_work_item: AzureDevOpsWorkItem | None = None
+    # xWiki specific fields
+    xwiki: XWiki | None = None
 
     @model_validator(mode="before")
     def before_init(cls, values):
@@ -417,6 +446,7 @@ class DataSource(BaseModel):
             DataSourceType.GOOGLE,
             DataSourceType.AZURE_DEVOPS_WIKI,
             DataSourceType.AZURE_DEVOPS_WORK_ITEM,
+            DataSourceType.XWIKI,
         ]:
             complete_state = values.get("complete_state", 0)
             if complete_state is not None:

{codemie_sdk_python-0.1.595 → codemie_sdk_python-0.1.613}/src/codemie_sdk/models/integration.py

@@ -41,6 +41,7 @@ class CredentialTypes(str, Enum):
     SHAREPOINT = "SharePoint"
     GOOGLE_OAUTH = "GoogleOAuth"
     XWIKI = "XWiki"
+    MS_TEAMS = "MSTeams"
 
 
 class IntegrationType(str, Enum):

{codemie_sdk_python-0.1.595 → codemie_sdk_python-0.1.613}/src/codemie_sdk/models/project.py

@@ -79,6 +79,7 @@ class ProjectListItem(BaseModel):
     cost_center_name: str | None = None
     spending: float | None = None
     budgets: list[ProjectBudgetItem] | None = []
+    chargeback_attribution: str = "project"
 
 
 class PaginationInfo(BaseModel):
@@ -116,6 +117,7 @@ class ProjectDetailResponse(BaseModel):
     cost_center_id: UUID | None = None
     cost_center_name: str | None = None
     members: list[ProjectMember]
+    chargeback_attribution: str = "project"
 
 
 class ProjectCreateRequest(BaseModel):
@@ -147,6 +149,7 @@ class ProjectCreateResponse(BaseModel):
     cost_center_id: UUID | None = None
     cost_center_name: str | None = None
     enforce_member_spend_limits: bool = False
+    chargeback_attribution: str = "project"
 
 
 # PATCH /v1/projects/{name} returns the same shape as the create response
@@ -168,6 +171,7 @@ class ProjectUpdateRequest(BaseModel):
     cost_center_id: UUID | None = None
     clear_cost_center: bool = False
     enforce_member_spend_limits: bool | None = None
+    chargeback_attribution: str | None = None
 
 
 class ProjectDeleteResponse(BaseModel):