codemie-sdk-python 0.1.247__tar.gz → 0.1.248__tar.gz

{codemie_sdk_python-0.1.247 → codemie_sdk_python-0.1.248}/PKG-INFO

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

{codemie_sdk_python-0.1.247 → codemie_sdk_python-0.1.248}/pyproject.toml

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

{codemie_sdk_python-0.1.247 → codemie_sdk_python-0.1.248}/src/codemie_sdk/models/datasource.py

@@ -260,6 +260,40 @@ class UpdateFileDataSourceRequest(BaseUpdateDataSourceRequest):
         super().__init__(type=DataSourceType.FILE, **data)
 
 
+class CodeAnalysisDataSourceRequest(BaseModel):
+    """Model for provider-based datasource creation requests"""
+
+    model_config = ConfigDict(extra="ignore", populate_by_name=True)
+
+    name: str = Field(..., description="Datasource name")
+    description: Optional[str] = Field(None, description="Datasource description")
+    project_name: str = Field(..., description="Project name")
+    project_space_visible: bool = Field(False, alias="projectSpaceVisible")
+    branch: Optional[str] = Field(None, description="Git branch")
+    api_url: str = Field(..., description="Repository URL")
+    access_token: str = Field(..., description="Access token for repository")
+    analyzer: Optional[str] = Field(
+        None, description="Code analyzer type (e.g., Java, Python)"
+    )
+    datasource_root: str = Field("/", description="Root directory to analyze")
+
+
+class CodeExplorationDataSourceRequest(BaseModel):
+    """Model for CodeExplorationToolkit datasource creation requests"""
+
+    model_config = ConfigDict(extra="ignore", populate_by_name=True)
+
+    name: str = Field(..., description="Datasource name")
+    description: Optional[str] = Field(None, description="Datasource description")
+    project_name: str = Field(..., description="Project name")
+    project_space_visible: bool = Field(False, alias="projectSpaceVisible")
+    code_analysis_datasource_ids: List[str] = Field(
+        ...,
+        alias="code_analysis_datasource_ids",
+        description="List of CodeAnalysisToolkit datasource IDs",
+    )
+
+
 class DataSource(BaseModel):
     model_config = ConfigDict(
         extra="ignore",

{codemie_sdk_python-0.1.247 → codemie_sdk_python-0.1.248}/src/codemie_sdk/services/datasource.py

@@ -15,6 +15,8 @@ from ..models.datasource import (
     UpdateCodeDataSourceRequest,
     BaseUpdateDataSourceRequest,
     FileDataSourceRequest,
+    CodeAnalysisDataSourceRequest,
+    CodeExplorationDataSourceRequest,
 )
 from ..models.assistant import AssistantListResponse
 from ..utils import ApiRequestHandler
@@ -225,3 +227,29 @@ class DatasourceService:
         return self._api.get(
             f"/v1/index/{datasource_id}/assistants", List[AssistantListResponse]
         )
+
+    def create_provider_datasource(
+        self,
+        toolkit_id: str,
+        provider_name: str,
+        request: Union[CodeAnalysisDataSourceRequest, CodeExplorationDataSourceRequest],
+    ) -> dict:
+        """Create a provider-based datasource.
+
+        Args:
+            toolkit_id: ID of the toolkit to use
+            provider_name: Name of the provider
+            request: Provider datasource creation request (CodeAnalysisDataSourceRequest or CodeExplorationDataSourceRequest)
+
+        Returns:
+            dict: Response from the server containing operation status
+        """
+        endpoint = (
+            f"/v1/index/provider?toolkit_id={toolkit_id}&provider_name={provider_name}"
+        )
+
+        return self._api.post(
+            endpoint,
+            dict,
+            json_data=request.model_dump(by_alias=True, exclude_none=True),
+        )