codemie-sdk-python 0.1.351__tar.gz → 0.1.352__tar.gz

{codemie_sdk_python-0.1.351 → codemie_sdk_python-0.1.352}/PKG-INFO

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

{codemie_sdk_python-0.1.351 → codemie_sdk_python-0.1.352}/pyproject.toml

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

{codemie_sdk_python-0.1.351 → codemie_sdk_python-0.1.352}/src/codemie_sdk/models/datasource.py

@@ -28,6 +28,7 @@ class DataSourceType(str, Enum):
     BEDROCK = "knowledge_base_bedrock"
     PLATFORM = "platform_marketplace_assistant"
     AZURE_DEVOPS_WIKI = "knowledge_base_azure_devops_wiki"
+    AZURE_DEVOPS_WORK_ITEM = "knowledge_base_azure_devops_work_item"
 
 
 class DataSourceStatus(str, Enum):
@@ -112,6 +113,16 @@ class AzureDevOpsWiki(BaseModel):
     model_config = ConfigDict(extra="ignore")
 
 
+class AzureDevOpsWorkItem(BaseModel):
+    """Model for Azure DevOps Work Item-specific response fields"""
+
+    wiql_query: Optional[str] = None
+    organization: Optional[str] = None
+    project: Optional[str] = None
+
+    model_config = ConfigDict(extra="ignore")
+
+
 class Code(BaseModel):
     """Model for code repository datasource creation"""
 
@@ -219,6 +230,13 @@ class AzureDevOpsWikiDataSourceRequest(BaseDataSourceRequest, AzureDevOpsWiki):
         super().__init__(type=DataSourceType.AZURE_DEVOPS_WIKI, **data)
 
 
+class AzureDevOpsWorkItemDataSourceRequest(BaseDataSourceRequest, AzureDevOpsWorkItem):
+    """Model for Azure DevOps Work Item datasource creation requests"""
+
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.AZURE_DEVOPS_WORK_ITEM, **data)
+
+
 class BaseUpdateDataSourceRequest(BaseDataSourceRequest):
     """Mixin update-specific reindex fields"""
 
@@ -298,6 +316,15 @@ class UpdateAzureDevOpsWikiDataSourceRequest(
         super().__init__(type=DataSourceType.AZURE_DEVOPS_WIKI, **data)
 
 
+class UpdateAzureDevOpsWorkItemDataSourceRequest(
+    BaseUpdateDataSourceRequest, AzureDevOpsWorkItem
+):
+    """Model for Azure DevOps Work Item datasource updates"""
+
+    def __init__(self, **data):
+        super().__init__(type=DataSourceType.AZURE_DEVOPS_WORK_ITEM, **data)
+
+
 class CodeAnalysisDataSourceRequest(BaseModel):
     """Model for provider-based datasource creation requests"""
 
@@ -368,6 +395,8 @@ class DataSource(BaseModel):
     google_doc_link: Optional[str] = None
     # Azure DevOps Wiki specific fields
     azure_devops_wiki: Optional[AzureDevOpsWiki] = None
+    # Azure DevOps Work Item specific fields
+    azure_devops_work_item: Optional[AzureDevOpsWorkItem] = None
 
     @model_validator(mode="before")
     def before_init(cls, values):
@@ -385,6 +414,7 @@ class DataSource(BaseModel):
             DataSourceType.JIRA,
             DataSourceType.GOOGLE,
             DataSourceType.AZURE_DEVOPS_WIKI,
+            DataSourceType.AZURE_DEVOPS_WORK_ITEM,
         ]:
             complete_state = values.get("complete_state", 0)
             if complete_state is not None: