codemie-sdk-python 0.1.30__py3-none-any.whl → 0.1.32__py3-none-any.whl

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/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/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")

{codemie_sdk_python-0.1.30.dist-info → codemie_sdk_python-0.1.32.dist-info}/METADATA

@@ -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.dist-info → codemie_sdk_python-0.1.32.dist-info}/RECORD

@@ -13,8 +13,8 @@ codemie_sdk/models/task.py,sha256=J4ZFRY3s8qBGrqB5NLQF0rMbInLh4s7OEZ0ZfmnW0Ho,14
 codemie_sdk/models/user.py,sha256=Q0rjimZh-IbeaPfq6b6fk6ZaCtwLqWHEIlU863suCS4,1777
 codemie_sdk/models/workflow.py,sha256=CyK-kBnOck2KiRX3gmWF_mcGtqxsGXH3FS3jwYbpz_A,2327
 codemie_sdk/models/workflow_state.py,sha256=CMYFQZ7sy4QxmnWmc83TFfqP7TG_3rW5MdH5fxsS9kY,1251
-codemie_sdk/services/assistant.py,sha256=-U5YpqOyKlVplSRPrnDItG8TUpg9qqhO6-F7c7Rz2xU,5110
-codemie_sdk/services/datasource.py,sha256=2LikRr_V-hgnflMjgbxydHugRMP1eO8_aI33DrVeXQ8,7834
+codemie_sdk/services/assistant.py,sha256=D3aYFPd97wZIpKylAmxF6qtMwSeKyM54OqeKoDOw1U4,5554
+codemie_sdk/services/datasource.py,sha256=wlfB_F68CvJeMBiPRBtD9o7gxo3SnZdMPX1DyNNxoyM,7011
 codemie_sdk/services/integration.py,sha256=vJnSkXk2C2l0ahX2SUsuA7fKhY2hTuAByynx5Lgh7Ls,4864
 codemie_sdk/services/llm.py,sha256=0-e4_7RvLHs2giCyoQ5U4KDTh6p5VXgPKNxnDP9ZDFU,1100
 codemie_sdk/services/task.py,sha256=3e9t8_LMkR4xfeMBwMCo7ZF87PxPS-ZbzDg85ilda2M,1031
@@ -23,7 +23,7 @@ codemie_sdk/services/workflow.py,sha256=aOV13WrFAqMXOPd2jVHbEhg3fpezk4XWo382D5aW
 codemie_sdk/services/workflow_execution.py,sha256=aGoT3rdTmh5-doAsrmBBjLEuOfvL5aqeo3g9th1_aAw,3647
 codemie_sdk/services/workflow_execution_state.py,sha256=tXoaa8yT09xgYEUNiHhVULe76TwGwVgZupMIUyyLxdo,2070
 codemie_sdk/utils/__init__.py,sha256=BXAJJfAzO89-kMYvWWo9wSNhSbGgF3vB1In9sePFhMM,109
-codemie_sdk/utils/http.py,sha256=FWU56W_-vBGbHfX1EC1zsoRjDoNCyNK2HCdNb4IkQVc,9233
-codemie_sdk_python-0.1.30.dist-info/METADATA,sha256=BidaXb3DI4hKUo17aa8VlOtXbWzgnXu5UYvVCydRdfs,21614
-codemie_sdk_python-0.1.30.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-codemie_sdk_python-0.1.30.dist-info/RECORD,,
+codemie_sdk/utils/http.py,sha256=YzE53y1r7PBUKcEQt-gGYduRY06I8xzqVRzlj9f_voA,10084
+codemie_sdk_python-0.1.32.dist-info/METADATA,sha256=c2iRrHVxmZdPwVjgJS8GASOvzbC2lI-Ey8ph7FvBYrU,21614
+codemie_sdk_python-0.1.32.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+codemie_sdk_python-0.1.32.dist-info/RECORD,,