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

codemie_test_harness/.env

@@ -0,0 +1,5 @@
+PROJECT_NAME=codemie
+DEFAULT_TIMEOUT=60
+CLEANUP_DATA=True
+LANGFUSE_TRACES_ENABLED=False
+VERIFY_SSL=False

codemie_test_harness/cli/runner.py

@@ -12,32 +12,41 @@ except Exception:
 from .constants import CONSOLE
 
 
-def _resolve_tests_path() -> str:
-    """Resolve absolute path to the installed tests package.
+def _resolve_tests_path_and_root() -> tuple[str, str]:
+    """Resolve tests path and working directory.
 
-    Priority:
-    1) importlib.resources.files("tests") if the package is installed
-    2) Fallback to repo layout: <repo_root>/tests relative to this file
+    Returns:
+        tuple: (tests_path, root_dir)
+            - tests_path: Path to tests for pytest collection
+            - root_dir: Directory to use as cwd so pytest.ini/.env are found
     """
     # 1) Use importlib.resources to locate installed package "tests"
     if pkg_files is not None:
         try:
-            tests_dir = pkg_files("tests")
-            # Convert to a concrete filesystem path string
-            return str(Path(tests_dir.joinpath("")))
+            # Try to locate codemie_test_harness.tests package
+            tests_dir = Path(str(pkg_files("codemie_test_harness.tests")))
+            # Root dir should be the codemie_test_harness package directory
+            # where pytest.ini and .env are located
+            root_dir = str(tests_dir.parent)  # codemie_test_harness package root
+            return str(tests_dir), root_dir
         except Exception:
             pass
 
     # 2) Fallback to repo layout when running from source
     # runner.py -> cli -> codemie_test_harness -> <repo_root>
-    repo_root = Path(__file__).resolve().parents[2]
-    return str(repo_root / "tests")
+    codemie_test_harness_root = (
+        Path(__file__).resolve().parents[1]
+    )  # codemie_test_harness directory
+    tests_path = str(codemie_test_harness_root / "tests")
+    # Use codemie_test_harness as root_dir since pytest.ini and .env are there
+    root_dir = str(codemie_test_harness_root)
+    return tests_path, root_dir
 
 
 def build_pytest_cmd(
     workers: int, marks: str, reruns: int, extra: Iterable[str] | None = None
-) -> List[str]:
-    tests_path = _resolve_tests_path()
+) -> tuple[List[str], str]:
+    tests_path, root_dir = _resolve_tests_path_and_root()
     cmd = [sys.executable, "-m", "pytest", tests_path]
     if workers:
         cmd += ["-n", str(workers)]
@@ -47,12 +56,12 @@ def build_pytest_cmd(
         cmd += ["--reruns", str(reruns)]
     if extra:
         cmd += list(extra)
-    return cmd
+    return cmd, root_dir
 
 
 def run_pytest(
     workers: int, marks: str, reruns: int, extra: Iterable[str] | None = None
 ) -> None:
-    cmd = build_pytest_cmd(workers, marks, reruns, extra)
-    CONSOLE.print("[cyan]Running:[/] " + " ".join(cmd))
-    raise SystemExit(subprocess.call(cmd))
+    cmd, root_dir = build_pytest_cmd(workers, marks, reruns, extra)
+    CONSOLE.print(f"[cyan]Running:[/] {' '.join(cmd)} (cwd={root_dir})")
+    raise SystemExit(subprocess.call(cmd, cwd=root_dir))

codemie_test_harness/pytest.ini

@@ -0,0 +1,6 @@
+[pytest]
+addopts = -v
+filterwarnings =
+    ignore::pytest.PytestUnknownMarkWarning
+    ignore::urllib3.exceptions.InsecureRequestWarning
+    ignore::pydantic.warnings.PydanticDeprecatedSince20

{tests → codemie_test_harness/tests}/__init__.py

@@ -5,7 +5,7 @@ from pathlib import Path
 
 from dotenv import load_dotenv
 
-from tests.utils.aws_parameters_store import AwsParameterStore
+from codemie_test_harness.tests.utils.aws_parameters_store import AwsParameterStore
 
 load_dotenv()
 

{tests → codemie_test_harness/tests}/assistant/datasource/test_code_datasource.py

@@ -4,8 +4,8 @@ import pytest
 from hamcrest import assert_that, equal_to
 from requests import HTTPError
 
-from tests.test_data.index_test_data import index_test_data
-from tests.utils.base_utils import get_random_name, assert_response
+from codemie_test_harness.tests.test_data.index_test_data import index_test_data
+from codemie_test_harness.tests.utils.base_utils import get_random_name, assert_response
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/assistant/datasource/test_confluence_datasource.py

@@ -5,11 +5,11 @@ import pytest
 from hamcrest import assert_that, equal_to
 from requests import HTTPError
 
-from tests.test_data.pm_tools_test_data import (
+from codemie_test_harness.tests.test_data.pm_tools_test_data import (
     CONFLUENCE_TOOL_PROMPT,
     RESPONSE_FOR_CONFLUENCE_TOOL,
 )
-from tests.utils.base_utils import get_random_name, assert_response
+from codemie_test_harness.tests.utils.base_utils import get_random_name, assert_response
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/assistant/datasource/test_file_indexing.py

@@ -6,16 +6,16 @@ from hamcrest import (
 )
 from requests import HTTPError
 
-from tests.test_data.file_test_data import (
+from codemie_test_harness.tests.test_data.file_test_data import (
     file_test_data,
     large_files_test_data,
     RESPONSE_FOR_TWO_FILES_INDEXED,
 )
-from tests.utils.base_utils import get_random_name, assert_response
-from tests.test_data.index_test_data import index_test_data
-from tests.utils.client_factory import get_client
-from tests.utils.constants import FILES_PATH
-from tests.utils.pytest_utils import check_mark
+from codemie_test_harness.tests.utils.base_utils import get_random_name, assert_response
+from codemie_test_harness.tests.test_data.index_test_data import index_test_data
+from codemie_test_harness.tests.utils.client_factory import get_client
+from codemie_test_harness.tests.utils.constants import FILES_PATH
+from codemie_test_harness.tests.utils.pytest_utils import check_mark
 
 
 def pytest_generate_tests(metafunc):

{tests → codemie_test_harness/tests}/assistant/datasource/test_google_datasource.py

@@ -4,12 +4,12 @@ import pytest
 from hamcrest import assert_that, equal_to
 from requests import HTTPError
 
-from tests.test_data.google_datasource_test_data import (
+from codemie_test_harness.tests.test_data.google_datasource_test_data import (
     GOOGLE_DOC_URL,
     RESPONSE_FOR_GOOGLE_DOC,
     USER_PROMPT,
 )
-from tests.utils.base_utils import get_random_name, assert_response
+from codemie_test_harness.tests.utils.base_utils import get_random_name, assert_response
 
 
 @pytest.mark.smoke

{tests → codemie_test_harness/tests}/assistant/datasource/test_jira_datasource.py

@@ -5,8 +5,11 @@ import pytest
 from hamcrest import assert_that, equal_to
 from requests import HTTPError
 
-from tests.test_data.pm_tools_test_data import JIRA_TOOL_PROMPT, RESPONSE_FOR_JIRA_TOOL
-from tests.utils.base_utils import get_random_name, assert_response
+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.base_utils import get_random_name, assert_response
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/assistant/default_integrations/test_default_integrations_for_tool.py

@@ -4,12 +4,15 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.enums.tools import Toolkit, ProjectManagementTool
-from tests.integrations import integrations
-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 import PROJECT
+from codemie_test_harness.tests.enums.tools import Toolkit, ProjectManagementTool
+from codemie_test_harness.tests.integrations import integrations
+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
 
 
 @pytest.mark.not_for_parallel_run

{tests → codemie_test_harness/tests}/assistant/default_integrations/test_default_integrations_for_tool_kit.py

@@ -4,12 +4,14 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import Toolkit, 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.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 Toolkit, 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.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}/assistant/default_integrations/test_default_integrations_for_tool_with_datasource.py

@@ -4,14 +4,14 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import Toolkit, 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 Toolkit, 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.constants import test_project_name
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 
 git_tool_prompt = list_branches_set_active_branch_test_data[0][2]

{tests → codemie_test_harness/tests}/assistant/test_assistants.py

@@ -13,16 +13,16 @@ from codemie_sdk.models.assistant import (
     ChatMessage,
     ChatRole,
 )
-from tests import TEST_USER, PROJECT, LANGFUSE_TRACES_ENABLED
-from tests.test_data.file_test_data import (
+from codemie_test_harness.tests import TEST_USER, PROJECT, LANGFUSE_TRACES_ENABLED
+from codemie_test_harness.tests.test_data.file_test_data import (
     files_with_different_types_test_data,
     file_test_data,
     RESPONSE_FOR_TWO_FILES_UPLOADED,
 )
-from tests.test_data.output_schema_test_data import output_schema
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import FILES_PATH
-from tests.utils.file_utils import save_file
+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.constants import FILES_PATH
+from codemie_test_harness.tests.utils.file_utils import save_file
 
 system_prompt_for_chatting_with_files = """Your task is to show the content of the file. You MUST NOT:
     - analyze the content

{tests → codemie_test_harness/tests}/assistant/tools/ado/test_assistant_for_ado_test_plan_tools.py

@@ -1,8 +1,8 @@
 import pytest
 from codemie_sdk.models.integration import CredentialTypes
 
-from tests.enums.tools import Toolkit, AzureDevOpsTestPlanTool
-from tests.test_data.ado_test_plan_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit, 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,
@@ -10,10 +10,10 @@ from tests.test_data.ado_test_plan_tools_test_data import (
     ADO_TEST_PLAN_DELETE_SUITE,
     ADO_TEST_PLAN_DELETE_PLAN,
 )
-from tests.utils.aws_parameters_store import CredentialsUtil
-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.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 ID_PATTERN
+from codemie_test_harness.tests.utils.json_utils import extract_id_from_ado_response
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/assistant/tools/ado/test_assistant_for_ado_wiki_tools.py

@@ -1,16 +1,16 @@
 import pytest
 
-from tests.enums.tools import Toolkit, AzureDevOpsWikiTool
+from codemie_test_harness.tests.enums.tools import Toolkit, AzureDevOpsWikiTool
 from codemie_sdk.models.integration import CredentialTypes
-from tests.test_data.ado_wiki_tools_test_data import (
+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.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}/assistant/tools/ado/test_assistant_for_ado_work_item_tools.py

@@ -1,17 +1,17 @@
 import pytest
 from codemie_sdk.models.integration import CredentialTypes
 
-from tests.enums.tools import Toolkit, AzureDevOpsWorkItemTool
-from tests.test_data.ado_work_item_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit, 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.aws_parameters_store import CredentialsUtil
-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.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 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}/assistant/tools/cloud/test_cloud_tools.py

@@ -1,7 +1,7 @@
 import pytest
 
-from tests.enums.tools import Toolkit, CloudTool
-from tests.test_data.cloud_tools_test_data import cloud_test_data
+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
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/assistant/tools/codebase/test_codebase_tools.py

@@ -1,10 +1,10 @@
 import pytest
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests.enums.tools import (
+from codemie_test_harness.tests.enums.tools import (
     CodeBaseTool,
 )
-from tests.test_data.codebase_tools_test_data import (
+from codemie_test_harness.tests.test_data.codebase_tools_test_data import (
     code_tools_test_data,
     sonar_tools_test_data,
 )

{tests → codemie_test_harness/tests}/assistant/tools/datamanagement/test_assistant_with_data_management_tools.py

@@ -4,11 +4,11 @@ import uuid
 import pytest
 from codemie_sdk.models.integration import CredentialTypes
 
-from tests.enums.tools import (
+from codemie_test_harness.tests.enums.tools import (
     Toolkit,
     DataManagementTool,
 )
-from tests.test_data.data_management_tools_test_data import (
+from codemie_test_harness.tests.test_data.data_management_tools_test_data import (
     ELASTIC_TOOL_TASK,
     RESPONSE_FOR_ELASTIC,
     sql_tools_test_data,
@@ -18,7 +18,7 @@ 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 codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/assistant/tools/filemanagement/test_assistant_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 Toolkit, FileManagementTool
-from tests.test_data.file_management_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit, 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,
@@ -16,7 +16,7 @@ from tests.test_data.file_management_tools_test_data import (
     file_editing_tools_test_data,
     show_diff_task,
 )
-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}/assistant/tools/git/test_assistant_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

{tests → codemie_test_harness/tests}/assistant/tools/mcp/test_cli_mcp_server.py

@@ -6,10 +6,10 @@ import pytest
 from hamcrest import assert_that, contains_string
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests import autotest_entity_prefix
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.base_utils import get_random_name
-from tests.test_data.mcp_server_test_data import (
+from codemie_test_harness.tests import autotest_entity_prefix
+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.test_data.mcp_server_test_data import (
     cli_mcp_server_test_data,
     CLI_MCP_SERVER,
 )

{tests → codemie_test_harness/tests}/assistant/tools/mcp/test_mcp_servers.py

@@ -2,7 +2,7 @@ import os
 
 import pytest
 
-from tests.test_data.mcp_server_test_data import (
+from codemie_test_harness.tests.test_data.mcp_server_test_data import (
     time_mcp_server_test_data,
     time_expected_response,
     FETCH_MCP_SERVER,

{tests → codemie_test_harness/tests}/assistant/tools/notification/test_assistant_notification_tools.py

@@ -4,9 +4,9 @@ import pytest
 from hamcrest import assert_that, equal_to
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests.enums.tools import Toolkit, NotificationTool
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.test_data.notification_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit, NotificationTool
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.test_data.notification_tools_test_data import (
     EMAIL_TOOL_PROMPT,
     EMAIL_RESPONSE,
     EMAIL_SUBJECT,

{tests → codemie_test_harness/tests}/assistant/tools/openapi/test_assistant_with_open_api_tools.py

@@ -2,8 +2,8 @@ import os
 
 import pytest
 
-from tests.enums.tools import Toolkit
-from tests.test_data.open_api_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit
+from codemie_test_harness.tests.test_data.open_api_tools_test_data import (
     open_api_tools_test_data,
 )
 

{tests → codemie_test_harness/tests}/assistant/tools/plugin/test_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}/assistant/tools/plugin/test_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}/assistant/tools/project_management/test_assistant_pm_tools.py

@@ -2,18 +2,20 @@ import pytest
 
 from codemie_sdk.models.assistant import ToolConfig
 from codemie_sdk.models.integration import CredentialTypes
-from tests.enums.tools import Toolkit, ProjectManagementTool
-from tests.test_data.pm_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit, ProjectManagementTool
+from codemie_test_harness.tests.test_data.pm_tools_test_data import (
     CONFLUENCE_TOOL_PROMPT,
     RESPONSE_FOR_CONFLUENCE_TOOL,
     JIRA_TOOL_PROMPT,
     RESPONSE_FOR_JIRA_TOOL,
     RESPONSE_FOR_CONFLUENCE_TOOL_UNAUTHORIZED,
 )
-from tests.test_data.project_management_test_data import pm_tools_test_data
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.base_utils import credentials_to_dict
-from tests.utils.constants import (
+from codemie_test_harness.tests.test_data.project_management_test_data import (
+    pm_tools_test_data,
+)
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.base_utils import credentials_to_dict
+from codemie_test_harness.tests.utils.constants import (
     project_management_integrations,
 )
 

{tests → codemie_test_harness/tests}/assistant/tools/research/test_assistant_research_tools.py

@@ -1,12 +1,12 @@
 import pytest
 from hamcrest import assert_that, greater_than
 
-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}/assistant/tools/servicenow/test_servicenow_tools.py

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

{tests → codemie_test_harness/tests}/assistant/tools/vcs/test_assistant_with_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}/conftest.py

@@ -22,23 +22,25 @@ from codemie_sdk.models.integration import (
     Integration,
 )
 from codemie_sdk.models.workflow import WorkflowCreateRequest, WorkflowMode, Workflow
-from tests import PROJECT, autotest_entity_prefix
-from tests.test_data.google_datasource_test_data import GOOGLE_DOC_URL
-from tests.utils.assistant_utils import AssistantUtils
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.base_utils import get_random_name, wait_for_entity
-from tests.utils.client_factory import get_client
-from tests.utils.constants import TESTS_PATH
-from tests.utils.datasource_utils import DataSourceUtils
-from tests.utils.gitbud_utils import GitBudUtils
-from tests.utils.integration_utils import IntegrationUtils
-from tests.utils.logger_util import setup_logger
-from tests.utils.notification_utils import GmailUtils
-from tests.utils.provider_utils import ProviderUtils
-from tests.utils.search_utils import SearchUtils
-from tests.utils.similarity_check import SimilarityCheck
-from tests.utils.workflow_utils import WorkflowUtils
-from tests.utils.yaml_utils import (
+from codemie_test_harness.tests import PROJECT, autotest_entity_prefix
+from codemie_test_harness.tests.test_data.google_datasource_test_data import (
+    GOOGLE_DOC_URL,
+)
+from codemie_test_harness.tests.utils.assistant_utils import AssistantUtils
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.base_utils import get_random_name, wait_for_entity
+from codemie_test_harness.tests.utils.client_factory import get_client
+from codemie_test_harness.tests.utils.constants import TESTS_PATH
+from codemie_test_harness.tests.utils.datasource_utils import DataSourceUtils
+from codemie_test_harness.tests.utils.gitbud_utils import GitBudUtils
+from codemie_test_harness.tests.utils.integration_utils import IntegrationUtils
+from codemie_test_harness.tests.utils.logger_util import setup_logger
+from codemie_test_harness.tests.utils.notification_utils import GmailUtils
+from codemie_test_harness.tests.utils.provider_utils import ProviderUtils
+from codemie_test_harness.tests.utils.search_utils import SearchUtils
+from codemie_test_harness.tests.utils.similarity_check import SimilarityCheck
+from codemie_test_harness.tests.utils.workflow_utils import WorkflowUtils
+from codemie_test_harness.tests.utils.yaml_utils import (
     WorkflowYamlModel,
     AssistantModel,
     ToolModel,