codemie-sdk-python 0.1.22__py3-none-any.whl → 0.1.28__py3-none-any.whl

codemie_sdk/models/assistant.py

@@ -3,9 +3,9 @@
 import uuid
 from datetime import datetime
 from enum import Enum
-from typing import List, Optional, Any, Union
+from typing import List, Optional, Any, Union, Dict
 
-from pydantic import BaseModel, Field, ConfigDict
+from pydantic import BaseModel, Field, ConfigDict, model_validator
 
 from .common import User
 from .integration import Integration
@@ -171,6 +171,28 @@ class ChatMessage(BaseModel):
     message: Optional[str] = Field(default="")
 
 
+class ToolConfig(BaseModel):
+    name: str
+    tool_creds: Optional[Dict[str, Any]] = None
+    integration_id: Optional[str] = None
+
+    @model_validator(mode="after")
+    def validate_credentials_provided(self) -> "ToolConfig":
+        """
+        Validate that either tool_creds or integration_id is provided.
+
+        At least one of these fields must be specified for the tool configuration
+        to be valid. This ensures that the tool has a way to obtain credentials.
+        """
+        if not self.tool_creds and not self.integration_id:
+            raise ValueError("Either tool_creds or integration_id must be provided")
+        if self.tool_creds and self.integration_id:
+            raise ValueError(
+                "Either tool_creds or integration_id must be provided, but not both"
+            )
+        return self
+
+
 class AssistantChatRequest(BaseModel):
     """Model for chat request to assistant."""
 
@@ -197,6 +219,7 @@ class AssistantChatRequest(BaseModel):
     metadata: Optional[dict[str, Any]] = Field(
         default=None, description="Provide additional metadata"
     )
+    tools_config: Optional[List[ToolConfig]] = None
 
 
 class BaseModelResponse(BaseModel):

codemie_sdk/models/datasource.py

@@ -1,7 +1,7 @@
 import re
 from datetime import datetime
 from enum import Enum
-from typing import Optional, List, Union
+from typing import Optional, List, Union, Tuple
 
 from pydantic import BaseModel, Field, model_validator, ConfigDict, field_validator
 
@@ -71,13 +71,23 @@ class Jira(BaseModel):
 
 
 class Google(BaseModel):
-    """Model for Jira-specific response fields"""
+    """Model for Google-specific response fields"""
 
     google_doc: str = Field(None, alias="googleDoc")
 
     model_config = ConfigDict(extra="ignore")
 
 
+class File(BaseModel):
+    """Model for File-specific response fields"""
+
+    files: Optional[List[Tuple[str, bytes, str]]] = (
+        None  # (filename, content, mime_type)
+    )
+
+    model_config = ConfigDict(extra="ignore")
+
+
 class Code(BaseModel):
     """Model for code repository datasource creation"""
 
@@ -171,6 +181,13 @@ class GoogleDataSourceRequest(BaseDataSourceRequest, Google):
         return ["google_doc"]
 
 
+class FileDataSourceRequest(BaseDataSourceRequest):
+    """Model for File datasource creation requests"""
+
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.FILE, **data)
+
+
 class BaseUpdateDataSourceRequest(BaseDataSourceRequest):
     """Mixin update-specific reindex fields"""
 
@@ -234,6 +251,13 @@ class UpdateGoogleDataSourceRequest(BaseUpdateDataSourceRequest):
         super().__init__(type=DataSourceType.GOOGLE, **data)
 
 
+class UpdateFileDataSourceRequest(BaseUpdateDataSourceRequest):
+    """Model for File datasource updates"""
+
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.FILE, **data)
+
+
 class DataSource(BaseModel):
     model_config = ConfigDict(
         extra="ignore",
@@ -293,5 +317,6 @@ class DataSource(BaseModel):
                 "summarization_prompt": values.get("prompt"),
                 "summarization_model": values.get("summarization_model"),
                 "summarization_docs_generation": values.get("docs_generation"),
+                "embeddings_model": values.get("embeddings_model"),
             }
         return values

codemie_sdk/services/datasource.py

@@ -1,7 +1,8 @@
 """DataSource service implementation."""
 
 import json
-from typing import Literal, List
+from typing import Literal, List, Union, Tuple
+from pathlib import Path
 
 from ..models.common import PaginationParams
 from ..models.datasource import (
@@ -12,6 +13,7 @@ from ..models.datasource import (
     CodeDataSourceRequest,
     UpdateCodeDataSourceRequest,
     BaseUpdateDataSourceRequest,
+    FileDataSourceRequest,
 )
 from ..utils import ApiRequestHandler
 
@@ -50,6 +52,71 @@ class DatasourceService:
             json_data=request.model_dump(by_alias=True, exclude_none=True),
         )
 
+    def create_file_datasource(
+        self,
+        request: FileDataSourceRequest,
+        files: List[Union[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 = []
+
+        for file_item in files:
+            if isinstance(file_item, str):
+                # File path provided - read file and detect MIME type
+                file_path = Path(file_item)
+                with open(file_path, "rb") as f:
+                    content = f.read()
+
+                # Basic MIME type detection
+                mime_type = self._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:
+                # (filename, content, mime_type) tuple provided
+                filename, content, mime_type = file_item
+                file_uploads.append(("files", (filename, content, mime_type)))
+
+            else:
+                raise ValueError(
+                    "Each file must be either a file path (str) or a tuple of (filename, content, mime_type)"
+                )
+
+        return self._api.post_multipart(
+            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

@@ -46,9 +46,16 @@ class ApiRequestHandler:
         self._token = token
         self._verify_ssl = verify_ssl
 
-    def _get_headers(self) -> dict:
-        """Gets request headers with auth token."""
-        headers = {"Content-Type": "application/json"}
+    def _get_headers(self, exclude_content_type: bool = False) -> dict:
+        """Gets request headers with auth token.
+
+        Args:
+            exclude_content_type: Whether to exclude Content-Type header (for multipart requests)
+        """
+        headers = {}
+        if not exclude_content_type:
+            headers["Content-Type"] = "application/json"
+
         if (
             "0.0.0.0" in self._base_url
             or "127.0.0.1" in self._base_url
@@ -176,6 +183,44 @@ class ApiRequestHandler:
 
         return self._parse_response(response, response_model, wrap_response)
 
+    @log_request
+    def post_multipart(
+        self,
+        endpoint: str,
+        response_model: Type[T],
+        params: Optional[Dict[str, Any]] = None,
+        data: Optional[Dict[str, Any]] = None,
+        files: Optional[List] = None,
+        wrap_response: bool = True,
+    ) -> T:
+        """Makes a POST 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
+            data: Data to be sent as multipart/form-data
+            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.post(
+            url=f"{self._base_url}{endpoint}",
+            headers=self._get_headers(
+                exclude_content_type=True
+            ),  # Let requests set multipart content-type
+            params=params,
+            data=data,
+            files=files,
+            verify=self._verify_ssl,
+        )
+        response.raise_for_status()
+
+        return self._parse_response(response, response_model, wrap_response)
+
     @log_request
     def put(
         self,

{codemie_sdk_python-0.1.22.dist-info → codemie_sdk_python-0.1.28.dist-info}/METADATA

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

{codemie_sdk_python-0.1.22.dist-info → codemie_sdk_python-0.1.28.dist-info}/RECORD

@@ -4,9 +4,9 @@ codemie_sdk/auth/credentials.py,sha256=u2eLNsD8fELTgreQghVb0g0kk82zchUkux0M5lzq-
 codemie_sdk/client/__init__.py,sha256=yf6C39MmrJ6gK9ZHMhBeynKwUUYVSUTQbKxU8-4qpKg,101
 codemie_sdk/client/client.py,sha256=dVcWqd-ruWd8GOZ3_33C3LnHgUyyxIoCKxKYPYbBPq8,4373
 codemie_sdk/exceptions.py,sha256=XoVPyognx-JmyVxLHkZPAcX1CMi1OoT1diBFJLU54so,1183
-codemie_sdk/models/assistant.py,sha256=nmvA302yHpSw0iwhdrkRHT1dOsz2smlTYCWa7CItNsM,6568
+codemie_sdk/models/assistant.py,sha256=RCVuYcKI_DzGyObJ3e1l4rrcVev-1Hq7crW9XY5pvus,7479
 codemie_sdk/models/common.py,sha256=V4sJCzwFTF8BPlrd2OKI44Em5HmPmn2Nm8fQNsnBZ_Q,1128
-codemie_sdk/models/datasource.py,sha256=Iun6O76o4m-_jbyyumMndpyr4OiN0lwqKb8u5AOTHdM,10037
+codemie_sdk/models/datasource.py,sha256=BrwMG8NLVwNtsJZunXT2HPc9oI-r8rpmc5OrdYHQqkA,10746
 codemie_sdk/models/integration.py,sha256=F1YBdCK1w5W_jKK8prEsg7A00NghkEBvJgkkr8KpC6w,1764
 codemie_sdk/models/llm.py,sha256=ppb9-1dx1UFhRuJpSR3ij7H6Pfhe9nO4C4BEOIbToy4,1192
 codemie_sdk/models/task.py,sha256=J4ZFRY3s8qBGrqB5NLQF0rMbInLh4s7OEZ0ZfmnW0Ho,1476
@@ -14,7 +14,7 @@ codemie_sdk/models/user.py,sha256=Q0rjimZh-IbeaPfq6b6fk6ZaCtwLqWHEIlU863suCS4,17
 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=hYH__M5LD33dfh7CCS7HYmEn8izsGkO0nfVa-dpoj6w,5118
+codemie_sdk/services/datasource.py,sha256=2LikRr_V-hgnflMjgbxydHugRMP1eO8_aI33DrVeXQ8,7834
 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=JfyX4gP-grQUI795QL0DJ72aBkiqNd1fs31fUqtSnYE,7757
-codemie_sdk_python-0.1.22.dist-info/METADATA,sha256=cH8AVifYpaYU7UjghY6i1VVnwbg3kpCwS5jWLXGo0W0,21614
-codemie_sdk_python-0.1.22.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-codemie_sdk_python-0.1.22.dist-info/RECORD,,
+codemie_sdk/utils/http.py,sha256=FWU56W_-vBGbHfX1EC1zsoRjDoNCyNK2HCdNb4IkQVc,9233
+codemie_sdk_python-0.1.28.dist-info/METADATA,sha256=BqZfGZdKdxn7RN5CX_9Eny6H1Na3_e9oedfeBlW5C6E,21614
+codemie_sdk_python-0.1.28.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+codemie_sdk_python-0.1.28.dist-info/RECORD,,