codemie-sdk-python 0.1.30__tar.gz → 0.1.32__tar.gz

{codemie_sdk_python-0.1.30 → codemie_sdk_python-0.1.32}/PKG-INFO

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

{codemie_sdk_python-0.1.30 → codemie_sdk_python-0.1.32}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.30"
+version = "0.1.32"
 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.30 → codemie_sdk_python-0.1.32}/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.30 → codemie_sdk_python-0.1.32}/src/codemie_sdk/services/datasource.py

@@ -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.
 

{codemie_sdk_python-0.1.30 → codemie_sdk_python-0.1.32}/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")