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

codemie_test_harness/cli/cli.py

@@ -20,21 +20,9 @@ from .constants import (
     KEY_AUTH_CLIENT_SECRET,
     KEY_AUTH_REALM_NAME,
     KEY_CODEMIE_API_DOMAIN,
+    KEY_AUTH_USERNAME,
+    KEY_AUTH_PASSWORD,
     # integrations
-    KEY_GIT_ENV,
-    KEY_GITLAB_URL,
-    KEY_GITLAB_TOKEN,
-    KEY_GITLAB_PROJECT,
-    KEY_GITLAB_PROJECT_ID,
-    KEY_GITHUB_URL,
-    KEY_GITHUB_TOKEN,
-    KEY_GITHUB_PROJECT,
-    KEY_JIRA_URL,
-    KEY_JIRA_TOKEN,
-    KEY_JQL,
-    KEY_CONFLUENCE_URL,
-    KEY_CONFLUENCE_TOKEN,
-    KEY_CQL,
     DEFAULT_MARKS,
     DEFAULT_XDIST_N,
     DEFAULT_RERUNS,
@@ -45,6 +33,8 @@ from .utils import get_config_value, ensure_env_from_config
 from .runner import run_pytest
 from .commands.config_cmd import config_cmd
 from .commands.run_cmd import run_cmd
+from .commands.assistant_cmd import assistant_cmd
+from .commands.workflow_cmd import workflow_cmd
 
 
 @click.group(context_settings=CONTEXT_SETTINGS)
@@ -65,40 +55,12 @@ from .commands.run_cmd import run_cmd
     "--auth-client-secret", envvar=KEY_AUTH_CLIENT_SECRET, help="Auth client secret"
 )
 @click.option("--auth-realm-name", envvar=KEY_AUTH_REALM_NAME, help="Auth realm name")
+@click.option("--auth-username", envvar=KEY_AUTH_USERNAME, help="Auth username")
+@click.option("--auth-password", envvar=KEY_AUTH_PASSWORD, help="Auth password")
 @click.option(
     "--api-domain", envvar=KEY_CODEMIE_API_DOMAIN, help="CodeMie API domain URL"
 )
-# Integrations
-@click.option(
-    "--git-env",
-    envvar=KEY_GIT_ENV,
-    type=click.Choice(["gitlab", "github"], case_sensitive=False),
-    help="Git provider env: gitlab or github",
-)
-# GitLab
-@click.option("--gitlab-url", envvar=KEY_GITLAB_URL, help="GitLab base URL")
-@click.option("--gitlab-token", envvar=KEY_GITLAB_TOKEN, help="GitLab access token")
-@click.option("--gitlab-project", envvar=KEY_GITLAB_PROJECT, help="GitLab project URL")
-@click.option(
-    "--gitlab-project-id",
-    envvar=KEY_GITLAB_PROJECT_ID,
-    help="GitLab project id",
-    type=str,
-)
-# GitHub
-@click.option("--github-url", envvar=KEY_GITHUB_URL, help="GitHub base URL")
-@click.option("--github-token", envvar=KEY_GITHUB_TOKEN, help="GitHub access token")
-@click.option("--github-project", envvar=KEY_GITHUB_PROJECT, help="GitHub project URL")
-# JIRA
-@click.option("--jira-url", envvar=KEY_JIRA_URL, help="Jira base URL")
-@click.option("--jira-token", envvar=KEY_JIRA_TOKEN, help="Jira token")
-@click.option("--jql", envvar=KEY_JQL, help="JQL query string")
-# Confluence
-@click.option("--confluence-url", envvar=KEY_CONFLUENCE_URL, help="Confluence base URL")
-@click.option(
-    "--confluence-token", envvar=KEY_CONFLUENCE_TOKEN, help="Confluence token"
-)
-@click.option("--cql", envvar=KEY_CQL, help="CQL query string")
+# Integration credentials are set via 'codemie-test-harness config set' command
 @click.pass_context
 def cli(
     ctx: click.Context,
@@ -109,25 +71,18 @@ def cli(
     auth_client_id: Optional[str],
     auth_client_secret: Optional[str],
     auth_realm_name: Optional[str],
+    auth_username: Optional[str],
+    auth_password: Optional[str],
     api_domain: Optional[str],
-    git_env: Optional[str],
-    gitlab_url: Optional[str],
-    gitlab_token: Optional[str],
-    gitlab_project: Optional[str],
-    gitlab_project_id: Optional[str],
-    github_url: Optional[str],
-    github_token: Optional[str],
-    github_project: Optional[str],
-    jira_url: Optional[str],
-    jira_token: Optional[str],
-    jql: Optional[str],
-    confluence_url: Optional[str],
-    confluence_token: Optional[str],
-    cql: Optional[str],
 ):
     """CodeMie Test Harness CLI.
 
     Without subcommand it will run pytest using configured defaults.
+
+    Integration credentials should be set using:
+      codemie-test-harness config set KEY VALUE
+
+    Use 'codemie-test-harness config vars' to see all available credentials.
     """
     ctx.ensure_object(dict)
 
@@ -150,23 +105,10 @@ def cli(
         KEY_AUTH_SERVER_URL: auth_server_url,
         KEY_AUTH_CLIENT_ID: auth_client_id,
         KEY_AUTH_CLIENT_SECRET: auth_client_secret,
+        KEY_AUTH_USERNAME: auth_username,
+        KEY_AUTH_PASSWORD: auth_password,
         KEY_AUTH_REALM_NAME: auth_realm_name,
         KEY_CODEMIE_API_DOMAIN: api_domain,
-        # integrations
-        KEY_GIT_ENV: git_env,
-        KEY_GITLAB_URL: gitlab_url,
-        KEY_GITLAB_TOKEN: gitlab_token,
-        KEY_GITLAB_PROJECT: gitlab_project,
-        KEY_GITLAB_PROJECT_ID: gitlab_project_id,
-        KEY_GITHUB_URL: github_url,
-        KEY_GITHUB_TOKEN: github_token,
-        KEY_GITHUB_PROJECT: github_project,
-        KEY_JIRA_URL: jira_url,
-        KEY_JIRA_TOKEN: jira_token,
-        KEY_JQL: jql,
-        KEY_CONFLUENCE_URL: confluence_url,
-        KEY_CONFLUENCE_TOKEN: confluence_token,
-        KEY_CQL: cql,
     }
     for k, v in provided.items():
         if v is not None and v != "":
@@ -186,6 +128,8 @@ def cli(
 # Register subcommands
 cli.add_command(config_cmd)
 cli.add_command(run_cmd)
+cli.add_command(assistant_cmd)
+cli.add_command(workflow_cmd)
 
 
 if __name__ == "__main__":  # pragma: no cover

codemie_test_harness/cli/commands/assistant_cmd.py

@@ -0,0 +1,104 @@
+"""Assistant CLI commands for interactive chat with assistants."""
+
+from __future__ import annotations
+
+import uuid
+from typing import Optional
+
+import click
+
+import os
+
+from codemie_sdk.models.assistant import AssistantChatRequest
+from codemie_test_harness.tests.utils.client_factory import get_client
+from ..constants import CONSOLE
+
+
+@click.group(name="assistant")
+def assistant_cmd():
+    """Interact with CodeMie assistants."""
+    pass
+
+
+@assistant_cmd.command(name="chat")
+@click.option(
+    "--assistant-id",
+    required=True,
+    help="Assistant ID to chat with",
+)
+@click.option(
+    "--conversation-id",
+    help="Conversation ID to continue existing chat (optional)",
+)
+@click.option(
+    "--message",
+    "-m",
+    required=True,
+    help="Message to send to the assistant",
+)
+@click.option(
+    "--stream/--no-stream",
+    default=False,
+    help="Stream the response (default: no-stream)",
+)
+def chat_cmd(
+    assistant_id: str,
+    conversation_id: Optional[str],
+    message: str,
+    stream: bool,
+):
+    """Chat with a specific assistant.
+
+    Example:
+        codemie-test-harness assistant chat --assistant-id "asst_123" -m "Hello, how can you help me?"
+        codemie-test-harness assistant chat --assistant-id "asst_123" --conversation-id "conv_456" -m "Continue our conversation"
+    """
+    try:
+        client = get_client()
+
+        # Generate conversation ID if not provided
+        if not conversation_id:
+            conversation_id = str(uuid.uuid4())
+            CONSOLE.print(
+                f"[green]New conversation started with ID: {conversation_id}[/green]"
+            )
+        else:
+            CONSOLE.print(f"[blue]Continuing conversation: {conversation_id}[/blue]")
+
+        # Create chat request
+        langfuse_enabled = (
+            os.getenv("LANGFUSE_TRACES_ENABLED", "false").lower() == "true"
+        )
+        chat_request = AssistantChatRequest(
+            text=message,
+            conversation_id=conversation_id,
+            stream=stream,
+            metadata={"langfuse_traces_enabled": langfuse_enabled},
+        )
+
+        CONSOLE.print(
+            f"[yellow]Sending message to assistant {assistant_id}...[/yellow]"
+        )
+
+        # Send chat request
+        response = client.assistants.chat(
+            assistant_id=assistant_id, request=chat_request
+        )
+
+        # Display response
+        if hasattr(response, "generated") and response.generated:
+            CONSOLE.print("[green]Assistant Response:[/green]")
+            CONSOLE.print(response.generated)
+        elif hasattr(response, "task_id") and response.task_id:
+            CONSOLE.print(
+                f"[yellow]Background task created with ID: {response.task_id}[/yellow]"
+            )
+            CONSOLE.print(
+                "[blue]You can check the task status using the task ID[/blue]"
+            )
+        else:
+            CONSOLE.print(f"[cyan]Raw response:[/cyan] {response}")
+
+    except Exception as e:
+        CONSOLE.print(f"[red]Error chatting with assistant: {e}[/red]")
+        raise click.ClickException(str(e))