codemie-test-harness 0.1.130__py3-none-any.whl → 0.1.132__py3-none-any.whl

{tests → codemie_test_harness/tests}/workflow/assistant_tools/plugin/test_workflow_with_assistant_with_development_plugin.py

@@ -2,13 +2,13 @@ import os
 
 import pytest
 
-from tests.enums.tools import Toolkit
-from tests.test_data.plugin_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit
+from codemie_test_harness.tests.test_data.plugin_tools_test_data import (
     list_files_plugin_tools_test_data,
     CREATE_READ_DELETE_FILE_TEST_DATA,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import TESTS_PATH
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import TESTS_PATH
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/plugin/test_workflow_with_assistant_with_plugin_and_mcp_servers.py

@@ -2,12 +2,12 @@ import os
 
 import pytest
 
-from tests.enums.tools import Toolkit
-from tests.test_data.mcp_server_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit
+from codemie_test_harness.tests.test_data.mcp_server_test_data import (
     cli_mcp_server_with_plugin_test_data,
     filesystem_mcp_server_with_plugin_test_data,
 )
-from tests.utils.constants import TESTS_PATH
+from codemie_test_harness.tests.utils.constants import TESTS_PATH
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/project_management/test_workflow_with_assistant_pm_tools.py

@@ -1,8 +1,10 @@
 import pytest
 
-from tests.enums.tools import Toolkit
-from tests.test_data.project_management_test_data import pm_tools_test_data
-from tests.utils.constants import (
+from codemie_test_harness.tests.enums.tools import Toolkit
+from codemie_test_harness.tests.test_data.project_management_test_data import (
+    pm_tools_test_data,
+)
+from codemie_test_harness.tests.utils.constants import (
     project_management_integrations,
 )
 

{tests → codemie_test_harness/tests}/workflow/assistant_tools/research/test_workflow_with_assistant_research_tools.py

@@ -1,12 +1,12 @@
 import pytest
 from hamcrest import assert_that, greater_than_or_equal_to
 
-from tests.enums.tools import Toolkit, ResearchToolName
-from tests.test_data.research_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit, ResearchToolName
+from codemie_test_harness.tests.test_data.research_tools_test_data import (
     search_tools_test_data,
     interactions_tools_test_data,
 )
-from tests.utils.base_utils import percent_of_relevant_titles
+from codemie_test_harness.tests.utils.base_utils import percent_of_relevant_titles
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/servicenow/test_workflow_with_servicenow_tools.py

@@ -1,7 +1,10 @@
 import pytest
 
-from tests.enums.tools import ServiceNowTool, Toolkit
-from tests.test_data.servicenow_tools_test_data import PROMPT, EXPECTED_RESPONSE
+from codemie_test_harness.tests.enums.tools import ServiceNowTool, Toolkit
+from codemie_test_harness.tests.test_data.servicenow_tools_test_data import (
+    PROMPT,
+    EXPECTED_RESPONSE,
+)
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/vcs/workflow_with_assistant_vcs_tools.py

@@ -1,12 +1,12 @@
 import pytest
 
-from tests.enums.tools import (
+from codemie_test_harness.tests.enums.tools import (
     Toolkit,
 )
-from tests.test_data.vcs_tools_test_data import (
+from codemie_test_harness.tests.test_data.vcs_tools_test_data import (
     vcs_tools_test_data,
 )
-from tests.utils.constants import vcs_integrations
+from codemie_test_harness.tests.utils.constants import vcs_integrations
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/config_validation/test_config_validation.py

@@ -4,8 +4,8 @@ import pytest
 from hamcrest import assert_that
 
 from codemie_sdk.models.workflow import WorkflowCreateRequest
-from tests import PROJECT
-from tests.test_data.workflow_validation_messages import (
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.test_data.workflow_validation_messages import (
     ASSISTANT_NOT_EXIST,
     ASSISTANT_NOT_EXIST_IN_STATE,
     INVALID_YAML,
@@ -29,7 +29,7 @@ from tests.test_data.workflow_validation_messages import (
     MISSING_STATES_NEXT_SWITCH_CASES_STATE_ID,
     INVALID_YAML_FORMAT_PROVIDED,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 # Map of test yaml file names to expected error messages (as in Java test)
 VALIDATION_TEST_DATA = [

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/default_integrations/test_default_integrations_for_tool.py

@@ -6,15 +6,15 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import ProjectManagementTool
-from tests.test_data.direct_tools.project_management_tools_test_data import (
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.integrations import integrations
+from codemie_test_harness.tests.enums.tools import ProjectManagementTool
+from codemie_test_harness.tests.test_data.direct_tools.project_management_tools_test_data import (
     project_management_tools_data,
 )
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 
 jira_tool_prompt = project_management_tools_data[0][2]

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/default_integrations/test_default_integrations_for_tool_kit.py

@@ -8,13 +8,15 @@ from codemie_sdk.models.integration import (
 )
 from hamcrest import equal_to, assert_that
 
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import AzureDevOpsWikiTool
-from tests.test_data.direct_tools.ado_wiki_tools_test_data import ado_wiki_get_test_data
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.integrations import integrations
+from codemie_test_harness.tests.enums.tools import AzureDevOpsWikiTool
+from codemie_test_harness.tests.test_data.direct_tools.ado_wiki_tools_test_data import (
+    ado_wiki_get_test_data,
+)
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 ado_wiki_prompt = ado_wiki_get_test_data[0][2]
 ado_wiki_answer = ado_wiki_get_test_data[0][3]

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_ado_test_plan_tools.py

@@ -4,10 +4,10 @@ import random
 
 import pytest
 
-from tests.test_data.direct_tools.ado_test_plan_tools_test_data import (
+from codemie_test_harness.tests.test_data.direct_tools.ado_test_plan_tools_test_data import (
     ado_test_plan_get_test_data,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_ado_wiki_tools.py

@@ -4,8 +4,10 @@ import random
 
 import pytest
 
-from tests.test_data.direct_tools.ado_wiki_tools_test_data import ado_wiki_get_test_data
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.test_data.direct_tools.ado_wiki_tools_test_data import (
+    ado_wiki_get_test_data,
+)
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_ado_work_item_tools.py

@@ -4,10 +4,10 @@ import random
 
 import pytest
 
-from tests.test_data.direct_tools.ado_work_item_tools_test_data import (
+from codemie_test_harness.tests.test_data.direct_tools.ado_work_item_tools_test_data import (
     ado_work_item_get_test_data,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_cloud_tools.py

@@ -4,9 +4,11 @@ import random
 
 import pytest
 
-from tests.enums.tools import Toolkit, CloudTool
-from tests.test_data.direct_tools.cloud_tools_test_data import cloud_test_data
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.enums.tools import Toolkit, CloudTool
+from codemie_test_harness.tests.test_data.direct_tools.cloud_tools_test_data import (
+    cloud_test_data,
+)
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_codebase_tools.py

@@ -5,12 +5,12 @@ import random
 import pytest
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests.enums.tools import CodeBaseTool
-from tests.test_data.direct_tools.codebase_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import CodeBaseTool
+from codemie_test_harness.tests.test_data.direct_tools.codebase_tools_test_data import (
     sonar_tools_test_data,
 )
-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
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_data_management_tools.py

@@ -7,11 +7,11 @@ import pytest
 from hamcrest import assert_that, contains_inanyorder
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests.test_data.direct_tools.data_management_tools_test_data import (
+from codemie_test_harness.tests.test_data.direct_tools.data_management_tools_test_data import (
     sql_tools_test_data,
 )
-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
 
 pytestmark = pytest.mark.skipif(
     os.getenv("ENV") == "local", reason="Skipping this tests on local environment"

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_file_management_tools.py

@@ -3,10 +3,10 @@ import json
 
 import pytest
 
-from tests.test_data.direct_tools.file_management_tools_test_data import (
+from codemie_test_harness.tests.test_data.direct_tools.file_management_tools_test_data import (
     file_management_tools_test_data,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_notification_tools.py

@@ -3,12 +3,12 @@ import json
 
 import pytest
 
-from tests.enums.tools import NotificationTool
-from tests.test_data.direct_tools.notification_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import NotificationTool
+from codemie_test_harness.tests.test_data.direct_tools.notification_tools_test_data import (
     notification_tools_test_data,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import notification_integrations
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import notification_integrations
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_open_api_tools.py

@@ -5,10 +5,10 @@ import random
 
 import pytest
 
-from tests.test_data.direct_tools.open_api_tools_test_data import (
+from codemie_test_harness.tests.test_data.direct_tools.open_api_tools_test_data import (
     open_api_tools_test_data,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 pytestmark = pytest.mark.skipif(
     os.getenv("ENV") == "azure", reason="Still have an issue with encoding long strings"

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_plugin_tools.py

@@ -3,12 +3,12 @@ import os
 
 import pytest
 
-from tests.test_data.direct_tools.direct_tools_test_data import (
+from codemie_test_harness.tests.test_data.direct_tools.direct_tools_test_data import (
     cli_tools_test_data,
     dev_plugin_tools_test_data,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import TESTS_PATH
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import TESTS_PATH
 
 
 @pytest.fixture(scope="session", autouse=True)

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_project_management_tools.py

@@ -4,11 +4,11 @@ import random
 
 import pytest
 
-from tests.test_data.direct_tools.project_management_tools_test_data import (
+from codemie_test_harness.tests.test_data.direct_tools.project_management_tools_test_data import (
     project_management_tools_data,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import project_management_integrations
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import project_management_integrations
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_research_tools.py

@@ -4,10 +4,10 @@ import random
 
 import pytest
 
-from tests.test_data.direct_tools.research_tools_test_data import (
+from codemie_test_harness.tests.test_data.direct_tools.research_tools_test_data import (
     research_tools_test_data,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_servicenow_tools.py

@@ -6,12 +6,12 @@ import copy
 
 from hamcrest import equal_to, assert_that
 
-from tests.enums.tools import ServiceNowTool
-from tests.test_data.direct_tools.servicenow_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import ServiceNowTool
+from codemie_test_harness.tests.test_data.direct_tools.servicenow_tools_test_data import (
     EXPECTED_RESPONSE,
     PROMPT,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/direct_tools_calling/test_workflow_with_vcs_tools.py

@@ -4,11 +4,11 @@ import random
 
 import pytest
 
-from tests.test_data.direct_tools.vcs_tools_test_data import (
+from codemie_test_harness.tests.test_data.direct_tools.vcs_tools_test_data import (
     vcs_tools_test_data,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import vcs_integrations
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import vcs_integrations
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/test_workflows.py

@@ -3,9 +3,9 @@ import json
 import pytest
 from hamcrest import assert_that, equal_to
 
-from tests.test_data.output_schema_test_data import output_schema
-from tests.utils.base_utils import get_random_name
-from tests.utils.yaml_utils import AssistantModel, StateModel
+from codemie_test_harness.tests.test_data.output_schema_test_data import output_schema
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.yaml_utils import AssistantModel, StateModel
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/ado/test_workflow_with_ado_test_plan_tools.py

@@ -1,7 +1,7 @@
 import pytest
 
-from tests.enums.tools import AzureDevOpsTestPlanTool
-from tests.test_data.ado_test_plan_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import AzureDevOpsTestPlanTool
+from codemie_test_harness.tests.test_data.ado_test_plan_tools_test_data import (
     ado_test_plan_get_test_data,
     ADO_TEST_PLAN_CREATE_TEST_PLAN,
     ADO_TEST_PLAN_CREATE_SUITE,
@@ -9,9 +9,9 @@ from tests.test_data.ado_test_plan_tools_test_data import (
     ADO_TEST_PLAN_DELETE_SUITE,
     ADO_TEST_PLAN_DELETE_PLAN,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import ID_PATTERN
-from tests.utils.json_utils import extract_id_from_ado_response
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import ID_PATTERN
+from codemie_test_harness.tests.utils.json_utils import extract_id_from_ado_response
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/ado/test_workflow_with_ado_wiki_tools.py

@@ -1,14 +1,14 @@
 import pytest
 
-from tests.enums.tools import AzureDevOpsWikiTool
-from tests.test_data.ado_wiki_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import AzureDevOpsWikiTool
+from codemie_test_harness.tests.test_data.ado_wiki_tools_test_data import (
     ado_wiki_get_test_data,
     ADO_WIKI_CREATE_PAGE,
     ADO_WIKI_RENAME_PAGE,
     ADO_WIKI_MODIFY_PAGE,
     ADO_WIKI_DELETE_PAGE,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/ado/test_workflow_with_ado_work_item_tools.py

@@ -1,15 +1,15 @@
 import pytest
 
-from tests.enums.tools import AzureDevOpsWorkItemTool
-from tests.test_data.ado_work_item_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import AzureDevOpsWorkItemTool
+from codemie_test_harness.tests.test_data.ado_work_item_tools_test_data import (
     ado_work_item_get_test_data,
     ADO_WORK_ITEM_CREATE,
     ADO_WORK_ITEM_UPDATE,
     ADO_WORK_ITEM_LINK,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import WORK_ITEM_ID_PATTERN
-from tests.utils.json_utils import extract_id_from_ado_response
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import WORK_ITEM_ID_PATTERN
+from codemie_test_harness.tests.utils.json_utils import extract_id_from_ado_response
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/cloud/test_workflow_with_cloud_tools.py

@@ -1,8 +1,8 @@
 import pytest
 
-from tests.enums.tools import Toolkit, CloudTool
-from tests.test_data.cloud_tools_test_data import cloud_test_data
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.enums.tools import Toolkit, CloudTool
+from codemie_test_harness.tests.test_data.cloud_tools_test_data import cloud_test_data
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/codebase/test_workflow_with_codebase_tools.py

@@ -1,13 +1,13 @@
 import pytest
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests.enums.tools import CodeBaseTool
-from tests.test_data.codebase_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import CodeBaseTool
+from codemie_test_harness.tests.test_data.codebase_tools_test_data import (
     code_tools_test_data,
     sonar_tools_test_data,
 )
-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
 
 
 @pytest.fixture(scope="module")

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/data_management/test_workflow_with_data_management_tools.py

@@ -3,8 +3,8 @@ import os
 import pytest
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests.enums.tools import DataManagementTool
-from tests.test_data.data_management_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import DataManagementTool
+from codemie_test_harness.tests.test_data.data_management_tools_test_data import (
     ELASTIC_TOOL_TASK,
     RESPONSE_FOR_ELASTIC,
     sql_tools_test_data,
@@ -14,8 +14,8 @@ from tests.test_data.data_management_tools_test_data import (
     SQL_TOOL_QUERY_TABLE_TASK,
     RESPONSE_FOR_SQL,
 )
-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
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/default_integrations/test_default_integrations_for_tool.py

@@ -4,13 +4,16 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import ProjectManagementTool
-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.base_utils import get_random_name
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.integrations import integrations
+from codemie_test_harness.tests.enums.tools import ProjectManagementTool
+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.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 
 @pytest.mark.not_for_parallel_run

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/default_integrations/test_default_integrations_for_tool_kit.py

@@ -4,13 +4,15 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import AzureDevOpsWikiTool
-from tests.test_data.ado_wiki_tools_test_data import ado_wiki_get_test_data
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.integrations import integrations
+from codemie_test_harness.tests.enums.tools import AzureDevOpsWikiTool
+from codemie_test_harness.tests.test_data.ado_wiki_tools_test_data import (
+    ado_wiki_get_test_data,
+)
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 ado_wiki_prompt = ado_wiki_get_test_data[0][2]
 ado_wiki_answer = ado_wiki_get_test_data[0][3]

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/default_integrations/test_default_integrations_for_tool_with_datasource.py

@@ -4,15 +4,15 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import GitTool
-from tests.test_data.git_tools_test_data import (
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.integrations import integrations
+from codemie_test_harness.tests.enums.tools import GitTool
+from codemie_test_harness.tests.test_data.git_tools_test_data import (
     list_branches_set_active_branch_test_data,
 )
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 git_tool_prompt = list_branches_set_active_branch_test_data[0][2]
 git_tool_answer = list_branches_set_active_branch_test_data[0][3]

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/file_management/test_workflow_with_file_management_tools.py

@@ -3,8 +3,8 @@ import os
 import pytest
 from hamcrest import assert_that, contains_string, is_not, all_of
 
-from tests.enums.tools import FileManagementTool
-from tests.test_data.file_management_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import FileManagementTool
+from codemie_test_harness.tests.test_data.file_management_tools_test_data import (
     file_management_tools_test_data,
     GENERATE_IMAGE_TOOL_TASK,
     WRITE_FILE_TASK,
@@ -17,7 +17,7 @@ from tests.test_data.file_management_tools_test_data import (
     RESPONSE_FOR_DIFF_UPDATE,
     RESPONSE_FOR_FILE_EDITOR,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/virtual_assistant_tools/git/test_workflow_with_git_tools.py

@@ -1,7 +1,7 @@
 import pytest
 from hamcrest import assert_that, equal_to, not_none
 
-from tests.test_data.git_tools_test_data import (
+from codemie_test_harness.tests.test_data.git_tools_test_data import (
     list_branches_set_active_branch_test_data,
     create_branch_test_data,
     create_file_test_data,
@@ -10,7 +10,7 @@ from tests.test_data.git_tools_test_data import (
     get_merge_request_changes_test_data,
     update_file_test_data,
 )
-from tests.utils.base_utils import get_random_name, to_camel_case
+from codemie_test_harness.tests.utils.base_utils import get_random_name, to_camel_case
 
 
 @pytest.mark.regression