codemie-sdk-python 0.1.199__tar.gz → 0.1.201__tar.gz

{codemie_sdk_python-0.1.199 → codemie_sdk_python-0.1.201}/PKG-INFO

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

{codemie_sdk_python-0.1.199 → codemie_sdk_python-0.1.201}/pyproject.toml

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

{codemie_sdk_python-0.1.199 → codemie_sdk_python-0.1.201}/src/codemie_sdk/client/client.py

@@ -11,6 +11,7 @@ from ..services.integration import IntegrationService
 from ..services.task import TaskService
 from ..services.user import UserService
 from ..services.workflow import WorkflowService
+from ..services.files import FileOperationService
 
 
 class CodeMieClient:
@@ -81,6 +82,9 @@ class CodeMieClient:
         self.conversations = ConversationService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.files = FileOperationService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
 
     @property
     def token(self) -> str:

codemie_sdk_python-0.1.201/src/codemie_sdk/models/file_operation.py

@@ -0,0 +1,25 @@
+"""File operation models."""
+
+from typing import Optional, List, Any, Dict
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class FileResponse(BaseModel):
+    """Individual file response model."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    file_url: str = Field(..., description="URL or identifier for the uploaded file")
+
+
+class FileBulkCreateResponse(BaseModel):
+    """Response model for bulk file creation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    files: List[FileResponse] = Field(
+        ..., description="List of successfully uploaded files with their URLs"
+    )
+    failed_files: Optional[List[Dict[str, Any]]] = Field(
+        None, description="List of files that failed to upload, null if no failures"
+    )

codemie_sdk_python-0.1.201/src/codemie_sdk/models/workflow_execution_payload.py

@@ -0,0 +1,17 @@
+"""Workflow execution payload models."""
+
+from typing import Optional
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class WorkflowExecutionCreateRequest(BaseModel):
+    """Request model for workflow execution creation."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    user_input: Optional[str] = Field(
+        None, description="User input for the workflow execution"
+    )
+    file_name: Optional[str] = Field(
+        None, description="File name associated with the workflow execution"
+    )

codemie_sdk_python-0.1.201/src/codemie_sdk/services/files.py

@@ -0,0 +1,61 @@
+"""File operation service implementation."""
+
+import mimetypes
+from pathlib import Path
+from typing import List
+
+from ..models.file_operation import FileBulkCreateResponse
+from ..utils import ApiRequestHandler
+
+
+class FileOperationService:
+    """Service for managing file operations."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the file operation service.
+
+        Args:
+            api_domain: Base URL for the CodeMie API
+            token: Authentication token
+            verify_ssl: Whether to verify SSL certificates
+        """
+        self._api = ApiRequestHandler(api_domain, token, verify_ssl)
+
+    def bulk_upload(self, files: List[Path]) -> FileBulkCreateResponse:
+        """Upload multiple files in a single operation.
+
+        Args:
+            files: List of File paths (required)
+
+        Returns:
+            FileBulkCreateResponse: Results of the bulk operation including
+
+        Raises:
+            ValueError: If items in files is not a Path
+            ApiError: If bulk creation fails or validation errors occur
+        """
+
+        files_to_upload = []
+
+        for file_path in files:
+            if isinstance(file_path, Path):
+                # File path provided - read file and detect MIME type
+                with open(file_path, "rb") as file:
+                    content = file.read()
+
+                # Basic MIME type detection
+                mime_type = (
+                    mimetypes.guess_type(file_path.name)[0]
+                    or "application/octet-stream"
+                )
+                files_to_upload.append(("files", (file_path.name, content, mime_type)))
+            else:
+                raise ValueError("Each item in list must be a Path")
+
+        response = self._api.post_multipart(
+            "/v1/files/bulk",
+            files=files_to_upload,
+            response_model=FileBulkCreateResponse,
+        )
+
+        return response

{codemie_sdk_python-0.1.199 → codemie_sdk_python-0.1.201}/src/codemie_sdk/services/workflow.py

@@ -130,8 +130,23 @@ class WorkflowService:
         """
         return self._api.delete(f"/v1/workflows/{workflow_id}", dict)
 
-    def run(self, workflow_id: str, user_input: str = "") -> dict:
-        return self.executions(workflow_id).create(user_input)
+    def run(
+        self,
+        workflow_id: str,
+        user_input: Optional[str] = None,
+        file_name: Optional[str] = None,
+    ) -> dict:
+        """Run a workflow with optional input parameters.
+
+        Args:
+            workflow_id: ID of the workflow to run
+            user_input: Optional user input for the workflow execution
+            file_name: Optional file name for the workflow execution
+
+        Returns:
+            dict: Created workflow execution details
+        """
+        return self.executions(workflow_id).create(user_input, file_name)
 
     def executions(self, workflow_id: str) -> WorkflowExecutionService:
         """Get workflow execution service for the specified workflow.

{codemie_sdk_python-0.1.199 → codemie_sdk_python-0.1.201}/src/codemie_sdk/services/workflow_execution.py

@@ -4,6 +4,7 @@ from typing import List, Optional
 
 from ..models.common import PaginationParams
 from ..models.workflow import WorkflowExecution
+from ..models.workflow_execution_payload import WorkflowExecutionCreateRequest
 from ..models.workflow_thoughts import WorkflowExecutionThought
 from .workflow_execution_state import WorkflowExecutionStateService
 from ..utils import ApiRequestHandler
@@ -44,19 +45,25 @@ class WorkflowExecutionService:
             params=params,
         )
 
-    def create(self, user_input: Optional[str] = "") -> dict:
+    def create(
+        self, user_input: Optional[str] = None, file_name: Optional[str] = None
+    ) -> dict:
         """Create a new workflow execution.
 
         Args:
             user_input: Optional input data for the workflow execution.
-                       Defaults to None.
+            file_name: Optional file name associated with the workflow execution.
 
         Returns:
             dict: Created workflow execution details
         """
-        payload = {"user_input": user_input}
+        payload = WorkflowExecutionCreateRequest(
+            user_input=user_input, file_name=file_name
+        )
         return self._api.post(
-            f"/v1/workflows/{self._workflow_id}/executions", dict, json_data=payload
+            f"/v1/workflows/{self._workflow_id}/executions",
+            dict,
+            json_data=payload.model_dump(),
         )
 
     def get(self, execution_id: str) -> WorkflowExecution: