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

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

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

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

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

{codemie_sdk_python-0.1.619 → 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"
@@ -146,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"""
 
@@ -201,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)
@@ -297,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"""
 
@@ -417,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
@@ -452,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"),