codemie-sdk-python 0.1.503__py3-none-any.whl → 0.1.504__py3-none-any.whl

codemie_sdk/__init__.py

@@ -148,7 +148,12 @@ from .models.skill import (
     SkillPagination,
 )
 from .services.skill import SkillService
-from .models.assistant import BuiltinSubagent
+from .models.assistant import (
+    BuiltinSubagent,
+    AssistantProjectFeature,
+    AssistantProjectMappingRequest,
+    AssistantProjectMappingResponse,
+)
 
 __version__ = "0.2.12"
 __all__ = [
@@ -264,4 +269,7 @@ __all__ = [
     "SkillPagination",
     "SkillService",
     "BuiltinSubagent",
+    "AssistantProjectFeature",
+    "AssistantProjectMappingRequest",
+    "AssistantProjectMappingResponse",
 ]

codemie_sdk/models/assistant.py

@@ -607,3 +607,26 @@ 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):
+    """Enum for assistant project feature discriminators."""
+
+    teams = "teams"
+
+
+class AssistantProjectMappingRequest(BaseModel):
+    """Request body for enabling an assistant for a project feature."""
+
+    model_config = ConfigDict(extra="ignore", use_enum_values=True)
+
+    project_name: str = Field(description="Project identifier")
+    feature: AssistantProjectFeature = Field(description="Feature discriminator")
+
+
+class AssistantProjectMappingResponse(BaseModel):
+    """Response for assistant project mapping operations."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    message: str = Field(description="Operation result message")

codemie_sdk/services/assistant.py

@@ -28,6 +28,9 @@ from ..models.assistant import (
     ExportAssistantPayload,
     AssistantEvaluationRequest,
     ReactionRequest,
+    AssistantProjectFeature,
+    AssistantProjectMappingRequest,
+    AssistantProjectMappingResponse,
 )
 from ..models.common import PaginationParams
 from ..utils import ApiRequestHandler, TokenSource
@@ -607,3 +610,81 @@ class AssistantService:
             Response from the API
         """
         return self._api.delete(f"/v1/assistants/{assistant_id}/reactions", dict)
+
+    def list_project_mapping(
+        self,
+        feature: Union[AssistantProjectFeature, str],
+        project: str,
+        page: int = 0,
+        per_page: int = 12,
+    ) -> list[Assistant]:
+        """List assistants enabled for a project feature.
+
+        Args:
+            feature: Feature discriminator (e.g. "teams")
+            project: Project name (application name)
+            page: 0-indexed page number
+            per_page: Items per page (max 100)
+
+        Returns:
+            List of assistants mapped to the project feature
+        """
+        params = {
+            "feature": feature if isinstance(feature, str) else feature.value,
+            "project": project,
+            "page": page,
+            "per_page": per_page,
+        }
+        return self._api.get(
+            "/v1/assistants/projects/mapping", list[Assistant], params=params
+        )
+
+    def enable_project_mapping(
+        self,
+        assistant_id: str,
+        request: AssistantProjectMappingRequest,
+    ) -> AssistantProjectMappingResponse:
+        """Enable an assistant for a project feature.
+
+        Idempotent: returns 200 whether the mapping already existed or not.
+
+        Args:
+            assistant_id: UUID of the assistant
+            request: Mapping request with project_name and feature
+
+        Returns:
+            AssistantProjectMappingResponse with confirmation message
+        """
+        return self._api.post(
+            f"/v1/assistants/{assistant_id}/projects/mapping",
+            AssistantProjectMappingResponse,
+            json_data=request.model_dump(exclude_none=True),
+            wrap_response=False,
+        )
+
+    def disable_project_mapping(
+        self,
+        assistant_id: str,
+        project: str,
+        feature: Union[AssistantProjectFeature, str],
+    ) -> AssistantProjectMappingResponse:
+        """Disable an assistant for a project feature.
+
+        Args:
+            assistant_id: UUID of the assistant
+            project: Project name
+            feature: Feature discriminator (e.g. "teams")
+
+        Returns:
+            AssistantProjectMappingResponse with confirmation message
+        """
+        params = {
+            "project": project,
+            "feature": feature if isinstance(feature, str) else feature.value,
+        }
+        return self._api.delete(
+            f"/v1/assistants/{assistant_id}/projects/mapping",
+            AssistantProjectMappingResponse,
+            params=params,
+            wrap_response=False,
+        )

{codemie_sdk_python-0.1.503.dist-info → codemie_sdk_python-0.1.504.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.503
+Version: 0.1.504
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com
@@ -353,6 +353,48 @@ 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.

{codemie_sdk_python-0.1.503.dist-info → codemie_sdk_python-0.1.504.dist-info}/RECORD

@@ -1,4 +1,4 @@
-codemie_sdk/__init__.py,sha256=etnHL9r4gggqlHhQ4hQPDxVHHYSNUT-L5UlWHXJ4Fpg,7693
+codemie_sdk/__init__.py,sha256=_UfLtQJ46-DnFEbmO46FbyASz6T1pUkXUjPmSzsIwzk,7912
 codemie_sdk/auth/__init__.py,sha256=IksEj223xEZtJ-cQ0AT9L0Bs9psIJ8QNzDXrPTUQ3xQ,126
 codemie_sdk/auth/credentials.py,sha256=hO0X3EQWZSlIm0ABhiJo4fIIndm2blB0JZWMaWvsuTE,7774
 codemie_sdk/client/__init__.py,sha256=yf6C39MmrJ6gK9ZHMhBeynKwUUYVSUTQbKxU8-4qpKg,101
@@ -7,7 +7,7 @@ codemie_sdk/exceptions.py,sha256=uuzvtvYBMzrNGt7Q4k5RVq0Pd8eJ_tR2tLy3pF2ICDs,114
 codemie_sdk/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_sdk/models/admin.py,sha256=2UGaO60ZrHHcDcV9thYBVxDyUCvtfPQhs10WswhGFJs,2298
 codemie_sdk/models/analytics.py,sha256=B29Cqt2Mvqohfjz-keRV5gFBpTKw_iIkF2ARxmEUv-Y,4953
-codemie_sdk/models/assistant.py,sha256=zdUOMEn9xG6STER6IfoltkC0hTA2UZXwkpaTRZiFC40,20784
+codemie_sdk/models/assistant.py,sha256=kq-2mDApNdAWtV4-Dx3BJhdvhLpw_IGaYly8Dwy9NFA,21475
 codemie_sdk/models/categories.py,sha256=eq7_sw-sevNLeWpjaW3N055Xcgbbevs0Bl_rj--maUk,2303
 codemie_sdk/models/common.py,sha256=ipNNdQlv4Ct1UyIAknjJbrdjjvv0lZJYK6dnPm8wai4,1170
 codemie_sdk/models/conversation.py,sha256=Sxgg2qvrHsDK2nKIeOrbv7wCdqZ9uSoq-e6nT4D1y_w,6288
@@ -33,7 +33,7 @@ codemie_sdk/models/workflow_state.py,sha256=uP1yLYwPpgPc17xpLwjanvlRzK6zT9qMZR-f
 codemie_sdk/models/workflow_thoughts.py,sha256=9fE380lMvC23fq677EI_JTk48Nf2jyPg3-mpj9VTx54,579
 codemie_sdk/services/admin.py,sha256=yk1xdBXDjpdW4OMOdBaURkZhAxak9h-TppKyhStLekc,4006
 codemie_sdk/services/analytics.py,sha256=HosMPk8B_kmuU5XcJhy5Mjx3ZjIwZAQB0LwDPsdH7SI,14882
-codemie_sdk/services/assistant.py,sha256=8rV41Eex0Hgqzsy2LFV3uSphfCyAvlGeSKhPBTdjlYs,20700
+codemie_sdk/services/assistant.py,sha256=1RWZl8LZEea1GB9AAUqu5W2CWL-8th0RENJL7Tcsf3A,23303
 codemie_sdk/services/categories.py,sha256=1qd98JCcMNdpbos0O1WpMApT6TurE7OuI7zoYewnwk8,5030
 codemie_sdk/services/codemie_guardrails.py,sha256=O3yCeIQYG8W7z-sVRR4Cw9UVncqP9lPVNxUSaUZrh7I,2833
 codemie_sdk/services/conversation.py,sha256=lYdNbiA1PeyN3qt0ZuHOIWDgVtP0keNXsDGQ-X-uluU,10447
@@ -57,6 +57,6 @@ codemie_sdk/services/workflow_execution.py,sha256=Nnf_ZJQrK9HiRyCRuUjUu3jXmbH3sn
 codemie_sdk/services/workflow_execution_state.py,sha256=u1xmGvaHjqvQS5ujhyvJ_Ac7ByjKem2Qe8q0swsmoWw,2045
 codemie_sdk/utils/__init__.py,sha256=EzK_UrS5KHmAnE73Hci9LaObB5omjVUhIq4svh0gfwo,137
 codemie_sdk/utils/http.py,sha256=yKWx-K9cFbpstVt0qran42e6MESm0f6XVd8MIctaD5Y,12546
-codemie_sdk_python-0.1.503.dist-info/METADATA,sha256=PqokQMw-4RB3JGTlE70W2LpA6HOmiUwrC2gDBAJ8f9Q,36801
-codemie_sdk_python-0.1.503.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-codemie_sdk_python-0.1.503.dist-info/RECORD,,
+codemie_sdk_python-0.1.504.dist-info/METADATA,sha256=0MZPRQ0QpfQOLEHMmswMVNiZ9XekC4BieKTRCIhvaFY,37915
+codemie_sdk_python-0.1.504.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+codemie_sdk_python-0.1.504.dist-info/RECORD,,