codemie-sdk-python 0.1.521__tar.gz → 0.1.522__tar.gz

{codemie_sdk_python-0.1.521 → codemie_sdk_python-0.1.522}/PKG-INFO

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

{codemie_sdk_python-0.1.521 → codemie_sdk_python-0.1.522}/pyproject.toml

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

{codemie_sdk_python-0.1.521 → codemie_sdk_python-0.1.522}/src/codemie_sdk/services/datasource.py

@@ -60,26 +60,20 @@ class DatasourceService:
             json_data=request.model_dump(by_alias=True, exclude_none=True),
         )
 
-    def create_file_datasource(
-        self,
-        request: FileDataSourceRequest,
+    @staticmethod
+    def _build_file_uploads(
         files: list[str | Tuple[str, bytes, str]],
-    ) -> dict:
-        """Create a new file datasource with file uploads.
+    ) -> list[tuple[str, tuple[str, bytes, str]]]:
+        """Build multipart file tuples from file paths or (filename, content, mime_type) tuples.
 
         Args:
-            request: File datasource creation request
             files: List of files to upload. Each item can be:
                    - str: file path (will read file and detect MIME type)
                    - Tuple[str, bytes, str]: (filename, content, mime_type)
 
         Returns:
-            dict: Response from the server containing operation status
+            list: Multipart-ready ("files", (filename, content, mime_type)) tuples
         """
-        endpoint = "/v1/index/knowledge_base/file"
-
-        # Prepare multipart form data
-        params = request.model_dump(by_alias=True, exclude_none=True)
         file_uploads = []
 
         for file_item in files:
@@ -106,15 +100,47 @@ class DatasourceService:
                     "Each file must be either a file path (str) or a tuple of (filename, content, mime_type)"
                 )
 
+        return file_uploads
+
+    def create_file_datasource(
+        self,
+        request: FileDataSourceRequest,
+        files: list[str | Tuple[str, bytes, str]],
+    ) -> dict:
+        """Create a new file datasource with file uploads.
+
+        Args:
+            request: File datasource creation request
+            files: List of files to upload. Each item can be:
+                   - str: file path (will read file and detect MIME type)
+                   - Tuple[str, bytes, str]: (filename, content, mime_type)
+
+        Returns:
+            dict: Response from the server containing operation status
+        """
+        endpoint = "/v1/index/knowledge_base/file"
+
+        # Prepare multipart form data
+        params = request.model_dump(by_alias=True, exclude_none=True)
+        file_uploads = self._build_file_uploads(files)
+
         return self._api.post_multipart(
             endpoint, dict, params=params, files=file_uploads
         )
 
-    def update(self, datasource_id: str, request: BaseUpdateDataSourceRequest) -> dict:
+    def update(
+        self,
+        datasource_id: str,
+        request: BaseUpdateDataSourceRequest,
+        files: list[str | Tuple[str, bytes, str]] | None = None,
+    ) -> dict:
         """Update an existing datasource.
 
         Args:
             request: Update datasource request
+            files: Optional list of files to add to a File datasource. Each item
+                   can be a file path (str) or a (filename, content, mime_type)
+                   tuple. Only supported for UpdateFileDataSourceRequest.
 
         Returns:
             dict: Response from the server containing operation status
@@ -140,6 +166,11 @@ class DatasourceService:
         # PUT /index/knowledge_base/file now uses Depends() — all fields are query params
         if isinstance(request, UpdateFileDataSourceRequest):
             params.update(data)
+            if files:
+                file_uploads = self._build_file_uploads(files)
+                return self._api.put_multipart(
+                    endpoint, dict, params=params, files=file_uploads
+                )
             return self._api.put(endpoint, dict, {}, params=params)
 
         return self._api.put(endpoint, dict, params=params, json_data=data)

{codemie_sdk_python-0.1.521 → codemie_sdk_python-0.1.522}/src/codemie_sdk/utils/http.py

@@ -274,6 +274,40 @@ class ApiRequestHandler:
 
         return self._parse_response(response, response_model, wrap_response)
 
+    @log_request
+    def put_multipart(
+        self,
+        endpoint: str,
+        response_model: Type[T],
+        params: dict[str, Any] | None = None,
+        files: list | None = None,
+        wrap_response: bool = True,
+    ) -> T:
+        """Makes a PUT multipart/form-data request and parses the response.
+
+        Args:
+            endpoint: API endpoint path
+            response_model: Pydantic model class or list[Model] for response
+            params: Url parameters
+            files: List of file tuples for upload
+            wrap_response: Whether response is wrapped in 'data' field
+
+        Returns:
+            Parsed response object or list of objects
+        """
+        response = requests.put(
+            url=f"{self._base_url}{endpoint}",
+            headers=self._get_headers(
+                exclude_content_type=True
+            ),  # Let requests set multipart content-type
+            params=params,
+            files=files,
+            verify=self._verify_ssl,
+        )
+        response.raise_for_status()
+
+        return self._parse_response(response, response_model, wrap_response)
+
     @log_request
     def put(
         self,