codemie-sdk-python 0.1.22__tar.gz → 0.1.28__tar.gz

{codemie_sdk_python-0.1.22 → codemie_sdk_python-0.1.28}/PKG-INFO

@@ -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 → codemie_sdk_python-0.1.28}/pyproject.toml

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

{codemie_sdk_python-0.1.22 → codemie_sdk_python-0.1.28}/src/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_python-0.1.22 → codemie_sdk_python-0.1.28}/src/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_python-0.1.22 → codemie_sdk_python-0.1.28}/src/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_python-0.1.22 → codemie_sdk_python-0.1.28}/src/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,