lucius-mcp 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

src/client/client.py

@@ -10,7 +10,7 @@ from collections.abc import Awaitable
 from typing import Literal, TypeVar, cast, overload
 
 import httpx
-from pydantic import SecretStr, ValidationError
+from pydantic import Field, SecretStr, ValidationError
 
 from src.client.exceptions import TestCaseNotFoundError
 from src.utils.config import settings
@@ -26,7 +26,9 @@ from .exceptions import (
 from .generated.api.custom_field_controller_api import CustomFieldControllerApi
 from .generated.api.custom_field_project_controller_api import CustomFieldProjectControllerApi
 from .generated.api.custom_field_project_controller_v2_api import CustomFieldProjectControllerV2Api
+from .generated.api.custom_field_value_controller_api import CustomFieldValueControllerApi
 from .generated.api.custom_field_value_project_controller_api import CustomFieldValueProjectControllerApi
+from .generated.api.integration_controller_api import IntegrationControllerApi
 from .generated.api.launch_controller_api import LaunchControllerApi
 from .generated.api.launch_search_controller_api import LaunchSearchControllerApi
 from .generated.api.shared_step_attachment_controller_api import SharedStepAttachmentControllerApi
@@ -46,17 +48,23 @@ from .generated.models.aql_validate_response_dto import AqlValidateResponseDto
 from .generated.models.attachment_step_dto import AttachmentStepDto
 from .generated.models.body_step_dto import BodyStepDto
 from .generated.models.custom_field_project_with_values_dto import CustomFieldProjectWithValuesDto
+from .generated.models.custom_field_value_project_create_dto import CustomFieldValueProjectCreateDto
+from .generated.models.custom_field_value_project_patch_dto import CustomFieldValueProjectPatchDto
 from .generated.models.custom_field_value_with_cf_dto import CustomFieldValueWithCfDto
 from .generated.models.custom_field_with_values_dto import CustomFieldWithValuesDto
 from .generated.models.find_all29200_response import FindAll29200Response
+from .generated.models.integration_dto import IntegrationDto
+from .generated.models.issue_dto import IssueDto
 from .generated.models.launch_create_dto import LaunchCreateDto
 from .generated.models.launch_dto import LaunchDto
+from .generated.models.page_custom_field_value_with_tc_count_dto import PageCustomFieldValueWithTcCountDto
 from .generated.models.page_launch_dto import PageLaunchDto
 from .generated.models.page_launch_preview_dto import PageLaunchPreviewDto
 from .generated.models.page_shared_step_dto import PageSharedStepDto
 from .generated.models.page_test_case_dto import PageTestCaseDto
 from .generated.models.scenario_step_create_dto import ScenarioStepCreateDto
 from .generated.models.scenario_step_created_response_dto import ScenarioStepCreatedResponseDto
+from .generated.models.scenario_step_patch_dto import ScenarioStepPatchDto
 from .generated.models.shared_step_attachment_row_dto import SharedStepAttachmentRowDto
 from .generated.models.shared_step_create_dto import SharedStepCreateDto
 from .generated.models.shared_step_dto import SharedStepDto
@@ -77,9 +85,13 @@ from .overridden.test_case_custom_fields_v2 import TestCaseCustomFieldV2Controll
 
 # Subclasses to add missing fields to generated models
 class TestCaseDtoWithCF(TestCaseDto):
-    """Subclass to support custom_fields access."""
+    """Subclass to support custom_fields and issues access."""
 
     custom_fields: list[CustomFieldValueWithCfDto] | None = None
+    issues: list[IssueDto] | None = None
+
+
+TestCaseDtoWithCF.model_rebuild()
 
 
 class BodyStepDtoWithSteps(BodyStepDto):
@@ -89,6 +101,14 @@ class BodyStepDtoWithSteps(BodyStepDto):
     id: int | None = None
 
 
+class StepWithExpected(BodyStepDto):
+    """Subclass to support expected results and nested steps."""
+
+    expected_result: str | None = Field(default=None, alias="expectedResult")
+    steps: list[SharedStepScenarioDtoStepsInner] | None = None
+    id: int | None = None
+
+
 class AttachmentStepDtoWithName(AttachmentStepDto):
     """Subclass to support name attribute and id."""
 
@@ -119,11 +139,13 @@ type ApiType = (
     | CustomFieldControllerApi
     | CustomFieldProjectControllerApi
     | CustomFieldProjectControllerV2Api
+    | CustomFieldValueControllerApi
     | CustomFieldValueProjectControllerApi
     | TestLayerControllerApi
     | TestLayerSchemaControllerApi
     | LaunchControllerApi
     | LaunchSearchControllerApi
+    | IntegrationControllerApi
 )
 
 type NormalizedScenarioDict = dict[str, object]
@@ -148,12 +170,14 @@ __all__ = [
     "PageTestCaseDto",
     "ScenarioStepCreateDto",
     "ScenarioStepCreatedResponseDto",
+    "ScenarioStepPatchDto",
     "SharedStepAttachmentRowDto",
     "SharedStepCreateDto",
     "SharedStepDto",
     "SharedStepPatchDto",
     "SharedStepScenarioDtoStepsInner",
     "SharedStepStepDtoWithId",
+    "StepWithExpected",
     "TestCaseAttachmentRowDto",
     "TestCaseCreateV2Dto",
     "TestCaseDto",
@@ -229,11 +253,13 @@ class AllureClient:
         self._custom_field_api: CustomFieldControllerApi | None = None
         self._custom_field_project_api: CustomFieldProjectControllerApi | None = None
         self._custom_field_project_v2_api: CustomFieldProjectControllerV2Api | None = None
+        self._custom_field_value_api: CustomFieldValueControllerApi | None = None
         self._custom_field_value_project_api: CustomFieldValueProjectControllerApi | None = None
         self._test_layer_api: TestLayerControllerApi | None = None
         self._test_layer_schema_api: TestLayerSchemaControllerApi | None = None
         self._launch_api: LaunchControllerApi | None = None
         self._launch_search_api: LaunchSearchControllerApi | None = None
+        self._integration_api: IntegrationControllerApi | None = None
         self._is_entered = False
 
     @classmethod
@@ -374,11 +400,13 @@ class AllureClient:
             self._custom_field_api = CustomFieldControllerApi(self._api_client)
             self._custom_field_project_api = CustomFieldProjectControllerApi(self._api_client)
             self._custom_field_project_v2_api = CustomFieldProjectControllerV2Api(self._api_client)
+            self._custom_field_value_api = CustomFieldValueControllerApi(self._api_client)
             self._custom_field_value_project_api = CustomFieldValueProjectControllerApi(self._api_client)
             self._test_layer_api = TestLayerControllerApi(self._api_client)
             self._test_layer_schema_api = TestLayerSchemaControllerApi(self._api_client)
             self._launch_api = LaunchControllerApi(self._api_client)
             self._launch_search_api = LaunchSearchControllerApi(self._api_client)
+            self._integration_api = IntegrationControllerApi(self._api_client)
 
     @property
     def api_client(self) -> ApiClient:
@@ -391,6 +419,21 @@ class AllureClient:
             raise RuntimeError("AllureClient must be used as an async context manager")
         return self._api_client
 
+    async def get_integrations(self) -> list[IntegrationDto]:
+        """Fetch all integrations."""
+        # Ensure we have a valid client
+        if self._integration_api is None:
+            raise RuntimeError("AllureClient must be used as an async context manager")
+
+        try:
+            # Fetch first page with reasonable size
+            page = await self._integration_api.get_integrations(page=0, size=100)
+            return page.content or []
+        except Exception:
+            # Log warning or re-raise depending on strictness.
+            # For now return empty list to act as fallback.
+            return []
+
     async def __aenter__(self) -> AllureClient:
         """Initialize the client session within an async context.
 
@@ -512,6 +555,11 @@ class AllureClient:
         self, attr_name: Literal["_custom_field_project_v2_api"], *, error_name: str | None = None
     ) -> CustomFieldProjectControllerV2Api: ...
 
+    @overload
+    async def _get_api(
+        self, attr_name: Literal["_custom_field_value_api"], *, error_name: str | None = None
+    ) -> CustomFieldValueControllerApi: ...
+
     @overload
     async def _get_api(
         self, attr_name: Literal["_custom_field_value_project_api"], *, error_name: str | None = None
@@ -831,6 +879,28 @@ class AllureClient:
                     raise AllureValidationError("Unexpected launch list response from API") from e
                 return FindAll29200Response(preview_data)
 
+    async def get_launch(self, launch_id: int) -> LaunchDto:
+        """Retrieve a specific launch by its ID.
+
+        Args:
+            launch_id: The unique ID of the launch.
+
+        Returns:
+            The launch data.
+
+        Raises:
+            AllureNotFoundError: If launch doesn't exist.
+            AllureValidationError: If input is invalid.
+            AllureAuthError: If unauthorized.
+            AllureAPIError: If the server returns an error.
+        """
+        api = await self._get_api("_launch_api", error_name="launch APIs")
+
+        if not isinstance(launch_id, int) or launch_id <= 0:
+            raise AllureValidationError("Launch ID must be a positive integer")
+
+        return await self._call_api(api.find_one23(id=launch_id, _request_timeout=self._timeout))
+
     async def search_launches_aql(
         self,
         project_id: int,
@@ -1088,6 +1158,165 @@ class AllureClient:
             with_expected_result=with_expected_result,
         )
 
+    async def list_custom_field_values(
+        self,
+        project_id: int,
+        custom_field_id: int,
+        *,
+        query: str | None = None,
+        var_global: bool | None = None,
+        test_case_search: str | None = None,
+        page: int | None = None,
+        size: int | None = None,
+        sort: list[str] | None = None,
+    ) -> PageCustomFieldValueWithTcCountDto:
+        """List custom field values for a project field.
+
+        Args:
+            project_id: Target project ID.
+            custom_field_id: Target custom field ID (project-scoped).
+            query: Optional search query.
+            var_global: Optional global flag filter.
+            test_case_search: Optional test case search filter.
+            page: Zero-based page index.
+            size: Page size.
+            sort: Optional sort criteria.
+
+        Returns:
+            Paginated custom field values with test case counts.
+
+        Raises:
+            AllureNotFoundError: If project or custom field doesn't exist.
+            AllureValidationError: If input data fails validation.
+            AllureAuthError: If unauthorized.
+            AllureAPIError: If the server returns an error.
+        """
+        api = await self._get_api("_custom_field_value_project_api")
+
+        if not isinstance(project_id, int) or project_id <= 0:
+            raise AllureValidationError("Project ID must be a positive integer")
+        if not isinstance(custom_field_id, int) or custom_field_id == 0:
+            raise AllureValidationError("Custom Field ID must be a non-zero integer")
+        if page is not None and (not isinstance(page, int) or page < 0):
+            raise AllureValidationError("Page must be a non-negative integer")
+        if size is not None and (not isinstance(size, int) or size <= 0 or size > 1000):
+            raise AllureValidationError("Size must be between 1 and 1000")
+
+        return await self._call_api(
+            api.find_all22(
+                project_id=project_id,
+                custom_field_id=custom_field_id,
+                query=query,
+                var_global=var_global,
+                test_case_search=test_case_search,
+                page=page,
+                size=size,
+                sort=sort,
+                _request_timeout=self._timeout,
+            )
+        )
+
+    async def create_custom_field_value(
+        self, project_id: int, data: CustomFieldValueProjectCreateDto
+    ) -> CustomFieldValueWithCfDto:
+        """Create a custom field value in a project.
+
+        Args:
+            project_id: Target project ID.
+            data: Custom field value payload.
+
+        Returns:
+            The created custom field value DTO.
+
+        Raises:
+            AllureNotFoundError: If project doesn't exist.
+            AllureValidationError: If input data fails validation.
+            AllureAuthError: If unauthorized.
+            AllureAPIError: If the server returns an error.
+        """
+        api = await self._get_api("_custom_field_value_project_api")
+
+        if not isinstance(project_id, int) or project_id <= 0:
+            raise AllureValidationError("Project ID must be a positive integer")
+
+        try:
+            return await self._call_api(
+                api.create26(
+                    project_id=project_id,
+                    custom_field_value_project_create_dto=data,
+                    _request_timeout=self._timeout,
+                )
+            )
+        except AllureAPIError as exc:
+            if exc.status_code == 409:
+                raise AllureValidationError(
+                    "Duplicate custom field value name.",
+                    status_code=exc.status_code,
+                    response_body=exc.response_body,
+                    suggestions=["Use a unique custom field value name"],
+                ) from exc
+            raise
+
+    async def update_custom_field_value(
+        self, project_id: int, cfv_id: int, data: CustomFieldValueProjectPatchDto
+    ) -> None:
+        """Update a custom field value in a project.
+
+        Args:
+            project_id: Target project ID.
+            cfv_id: Target custom field value ID.
+            data: Patch payload for the custom field value.
+
+        Raises:
+            AllureNotFoundError: If project or value doesn't exist.
+            AllureValidationError: If input data fails validation.
+            AllureAuthError: If unauthorized.
+            AllureAPIError: If the server returns an error.
+        """
+        api = await self._get_api("_custom_field_value_project_api")
+
+        if not isinstance(project_id, int) or project_id <= 0:
+            raise AllureValidationError("Project ID must be a positive integer")
+        if not isinstance(cfv_id, int) or cfv_id <= 0:
+            raise AllureValidationError("Custom Field Value ID must be a positive integer")
+
+        await self._call_api(
+            api.patch23(
+                project_id=project_id,
+                cfv_id=cfv_id,
+                custom_field_value_project_patch_dto=data,
+                _request_timeout=self._timeout,
+            )
+        )
+
+    async def delete_custom_field_value(self, project_id: int, cfv_id: int) -> None:
+        """Delete a custom field value in a project.
+
+        Args:
+            project_id: Target project ID.
+            cfv_id: Target custom field value ID.
+
+        Raises:
+            AllureNotFoundError: If project or value doesn't exist.
+            AllureValidationError: If input data fails validation.
+            AllureAuthError: If unauthorized.
+            AllureAPIError: If the server returns an error.
+        """
+        api = await self._get_api("_custom_field_value_project_api")
+
+        if not isinstance(project_id, int) or project_id <= 0:
+            raise AllureValidationError("Project ID must be a positive integer")
+        if not isinstance(cfv_id, int) or cfv_id <= 0:
+            raise AllureValidationError("Custom Field Value ID must be a positive integer")
+
+        await self._call_api(
+            api.delete47(
+                project_id=project_id,
+                id=cfv_id,
+                _request_timeout=self._timeout,
+            )
+        )
+
     async def get_custom_fields_with_values(self, project_id: int) -> list[CustomFieldProjectWithValuesDto]:
         """Fetch all custom fields and their allowed values for a project.
 
@@ -1197,14 +1426,14 @@ class AllureClient:
             AllureAPIError: If the API request fails.
         """
         scenario_api = await self._get_api("_scenario_api")
-        await self._call_api(
-            scenario_api.delete_by_id1(
+        await self._call_api_raw(
+            scenario_api.delete_by_id1_without_preload_content(
                 id=step_id,
                 _request_timeout=self._timeout,
             )
         )
 
-    async def get_test_case(self, test_case_id: int) -> TestCaseDto:
+    async def get_test_case(self, test_case_id: int) -> TestCaseDtoWithCF:
         """Retrieve a specific test case by its ID.
 
         Args:
@@ -1237,6 +1466,8 @@ class AllureClient:
                 )
                 if overview.custom_fields:
                     case.custom_fields = overview.custom_fields
+                if overview.issues:
+                    case.issues = overview.issues
             except Exception as e:
                 logger.warning(f"Failed to fetch overview for test case {test_case_id}: {e}")
 
@@ -1377,13 +1608,14 @@ class AllureClient:
                     child_ids = step_def.get("children") or []
                     child_steps = build_steps(child_ids) if child_ids else None
 
-                    # Build BodyStepDtoWithSteps
+                    # Build StepWithExpected
                     steps_list.append(
                         SharedStepScenarioDtoStepsInner(
-                            actual_instance=BodyStepDtoWithSteps.model_construct(
+                            actual_instance=StepWithExpected.model_construct(
                                 type="BodyStepDto",
                                 body=body,
                                 body_json=None,  # Skip complex rich-text
+                                expected_result=step_def.get("expectedResult"),
                                 steps=child_steps,
                                 id=sid,
                             )
@@ -1482,6 +1714,38 @@ class AllureClient:
 
         return await self._create_scenario_step_via_api(shared_step_scenario_api, step)
 
+    async def patch_test_case_scenario_step(
+        self,
+        step_id: int,
+        patch: ScenarioStepPatchDto,
+    ) -> None:
+        """Patch a specific scenario step within a test case."""
+        scenario_api = await self._get_api("_scenario_api")
+        await self._call_api_raw(
+            scenario_api.patch_by_id_without_preload_content(
+                id=step_id,
+                scenario_step_patch_dto=patch,
+                with_expected_result=False,
+                _request_timeout=self._timeout,
+            )
+        )
+
+    async def patch_shared_step_scenario_step(
+        self,
+        step_id: int,
+        patch: ScenarioStepPatchDto,
+    ) -> None:
+        """Patch a specific scenario step within a shared step."""
+        shared_step_scenario_api = await self._get_api("_shared_step_scenario_api")
+        await self._call_api_raw(
+            shared_step_scenario_api.patch_by_id1_without_preload_content(
+                id=step_id,
+                scenario_step_patch_dto=patch,
+                with_expected_result=False,
+                _request_timeout=self._timeout,
+            )
+        )
+
     async def upload_shared_step_attachment(
         self,
         shared_step_id: int,

src/client/exceptions.py

@@ -42,6 +42,28 @@ class TestCaseNotFoundError(ResourceNotFoundError):
         )
 
 
+class LaunchNotFoundError(ResourceNotFoundError):
+    """Resource not found for a specific launch."""
+
+    def __init__(
+        self,
+        launch_id: int,
+        status_code: int | None = None,
+        response_body: str | None = None,
+    ) -> None:
+        suggestions = [
+            "Verify the launch ID",
+            "Use list_launches to find valid IDs",
+            "Check access to the project containing this launch",
+        ]
+        super().__init__(
+            message=f"Launch ID {launch_id} not found",
+            status_code=status_code,
+            response_body=response_body,
+            suggestions=suggestions,
+        )
+
+
 class AllureAuthError(AuthenticationError):
     """Authentication failed (401/403)."""
 
@@ -57,5 +79,6 @@ __all__ = [
     "AllureNotFoundError",
     "AllureRateLimitError",
     "AllureValidationError",
+    "LaunchNotFoundError",
     "TestCaseNotFoundError",
 ]

src/client/generated/README.md

@@ -107,6 +107,24 @@ Class | Method | HTTP request | Description
 *CustomFieldValueProjectControllerApi* | [**merge_custom_fields_to_existing_record**](src/client/generated/docs/CustomFieldValueProjectControllerApi.md#merge_custom_fields_to_existing_record) | **POST** /api/project/{projectId}/cfv/merge-to/{toCfvId} | Merge custom field values to existing record by id
 *CustomFieldValueProjectControllerApi* | [**merge_custom_fields_to_new_record**](src/client/generated/docs/CustomFieldValueProjectControllerApi.md#merge_custom_fields_to_new_record) | **POST** /api/project/{projectId}/cfv/merge | Merge custom field values to new record
 *CustomFieldValueProjectControllerApi* | [**patch23**](src/client/generated/docs/CustomFieldValueProjectControllerApi.md#patch23) | **PATCH** /api/project/{projectId}/cfv/{cfvId} | Patch specified custom field value, test results won't be affected
+*IntegrationControllerApi* | [**create37**](src/client/generated/docs/IntegrationControllerApi.md#create37) | **POST** /api/integration | 
+*IntegrationControllerApi* | [**create_project_integration**](src/client/generated/docs/IntegrationControllerApi.md#create_project_integration) | **POST** /api/integration/project | 
+*IntegrationControllerApi* | [**delete_by_id3**](src/client/generated/docs/IntegrationControllerApi.md#delete_by_id3) | **DELETE** /api/integration/{id} | 
+*IntegrationControllerApi* | [**delete_project_integration**](src/client/generated/docs/IntegrationControllerApi.md#delete_project_integration) | **DELETE** /api/integration/{integrationId}/project/{projectId} | 
+*IntegrationControllerApi* | [**find_one_by_id**](src/client/generated/docs/IntegrationControllerApi.md#find_one_by_id) | **GET** /api/integration/{id} | 
+*IntegrationControllerApi* | [**find_project_integration_by_id**](src/client/generated/docs/IntegrationControllerApi.md#find_project_integration_by_id) | **GET** /api/integration/{integrationId}/project/{projectId} | 
+*IntegrationControllerApi* | [**get_available_integrations**](src/client/generated/docs/IntegrationControllerApi.md#get_available_integrations) | **GET** /api/integration/available | 
+*IntegrationControllerApi* | [**get_global_fields**](src/client/generated/docs/IntegrationControllerApi.md#get_global_fields) | **GET** /api/integration/globalfields | 
+*IntegrationControllerApi* | [**get_integration_projects**](src/client/generated/docs/IntegrationControllerApi.md#get_integration_projects) | **GET** /api/integration/{id}/project | 
+*IntegrationControllerApi* | [**get_integrations**](src/client/generated/docs/IntegrationControllerApi.md#get_integrations) | **GET** /api/integration | 
+*IntegrationControllerApi* | [**get_project_available_integrations**](src/client/generated/docs/IntegrationControllerApi.md#get_project_available_integrations) | **GET** /api/integration/project/{projectId}/available | 
+*IntegrationControllerApi* | [**get_project_integration_fields**](src/client/generated/docs/IntegrationControllerApi.md#get_project_integration_fields) | **GET** /api/integration/projectfields | 
+*IntegrationControllerApi* | [**get_project_integrations**](src/client/generated/docs/IntegrationControllerApi.md#get_project_integrations) | **GET** /api/integration/project/{projectId} | 
+*IntegrationControllerApi* | [**patch34**](src/client/generated/docs/IntegrationControllerApi.md#patch34) | **PATCH** /api/integration/{id} | 
+*IntegrationControllerApi* | [**patch_project_integration**](src/client/generated/docs/IntegrationControllerApi.md#patch_project_integration) | **PATCH** /api/integration/{integrationId}/project/{projectId} | 
+*IntegrationControllerApi* | [**suggest15**](src/client/generated/docs/IntegrationControllerApi.md#suggest15) | **GET** /api/integration/suggest | Suggest integrations
+*IntegrationControllerApi* | [**validate1**](src/client/generated/docs/IntegrationControllerApi.md#validate1) | **POST** /api/integration/validate | 
+*IntegrationControllerApi* | [**validate2**](src/client/generated/docs/IntegrationControllerApi.md#validate2) | **POST** /api/integration/project/validate | 
 *LaunchControllerApi* | [**add_test_cases**](src/client/generated/docs/LaunchControllerApi.md#add_test_cases) | **POST** /api/launch/{id}/testcase/add | Add test cases to launch
 *LaunchControllerApi* | [**add_test_plan**](src/client/generated/docs/LaunchControllerApi.md#add_test_plan) | **POST** /api/launch/{id}/testplan/add | Add test plan to launch
 *LaunchControllerApi* | [**apply_defect_matchers**](src/client/generated/docs/LaunchControllerApi.md#apply_defect_matchers) | **POST** /api/launch/{id}/defect/apply | Apply defect matchers to launch

src/client/generated/__init__.py

@@ -24,6 +24,7 @@ __all__ = [
     "CustomFieldSchemaControllerApi",
     "CustomFieldValueControllerApi",
     "CustomFieldValueProjectControllerApi",
+    "IntegrationControllerApi",
     "LaunchControllerApi",
     "LaunchSearchControllerApi",
     "ProjectControllerApi",
@@ -742,6 +743,7 @@ from src.client.generated.api.custom_field_project_controller_v2_api import Cust
 from src.client.generated.api.custom_field_schema_controller_api import CustomFieldSchemaControllerApi as CustomFieldSchemaControllerApi
 from src.client.generated.api.custom_field_value_controller_api import CustomFieldValueControllerApi as CustomFieldValueControllerApi
 from src.client.generated.api.custom_field_value_project_controller_api import CustomFieldValueProjectControllerApi as CustomFieldValueProjectControllerApi
+from src.client.generated.api.integration_controller_api import IntegrationControllerApi as IntegrationControllerApi
 from src.client.generated.api.launch_controller_api import LaunchControllerApi as LaunchControllerApi
 from src.client.generated.api.launch_search_controller_api import LaunchSearchControllerApi as LaunchSearchControllerApi
 from src.client.generated.api.project_controller_api import ProjectControllerApi as ProjectControllerApi

src/client/generated/api/__init__.py

@@ -7,6 +7,7 @@ from src.client.generated.api.custom_field_project_controller_v2_api import Cust
 from src.client.generated.api.custom_field_schema_controller_api import CustomFieldSchemaControllerApi
 from src.client.generated.api.custom_field_value_controller_api import CustomFieldValueControllerApi
 from src.client.generated.api.custom_field_value_project_controller_api import CustomFieldValueProjectControllerApi
+from src.client.generated.api.integration_controller_api import IntegrationControllerApi
 from src.client.generated.api.launch_controller_api import LaunchControllerApi
 from src.client.generated.api.launch_search_controller_api import LaunchSearchControllerApi
 from src.client.generated.api.project_controller_api import ProjectControllerApi