codemie-sdk-python 0.1.199__py3-none-any.whl → 0.1.201__py3-none-any.whl

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/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/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/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/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/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:

{codemie_sdk_python-0.1.199.dist-info → codemie_sdk_python-0.1.201.dist-info}/METADATA

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

@@ -2,32 +2,35 @@ codemie_sdk/__init__.py,sha256=leeACpTc2113BsY_IgO-ceFHPrQKC2WhuOfuuyyIqDE,567
 codemie_sdk/auth/__init__.py,sha256=IksEj223xEZtJ-cQ0AT9L0Bs9psIJ8QNzDXrPTUQ3xQ,126
 codemie_sdk/auth/credentials.py,sha256=OzR_CXPBNTEC6VmNdzcCHF7rWWGrVf3agAlGKgPtTiU,4361
 codemie_sdk/client/__init__.py,sha256=yf6C39MmrJ6gK9ZHMhBeynKwUUYVSUTQbKxU8-4qpKg,101
-codemie_sdk/client/client.py,sha256=6yXXNYF4m1vh_0UMPENaxL39hmt_wdPF_b5NtsKMJQQ,5174
+codemie_sdk/client/client.py,sha256=kEXMztYJr3Ti1soZuAz1IxMgeLqiwLgBHWIL3PqVqfw,5348
 codemie_sdk/exceptions.py,sha256=XoVPyognx-JmyVxLHkZPAcX1CMi1OoT1diBFJLU54so,1183
 codemie_sdk/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_sdk/models/assistant.py,sha256=zb_k9EZ7rVFD7T2BwSqu_UA0psLKn7VHbY6JB6SyMOo,10946
 codemie_sdk/models/common.py,sha256=gmZ-ps8TbaieNKr0kUKoQEjhVrHD2CAYomOpZQRatH8,1195
 codemie_sdk/models/conversation.py,sha256=tGlqtVnoRCbTm8J8Y1BUmBvMpLW3IRFE0CIHJYoNU_Y,4238
 codemie_sdk/models/datasource.py,sha256=yCFB_wg9Lo2V6mzF2N3lsVBXJoyc8pyfMgMZUorn0ng,10852
+codemie_sdk/models/file_operation.py,sha256=D2hnsxOO9lzVMw0BNCgKPfHf28bWQmbw5rccgaTXI90,747
 codemie_sdk/models/integration.py,sha256=IxTbbyEHsD8HWWFDtZDoWuE1ZwR9wwiBk3XUwlf1KZI,2328
 codemie_sdk/models/llm.py,sha256=ppb9-1dx1UFhRuJpSR3ij7H6Pfhe9nO4C4BEOIbToy4,1192
 codemie_sdk/models/task.py,sha256=J4ZFRY3s8qBGrqB5NLQF0rMbInLh4s7OEZ0ZfmnW0Ho,1476
 codemie_sdk/models/user.py,sha256=Q0rjimZh-IbeaPfq6b6fk6ZaCtwLqWHEIlU863suCS4,1777
 codemie_sdk/models/workflow.py,sha256=qfk0rBJnFUMpcEDq_E5GB3hzYKbe_bb2NYJlLZJwUEE,2453
+codemie_sdk/models/workflow_execution_payload.py,sha256=iEGkdw1jm09aniZiXswbQeLtoiUYIxhc3vsBvZL00JE,515
 codemie_sdk/models/workflow_state.py,sha256=okEMKzkiBU3GHs9VNBoiEMOnOeZRMXGYtpL0NYSg-FY,1374
 codemie_sdk/models/workflow_thoughts.py,sha256=CjHaIPgca9kQAjpoq8IpX4MuDeql1SvaoKS5RmyU2SE,616
 codemie_sdk/services/assistant.py,sha256=BsTR_uoPoZtta6icfw2M0Ea6w2VIhYndDUVqsHJuZJo,7888
 codemie_sdk/services/conversation.py,sha256=Ss0mkGaBi4u8-YKzhRYUJOfoFqJueDp9JurI5DRCGT8,2566
 codemie_sdk/services/datasource.py,sha256=d8LImpireGOnXKvj4h8iafq_ZQMn177_GGLLiE5g7wU,7405
+codemie_sdk/services/files.py,sha256=MKLJEv208yUkeRO2JfC4XWPeJ4TKg5ZGaxaBU5GXzoM,1955
 codemie_sdk/services/integration.py,sha256=SdwFwR3hCPyJYilzzlkpKPLNbO89nfqmIXXoT7bDEBI,5410
 codemie_sdk/services/llm.py,sha256=0-e4_7RvLHs2giCyoQ5U4KDTh6p5VXgPKNxnDP9ZDFU,1100
 codemie_sdk/services/task.py,sha256=3e9t8_LMkR4xfeMBwMCo7ZF87PxPS-ZbzDg85ilda2M,1031
 codemie_sdk/services/user.py,sha256=7B-Qw451qKPD5Io6qLda-kbFDaPRQ3TamJamiGwCQu4,1013
-codemie_sdk/services/workflow.py,sha256=0d5grAsBIH2NZY66UbVhlDqEFYxEYSBByj3GNgcz_DQ,4965
-codemie_sdk/services/workflow_execution.py,sha256=fvDjn0GEm8ObL8N2LUO0kk6zwTl0-DYBr2jZBx9OeLE,4377
+codemie_sdk/services/workflow.py,sha256=cAGv2jEnb3dOSk5xxqg3L15mTcSkAVxaZHVZwTYjT-w,5407
+codemie_sdk/services/workflow_execution.py,sha256=P57fz3fsUnKLg8qUYszMxCn_ykovh22BQuUk0EGnC9I,4654
 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=ZtXb-hlwqXmjSTKRncgO-9MYoXnmHt2Dljno0kPy7mA,9506
-codemie_sdk_python-0.1.199.dist-info/METADATA,sha256=6CdEAafnXkN6eKLG8bJpVctfDuWmw8bNmItHa7bBiuE,24494
-codemie_sdk_python-0.1.199.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-codemie_sdk_python-0.1.199.dist-info/RECORD,,
+codemie_sdk_python-0.1.201.dist-info/METADATA,sha256=D1b21D4ExSKWNq3jEKxOhZYYFat6temEKVBN7KV459I,24494
+codemie_sdk_python-0.1.201.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+codemie_sdk_python-0.1.201.dist-info/RECORD,,