codemie-test-harness 0.1.169__py3-none-any.whl → 0.1.171__py3-none-any.whl

codemie_test_harness/cli/commands/workflow_cmd.py

@@ -0,0 +1,64 @@
+"""Workflow CLI commands for executing and managing workflows."""
+
+from __future__ import annotations
+
+
+import click
+
+from codemie_test_harness.tests.utils.client_factory import get_client
+from ..constants import CONSOLE
+
+
+@click.group(name="workflow")
+def workflow_cmd():
+    """Interact with CodeMie workflows."""
+    pass
+
+
+@workflow_cmd.command(name="execute")
+@click.option(
+    "--workflow-id",
+    required=True,
+    help="Workflow ID to execute",
+)
+@click.option(
+    "--user-input",
+    "-i",
+    default="",
+    help="User input for the workflow execution (optional)",
+)
+def execute_cmd(
+    workflow_id: str,
+    user_input: str,
+):
+    """Execute a workflow.
+
+    Example:
+        codemie-test-harness workflow execute --workflow-id "wf_123"
+        codemie-test-harness workflow execute --workflow-id "wf_123" --user-input "process this data"
+    """
+    try:
+        client = get_client()
+
+        CONSOLE.print(f"[yellow]Executing workflow {workflow_id}...[/yellow]")
+        if user_input:
+            CONSOLE.print(f"[blue]User input: {user_input}[/blue]")
+
+        # Execute workflow
+        response = client.workflows.run(workflow_id=workflow_id, user_input=user_input)
+
+        # Display response
+        CONSOLE.print("[green]Workflow execution started successfully![/green]")
+
+        if hasattr(response, "execution_id"):
+            CONSOLE.print(f"[cyan]Execution ID: {response.execution_id}[/cyan]")
+
+        if hasattr(response, "status"):
+            CONSOLE.print(f"[cyan]Status: {response.status}[/cyan]")
+
+        # Display the full response for debugging
+        CONSOLE.print(f"[dim]Response: {response}[/dim]")
+
+    except Exception as e:
+        CONSOLE.print(f"[red]Error executing workflow: {e}[/red]")
+        raise click.ClickException(str(e))

codemie_test_harness/cli/constants.py

@@ -12,13 +12,16 @@ CONFIG_FILE = CONFIG_DIR / "test-harness.json"
 KEY_AUTH_SERVER_URL = "AUTH_SERVER_URL"
 KEY_AUTH_CLIENT_ID = "AUTH_CLIENT_ID"
 KEY_AUTH_CLIENT_SECRET = "AUTH_CLIENT_SECRET"
+KEY_AUTH_PASSWORD = "AUTH_PASSWORD"
+KEY_AUTH_USERNAME = "AUTH_USERNAME"
 KEY_AUTH_REALM_NAME = "AUTH_REALM_NAME"
 KEY_CODEMIE_API_DOMAIN = "CODEMIE_API_DOMAIN"
 KEY_MARKS = "PYTEST_MARKS"
 KEY_XDIST_N = "PYTEST_N"
 KEY_RERUNS = "PYTEST_RERUNS"
 
-# Integrations / external systems
+# === COMPLETE INTEGRATION CREDENTIALS KEYS ===
+# Version Control Systems (GitLab, GitHub)
 KEY_GIT_ENV = "GIT_ENV"  # e.g. gitlab or github
 # GitLab
 KEY_GITLAB_URL = "GITLAB_URL"
@@ -29,14 +32,115 @@ KEY_GITLAB_PROJECT_ID = "GITLAB_PROJECT_ID"
 KEY_GITHUB_URL = "GITHUB_URL"
 KEY_GITHUB_TOKEN = "GITHUB_TOKEN"
 KEY_GITHUB_PROJECT = "GITHUB_PROJECT"
-# JIRA
+
+# Project Management (JIRA, Confluence)
+# JIRA Server
 KEY_JIRA_URL = "JIRA_URL"
 KEY_JIRA_TOKEN = "JIRA_TOKEN"
-KEY_JQL = "JQL"
-# Confluence
+KEY_JQL = "JIRA_JQL"
+# JIRA Cloud
+KEY_JIRA_CLOUD_URL = "JIRA_CLOUD_URL"
+KEY_JIRA_CLOUD_EMAIL = "JIRA_CLOUD_EMAIL"
+KEY_JIRA_CLOUD_TOKEN = "JIRA_CLOUD_TOKEN"
+KEY_JIRA_CLOUD_JQL = "JIRA_CLOUD_JQL"
+# Confluence Server
 KEY_CONFLUENCE_URL = "CONFLUENCE_URL"
 KEY_CONFLUENCE_TOKEN = "CONFLUENCE_TOKEN"
-KEY_CQL = "CQL"
+KEY_CQL = "CONFLUENCE_CQL"
+# Confluence Cloud
+KEY_CONFLUENCE_CLOUD_URL = "CONFLUENCE_CLOUD_URL"
+KEY_CONFLUENCE_CLOUD_EMAIL = "CONFLUENCE_CLOUD_EMAIL"
+KEY_CONFLUENCE_CLOUD_TOKEN = "CONFLUENCE_CLOUD_TOKEN"
+KEY_CONFLUENCE_CLOUD_CQL = "CONFLUENCE_CLOUD_CQL"
+
+# Cloud Providers (AWS, Azure, GCP)
+# AWS
+KEY_AWS_ACCESS_KEY_ID = "AWS_ACCESS_KEY_ID"
+KEY_AWS_SECRET_ACCESS_KEY = "AWS_SECRET_ACCESS_KEY"
+KEY_AWS_REGION = "AWS_REGION"
+KEY_AWS_SESSION_TOKEN = "AWS_SESSION_TOKEN"
+# Azure
+KEY_AZURE_CLIENT_ID = "AZURE_CLIENT_ID"
+KEY_AZURE_CLIENT_SECRET = "AZURE_CLIENT_SECRET"
+KEY_AZURE_TENANT_ID = "AZURE_TENANT_ID"
+KEY_AZURE_SUBSCRIPTION_ID = "AZURE_SUBSCRIPTION_ID"
+# GCP
+KEY_GCP_SA_KEY_BASE64 = "GCP_SA_KEY_BASE64"
+
+# Development Tools (Azure DevOps, ServiceNow, Keycloak, SonarQube)
+# Azure DevOps
+KEY_AZURE_DEVOPS_URL = "AZURE_DEVOPS_URL"
+KEY_AZURE_DEVOPS_TOKEN = "AZURE_DEVOPS_TOKEN"
+KEY_AZURE_DEVOPS_PROJECT_NAME = "AZURE_DEVOPS_PROJECT_NAME"
+KEY_AZURE_DEVOPS_ORGANIZATION_NAME = "AZURE_DEVOPS_ORGANIZATION_NAME"
+# ServiceNow
+KEY_SERVICENOW_URL = "SERVICENOW_URL"
+KEY_SERVICENOW_TOKEN = "SERVICENOW_TOKEN"
+# Keycloak
+KEY_KEYCLOAK_URL = "KEYCLOAK_URL"
+KEY_KEYCLOAK_REALM = "KEYCLOAK_REALM"
+KEY_KEYCLOAK_CLIENT_ID = "KEYCLOAK_CLIENT_ID"
+KEY_KEYCLOAK_CLIENT_SECRET = "KEYCLOAK_CLIENT_SECRET"
+# SonarQube Server
+KEY_SONAR_URL = "SONAR_URL"
+KEY_SONAR_TOKEN = "SONAR_TOKEN"
+KEY_SONAR_PROJECT_KEY = "SONAR_PROJECT_KEY"
+# SonarCloud
+KEY_SONAR_CLOUD_URL = "SONAR_CLOUD_URL"
+KEY_SONAR_CLOUD_TOKEN = "SONAR_CLOUD_TOKEN"
+KEY_SONAR_CLOUD_PROJECT_KEY = "SONAR_CLOUD_PROJECT_KEY"
+
+# Notifications (Email/Gmail, OAuth, Telegram)
+# Email/Gmail
+KEY_GMAIL_URL = "GMAIL_URL"
+KEY_SMTP_USERNAME = "SMTP_USERNAME"
+KEY_SMTP_PASSWORD = "SMTP_PASSWORD"
+# OAuth
+KEY_OAUTH_URL = "OAUTH_URL"
+KEY_OAUTH_CLIENT_ID = "OAUTH_CLIENT_ID"
+KEY_OAUTH_CLIENT_SECRET = "OAUTH_CLIENT_SECRET"
+KEY_OAUTH_REFRESH_TOKEN = "OAUTH_REFRESH_TOKEN"
+# Telegram
+KEY_TELEGRAM_TOKEN = "TELEGRAM_TOKEN"
+KEY_TELEGRAM_CHAT_ID = "TELEGRAM_CHAT_ID"
+
+# Research Tools (Kubernetes, Report Portal, Elasticsearch)
+# Kubernetes
+KEY_KUBERNETES_URL = "KUBERNETES_URL"
+KEY_KUBERNETES_TOKEN = "KUBERNETES_TOKEN"
+# Report Portal
+KEY_REPORT_PORTAL_URL = "REPORT_PORTAL_URL"
+KEY_REPORT_PORTAL_API_KEY = "REPORT_PORTAL_API_KEY"
+KEY_REPORT_PORTAL_PROJECT = "REPORT_PORTAL_PROJECT"
+# Elasticsearch
+KEY_ELASTICSEARCH_URL = "ELASTICSEARCH_URL"
+KEY_ELASTICSEARCH_API_KEY_ID = "ELASTICSEARCH_API_KEY_ID"
+KEY_ELASTICSEARCH_API_KEY = "ELASTICSEARCH_API_KEY"
+
+# Data Management (SQL databases, LiteLLM)
+# LiteLLM
+KEY_LITE_LLM_API_KEY = "LITE_LLM_API_KEY"
+# MySQL
+KEY_MYSQL_DIALECT = "MYSQL_DIALECT"
+KEY_MYSQL_URL = "MYSQL_URL"
+KEY_MYSQL_PORT = "MYSQL_PORT"
+KEY_MYSQL_DATABASE_NAME = "MYSQL_DATABASE_NAME"
+KEY_MYSQL_USERNAME = "MYSQL_USERNAME"
+KEY_MYSQL_PASSWORD = "MYSQL_PASSWORD"
+# PostgreSQL
+KEY_POSTGRES_DIALECT = "POSTGRES_DIALECT"
+KEY_POSTGRES_URL = "POSTGRES_URL"
+KEY_POSTGRES_PORT = "POSTGRES_PORT"
+KEY_POSTGRES_DATABASE_NAME = "POSTGRES_DATABASE_NAME"
+KEY_POSTGRES_USERNAME = "POSTGRES_USERNAME"
+KEY_POSTGRES_PASSWORD = "POSTGRES_PASSWORD"
+# MSSQL
+KEY_MSSQL_DIALECT = "MSSQL_DIALECT"
+KEY_MSSQL_URL = "MSSQL_URL"
+KEY_MSSQL_PORT = "MSSQL_PORT"
+KEY_MSSQL_DATABASE_NAME = "MSSQL_DATABASE_NAME"
+KEY_MSSQL_USERNAME = "MSSQL_USERNAME"
+KEY_MSSQL_PASSWORD = "MSSQL_PASSWORD"
 
 DEFAULT_MARKS = "smoke"
 DEFAULT_XDIST_N = 8
@@ -46,11 +150,14 @@ AUTH_KEYS = [
     KEY_AUTH_SERVER_URL,
     KEY_AUTH_CLIENT_ID,
     KEY_AUTH_CLIENT_SECRET,
+    KEY_AUTH_USERNAME,
+    KEY_AUTH_PASSWORD,
     KEY_AUTH_REALM_NAME,
     KEY_CODEMIE_API_DOMAIN,
 ]
 
-INTEGRATION_KEYS = [
+# === CREDENTIAL CATEGORIES ===
+VERSION_CONTROL_KEYS = [
     KEY_GIT_ENV,
     KEY_GITLAB_URL,
     KEY_GITLAB_TOKEN,
@@ -59,10 +166,282 @@ INTEGRATION_KEYS = [
     KEY_GITHUB_URL,
     KEY_GITHUB_TOKEN,
     KEY_GITHUB_PROJECT,
+]
+
+PROJECT_MANAGEMENT_KEYS = [
+    # JIRA Server
     KEY_JIRA_URL,
     KEY_JIRA_TOKEN,
     KEY_JQL,
+    # JIRA Cloud
+    KEY_JIRA_CLOUD_URL,
+    KEY_JIRA_CLOUD_EMAIL,
+    KEY_JIRA_CLOUD_TOKEN,
+    KEY_JIRA_CLOUD_JQL,
+    # Confluence Server
     KEY_CONFLUENCE_URL,
     KEY_CONFLUENCE_TOKEN,
     KEY_CQL,
+    # Confluence Cloud
+    KEY_CONFLUENCE_CLOUD_URL,
+    KEY_CONFLUENCE_CLOUD_EMAIL,
+    KEY_CONFLUENCE_CLOUD_TOKEN,
+    KEY_CONFLUENCE_CLOUD_CQL,
+]
+
+CLOUD_PROVIDERS_KEYS = [
+    # AWS
+    KEY_AWS_ACCESS_KEY_ID,
+    KEY_AWS_SECRET_ACCESS_KEY,
+    KEY_AWS_REGION,
+    KEY_AWS_SESSION_TOKEN,
+    # Azure
+    KEY_AZURE_CLIENT_ID,
+    KEY_AZURE_CLIENT_SECRET,
+    KEY_AZURE_TENANT_ID,
+    KEY_AZURE_SUBSCRIPTION_ID,
+    # GCP
+    KEY_GCP_SA_KEY_BASE64,
+]
+
+DEVELOPMENT_TOOLS_KEYS = [
+    # Azure DevOps
+    KEY_AZURE_DEVOPS_URL,
+    KEY_AZURE_DEVOPS_TOKEN,
+    KEY_AZURE_DEVOPS_PROJECT_NAME,
+    KEY_AZURE_DEVOPS_ORGANIZATION_NAME,
+    # ServiceNow
+    KEY_SERVICENOW_URL,
+    KEY_SERVICENOW_TOKEN,
+    # Keycloak
+    KEY_KEYCLOAK_URL,
+    KEY_KEYCLOAK_REALM,
+    KEY_KEYCLOAK_CLIENT_ID,
+    KEY_KEYCLOAK_CLIENT_SECRET,
+    # SonarQube Server
+    KEY_SONAR_URL,
+    KEY_SONAR_TOKEN,
+    KEY_SONAR_PROJECT_KEY,
+    # SonarCloud
+    KEY_SONAR_CLOUD_URL,
+    KEY_SONAR_CLOUD_TOKEN,
+    KEY_SONAR_CLOUD_PROJECT_KEY,
+]
+
+NOTIFICATIONS_KEYS = [
+    # Email/Gmail
+    KEY_GMAIL_URL,
+    KEY_SMTP_USERNAME,
+    KEY_SMTP_PASSWORD,
+    # OAuth
+    KEY_OAUTH_URL,
+    KEY_OAUTH_CLIENT_ID,
+    KEY_OAUTH_CLIENT_SECRET,
+    KEY_OAUTH_REFRESH_TOKEN,
+    # Telegram
+    KEY_TELEGRAM_TOKEN,
+    KEY_TELEGRAM_CHAT_ID,
+]
+
+RESEARCH_TOOLS_KEYS = [
+    # Kubernetes
+    KEY_KUBERNETES_URL,
+    KEY_KUBERNETES_TOKEN,
+    # Report Portal
+    KEY_REPORT_PORTAL_URL,
+    KEY_REPORT_PORTAL_API_KEY,
+    KEY_REPORT_PORTAL_PROJECT,
+    # Elasticsearch
+    KEY_ELASTICSEARCH_URL,
+    KEY_ELASTICSEARCH_API_KEY_ID,
+    KEY_ELASTICSEARCH_API_KEY,
 ]
+
+DATA_MANAGEMENT_KEYS = [
+    # LiteLLM
+    KEY_LITE_LLM_API_KEY,
+    # MySQL
+    KEY_MYSQL_DIALECT,
+    KEY_MYSQL_URL,
+    KEY_MYSQL_PORT,
+    KEY_MYSQL_DATABASE_NAME,
+    KEY_MYSQL_USERNAME,
+    KEY_MYSQL_PASSWORD,
+    # PostgreSQL
+    KEY_POSTGRES_DIALECT,
+    KEY_POSTGRES_URL,
+    KEY_POSTGRES_PORT,
+    KEY_POSTGRES_DATABASE_NAME,
+    KEY_POSTGRES_USERNAME,
+    KEY_POSTGRES_PASSWORD,
+    # MSSQL
+    KEY_MSSQL_DIALECT,
+    KEY_MSSQL_URL,
+    KEY_MSSQL_PORT,
+    KEY_MSSQL_DATABASE_NAME,
+    KEY_MSSQL_USERNAME,
+    KEY_MSSQL_PASSWORD,
+]
+
+# Combined lists for backwards compatibility
+INTEGRATION_KEYS = (
+    VERSION_CONTROL_KEYS
+    + PROJECT_MANAGEMENT_KEYS
+    + CLOUD_PROVIDERS_KEYS
+    + DEVELOPMENT_TOOLS_KEYS
+    + NOTIFICATIONS_KEYS
+    + RESEARCH_TOOLS_KEYS
+    + DATA_MANAGEMENT_KEYS
+)
+
+# Category mapping for CLI
+CREDENTIAL_CATEGORIES = {
+    "version-control": {
+        "name": "Version Control Systems",
+        "description": "GitLab, GitHub, and other version control platforms",
+        "keys": VERSION_CONTROL_KEYS,
+    },
+    "code-base": {
+        "name": "Code Base",
+        "description": "SonarQube and code analysis tools",
+        "keys": [
+            KEY_SONAR_URL,
+            KEY_SONAR_TOKEN,
+            KEY_SONAR_PROJECT_KEY,
+            KEY_SONAR_CLOUD_URL,
+            KEY_SONAR_CLOUD_TOKEN,
+            KEY_SONAR_CLOUD_PROJECT_KEY,
+        ],
+    },
+    "project-management": {
+        "name": "Project Management",
+        "description": "JIRA, Confluence, and project management tools",
+        "keys": PROJECT_MANAGEMENT_KEYS,
+    },
+    "cloud": {
+        "name": "Cloud",
+        "description": "AWS, GCP cloud services and Kubernetes configurations",
+        "keys": [
+            # AWS
+            KEY_AWS_ACCESS_KEY_ID,
+            KEY_AWS_SECRET_ACCESS_KEY,
+            KEY_AWS_REGION,
+            KEY_AWS_SESSION_TOKEN,
+            # GCP
+            KEY_GCP_SA_KEY_BASE64,
+            # Kubernetes
+            KEY_KUBERNETES_URL,
+            KEY_KUBERNETES_TOKEN,
+            # Azure Cloud
+            KEY_AZURE_CLIENT_ID,
+            KEY_AZURE_CLIENT_SECRET,
+            KEY_AZURE_TENANT_ID,
+            KEY_AZURE_SUBSCRIPTION_ID,
+        ],
+    },
+    "azure-devops": {
+        "name": "Azure DevOps",
+        "description": "Azure DevOps and Azure cloud platform services",
+        "keys": [
+            # Azure DevOps
+            KEY_AZURE_DEVOPS_URL,
+            KEY_AZURE_DEVOPS_TOKEN,
+            KEY_AZURE_DEVOPS_PROJECT_NAME,
+            KEY_AZURE_DEVOPS_ORGANIZATION_NAME,
+        ],
+    },
+    "access-management": {
+        "name": "Access Management",
+        "description": "Keycloak and identity management systems",
+        "keys": [
+            KEY_KEYCLOAK_URL,
+            KEY_KEYCLOAK_REALM,
+            KEY_KEYCLOAK_CLIENT_ID,
+            KEY_KEYCLOAK_CLIENT_SECRET,
+        ],
+    },
+    "notification-systems": {
+        "name": "Notification Systems",
+        "description": "Email, OAuth, Telegram, and communication platforms",
+        "keys": NOTIFICATIONS_KEYS,
+    },
+    "data-management": {
+        "name": "Data Management",
+        "description": "SQL databases, Elasticsearch, and data storage systems",
+        "keys": DATA_MANAGEMENT_KEYS
+        + [
+            KEY_ELASTICSEARCH_URL,
+            KEY_ELASTICSEARCH_API_KEY_ID,
+            KEY_ELASTICSEARCH_API_KEY,
+        ],
+    },
+    "it-service-management": {
+        "name": "IT Service Management",
+        "description": "ServiceNow and IT service management platforms",
+        "keys": [
+            KEY_SERVICENOW_URL,
+            KEY_SERVICENOW_TOKEN,
+        ],
+    },
+    "quality-assurance": {
+        "name": "Quality Assurance",
+        "description": "Report Portal and quality assurance tools",
+        "keys": [
+            KEY_REPORT_PORTAL_URL,
+            KEY_REPORT_PORTAL_API_KEY,
+            KEY_REPORT_PORTAL_PROJECT,
+        ],
+    },
+}
+
+# Sensitive fields that should be masked in input/output
+SENSITIVE_KEYS = {
+    KEY_AUTH_CLIENT_SECRET,
+    KEY_AUTH_PASSWORD,
+    KEY_GITLAB_TOKEN,
+    KEY_GITHUB_TOKEN,
+    KEY_JIRA_TOKEN,
+    KEY_JIRA_CLOUD_TOKEN,
+    KEY_CONFLUENCE_TOKEN,
+    KEY_CONFLUENCE_CLOUD_TOKEN,
+    KEY_AWS_SECRET_ACCESS_KEY,
+    KEY_AWS_SESSION_TOKEN,
+    KEY_AZURE_CLIENT_SECRET,
+    KEY_GCP_SA_KEY_BASE64,
+    KEY_AZURE_DEVOPS_TOKEN,
+    KEY_SERVICENOW_TOKEN,
+    KEY_KEYCLOAK_CLIENT_SECRET,
+    KEY_SONAR_TOKEN,
+    KEY_SONAR_CLOUD_TOKEN,
+    KEY_SMTP_PASSWORD,
+    KEY_OAUTH_CLIENT_SECRET,
+    KEY_OAUTH_REFRESH_TOKEN,
+    KEY_TELEGRAM_TOKEN,
+    KEY_KUBERNETES_TOKEN,
+    KEY_REPORT_PORTAL_API_KEY,
+    KEY_ELASTICSEARCH_API_KEY,
+    KEY_LITE_LLM_API_KEY,
+    KEY_MYSQL_PASSWORD,
+    KEY_POSTGRES_PASSWORD,
+    KEY_MSSQL_PASSWORD,
+}
+
+
+def is_sensitive_key(key: str) -> bool:
+    """Check if a credential key contains sensitive information."""
+    return key in SENSITIVE_KEYS
+
+
+def mask_sensitive_value(value: str, show_real: bool = False) -> str:
+    """Mask sensitive credential values for display.
+
+    Args:
+        value: The value to potentially mask
+        show_real: If True, return the real value instead of masking
+    """
+    if show_real:
+        return value
+
+    if not value or len(value) < 4:
+        return "*" * len(value) if value else ""
+    return value[:2] + "*" * (len(value) - 4) + value[-2:]

codemie_test_harness/cli/utils.py

@@ -43,3 +43,12 @@ def ensure_env_from_config(keys: list[str]) -> None:
     for k in keys:
         if k not in os.environ and k in cfg:
             os.environ[k] = str(cfg[k])
+
+
+def unset_config_key(key: str) -> bool:
+    cfg = load_config()
+    if key in cfg:
+        del cfg[key]
+        save_config(cfg)
+        return True
+    return False

codemie_test_harness/tests/assistant/test_assistants.py

@@ -14,6 +14,9 @@ from codemie_sdk.models.assistant import (
     ChatRole,
 )
 from codemie_test_harness.tests import TEST_USER, PROJECT, LANGFUSE_TRACES_ENABLED
+from codemie_test_harness.tests.test_data.assistant_test_data import (
+    EXCEL_TOOL_TEST_DATA,
+)
 from codemie_test_harness.tests.test_data.file_test_data import (
     files_with_different_types_test_data,
     file_test_data,
@@ -390,3 +393,42 @@ def test_create_assistant_and_prompt_with_multiple_files(
         assistant, prompt, conversation_id=conversation_id
     )
     similarity_check.check_similarity(response, RESPONSE_FOR_TWO_FILES_UPLOADED)
+
+
+@pytest.mark.assistant
+@pytest.mark.file
+@pytest.mark.regression
+@pytest.mark.smoke
+@pytest.mark.parametrize("prompt,expected_response", EXCEL_TOOL_TEST_DATA)
+def test_excel_tool_extended_functionality(
+    assistant_utils, assistant, similarity_check, prompt, expected_response
+):
+    """
+    Test extended Excel tool functionality with various scenarios.
+
+    This test covers:
+    - Data extraction from visible sheets only
+    - All data including hidden sheets
+    - Sheet name listing functionality
+    - File statistics and structure analysis
+    - Single sheet extraction by index and name
+    - Data cleaning and normalization
+    - Hidden sheet visibility control
+    - Column structure and data type analysis
+    - Tabular structure normalization
+    - Multi-sheet comprehensive analysis
+
+    """
+    assistant_instance = assistant()
+
+    uploaded_file = assistant_utils.upload_file_to_chat(
+        FILES_PATH / "test_extended.xlsx"
+    )
+    file_url = uploaded_file.get("file_url")
+
+    # Send the prompt with the uploaded file
+    response = assistant_utils.ask_assistant(
+        assistant_instance, prompt, file_urls=[file_url]
+    )
+
+    similarity_check.check_similarity(response, expected_response)