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

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

@@ -1,6 +1,6 @@
 import pytest
 
-from tests.enums.tools import ResearchToolName
+from codemie_test_harness.tests.enums.tools import ResearchToolName
 
 research_tools_test_data = [
     (

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

@@ -1,4 +1,4 @@
-from tests.enums.tools import VcsTool, Toolkit
+from codemie_test_harness.tests.enums.tools import VcsTool, Toolkit
 
 
 vcs_tools_test_data = [

{tests → codemie_test_harness/tests}/test_data/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
 
 CODE_INTERPRETER_TOOL_TASK = """
 execute:

codemie_test_harness/tests/test_data/files/provider_payload.json

@@ -0,0 +1,382 @@
+   {
+      "id":"3928ea28-e9e2-4382-aae0-09d5a24dc2am",
+      "date":"2025-04-02T12:54:19.693921",
+      "update_date":"2025-04-02T13:50:36.917335",
+      "name":"test",
+      "service_location_url":"https://codemie-preview.lab.epam.com/codemie-provider-stub-api/",
+      "configuration":{
+         "auth_type":"Bearer"
+      },
+      "provided_toolkits":[
+         {
+            "toolkit_id":"023953e9-b4b2-4264-aa24-7a20ab7cb201",
+            "name":"TestAutomationCodeAnalysesToolkit",
+            "description":"This ToolKit provides tools for indexing the source code of provided repository and set of methods to get insights of the code and code snippets.",
+            "toolkit_config":{
+               "type":"Code Analyses Datasource Configuration",
+               "description":"Configuration for connecting to GitHub repositories to access code for indexing and analysis.",
+               "parameters":{
+                  "access_token":{
+                     "description":"Github/Gitlab project access token with appropriate scopes for repository access",
+                     "type":"Secret",
+                     "required":true,
+                     "enum":null
+                  },
+                  "datasource_id":{
+                     "description":"Unique identifier for the datasource",
+                     "type":"UUID",
+                     "required":true,
+                     "enum":null
+                  },
+                  "api_url":{
+                     "description":"Git API URL (use your Git Enterprise API URL)",
+                     "type":"URL",
+                     "required":true,
+                     "enum":null
+                  },
+                  "branch":{
+                     "description":"Branch to index, defaults to master",
+                     "type":"String",
+                     "required":false,
+                     "enum":null
+                  }
+               }
+            },
+            "provided_tools":[
+               {
+                  "name":"create_datasource",
+                  "description":"Creates a new datasource by indexing a Git repository for code analysis. The indexing process extracts code structure, dependencies, and other metadata to enable subsequent queries.",
+                  "args_schema":{
+                     "exclude_glob":{
+                        "type":"String",
+                        "required":false,
+                        "description":"Comma separated string of file paths to exclude from indexing",
+                        "enum":null
+                     },
+                     "analyzer":{
+                        "type":"List",
+                        "required":false,
+                        "description":"Type of code analyzer to use for the project. Supported analyzers Java/TS/JS/C#/C++. If not specified best matching analyzer will be chosen automatically.",
+                        "enum":[
+                           "Java",
+                           "TS",
+                           "JS",
+                           "C#",
+                           "C++",
+                           "Other"
+                        ]
+                     },
+                     "datasource_root":{
+                        "type":"String",
+                        "required":true,
+                        "description":"Root directory of the project to be indexed",
+                        "enum":null
+                     }
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"life_cycle_management",
+                     "tool_action_type":"create"
+                  },
+                  "tool_result_type":"Json",
+                  "sync_invocation_supported":false,
+                  "async_invocation_supported":true
+               },
+               {
+                  "name":"reindex_datasource",
+                  "description":"Updates an existing datasource by re-indexing the associated Git repository to incorporate the latest changes and maintain synchronization with the current codebase.",
+                  "args_schema":{
+
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"life_cycle_management",
+                     "tool_action_type":"modify"
+                  },
+                  "tool_result_type":"Json",
+                  "sync_invocation_supported":false,
+                  "async_invocation_supported":true
+               },
+               {
+                  "name":"delete_datasource",
+                  "description":"Removes an existing datasource and all its indexed data from the system, freeing up resources and cleaning up stale references.",
+                  "args_schema":{
+
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"life_cycle_management",
+                     "tool_action_type":"remove"
+                  },
+                  "tool_result_type":"Json",
+                  "sync_invocation_supported":false,
+                  "async_invocation_supported":true
+               },
+               {
+                  "name":"get_files_tree",
+                  "description":"Retrieves a hierarchical representation of the file system structure of the indexed repository, showing directories and files in a tree format for easy navigation.",
+                  "args_schema":{
+                     "path":{
+                        "type":"String",
+                        "required":true,
+                        "description":"Relative path within the repository to start the tree structure from",
+                        "enum":null
+                     },
+                     "level":{
+                        "type":"Number",
+                        "required":false,
+                        "description":"Maximum depth of the tree structure, defaults to full depth",
+                        "enum":null
+                     },
+                     "limit":{
+                        "type":"Number",
+                        "required":false,
+                        "description":"Maximum number of files to include in the tree structure, defaults to all files",
+                        "enum":null
+                     }
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"data_retrieval",
+                     "tool_action_type":null
+                  },
+                  "tool_result_type":"String",
+                  "sync_invocation_supported":true,
+                  "async_invocation_supported":false
+               },
+               {
+                  "name":"get_files_list",
+                  "description":"Retrieves a list of files and directories within the specified path in the indexed repository, showing file names and types for easy reference.",
+                  "args_schema":{
+                     "path":{
+                        "type":"String",
+                        "required":true,
+                        "description":"Relative path within the repository to list files from",
+                        "enum":null
+                     },
+                     "limit":{
+                        "type":"Number",
+                        "required":false,
+                        "description":"Maximum number of files to include in the list, defaults to all files",
+                        "enum":null
+                     },
+                     "recursive":{
+                        "type":"Boolean",
+                        "required":false,
+                        "description":"Flag to include files from subdirectories recursively, defaults to false",
+                        "enum":null
+                     }
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"data_retrieval",
+                     "tool_action_type":null
+                  },
+                  "tool_result_type":"Json",
+                  "sync_invocation_supported":true,
+                  "async_invocation_supported":false
+               },
+               {
+                  "name":"get_code_members",
+                  "description":"Extracts and returns structured information about code components such as classes, methods, functions, and variables within specified file in the indexed repository.",
+                  "args_schema":{
+                     "file_path":{
+                        "type":"String",
+                        "required":true,
+                        "description":"Path of the file to extract code members from",
+                        "enum":null
+                     },
+                     "start_line":{
+                        "type":"Number",
+                        "required":false,
+                        "description":"Starting line number to extract code members from, defaults to 1",
+                        "enum":null
+                     },
+                     "end_line":{
+                        "type":"Number",
+                        "required":false,
+                        "description":"Ending line number to extract code members till, defaults to end of file",
+                        "enum":null
+                     }
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"data_retrieval",
+                     "tool_action_type":null
+                  },
+                  "tool_result_type":"Json",
+                  "sync_invocation_supported":false,
+                  "async_invocation_supported":true
+               },
+               {
+                  "name":"get_trimmed_code",
+                  "description":"Retrieves a simplified version of code from specified files with non-essential elements (like comments and some whitespace) removed to focus on the core functionality.",
+                  "args_schema":{
+                     "file_path":{
+                        "type":"String",
+                        "required":true,
+                        "description":"Path of the file to extract trimmed code from",
+                        "enum":null
+                     },
+                     "exclude_private":{
+                        "type":"Boolean",
+                        "required":false,
+                        "description":"Flag to exclude private members from the trimmed code, defaults to false",
+                        "enum":null
+                     },
+                     "show_line_numbers":{
+                        "type":"Boolean",
+                        "required":false,
+                        "description":"Flag to include line numbers in the trimmed code, defaults to false",
+                        "enum":null
+                     }
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"data_retrieval",
+                     "tool_action_type":null
+                  },
+                  "tool_result_type":"String",
+                  "sync_invocation_supported":true,
+                  "async_invocation_supported":false
+               },
+               {
+                  "name":"get_code",
+                  "description":"Fetches the complete, unmodified source code of specified files or code segments from the indexed repository, preserving all original formatting and comments.",
+                  "args_schema":{
+                     "file_path":{
+                        "type":"String",
+                        "required":true,
+                        "description":"Path of the file to extract code from",
+                        "enum":null
+                     },
+                     "start_line":{
+                        "type":"Number",
+                        "required":false,
+                        "description":"Starting line number to extract code from, defaults to 1",
+                        "enum":null
+                     },
+                     "show_line_numbers":{
+                        "type":"Boolean",
+                        "required":false,
+                        "description":"Flag to include line numbers in the code, defaults to false",
+                        "enum":null
+                     },
+                     "end_line":{
+                        "type":"Number",
+                        "required":false,
+                        "description":"Ending line number to extract code till, defaults to end of file",
+                        "enum":null
+                     }
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"data_retrieval",
+                     "tool_action_type":null
+                  },
+                  "tool_result_type":"String",
+                  "sync_invocation_supported":true,
+                  "async_invocation_supported":false
+               },
+               {
+                  "name":"get_metadata",
+                  "description":"Retrieves metadata information about specified files in the indexed repository, including file size, line count, and other relevant details.",
+                  "args_schema":{
+                     "file_path":{
+                        "type":"String",
+                        "required":true,
+                        "description":"Path of the file to extract code from",
+                        "enum":null
+                     }
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"data_retrieval",
+                     "tool_action_type":null
+                  },
+                  "tool_result_type":"String",
+                  "sync_invocation_supported":true,
+                  "async_invocation_supported":false
+               },
+               {
+                  "name":"get_outgoing_dependencies",
+                  "description":"Analyzes and returns information about external dependencies and imports used by specified files or code components, helping to understand code relationships and dependency chains.",
+                  "args_schema":{
+                     "file_path":{
+                        "type":"String",
+                        "required":true,
+                        "description":"Path of the file to get outgoing dependencies from",
+                        "enum":null
+                     },
+                     "start_line":{
+                        "type":"Number",
+                        "required":false,
+                        "description":"Starting line number to get outgoing dependencies from, defaults to 1",
+                        "enum":null
+                     },
+                     "end_line":{
+                        "type":"Number",
+                        "required":false,
+                        "description":"Ending line number to get outgoing dependencies till, defaults to end of file",
+                        "enum":null
+                     }
+                  },
+                  "tool_metadata":{
+                     "tool_type":"stateful",
+                     "tool_purpose":"data_retrieval",
+                     "tool_action_type":null
+                  },
+                  "tool_result_type":"Json",
+                  "sync_invocation_supported":true,
+                  "async_invocation_supported":false
+               }
+            ]
+         },
+         {
+            "toolkit_id":"98212b72-8b83-4d70-852f-49f0df28a0d4",
+            "name":"TestToolkit",
+            "description":"Tst toolkit",
+            "toolkit_config":{
+               "type":"test toolkit",
+               "description":"Configuration for connecting to GitHub repositories to access code for indexing and analysis.",
+               "parameters":{
+
+               }
+            },
+            "provided_tools":[
+               {
+                  "name":"test_tool",
+                  "description":"Some test tools",
+                  "args_schema":{
+                     "service_url":{
+                        "type":"String",
+                        "required":true,
+                        "description":"The url of the service",
+                        "enum":null
+                     },
+                     "exclude_glob":{
+                        "type":"String",
+                        "required":false,
+                        "description":"Comma separated string of file paths to exclude from indexing",
+                        "enum":null
+                     },
+                     "service_method":{
+                        "type":"String",
+                        "required":true,
+                        "description":"The token to access the service",
+                        "enum":null
+                     }
+                  },
+                  "tool_metadata":{
+                     "tool_type":null,
+                     "tool_purpose":null,
+                     "tool_action_type":null
+                  },
+                  "tool_result_type":"Json",
+                  "sync_invocation_supported":false,
+                  "async_invocation_supported":true
+               }
+            ]
+         }
+      ]
+   }

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

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

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

@@ -3,8 +3,8 @@ import os
 import pytest
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests.enums.integrations import DataBaseDialect
-from tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.enums.integrations import DataBaseDialect
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
 
 valid_integrations = [
     (

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

@@ -2,7 +2,7 @@
 
 from dataclasses import dataclass
 from typing import List
-from tests.enums.model_types import ModelTypes
+from codemie_test_harness.tests.enums.model_types import ModelTypes
 
 
 @dataclass

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

@@ -1,7 +1,7 @@
 from codemie_sdk.models.assistant import MCPServerDetails, MCPServerConfig
-from tests.enums.tools import PluginTool
+from codemie_test_harness.tests.enums.tools import PluginTool
 
-from tests.utils.constants import TESTS_PATH
+from codemie_test_harness.tests.utils.constants import TESTS_PATH
 
 cli_mcp_server_test_data = [
     (

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

@@ -1,4 +1,4 @@
-from tests import AwsParameterStore
+from codemie_test_harness.tests import AwsParameterStore
 
 CHAT_ID = (
     AwsParameterStore.get_cloud_provider_credentials("telegram")

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

@@ -1,4 +1,4 @@
-from tests.enums.tools import OpenApiTool
+from codemie_test_harness.tests.enums.tools import OpenApiTool
 
 INVOKE_API_TOOL_TASK = (
     "Invoke GET /v1/info endpoint. Url you can get from tools available"

{tests → codemie_test_harness/tests}/test_data/plugin_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
 
 list_files_plugin_tools_test_data = [
     (

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

@@ -1,5 +1,5 @@
-from tests.enums.tools import ProjectManagementTool
-from tests.test_data.pm_tools_test_data import (
+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,
     CONFLUENCE_TOOL_PROMPT,
@@ -9,7 +9,7 @@ from tests.test_data.pm_tools_test_data import (
     CONFLUENCE_CLOUD_TOOL_PROMPT,
     RESPONSE_FOR_CONFLUENCE_CLOUD_TOOL,
 )
-from tests.utils.constants import ProjectManagementIntegrationType
+from codemie_test_harness.tests.utils.constants import ProjectManagementIntegrationType
 
 pm_tools_test_data = [
     (

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

@@ -1,6 +1,6 @@
 import pytest
 
-from tests.enums.tools import ResearchToolName
+from codemie_test_harness.tests.enums.tools import ResearchToolName
 
 search_tools_test_data = [
     (

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

@@ -1,6 +1,6 @@
 import os
 
-from tests.enums.tools import VcsTool
+from codemie_test_harness.tests.enums.tools import VcsTool
 
 GITHUB_TOOL_TASK = f"Using github tool get info about issue №5 for the repo {os.getenv('GITHUB_PROJECT')}"
 

{tests → codemie_test_harness/tests}/ui/conftest.py

@@ -3,7 +3,7 @@ import os
 import pytest
 from datetime import datetime
 
-from tests.ui.pageobject.login_page import LoginPage
+from codemie_test_harness.tests.ui.pageobject.login_page import LoginPage
 
 from reportportal_client import RPLogger, RPLogHandler
 

{tests → codemie_test_harness/tests}/ui/pageobject/assistants/assistants_page.py

@@ -1,7 +1,7 @@
 from reportportal_client import step
 from playwright.sync_api import expect
 
-from tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
 
 
 class AssistantsPage(BasePage):

{tests → codemie_test_harness/tests}/ui/pageobject/base_page.py

@@ -3,8 +3,8 @@ import re
 from playwright.sync_api import expect
 from reportportal_client import step
 
-from tests.ui.pageobject.components.pop_up import PopUp
-from tests.ui.pageobject.components.header import Header
+from codemie_test_harness.tests.ui.pageobject.components.pop_up import PopUp
+from codemie_test_harness.tests.ui.pageobject.components.header import Header
 
 
 class BasePage:

{tests → codemie_test_harness/tests}/ui/pageobject/login_page.py

@@ -1,8 +1,10 @@
 from reportportal_client import step
 from playwright.sync_api import expect
 
-from tests.ui.pageobject.assistants.assistants_page import AssistantsPage
-from tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.assistants.assistants_page import (
+    AssistantsPage,
+)
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
 
 
 class LoginPage(BasePage):

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/base_workflow_form_page.py

@@ -3,8 +3,8 @@ from playwright.sync_api import Page, Locator, expect
 from reportportal_client import step
 from typing import Optional
 
-from tests.ui.pageobject.base_page import BasePage
-from tests.ui.pageobject.components import WorkflowSidebar
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.components import WorkflowSidebar
 
 
 class BaseWorkflowFormPage(BasePage):

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/create_workflow_page.py

@@ -1,8 +1,10 @@
 from playwright.sync_api import expect
 from reportportal_client import step
 
-from tests.ui.pageobject.workflows.base_workflow_form_page import BaseWorkflowFormPage
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.ui.pageobject.workflows.base_workflow_form_page import (
+    BaseWorkflowFormPage,
+)
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 class CreateWorkflowPage(BaseWorkflowFormPage):

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/edit_workflow_page.py

@@ -3,7 +3,9 @@ from reportportal_client import step
 from typing import Optional
 import re
 
-from tests.ui.pageobject.workflows.base_workflow_form_page import BaseWorkflowFormPage
+from codemie_test_harness.tests.ui.pageobject.workflows.base_workflow_form_page import (
+    BaseWorkflowFormPage,
+)
 
 
 class EditWorkflowPage(BaseWorkflowFormPage):

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/workflow_details_page.py

@@ -6,9 +6,11 @@ from playwright.sync_api import expect, Locator
 from reportportal_client import step
 from typing import List, Optional
 
-from tests.ui.pageobject.base_page import BasePage
-from tests.ui.pageobject.components import WorkflowSidebar
-from tests.ui.pageobject.components.execution_history_row import ExecutionHistoryRow
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.components import WorkflowSidebar
+from codemie_test_harness.tests.ui.pageobject.components.execution_history_row import (
+    ExecutionHistoryRow,
+)
 
 
 class WorkflowDetailsPage(BasePage):

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/workflow_executions_page.py

@@ -2,11 +2,11 @@ from typing import List, Optional
 from playwright.sync_api import expect, Locator
 from reportportal_client import step
 
-from tests.ui.pageobject.base_page import BasePage
-from tests.ui.pageobject.components.workflow_execution_history_item import (
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.components.workflow_execution_history_item import (
     WorkflowExecutionHistoryItem,
 )
-from tests.ui.pageobject.components.workflow_execution_state import (
+from codemie_test_harness.tests.ui.pageobject.components.workflow_execution_state import (
     WorkflowExecutionState,
 )
 

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/workflow_template_details.py

@@ -1,7 +1,7 @@
 from playwright.sync_api import expect
 from reportportal_client import step
 
-from tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
 
 
 class WorkflowTemplateDetailsPage(BasePage):

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/workflow_templates_page.py

@@ -1,9 +1,11 @@
 from reportportal_client import step
 
-from tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
 from playwright.sync_api import Page, expect
 
-from tests.ui.pageobject.workflows.workflows_page import WorkflowsPage
+from codemie_test_harness.tests.ui.pageobject.workflows.workflows_page import (
+    WorkflowsPage,
+)
 
 
 class WorkflowTemplatesPage(BasePage):