codemie-sdk-python 0.1.618__tar.gz → 0.1.620__tar.gz

{codemie_sdk_python-0.1.618 → codemie_sdk_python-0.1.620}/PKG-INFO

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

{codemie_sdk_python-0.1.618 → codemie_sdk_python-0.1.620}/pyproject.toml

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

{codemie_sdk_python-0.1.618 → codemie_sdk_python-0.1.620}/src/codemie_sdk/models/datasource.py

@@ -21,6 +21,7 @@ class DataSourceType(str, Enum):
     FILE = "knowledge_base_file"
     XRAY = "knowledge_base_xray"
     GOOGLE = "llm_routing_google"
+    GIT_FAQ = "knowledge_base_git_faq"
     PROVIDER = "provider"
     SUMMARY = "summary"
     CHUNK_SUMMARY = "chunk-summary"
@@ -82,6 +83,7 @@ class Jira(BaseModel):
     """Model for Jira-specific response fields"""
 
     jql: str | None = None
+    custom_fields: list[str] | None = None
 
     model_config = ConfigDict(extra="ignore")
 
@@ -145,6 +147,15 @@ class Code(BaseModel):
     docs_generation: bool = Field(False, alias="docsGeneration")
 
 
+class GitFaq(BaseModel):
+    """Model for the Git FAQ (knowledge_base_git_faq) datasource creation"""
+
+    link: str | None = Field(..., min_length=1, max_length=1000)
+    branch: str | None = Field(..., min_length=1, max_length=1000)
+    files_filter: str | None = Field(None)
+    embedding_model: str | None = Field(None)
+
+
 class BaseDataSourceRequest(BaseModel):
     """Base model for all datasource creation requests"""
 
@@ -200,6 +211,15 @@ class CodeDataSourceRequest(BaseDataSourceRequest, Code):
         return ["link", "embeddings_model", "branch", "index_type"]
 
 
+class GitFaqDataSourceRequest(BaseDataSourceRequest, GitFaq):
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.GIT_FAQ, **data)
+
+    @classmethod
+    def required_fields(cls) -> list[str]:
+        return ["link", "branch"]
+
+
 class JiraDataSourceRequest(BaseDataSourceRequest, Jira):
     def __init__(self, **data):
         super().__init__(type=DataSourceType.JIRA, **data)
@@ -296,6 +316,18 @@ class BaseUpdateDataSourceRequest(BaseDataSourceRequest):
         return self
 
 
+class UpdateGitFaqDataSourceRequest(BaseUpdateDataSourceRequest):
+    """Model for Git FAQ datasource updates; source fields require full_reindex=True"""
+
+    link: str | None = Field(None)
+    branch: str | None = Field(None)
+    files_filter: str | None = Field(None)
+    setting_id: str | None = Field(None)
+
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.GIT_FAQ, **data)
+
+
 class UpdateCodeDataSourceRequest(BaseUpdateDataSourceRequest, Code):
     """Model for code repository datasource updates"""
 
@@ -416,6 +448,7 @@ class DataSource(BaseModel):
     tokens_usage: TokensUsage | None = Field(None)
     # Code specific fields
     code: Code | None = None
+    git_faq: GitFaq | None = None
     # Jira specific fields
     jira: Jira | None = None
     # Confluence specific fields
@@ -451,6 +484,13 @@ class DataSource(BaseModel):
             complete_state = values.get("complete_state", 0)
             if complete_state is not None:
                 values["processing_info"] = {"documents_count_key": complete_state}
+        elif values.get("index_type") == DataSourceType.GIT_FAQ:
+            values["git_faq"] = {
+                "link": values.get("link"),
+                "branch": values.get("branch"),
+                "files_filter": values.get("files_filter"),
+                "setting_id": values.get("setting_id"),
+            }
         elif values.get("index_type") == DataSourceType.CODE:
             values["code"] = {
                 "link": values.get("link"),

{codemie_sdk_python-0.1.618 → codemie_sdk_python-0.1.620}/src/codemie_sdk/services/datasource.py

@@ -302,19 +302,24 @@ class DatasourceService:
         project_name: str = None,
         space: str = None,
         wiki: str = None,
+        **extra_kwargs,
     ) -> dict:
         """POST v1/index/health — probe datasource connectivity without indexing.
 
         Args:
-            index_type: Type of the datasource index (e.g., "git", "xwiki")
+            index_type: Type of the datasource index (e.g., "git", "xwiki", "jira")
             git_url: Git repository URL (required for git type)
             setting_id: Optional integration setting ID (for authenticated repos)
             project_name: Optional project name context (defaults to empty string)
             space: xWiki space id to probe (required for xwiki type)
             wiki: xWiki wiki id to probe (defaults to "xwiki" on the backend)
+            **extra_kwargs: Additional payload fields forwarded verbatim (e.g.
+                ``jql`` and ``custom_fields`` for Jira KB health checks).
 
         Returns:
-            dict: Response containing 'error' (None on success) and 'documents_count'
+            dict: Response containing 'error' (None on success), 'documents_count',
+                and optionally 'field_error' / 'help' when a field-level validation
+                error is detected.
         """
         payload: dict = {"index_type": index_type, "project_name": project_name or ""}
         if git_url is not None:
@@ -325,8 +330,33 @@ class DatasourceService:
             payload["space"] = space
         if wiki is not None:
             payload["wiki"] = wiki
+        payload.update(extra_kwargs)
         return self._api.post("/v1/index/health", dict, json_data=payload)
 
+    def get_jira_fields(self, setting_id: str, project_name: str = "") -> list:
+        """GET /v1/index/jira/fields — retrieve available Jira custom fields.
+
+        Returns the list of non-default Jira fields available for the given
+        integration, sorted custom-first then alphabetically. Used by the UI
+        field picker for the ``custom_fields`` datasource configuration.
+
+        Args:
+            setting_id: The Jira integration setting ID whose Jira instance
+                should be queried for available fields.
+            project_name: Project name context for the request.
+
+        Returns:
+            list: Sorted list of field dicts (``id``, ``name``, ``custom``).
+
+        Raises:
+            ApiError: HTTP 422 when the Jira instance is unreachable
+                (ConnectionException maps to VPN-specific help text).
+        """
+        return self._api.get(
+            f"/v1/index/jira/fields?setting_id={setting_id}&project_name={project_name}",
+            list[dict],
+        )
+
     def get_elasticsearch_stats(self, datasource_id: str) -> ElasticsearchStatsResponse:
         """Get Elasticsearch statistics for a specific datasource index.