codemie-test-harness 0.1.184__py3-none-any.whl → 0.1.198__py3-none-any.whl

codemie_test_harness/tests/utils/credentials_manager.py

@@ -9,9 +9,10 @@ import os
 import re
 from typing import Optional, List, Dict, Any
 
+from codemie_sdk.models.integration import CredentialValues
+
 from codemie_test_harness.tests.utils.aws_parameters_store import AwsParameterStore
 from codemie_test_harness.tests.utils.env_resolver import EnvironmentResolver
-from codemie_sdk.models.integration import CredentialValues
 
 # Constants
 INTEGRATIONS_PARAMETER_PATH = "/codemie/autotests/integrations/"
@@ -27,7 +28,7 @@ class CredentialsManager:
     3. Default values (final fallback)
     """
 
-    AUTO_GENERATED = "AUTO_GENERATED"
+    AUTO_GENERATED = "AutoGenerated"
 
     # Configuration mapping: env_var -> AWS parameter details
     CREDENTIAL_MAPPINGS = {
@@ -541,6 +542,37 @@ class CredentialsManager:
             "aws_param": f"{INTEGRATIONS_PARAMETER_PATH}sql",
             "aws_path": "sandbox.mssql.password",
         },
+        # Sandbox InfluxDB (only available in sandbox)
+        "SANDBOX_INFLUX_DIALECT": {
+            "env": "INFLUX_DIALECT",
+            "aws_param": f"{INTEGRATIONS_PARAMETER_PATH}sql",
+            "aws_path": "sandbox.influxdb.dialect",
+        },
+        "SANDBOX_INFLUX_URL": {
+            "env": "INFLUX_URL",
+            "aws_param": f"{INTEGRATIONS_PARAMETER_PATH}sql",
+            "aws_path": "sandbox.influxdb.url",
+        },
+        "SANDBOX_INFLUX_PORT": {
+            "env": "INFLUX_PORT",
+            "aws_param": f"{INTEGRATIONS_PARAMETER_PATH}sql",
+            "aws_path": "sandbox.influxdb.port",
+        },
+        "SANDBOX_INFLUX_TOKEN": {
+            "env": "INFLUX_TOKEN",
+            "aws_param": f"{INTEGRATIONS_PARAMETER_PATH}sql",
+            "aws_path": "sandbox.influxdb.token",
+        },
+        "SANDBOX_INFLUX_ORG_NAME": {
+            "env": "INFLUX_ORG_NAME",
+            "aws_param": f"{INTEGRATIONS_PARAMETER_PATH}sql",
+            "aws_path": "sandbox.influxdb.org_name",
+        },
+        "SANDBOX_INFLUX_BUCKET": {
+            "env": "INFLUX_BUCKET",
+            "aws_param": f"{INTEGRATIONS_PARAMETER_PATH}sql",
+            "aws_path": "sandbox.influxdb.bucket",
+        },
     }
 
     # Cache for AWS Parameter Store data to avoid repeated calls
@@ -1126,6 +1158,28 @@ class CredentialsManager:
             CredentialValues(key="password", value=password),
         ]
 
+    @staticmethod
+    def influx_credentials() -> List[CredentialValues]:
+        """Create INFLUX credentials using configuration-driven approach.
+
+        INFLUX DB is only available in sandbox environment.
+        """
+        dialect = CredentialsManager.get_parameter("SANDBOX_INFLUX_DIALECT")
+        url = CredentialsManager.get_parameter("SANDBOX_INFLUX_URL")
+        port = CredentialsManager.get_parameter("SANDBOX_INFLUX_PORT")
+        token = CredentialsManager.get_parameter("SANDBOX_INFLUX_TOKEN")
+        org = CredentialsManager.get_parameter("SANDBOX_INFLUX_ORG_NAME")
+        bucket = CredentialsManager.get_parameter("SANDBOX_INFLUX_BUCKET")
+
+        return [
+            CredentialValues(key="dialect", value=dialect),
+            CredentialValues(key="url", value=url),
+            CredentialValues(key="port", value=port),
+            CredentialValues(key="token", value=token),
+            CredentialValues(key="org", value=org),
+            CredentialValues(key="bucket", value=bucket),
+        ]
+
     @staticmethod
     def sql_credentials(db_dialect: str) -> list[CredentialValues] | None:
         """Create SQL credentials using environment-aware configuration-driven approach.
@@ -1136,12 +1190,16 @@ class CredentialsManager:
         Returns:
             List of credential values for the specified database dialect
         """
-        if db_dialect.lower() == "mysql":
-            return CredentialsManager.mysql_credentials()
-        elif db_dialect.lower() in ["postgres", "postgresql"]:
-            return CredentialsManager.postgres_credentials()
-        elif db_dialect.lower() in ["mssql", "sqlserver"]:
-            return CredentialsManager.mssql_credentials()
+        sql_credentials_map = {
+            "mysql": CredentialsManager.mysql_credentials(),
+            "postgres": CredentialsManager.postgres_credentials(),
+            "postgresql": CredentialsManager.postgres_credentials(),
+            "mssql": CredentialsManager.mssql_credentials(),
+            "sqlserver": CredentialsManager.mssql_credentials(),
+            "influxdb": CredentialsManager.influx_credentials(),
+        }
+
+        return sql_credentials_map.get(db_dialect.lower())
 
     @staticmethod
     def open_api_credentials(token: str) -> List[CredentialValues]:

codemie_test_harness/tests/utils/workflow_utils.py

@@ -178,3 +178,94 @@ class WorkflowUtils(BaseUtils):
 
     def get_first_execution(self, workflow):
         return self.get_executions(workflow)[0]
+
+    def extract_triggered_tools_from_execution(self, workflow, execution=None):
+        """
+        Extract triggered tools from workflow execution thoughts.
+
+        Args:
+            workflow: Workflow object
+            execution: Execution object
+
+        Returns:
+            list: List of triggered tool names in lowercase
+        """
+
+        if execution:
+            execution_id = execution.execution_id
+        else:
+            executions = self.get_executions(workflow)
+            assert len(executions) > 0, "No workflow executions found"
+            first_execution = executions[0]
+            execution_id = first_execution.execution_id
+
+        triggered_tools = []
+
+        # Get execution states to find thought IDs
+        states_service = self.client.workflows.executions(workflow.id).states(
+            execution_id
+        )
+        states = states_service.list()
+
+        if not states:
+            return triggered_tools
+
+        # Collect all thought IDs from states
+        thought_ids = []
+        for state in states:
+            if hasattr(state, "thoughts") and state.thoughts:
+                for thought in state.thoughts:
+                    thought_ids.append(thought.id)
+
+        if not thought_ids:
+            return triggered_tools
+
+        # Get detailed thoughts information
+        executions_service = self.client.workflows.executions(workflow.id)
+        thoughts_data = executions_service.get_thoughts(execution_id, thought_ids)
+
+        # Convert thoughts objects to dictionaries for processing
+        thoughts_dict_data = [thought.model_dump() for thought in thoughts_data]
+
+        # Extract triggered tools from thoughts
+        triggered_tools = self._extract_tools_from_thoughts(thoughts_dict_data)
+
+        return triggered_tools
+
+    @staticmethod
+    def _extract_tools_from_thoughts(thoughts_data):
+        """
+        Extract tool names from thoughts data recursively.
+
+        Args:
+            thoughts_data: List of thought objects with nested children
+
+        Returns:
+            list: List of triggered tool names in lowercase
+        """
+        triggered_tools = []
+
+        def process_thought(_thought):
+            # Check if this thought represents a tool call
+            author_name = _thought.get("author_name", "")
+            author_type = _thought.get("author_type", "")
+
+            # Filter out 'Codemie Thoughts' and non-tool entries
+            if (
+                author_type == "Tool"
+                and author_name
+                and author_name != "Codemie Thoughts"
+            ):
+                triggered_tools.append(author_name.lower())
+
+            # Recursively process children
+            children = _thought.get("children", [])
+            for child in children:
+                process_thought(child)
+
+        # Process all top-level thoughts
+        if isinstance(thoughts_data, list):
+            for thought in thoughts_data:
+                process_thought(thought)
+
+        return triggered_tools

codemie_test_harness/tests/workflow/assistant_tools/access_management/test_workflow_with_assistant_with_keycloak_tool.py

@@ -5,6 +5,7 @@ from codemie_test_harness.tests.test_data.keycloak_tool_test_data import (
     KEYCLOAK_TOOL_PROMPT,
     KEYCLOAK_TOOL_RESPONSE,
 )
+from codemie_test_harness.tests.utils.base_utils import assert_tool_triggered
 
 
 @pytest.mark.workflow
@@ -28,4 +29,10 @@ def test_workflow_with_assistant_with_keycloak_tool(
     response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name
     )
+
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(AccessManagementTool.KEYCLOAK, triggered_tools)
+
     similarity_check.check_similarity(response, KEYCLOAK_TOOL_RESPONSE)

codemie_test_harness/tests/workflow/assistant_tools/ado/test_workflow_with_assistant_with_ado_test_plan_tools.py

@@ -9,7 +9,10 @@ from codemie_test_harness.tests.test_data.ado_test_plan_tools_test_data import (
     ADO_TEST_PLAN_DELETE_SUITE,
     ADO_TEST_PLAN_DELETE_PLAN,
 )
-from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
+)
 from codemie_test_harness.tests.utils.constants import ID_PATTERN
 from codemie_test_harness.tests.utils.json_utils import extract_id_from_ado_response
 
@@ -40,6 +43,11 @@ def test_workflow_with_assistant_with_ado_test_plan_get_tools(
         workflow_with_assistant.id, assistant.name
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)
 
 
@@ -76,6 +84,14 @@ def test_workflow_with_assistant_with_ado_test_plan_tools(
         workflow_with_assistant.id, assistant.name, create_prompt
     )
 
+    # Assert create test plan tool was triggered
+    create_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(
+        AzureDevOpsTestPlanTool.CREATE_TEST_PLAN, create_triggered_tools
+    )
+
     # Extract the test plan ID from the response
     test_plan_id = extract_id_from_ado_response(create_response, ID_PATTERN)
     create_expected = ADO_TEST_PLAN_CREATE_TEST_PLAN["expected_llm_answer"].format(
@@ -94,6 +110,14 @@ def test_workflow_with_assistant_with_ado_test_plan_tools(
         workflow_with_assistant.id, assistant.name, create_suite_prompt
     )
 
+    # Assert create test suite tool was triggered
+    create_suite_triggered_tools = (
+        workflow_utils.extract_triggered_tools_from_execution(workflow_with_assistant)
+    )
+    assert_tool_triggered(
+        AzureDevOpsTestPlanTool.CREATE_TEST_SUITE, create_suite_triggered_tools
+    )
+
     # Extract the test suite ID from the response
     test_suite_id = extract_id_from_ado_response(create_suite_response, ID_PATTERN)
     create_suite_expected = ADO_TEST_PLAN_CREATE_SUITE["expected_llm_answer"].format(
@@ -108,6 +132,14 @@ def test_workflow_with_assistant_with_ado_test_plan_tools(
     add_case_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, add_case_prompt
     )
+
+    # Assert add test case tool was triggered
+    add_case_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(
+        AzureDevOpsTestPlanTool.ADD_TEST_CASE, add_case_triggered_tools
+    )
     add_case_expected = ADO_TEST_PLAN_ADD_TEST_CASE["expected_llm_answer"].format(
         test_suite_id, test_plan_id
     )
@@ -120,6 +152,14 @@ def test_workflow_with_assistant_with_ado_test_plan_tools(
     delete_suite_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, delete_suite_prompt
     )
+
+    # Assert delete test suite tool was triggered
+    delete_suite_triggered_tools = (
+        workflow_utils.extract_triggered_tools_from_execution(workflow_with_assistant)
+    )
+    assert_tool_triggered(
+        AzureDevOpsTestPlanTool.DELETE_TEST_SUITE, delete_suite_triggered_tools
+    )
     delete_suite_expected = ADO_TEST_PLAN_DELETE_SUITE["expected_llm_answer"].format(
         test_suite_id, test_plan_id
     )
@@ -132,6 +172,15 @@ def test_workflow_with_assistant_with_ado_test_plan_tools(
     delete_plan_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, delete_plan_prompt
     )
+
+    # Assert delete test plan tool was triggered
+    delete_plan_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(
+        AzureDevOpsTestPlanTool.DELETE_TEST_PLAN, delete_plan_triggered_tools
+    )
+
     delete_plan_expected = ADO_TEST_PLAN_DELETE_PLAN["expected_llm_answer"].format(
         test_plan_id
     )

codemie_test_harness/tests/workflow/assistant_tools/ado/test_workflow_with_assistant_with_ado_wiki_tools.py

@@ -8,7 +8,10 @@ from codemie_test_harness.tests.test_data.ado_wiki_tools_test_data import (
     ADO_WIKI_MODIFY_PAGE,
     ADO_WIKI_DELETE_PAGE,
 )
-from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
+)
 
 
 @pytest.mark.workflow
@@ -37,6 +40,11 @@ def test_workflow_with_assistant_with_ado_wiki_get_tools(
         workflow_with_assistant.id, assistant.name
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)
 
 
@@ -69,6 +77,13 @@ def test_workflow_with_assistant_with_ado_wiki_modify_tools(
     create_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, create_prompt
     )
+
+    # Assert that modify wiki page tool was triggered
+    create_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(AzureDevOpsWikiTool.MODIFY_WIKI_PAGE, create_triggered_tools)
+
     similarity_check.check_similarity(create_response, create_expected)
 
     # 2. Rename the page
@@ -81,6 +96,13 @@ def test_workflow_with_assistant_with_ado_wiki_modify_tools(
     rename_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, rename_prompt
     )
+
+    # Assert that rename wiki page tool was triggered
+    rename_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(AzureDevOpsWikiTool.RENAME_WIKI_PAGE, rename_triggered_tools)
+
     similarity_check.check_similarity(rename_response, rename_expected)
 
     # 3. Modify the page
@@ -93,6 +115,13 @@ def test_workflow_with_assistant_with_ado_wiki_modify_tools(
     modify_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, modify_prompt
     )
+
+    # Assert that modify wiki page tool was triggered
+    modify_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(AzureDevOpsWikiTool.MODIFY_WIKI_PAGE, modify_triggered_tools)
+
     similarity_check.check_similarity(modify_response, modify_expected)
 
     # 4. Delete the page
@@ -105,4 +134,13 @@ def test_workflow_with_assistant_with_ado_wiki_modify_tools(
     delete_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, delete_prompt
     )
+
+    # Assert that delete wiki page tool was triggered
+    delete_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(
+        AzureDevOpsWikiTool.DELETE_WIKI_PAGE_BY_PATH, delete_triggered_tools
+    )
+
     similarity_check.check_similarity(delete_response, delete_expected)

codemie_test_harness/tests/workflow/assistant_tools/ado/test_workflow_with_assistant_with_ado_work_item_tools.py

@@ -7,7 +7,10 @@ from codemie_test_harness.tests.test_data.ado_work_item_tools_test_data import (
     ADO_WORK_ITEM_UPDATE,
     ADO_WORK_ITEM_LINK,
 )
-from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
+)
 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
 
@@ -38,6 +41,11 @@ def test_workflow_with_assistant_with_ado_work_item_get_tools(
         workflow_with_assistant.id, assistant.name
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)
 
 
@@ -74,6 +82,15 @@ def test_workflow_with_assistant_with_ado_work_item_modify_tools(
     create_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, create_prompt
     )
+
+    # Assert that create work item tool was triggered
+    create_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(
+        AzureDevOpsWorkItemTool.CREATE_WORK_ITEM, create_triggered_tools
+    )
+
     work_item_id = extract_id_from_ado_response(create_response, WORK_ITEM_ID_PATTERN)
     similarity_check.check_similarity(create_response, create_expected)
 
@@ -89,6 +106,15 @@ def test_workflow_with_assistant_with_ado_work_item_modify_tools(
     update_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, update_prompt
     )
+
+    # Assert that update work item tool was triggered
+    update_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(
+        AzureDevOpsWorkItemTool.UPDATE_WORK_ITEM, update_triggered_tools
+    )
+
     similarity_check.check_similarity(update_response, update_expected)
 
     # 3. Link the item
@@ -100,4 +126,11 @@ def test_workflow_with_assistant_with_ado_work_item_modify_tools(
     link_response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, link_prompt
     )
+
+    # Assert that link work items tool was triggered
+    link_triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(AzureDevOpsWorkItemTool.LINK_WORK_ITEMS, link_triggered_tools)
+
     similarity_check.check_similarity(link_response, link_expected)

codemie_test_harness/tests/workflow/assistant_tools/cloud/test_workflow_with_assistant_cloud_tools.py

@@ -1,6 +1,7 @@
 import pytest
 
 from codemie_test_harness.tests.test_data.cloud_tools_test_data import cloud_test_data
+from codemie_test_harness.tests.utils.base_utils import assert_tool_triggered
 
 
 @pytest.mark.workflow
@@ -32,4 +33,9 @@ def test_workflow_with_cloud_tools(
         workflow_with_assistant.id, assistant.name
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)

codemie_test_harness/tests/workflow/assistant_tools/codebase/test_worfklow_with_assistant_codebase_tools.py

@@ -6,6 +6,7 @@ from codemie_test_harness.tests.test_data.codebase_tools_test_data import (
     code_tools_test_data,
     sonar_tools_test_data,
 )
+from codemie_test_harness.tests.utils.base_utils import assert_tool_triggered
 
 
 @pytest.mark.workflow
@@ -35,6 +36,11 @@ def test_workflow_with_codebase_tools(
         workflow_with_assistant.id, assistant.name
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)
 
 
@@ -72,4 +78,9 @@ def test_workflow_with_sonar_tools(
         workflow_with_assistant.id, assistant.name
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(CodeBaseTool.SONAR, triggered_tools)
+
     similarity_check.check_similarity(response, expected)

codemie_test_harness/tests/workflow/assistant_tools/data_management/test_workflow_with_assistant_with_data_management_tools.py

@@ -1,7 +1,9 @@
 import pytest
 from codemie_sdk.models.integration import CredentialTypes
 
+from codemie_test_harness.tests.enums.integrations import DataBaseDialect
 from codemie_test_harness.tests.enums.tools import DataManagementTool, Toolkit
+from codemie_test_harness.tests.utils.base_utils import assert_tool_triggered
 from codemie_test_harness.tests.test_data.data_management_tools_test_data import (
     ELASTIC_TOOL_TASK,
     RESPONSE_FOR_ELASTIC,
@@ -11,6 +13,8 @@ from codemie_test_harness.tests.test_data.data_management_tools_test_data import
     SQL_TOOL_INSERT_TABLE_TASK,
     SQL_TOOL_QUERY_TABLE_TASK,
     RESPONSE_FOR_SQL,
+    INFLUXDB_QUERY_MEASUREMENT_TASK,
+    RESPONSE_FOR_INFLUXDB,
 )
 from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 from codemie_test_harness.tests.utils.env_resolver import EnvironmentResolver
@@ -45,6 +49,12 @@ def test_workflow_with_assistant_with_elastic_tools(
     response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name
     )
+
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(DataManagementTool.ELASTIC, triggered_tools)
+
     similarity_check.check_similarity(response, RESPONSE_FOR_ELASTIC)
 
 
@@ -71,21 +81,82 @@ def test_workflow_with_assistant_with_sql_tools(
     )
 
     assistant = assistant(
-        Toolkit.DATA_MANAGEMENT, DataManagementTool.SQL, settings=settings
+        Toolkit.DATA_MANAGEMENT,
+        DataManagementTool.SQL,
+        settings=settings,
+        system_prompt="Always run tools for user prompt",
     )
     workflow_with_assistant = workflow_with_assistant(assistant, "Run")
 
+    # Step 1: Create table
     workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, SQL_TOOL_CREATE_TABLE_TASK
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(DataManagementTool.SQL, triggered_tools)
+
+    # Step 2: Insert data
     workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, SQL_TOOL_INSERT_TABLE_TASK
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(DataManagementTool.SQL, triggered_tools)
+
+    # Step 3: Query data
     response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, SQL_TOOL_QUERY_TABLE_TASK
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(DataManagementTool.SQL, triggered_tools)
+
+    # Step 4: Delete table
     workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name, SQL_TOOL_DELETE_TABLE_TASK
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(DataManagementTool.SQL, triggered_tools)
 
     similarity_check.check_similarity(response, RESPONSE_FOR_SQL)
+
+
+@pytest.mark.workflow
+@pytest.mark.workflow_with_assistant
+@pytest.mark.influx
+@pytest.mark.api
+@pytest.mark.skipif(
+    not EnvironmentResolver.is_sandbox(),
+    reason="InfluxDB is only available in sandbox environments",
+)
+def test_workflow_with_assistant_with_influxdb_tools(
+    assistant,
+    workflow_with_assistant,
+    workflow_utils,
+    integration_utils,
+    similarity_check,
+):
+    """Test workflow execution with InfluxDB tools via assistant."""
+
+    credential_values = CredentialsManager.sql_credentials(DataBaseDialect.INFLUX)
+    settings = integration_utils.create_integration(
+        CredentialTypes.SQL, credential_values
+    )
+
+    assistant = assistant(
+        Toolkit.DATA_MANAGEMENT, DataManagementTool.SQL, settings=settings
+    )
+    workflow_with_assistant = workflow_with_assistant(assistant, "Run")
+
+    # Query the measurement
+    response = workflow_utils.execute_workflow(
+        workflow_with_assistant.id, assistant.name, INFLUXDB_QUERY_MEASUREMENT_TASK
+    )
+
+    similarity_check.check_similarity(response, RESPONSE_FOR_INFLUXDB)