codemie-test-harness 0.1.129__py3-none-any.whl → 0.1.131__py3-none-any.whl

{tests → codemie_test_harness/tests}/integrations/project/test_default_integrations.py

@@ -6,26 +6,33 @@ from codemie_sdk.models.integration import (
     IntegrationType,
     CredentialTypes,
 )
-from tests.enums.tools import (
+from codemie_test_harness.tests.enums.tools import (
     Toolkit,
     NotificationTool,
     ProjectManagementTool,
     CodeBaseTool,
     CloudTool,
 )
-from tests.test_data.ado_test_plan_tools_test_data import ado_test_plan_get_test_data
-from tests.test_data.cloud_tools_test_data import cloud_test_data
-from tests.test_data.codebase_tools_test_data import sonar_tools_test_data
-from tests.test_data.git_tools_test_data import (
+from codemie_test_harness.tests.test_data.ado_test_plan_tools_test_data import (
+    ado_test_plan_get_test_data,
+)
+from codemie_test_harness.tests.test_data.cloud_tools_test_data import cloud_test_data
+from codemie_test_harness.tests.test_data.codebase_tools_test_data import (
+    sonar_tools_test_data,
+)
+from codemie_test_harness.tests.test_data.git_tools_test_data import (
     list_branches_set_active_branch_test_data,
 )
-from tests.test_data.notification_tools_test_data import (
+from codemie_test_harness.tests.test_data.notification_tools_test_data import (
     EMAIL_TOOL_PROMPT,
     EMAIL_RESPONSE,
 )
-from tests.test_data.pm_tools_test_data import JIRA_TOOL_PROMPT, RESPONSE_FOR_JIRA_TOOL
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests.test_data.pm_tools_test_data import (
+    JIRA_TOOL_PROMPT,
+    RESPONSE_FOR_JIRA_TOOL,
+)
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 pytestmark = pytest.mark.skipif(
     os.getenv("ENV") in ["local", "gcp"],

{tests → codemie_test_harness/tests}/integrations/project/test_project_integrations.py

@@ -9,13 +9,16 @@ from codemie_sdk.models.integration import (
     IntegrationTestResponse,
     CredentialTypes,
 )
-from tests import PROJECT
-from tests.test_data.integrations_test_data import (
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.test_data.integrations_test_data import (
     valid_integrations,
     invalid_integrations,
     testable_integrations,
 )
-from tests.utils.base_utils import get_random_name, assert_error_details
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_error_details,
+)
 
 
 @pytest.mark.parametrize(

{tests → codemie_test_harness/tests}/integrations/user/test_default_integrations.py

@@ -6,26 +6,33 @@ from codemie_sdk.models.integration import (
     IntegrationType,
     CredentialTypes,
 )
-from tests.enums.tools import (
+from codemie_test_harness.tests.enums.tools import (
     Toolkit,
     NotificationTool,
     ProjectManagementTool,
     CodeBaseTool,
     CloudTool,
 )
-from tests.test_data.ado_test_plan_tools_test_data import ado_test_plan_get_test_data
-from tests.test_data.cloud_tools_test_data import cloud_test_data
-from tests.test_data.codebase_tools_test_data import sonar_tools_test_data
-from tests.test_data.git_tools_test_data import (
+from codemie_test_harness.tests.test_data.ado_test_plan_tools_test_data import (
+    ado_test_plan_get_test_data,
+)
+from codemie_test_harness.tests.test_data.cloud_tools_test_data import cloud_test_data
+from codemie_test_harness.tests.test_data.codebase_tools_test_data import (
+    sonar_tools_test_data,
+)
+from codemie_test_harness.tests.test_data.git_tools_test_data import (
     list_branches_set_active_branch_test_data,
 )
-from tests.test_data.notification_tools_test_data import (
+from codemie_test_harness.tests.test_data.notification_tools_test_data import (
     EMAIL_TOOL_PROMPT,
     EMAIL_RESPONSE,
 )
-from tests.test_data.pm_tools_test_data import JIRA_TOOL_PROMPT, RESPONSE_FOR_JIRA_TOOL
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests.test_data.pm_tools_test_data import (
+    JIRA_TOOL_PROMPT,
+    RESPONSE_FOR_JIRA_TOOL,
+)
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 pytestmark = pytest.mark.skipif(
     os.getenv("ENV") in ["local", "gcp"],

{tests → codemie_test_harness/tests}/integrations/user/test_user_integrations.py

@@ -9,13 +9,13 @@ from codemie_sdk.models.integration import (
     IntegrationTestResponse,
     CredentialTypes,
 )
-from tests import PROJECT
-from tests.test_data.integrations_test_data import (
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.test_data.integrations_test_data import (
     valid_integrations,
     invalid_integrations,
     testable_integrations,
 )
-from tests.utils.base_utils import (
+from codemie_test_harness.tests.utils.base_utils import (
     get_random_name,
     assert_error_details,
 )

{tests → codemie_test_harness/tests}/llm/assistants/test_llm.py

@@ -1,10 +1,10 @@
 import os
 import pytest
 from hamcrest import assert_that, has_item
-from tests.enums.model_types import ModelTypes
-from tests.test_data.llm_test_data import MODEL_RESPONSES
-from tests.utils.client_factory import get_client
-from tests.utils.pytest_utils import check_mark
+from codemie_test_harness.tests.enums.model_types import ModelTypes
+from codemie_test_harness.tests.test_data.llm_test_data import MODEL_RESPONSES
+from codemie_test_harness.tests.utils.client_factory import get_client
+from codemie_test_harness.tests.utils.pytest_utils import check_mark
 
 
 def pytest_generate_tests(metafunc):

codemie_test_harness/tests/providers/test_providers_endpoints.py

@@ -0,0 +1,239 @@
+import pytest
+from hamcrest import (
+    assert_that,
+    is_,
+    equal_to,
+    has_key,
+    all_of,
+)
+
+from tests.utils.base_utils import get_random_name
+from tests.utils.provider_utils import ProviderUtils
+
+
+@pytest.fixture(scope="session")
+def providers_utils(client):
+    return ProviderUtils(client)
+
+
+nonexistent_id = "nonexistent-provider-id-12345"
+
+# =============================================================================
+# GET /v1/providers - List all providers
+# =============================================================================
+
+
+@pytest.mark.regression
+def test_get_providers_endpoint(providers_utils):
+    response = providers_utils.send_get_request_to_providers_endpoint()
+
+    assert_that(response.status_code, equal_to(200))
+
+    response_data = response.json()
+
+    # Validate response structure
+    assert_that(isinstance(response_data, list), is_(True))
+    for provider in response_data:
+        assert_that(
+            provider,
+            all_of(has_key("provided_toolkits"), has_key("service_location_url")),
+        )
+
+
+# =============================================================================
+# POST /v1/providers - Create new provider
+# =============================================================================
+
+
+@pytest.mark.regression
+def test_post_providers_endpoint(providers_utils):
+    request_json = providers_utils.provider_request_json()
+    # Ensure unique provider name to avoid conflicts
+    provider_name = get_random_name()
+    request_json["name"] = provider_name
+
+    create_response = providers_utils.send_post_request_to_providers_endpoint(
+        request_json
+    )
+
+    assert_that(create_response.status_code, equal_to(200))
+
+    # get provider to verify it was created
+    get_response = providers_utils.send_get_request_to_providers_endpoint()
+
+    providers = get_response.json()
+    created_provider = next((p for p in providers if p["name"] == provider_name), None)
+
+    assert_that(
+        created_provider is not None,
+        is_(True),
+        "Created provider should be in the list",
+    )
+
+
+@pytest.mark.regression
+def test_post_providers_endpoint_validation_error(providers_utils):
+    """Test POST /v1/providers with invalid provided_toolkits node"""
+    request_json = providers_utils.provider_request_json()
+    request_json["provided_toolkits"] = "invalid_value"
+
+    response = providers_utils.send_post_request_to_providers_endpoint(request_json)
+
+    assert_that(response.status_code, equal_to(422))
+    assert_that(
+        response.json()["error"]["details"][0]["msg"],
+        equal_to("Input should be a valid list"),
+    )
+
+
+@pytest.mark.regression
+def test_post_providers_endpoint_with_existing_name(providers_utils):
+    provider_id = providers_utils.create_provider()
+
+    provider = providers_utils.get_provider_by_id(provider_id).json()
+
+    existing_name = provider["name"]
+    request_json = providers_utils.provider_request_json()
+    request_json["name"] = existing_name
+
+    create_response = providers_utils.send_post_request_to_providers_endpoint(
+        request_json
+    )
+
+    assert_that(create_response.status_code, equal_to(409))
+    assert_that(
+        create_response.json()["error"]["details"],
+        equal_to(f"A provider with the name [{existing_name}] already exists."),
+    )
+
+
+# =============================================================================
+# GET /v1/providers/datasource_schemas - Get datasource schemas
+# =============================================================================
+
+
+@pytest.mark.regression
+def test_get_providers_datasource_schema_endpoint(providers_utils):
+    response = providers_utils.send_get_request_datasource_schemas_endpoint()
+
+    assert_that(response.status_code, equal_to(200))
+
+    response_data = response.json()
+
+    # Validate response structure
+    assert_that(isinstance(response_data, list), is_(True))
+    for provider in response_data:
+        assert_that(
+            provider,
+            all_of(
+                has_key("id"),
+                has_key("provider_name"),
+                has_key("name"),
+                has_key("base_schema"),
+                has_key("create_schema"),
+            ),
+        )
+
+
+# =============================================================================
+# GET /v1/providers/{provider_id} - Get specific provider
+# =============================================================================
+
+
+@pytest.mark.regression
+def test_get_provider_by_id_endpoint(providers_utils):
+    provider_id = providers_utils.create_provider()
+
+    provider_response = providers_utils.get_provider_by_id(provider_id)
+
+    assert_that(provider_response.status_code, equal_to(200))
+    assert_that(provider_response.json()["id"], equal_to(provider_id))
+
+
+@pytest.mark.regression
+def test_get_nonexistent_provider(providers_utils):
+    """Test error handling when requesting nonexistent provider."""
+    response = providers_utils.get_provider_by_id(nonexistent_id)
+
+    assert_that(response.status_code, equal_to(404))
+
+    assert_that(
+        response.json()["error"]["details"],
+        equal_to(
+            f"The provider with ID [{nonexistent_id}] could not be found in the system."
+        ),
+    )
+
+
+# =============================================================================
+# PUT /v1/providers/{provider_id} - Update specific provider
+# =============================================================================
+
+
+@pytest.mark.regression
+def test_put_provider_endpoint(providers_utils):
+    provider_id = providers_utils.create_provider()
+    provider = providers_utils.get_provider_by_id(provider_id).json()
+
+    request_json = providers_utils.provider_request_json()
+    toolkits = provider["provided_toolkits"]
+    toolkits[0]["name"] = "Updated toolkit name"
+    toolkits[0]["description"] = "Updated toolkit description"
+    request_json["provided_toolkits"] = toolkits
+    request_json["name"] = f"{provider['name']} - updated"
+    update_response = providers_utils.update_provider(provider_id, request_json)
+
+    assert_that(update_response.status_code, equal_to(200))
+    assert_that(update_response.json()["provided_toolkits"], equal_to(toolkits))
+
+
+@pytest.mark.regression
+def test_put_nonexistent_provider(providers_utils):
+    response = providers_utils.update_provider(nonexistent_id, {})
+
+    assert_that(response.status_code, equal_to(404))
+    assert_that(
+        response.json()["error"]["details"],
+        equal_to(
+            f"The provider with ID [{nonexistent_id}] could not be found in the system."
+        ),
+    )
+
+
+# =============================================================================
+# DELETE /v1/providers/{provider_id} - Delete specific provider
+# =============================================================================
+
+
+@pytest.mark.regression
+def test_delete_provider_endpoint(providers_utils):
+    provider_id = providers_utils.create_provider()
+    delete_response = providers_utils.send_delete_provider_request(provider_id)
+
+    assert_that(delete_response.status_code, equal_to(204))
+
+    response = providers_utils.get_provider_by_id(provider_id)
+
+    assert_that(response.status_code, equal_to(404))
+
+    assert_that(
+        response.json()["error"]["details"],
+        equal_to(
+            f"The provider with ID [{provider_id}] could not be found in the system."
+        ),
+    )
+
+
+@pytest.mark.regression
+def test_delete_nonexistent_provider(providers_utils):
+    """Test error handling when requesting nonexistent provider."""
+    response = providers_utils.send_delete_provider_request(nonexistent_id)
+
+    assert_that(response.status_code, equal_to(404))
+
+    assert_that(
+        response.json()["error"]["details"],
+        equal_to(
+            f"The provider with ID [{nonexistent_id}] could not be found in the system."
+        ),
+    )

{tests → codemie_test_harness/tests}/search/test_search_assistant.py

@@ -1,9 +1,9 @@
 import pytest
 from hamcrest import assert_that
 
-from tests import TEST_USER
-from tests.utils.base_utils import get_random_name
-from tests.utils.pytest_utils import check_mark
+from codemie_test_harness.tests import TEST_USER
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.pytest_utils import check_mark
 
 
 @pytest.fixture(scope="session")

{tests → codemie_test_harness/tests}/search/test_search_datasource.py

@@ -3,9 +3,9 @@ from hamcrest import assert_that
 
 from codemie_sdk.models.datasource import DataSourceType, DataSourceStatus
 from codemie_sdk.models.integration import CredentialTypes
-from tests import TEST_USER, PROJECT
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests import TEST_USER, PROJECT
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 ASSERTION_ERROR = "Datasource is not found in search results."
 

{tests → codemie_test_harness/tests}/search/test_search_integration.py

@@ -2,8 +2,8 @@ import pytest
 from hamcrest import assert_that
 
 from codemie_sdk.models.integration import CredentialTypes, IntegrationType
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 ASSERTION_ERROR = "Integration is not found in search results."
 

{tests → codemie_test_harness/tests}/search/test_search_workflow.py

@@ -1,9 +1,9 @@
 import pytest
 from hamcrest import assert_that
 
-from tests import PROJECT
-from tests.utils.base_utils import get_random_name
-from tests.utils.yaml_utils import AssistantModel, StateModel
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.yaml_utils import AssistantModel, StateModel
 
 
 @pytest.fixture(scope="session")

{tests → codemie_test_harness/tests}/service/test_assistant_service.py

@@ -20,8 +20,8 @@ from codemie_sdk.models.assistant import (
     ChatRole,
     BaseModelResponse,
 )
-from tests import PROJECT, LANGFUSE_TRACES_ENABLED
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests import PROJECT, LANGFUSE_TRACES_ENABLED
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 def test_get_tools(client: CodeMieClient):

{tests → codemie_test_harness/tests}/service/test_datasource_service.py

@@ -28,8 +28,8 @@ from codemie_sdk.models.integration import (
     Integration,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.fixture

{tests → codemie_test_harness/tests}/service/test_integration_service.py

@@ -11,8 +11,8 @@ from codemie_sdk.models.integration import (
     CredentialValues,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 def test_list_project_integrations_minimal(client: CodeMieClient):

{tests → codemie_test_harness/tests}/service/test_task_service.py

@@ -16,8 +16,8 @@ from codemie_sdk.models.assistant import (
     ChatRole,
 )
 from codemie_sdk.models.task import BackgroundTaskEntity
-from tests import PROJECT, LANGFUSE_TRACES_ENABLED
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests import PROJECT, LANGFUSE_TRACES_ENABLED
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 def test_background_task_flow(client: CodeMieClient, default_llm):

{tests → codemie_test_harness/tests}/service/test_workflow_execution_service.py

@@ -11,8 +11,8 @@ from codemie_sdk.models.workflow import (
     WorkflowMode,
     ExecutionStatus,
 )
-from tests import PROJECT
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.fixture

{tests → codemie_test_harness/tests}/service/test_workflow_service.py

@@ -11,8 +11,8 @@ from codemie_sdk.models.workflow import (
     WorkflowUpdateRequest,
     WorkflowMode,
 )
-from tests import PROJECT
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.fixture

{tests → codemie_test_harness/tests}/test_data/ado_test_plan_tools_test_data.py

@@ -1,4 +1,4 @@
-from tests.enums.tools import Toolkit, AzureDevOpsTestPlanTool
+from codemie_test_harness.tests.enums.tools import Toolkit, AzureDevOpsTestPlanTool
 
 ado_test_plan_get_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/ado_wiki_tools_test_data.py

@@ -1,4 +1,4 @@
-from tests.enums.tools import AzureDevOpsWikiTool, Toolkit
+from codemie_test_harness.tests.enums.tools import AzureDevOpsWikiTool, Toolkit
 
 ado_wiki_get_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/ado_work_item_tools_test_data.py

@@ -1,4 +1,4 @@
-from tests.enums.tools import Toolkit, AzureDevOpsWorkItemTool
+from codemie_test_harness.tests.enums.tools import Toolkit, AzureDevOpsWorkItemTool
 
 ado_work_item_get_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/cloud_tools_test_data.py

@@ -2,9 +2,9 @@ import os
 
 import pytest
 
-from tests.enums.tools import Toolkit, CloudTool
+from codemie_test_harness.tests.enums.tools import Toolkit, CloudTool
 from codemie_sdk.models.integration import CredentialTypes
-from tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
 
 cloud_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/codebase_tools_test_data.py

@@ -1,5 +1,5 @@
-from tests.enums.tools import Toolkit, CodeBaseTool
-from tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.enums.tools import Toolkit, CodeBaseTool
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
 
 code_tools_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/data_management_tools_test_data.py

@@ -1,4 +1,4 @@
-from tests.enums.integrations import DataBaseDialect
+from codemie_test_harness.tests.enums.integrations import DataBaseDialect
 
 ELASTIC_TOOL_TASK = """
     Search for Document ID: Zx4eF5cBvnDA_Y45jlWN in codemie_metrics_logs index

{tests → codemie_test_harness/tests}/test_data/direct_tools/ado_test_plan_tools_test_data.py

@@ -1,4 +1,4 @@
-from tests.enums.tools import Toolkit, AzureDevOpsTestPlanTool
+from codemie_test_harness.tests.enums.tools import Toolkit, AzureDevOpsTestPlanTool
 
 ado_test_plan_get_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/direct_tools/ado_wiki_tools_test_data.py

@@ -1,4 +1,4 @@
-from tests.enums.tools import Toolkit, AzureDevOpsWikiTool
+from codemie_test_harness.tests.enums.tools import Toolkit, AzureDevOpsWikiTool
 
 ado_wiki_get_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/direct_tools/ado_work_item_tools_test_data.py

@@ -1,4 +1,4 @@
-from tests.enums.tools import Toolkit, AzureDevOpsWorkItemTool
+from codemie_test_harness.tests.enums.tools import Toolkit, AzureDevOpsWorkItemTool
 
 ado_work_item_get_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/direct_tools/cloud_tools_test_data.py

@@ -2,9 +2,9 @@ import os
 
 import pytest
 
-from tests.enums.tools import Toolkit, CloudTool
+from codemie_test_harness.tests.enums.tools import Toolkit, CloudTool
 from codemie_sdk.models.integration import CredentialTypes
-from tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
 
 cloud_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/direct_tools/codebase_tools_test_data.py

@@ -1,4 +1,4 @@
-from tests.enums.tools import Toolkit, CodeBaseTool
+from codemie_test_harness.tests.enums.tools import Toolkit, CodeBaseTool
 
 sonar_tools_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/direct_tools/data_management_tools_test_data.py

@@ -1,5 +1,5 @@
-from tests.enums.tools import DataManagementTool, Toolkit
-from tests.enums.integrations import DataBaseDialect
+from codemie_test_harness.tests.enums.tools import DataManagementTool, Toolkit
+from codemie_test_harness.tests.enums.integrations import DataBaseDialect
 
 sql_tools_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/direct_tools/direct_tools_test_data.py

@@ -1,5 +1,5 @@
-from tests.enums.tools import PluginTool
-from tests.utils.constants import TESTS_PATH
+from codemie_test_harness.tests.enums.tools import PluginTool
+from codemie_test_harness.tests.utils.constants import TESTS_PATH
 
 cli_tools_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/direct_tools/file_management_tools_test_data.py

@@ -2,7 +2,7 @@ import os
 
 import pytest
 
-from tests.enums.tools import FileManagementTool
+from codemie_test_harness.tests.enums.tools import FileManagementTool
 
 file_management_tools_test_data = [
     (

{tests → codemie_test_harness/tests}/test_data/direct_tools/notification_tools_test_data.py

@@ -2,8 +2,8 @@ import os
 
 import pytest
 
-from tests.enums.tools import NotificationTool
-from tests.test_data.notification_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import NotificationTool
+from codemie_test_harness.tests.test_data.notification_tools_test_data import (
     EMAIL_SUBJECT,
     EMAIL_BODY,
     EMAIL_RESPONSE,

{tests → codemie_test_harness/tests}/test_data/direct_tools/open_api_tools_test_data.py

@@ -1,6 +1,6 @@
 import os
 
-from tests.enums.tools import Toolkit, OpenApiTool
+from codemie_test_harness.tests.enums.tools import Toolkit, OpenApiTool
 
 # OpenAPI tools test data for direct tools calling
 open_api_tools_test_data = [

{tests → codemie_test_harness/tests}/test_data/direct_tools/project_management_tools_test_data.py

@@ -1,5 +1,5 @@
-from tests.enums.tools import Toolkit, ProjectManagementTool
-from tests.utils.constants import ProjectManagementIntegrationType
+from codemie_test_harness.tests.enums.tools import Toolkit, ProjectManagementTool
+from codemie_test_harness.tests.utils.constants import ProjectManagementIntegrationType
 
 project_management_tools_data = [
     (