codemie-sdk-python 0.1.69__tar.gz → 0.1.70__tar.gz

{codemie_sdk_python-0.1.69 → codemie_sdk_python-0.1.70}/PKG-INFO

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

{codemie_sdk_python-0.1.69 → codemie_sdk_python-0.1.70}/pyproject.toml

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

{codemie_sdk_python-0.1.69 → codemie_sdk_python-0.1.70}/src/codemie_sdk/__init__.py

@@ -19,5 +19,5 @@ Basic usage:
 
 from .client.client import CodeMieClient
 
-__version__ = "0.1.19"
+__version__ = "0.1.20"
 __all__ = ["CodeMieClient"]

{codemie_sdk_python-0.1.69 → codemie_sdk_python-0.1.70}/src/codemie_sdk/models/integration.py

@@ -1,8 +1,8 @@
 """Models for assistant-related data structures."""
 
+from datetime import datetime
 from enum import Enum
 from typing import List, Optional, Any
-from datetime import datetime
 
 from pydantic import BaseModel, Field, ConfigDict, field_serializer
 
@@ -75,3 +75,21 @@ class Integration(BaseModel):
     @field_serializer("date", "update_date")
     def serialize_dt(self, dt: datetime, _info):
         return dt.isoformat()
+
+
+class TestIntegrationRequest(BaseModel):
+    """Model for test integration request."""
+
+    credential_type: str
+    credential_values: Optional[List[CredentialValues]] = None
+    setting_id: Optional[str] = None
+
+
+class TestIntegrationResponse(BaseModel):
+    """Model for test integration response."""
+
+    def __getitem__(self, key):
+        return getattr(self, key)
+
+    message: str
+    success: bool

{codemie_sdk_python-0.1.69 → codemie_sdk_python-0.1.70}/src/codemie_sdk/services/integration.py

@@ -5,7 +5,11 @@ from typing import List, Optional, Dict, Any
 
 from ..exceptions import NotFoundError
 from ..models.common import PaginationParams
-from ..models.integration import Integration, IntegrationType
+from ..models.integration import (
+    Integration,
+    IntegrationType,
+    TestIntegrationRequest,
+)
 from ..utils import ApiRequestHandler
 
 
@@ -150,3 +154,19 @@ class IntegrationService:
         return self._api.delete(
             f"{self._get_base_path(setting_type)}/{setting_id}", dict
         )
+
+    def test(self, integration: TestIntegrationRequest, response_type: Any) -> Any:
+        """Test an integration.
+
+        Args:
+            integration: TestIntegrationRequest - integration to test
+            response_type: Type of response expected
+
+        Returns:
+            Test integration response
+        """
+        return self._api.post(
+            "/v1/settings/test/",
+            response_model=response_type,
+            json_data=integration.model_dump(exclude_none=True),
+        )