codemie-sdk-python 0.1.31__tar.gz → 0.1.33__tar.gz

{codemie_sdk_python-0.1.31 → codemie_sdk_python-0.1.33}/PKG-INFO

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

{codemie_sdk_python-0.1.31 → codemie_sdk_python-0.1.33}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.31"
+version = "0.1.33"
 description = "CodeMie SDK for Python"
 authors = [
     "Vadym Vlasenko <vadym_vlasenko@epam.com>",
@@ -28,6 +28,7 @@ python-gitlab = "^5.6.0"
 boto3 = "^1.38.20"
 aws-assume-role-lib = "^2.10.0"
 pyyaml = "^6.0.2"
+codemie-plugins = "^0.1.118"
 
 [build-system]
 requires = ["poetry-core"]

{codemie_sdk_python-0.1.31 → codemie_sdk_python-0.1.33}/src/codemie_sdk/models/datasource.py

@@ -264,6 +264,9 @@ class DataSource(BaseModel):
         populate_by_name=True,
     )
 
+    def __getitem__(self, key):
+        return getattr(self, key)
+
     id: str
     project_name: str
     name: str = Field(None, alias="repo_name")

{codemie_sdk_python-0.1.31 → codemie_sdk_python-0.1.33}/src/codemie_sdk/models/integration.py

@@ -56,6 +56,9 @@ class CredentialValues(BaseModel):
 class Integration(BaseModel):
     """Model for settings configuration."""
 
+    def __getitem__(self, key):
+        return getattr(self, key)
+
     model_config = ConfigDict(extra="ignore")
 
     id: Optional[str] = None

{codemie_sdk_python-0.1.31 → codemie_sdk_python-0.1.33}/src/codemie_sdk/models/workflow.py

@@ -46,6 +46,9 @@ class WorkflowUpdateRequest(BaseModel):
 class Workflow(BaseModel):
     """Workflow template model."""
 
+    def __getitem__(self, key):
+        return getattr(self, key)
+
     model_config = ConfigDict(populate_by_name=True)
 
     id: Optional[str] = None

{codemie_sdk_python-0.1.31 → codemie_sdk_python-0.1.33}/src/codemie_sdk/services/assistant.py

@@ -1,6 +1,7 @@
 """Assistant service implementation."""
 
 import json
+from pathlib import Path
 from typing import List, Union, Optional, Dict, Any, Literal
 
 import requests
@@ -171,3 +172,14 @@ class AssistantService:
             json_data=request.model_dump(exclude_none=True, by_alias=True),
             stream=request.stream,
         )
+
+    def upload_file_to_chat(self, file_path: Path):
+        """Upload a file to assistant chat and return the response containing file_url."""
+
+        with open(file_path, "rb") as file:
+            files = [
+                ("file", (file_path.name, file, self._api.detect_mime_type(file_path)))
+            ]
+            response = self._api.post_multipart("/v1/files/", dict, files=files)
+
+        return response

{codemie_sdk_python-0.1.31 → codemie_sdk_python-0.1.33}/src/codemie_sdk/services/datasource.py

@@ -1,7 +1,7 @@
 """DataSource service implementation."""
 
 import json
-from typing import Literal, List, Union, Tuple
+from typing import Literal, List, Union, Tuple, Optional, Dict, Any
 from pathlib import Path
 
 from ..models.common import PaginationParams
@@ -82,7 +82,7 @@ class DatasourceService:
                     content = f.read()
 
                 # Basic MIME type detection
-                mime_type = self._detect_mime_type(file_path)
+                mime_type = self._api.detect_mime_type(file_path)
                 file_uploads.append(("files", (file_path.name, content, mime_type)))
 
             elif isinstance(file_item, tuple) and len(file_item) == 3:
@@ -99,24 +99,6 @@ class DatasourceService:
             endpoint, dict, params=params, files=file_uploads
         )
 
-    @staticmethod
-    def _detect_mime_type(file_path: Path) -> str:
-        """Detect MIME type based on file extension."""
-        extension = file_path.suffix.lower()
-        mime_types = {
-            ".txt": "text/plain",
-            ".csv": "text/csv",
-            ".json": "application/json",
-            ".yaml": "application/x-yaml",
-            ".yml": "application/x-yaml",
-            ".xml": "application/xml",
-            ".pdf": "application/pdf",
-            ".pptx": "application/vnd.openxmlformats-officedocument.presentationml.presentation",
-            ".docx": "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
-            ".xlsx": "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
-        }
-        return mime_types.get(extension, "application/octet-stream")
-
     def update(self, datasource_id: str, request: BaseUpdateDataSourceRequest) -> dict:
         """Update an existing datasource.
 
@@ -156,6 +138,7 @@ class DatasourceService:
         projects: List[str] = None,
         owner: str = None,
         status: DataSourceStatus = None,
+        filters: Optional[Dict[str, Any]] = None,
     ) -> List[DataSource]:
         """
         List datasources with pagination and sorting support.
@@ -169,6 +152,7 @@ class DatasourceService:
             projects: Optional projects to filter by.
             owner: Optional owner to filter by in format: FirstName LastName.
             status: Optional data source status to filter by.
+            filters: Optional filters to apply. Should be a dictionary with filter criteria.
 
         Returns:
             DataSourceListResponse object containing list of datasources and pagination information.
@@ -177,15 +161,14 @@ class DatasourceService:
         params["sort_key"] = sort_key
         params["sort_order"] = sort_order
 
-        filters = {}
         if datasource_types:
-            filters["index_type"] = datasource_types
+            params["index_type"] = datasource_types
         if projects:
-            filters["project"] = projects
+            params["project"] = projects
         if status:
-            filters["status"] = status.value
+            params["status"] = status.value
         if owner:
-            filters["created_by"] = owner
+            params["created_by"] = owner
         if filters:
             params["filters"] = json.dumps(filters)
         return self._api.get("/v1/index", List[DataSource], params=params)

{codemie_sdk_python-0.1.31 → codemie_sdk_python-0.1.33}/src/codemie_sdk/services/workflow.py

@@ -1,7 +1,7 @@
 """Workflow service implementation."""
 
 import json
-from typing import List, Optional
+from typing import List, Optional, Any, Dict
 
 from .workflow_execution import WorkflowExecutionService
 from ..models.common import PaginationParams
@@ -81,6 +81,7 @@ class WorkflowService:
         page: int = 0,
         per_page: int = 10,
         projects: Optional[List[str]] = None,
+        filters: Optional[Dict[str, Any]] = None,
     ) -> List[Workflow]:
         """List workflows with filtering and pagination support.
 
@@ -88,6 +89,7 @@ class WorkflowService:
             page: Page number (0-based). Must be >= 0. Defaults to 0.
             per_page: Number of items per page. Must be > 0. Defaults to 10.
             projects: Optional projects to filter by.
+            filters: Optional filters to apply. Should be a dictionary with filter criteria.
 
         Returns:
             List of Workflow objects containing workflow information and pagination metadata.
@@ -96,9 +98,8 @@ class WorkflowService:
 
         params = PaginationParams(page=page, per_page=per_page).to_dict()
 
-        filters = {}
         if projects:
-            filters["project"] = projects
+            params["project"] = projects
         if filters:
             params["filters"] = json.dumps(filters)
 

{codemie_sdk_python-0.1.31 → codemie_sdk_python-0.1.33}/src/codemie_sdk/utils/http.py

@@ -1,5 +1,6 @@
 """HTTP utilities for CodeMie SDK."""
 
+from pathlib import Path
 from typing import TypeVar, Type, Optional, Any, Union, Dict, List, get_origin, get_args
 from pydantic import BaseModel
 import requests
@@ -275,3 +276,21 @@ class ApiRequestHandler:
         response.raise_for_status()
 
         return self._parse_response(response, response_model, wrap_response)
+
+    @staticmethod
+    def detect_mime_type(file_path: Path) -> str:
+        """Detect MIME type based on file extension."""
+        extension = file_path.suffix.lower()
+        mime_types = {
+            ".txt": "text/plain",
+            ".csv": "text/csv",
+            ".json": "application/json",
+            ".yaml": "application/x-yaml",
+            ".yml": "application/x-yaml",
+            ".xml": "application/xml",
+            ".pdf": "application/pdf",
+            ".pptx": "application/vnd.openxmlformats-officedocument.presentationml.presentation",
+            ".docx": "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+            ".xlsx": "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+        }
+        return mime_types.get(extension, "application/octet-stream")