codemie-sdk-python 0.1.503__tar.gz → 0.1.505__tar.gz

codemie_sdk_python-0.1.503/README.md → codemie_sdk_python-0.1.505/PKG-INFO

@@ -1,3 +1,16 @@
+Metadata-Version: 2.1
+Name: codemie-sdk-python
+Version: 0.1.505
+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,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.
@@ -1388,4 +1443,4 @@ poetry build
 
 # Publish to PyPI
 make publish
-```
+```

codemie_sdk_python-0.1.503/PKG-INFO → codemie_sdk_python-0.1.505/README.md

@@ -1,16 +1,3 @@
-Metadata-Version: 2.1
-Name: codemie-sdk-python
-Version: 0.1.503
-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,6 +340,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.
@@ -1401,4 +1430,4 @@ poetry build
 
 # Publish to PyPI
 make publish
-```
+```

{codemie_sdk_python-0.1.503 → codemie_sdk_python-0.1.505}/pyproject.toml

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

{codemie_sdk_python-0.1.503 → codemie_sdk_python-0.1.505}/src/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_python-0.1.503 → codemie_sdk_python-0.1.505}/src/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_python-0.1.503 → codemie_sdk_python-0.1.505}/src/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,
+        )