indent 0.1.26__py3-none-any.whl

exponent/commands/config_commands.py

@@ -0,0 +1,334 @@
+import json
+import sys
+
+import click
+
+from exponent.commands.common import (
+    redirect_to_login,
+    refresh_api_key_task,
+    run_until_complete,
+    set_login_complete,
+)
+from exponent.commands.settings import use_settings
+from exponent.commands.types import exponent_cli_group
+from exponent.core.config import Settings
+from exponent.core.graphql.client import GraphQLClient
+from exponent.core.graphql.get_chats_query import GET_CHATS_QUERY
+from exponent.core.graphql.subscriptions import AUTHENTICATED_USER_SUBSCRIPTION
+from exponent.utils.version import (
+    get_installed_metadata,
+    get_installed_version,
+    get_installer,
+    get_python_path,
+    get_sys_executable,
+)
+
+
+@exponent_cli_group()
+def config_cli() -> None:
+    """Manage Indent configuration settings."""
+    pass
+
+
+@config_cli.command(hidden=True)
+@use_settings
+def debug(
+    settings: Settings,
+) -> None:
+    click.echo("Settings:")
+    click.echo(settings)
+    click.echo("\nInstalled Version:")
+    click.echo(get_installed_version())
+    click.echo("\nPython Version:")
+    click.echo(sys.version)
+    click.echo("\nPython Path:")
+    click.echo(get_python_path())
+    click.echo("\nSys Executable Path:")
+    click.echo(get_sys_executable())
+    click.echo("\nInstaller:")
+    click.echo(get_installer())
+    click.echo("\nInstalled Metadata:")
+    click.echo(get_installed_metadata())
+
+
+@config_cli.command(hidden=True)
+@click.option(
+    "--set-git-warning-disabled",
+    is_flag=True,
+    help="Disable the git warning for running Indent from a non-git repository",
+)
+@click.option(
+    "--set-git-warning-enabled",
+    is_flag=True,
+    help="Enable the git warning for running Indent from a non-git repository",
+)
+@click.option(
+    "--set-auto-upgrade-enabled",
+    is_flag=True,
+    help="Enable automatic upgrades",
+)
+@click.option(
+    "--set-auto-upgrade-disabled",
+    is_flag=True,
+    help="Disable automatic upgrades",
+)
+@click.option(
+    "--set-base-api-url-override",
+    required=False,
+    hidden=True,
+    help="Override base API URL",
+)
+@click.option(
+    "--clear-base-api-url-override",
+    is_flag=True,
+    hidden=True,
+    help="Clear base API URL override",
+)
+@click.option(
+    "--set-base-ws-url-override",
+    required=False,
+    hidden=True,
+    help="Override base WS URL",
+)
+@click.option(
+    "--clear-base-ws-url-override",
+    is_flag=True,
+    hidden=True,
+    help="Clear base WS URL override",
+)
+@click.option(
+    "--set-use-default-colors",
+    is_flag=True,
+    help="Use default colors",
+)
+@click.option(
+    "--clear-use-default-colors",
+    is_flag=True,
+    help="Clear use default colors",
+)
+@use_settings
+def config(  # noqa: PLR0913
+    settings: Settings,
+    set_git_warning_disabled: bool,
+    set_git_warning_enabled: bool,
+    set_auto_upgrade_enabled: bool,
+    set_auto_upgrade_disabled: bool,
+    clear_base_api_url_override: bool,
+    clear_base_ws_url_override: bool,
+    set_use_default_colors: bool,
+    clear_use_default_colors: bool,
+    set_base_api_url_override: str | None = None,
+    set_base_ws_url_override: str | None = None,
+) -> None:
+    """Display current Indent configuration."""
+
+    num_options_set = sum(
+        [
+            set_git_warning_disabled,
+            set_git_warning_enabled,
+            set_auto_upgrade_enabled,
+            set_auto_upgrade_disabled,
+            clear_base_ws_url_override,
+            clear_base_api_url_override,
+            set_base_api_url_override is not None,
+            set_base_ws_url_override is not None,
+            set_use_default_colors,
+            clear_use_default_colors,
+        ]
+    )
+    if num_options_set == 0:
+        click.secho(
+            json.dumps(settings.get_config_file_settings(), indent=2),
+            fg="green",
+        )
+        return
+
+    if num_options_set > 1:
+        click.secho("Cannot set multiple options at the same time.", fg="red")
+        return
+
+    if set_git_warning_enabled:
+        settings.options.git_warning_disabled = False
+        settings.write_settings_to_config_file()
+
+        click.secho(
+            "Git warning enabled. Indent will now check for a git repository.\n",
+            fg="yellow",
+        )
+
+    if set_git_warning_disabled:
+        settings.options.git_warning_disabled = True
+        settings.write_settings_to_config_file()
+
+        click.secho(
+            "Git warning disabled. Indent will no longer check for a git repository.\n",
+            fg="yellow",
+        )
+
+    if set_auto_upgrade_enabled:
+        settings.options.auto_upgrade = True
+        settings.write_settings_to_config_file()
+
+        click.secho(
+            "Automatic upgrades enabled. Indent will now check for updates.\n",
+            fg="yellow",
+        )
+
+    if set_auto_upgrade_disabled:
+        settings.options.auto_upgrade = False
+        settings.write_settings_to_config_file()
+
+        click.secho(
+            "Automatic upgrades disabled. Indent will no longer check for updates.\n",
+            fg="yellow",
+        )
+
+    if clear_base_api_url_override:
+        settings.options.base_api_url_override = None
+        settings.write_settings_to_config_file()
+
+        click.secho(
+            "API URL override cleared.",
+            fg="yellow",
+        )
+
+    if clear_base_ws_url_override:
+        settings.options.base_ws_url_override = None
+        settings.write_settings_to_config_file()
+
+        click.secho(
+            "WS URL override cleared.",
+            fg="yellow",
+        )
+
+    if set_base_api_url_override:
+        settings.options.base_api_url_override = set_base_api_url_override
+        settings.write_settings_to_config_file()
+
+        click.secho(
+            "API URL override set.",
+            fg="yellow",
+        )
+
+    if set_base_ws_url_override:
+        settings.options.base_ws_url_override = set_base_ws_url_override
+        settings.write_settings_to_config_file()
+
+        click.secho(
+            "WS URL override set.",
+            fg="yellow",
+        )
+
+    if set_use_default_colors:
+        settings.options.use_default_colors = True
+        settings.write_settings_to_config_file()
+
+        click.secho(
+            "Use default colors set.",
+            fg="yellow",
+        )
+
+    if clear_use_default_colors:
+        settings.options.use_default_colors = False
+        settings.write_settings_to_config_file()
+
+
+@config_cli.command()
+@click.option("--key", help="Your Indent API Key")
+@use_settings
+def login(settings: Settings, key: str) -> None:
+    """Log in to Indent."""
+
+    if not key:
+        redirect_to_login(settings, "provided")
+        return
+
+    click.echo("Verifying API key...")
+    run_until_complete(
+        set_login_complete(key, settings.get_base_api_url(), settings.get_base_ws_url())
+    )
+    click.secho("Success!", fg="green")
+
+    click.echo(f"Saving API Key to {settings.config_file_path}")
+
+    if settings.api_key and settings.api_key != key:
+        click.confirm("Detected existing API Key, continue? ", default=True, abort=True)
+
+    settings.update_api_key(key)
+    settings.write_settings_to_config_file()
+
+    click.echo("API Key saved.")
+
+
+@config_cli.command(hidden=True)
+@use_settings
+def get_chats(
+    settings: Settings,
+) -> None:
+    if not settings.api_key:
+        redirect_to_login(settings)
+        return
+
+    run_until_complete(
+        get_chats_task(
+            api_key=settings.api_key,
+            base_api_url=settings.get_base_api_url(),
+            base_ws_url=settings.get_base_ws_url(),
+        )
+    )
+
+
+@config_cli.command(hidden=True)
+@use_settings
+def get_authenticated_user(
+    settings: Settings,
+) -> None:
+    if not settings.api_key:
+        redirect_to_login(settings)
+        return
+
+    run_until_complete(
+        get_authenticated_user_task(
+            api_key=settings.api_key,
+            base_api_url=settings.get_base_api_url(),
+            base_ws_url=settings.get_base_ws_url(),
+        )
+    )
+
+
+async def get_chats_task(
+    api_key: str,
+    base_api_url: str,
+    base_ws_url: str,
+) -> None:
+    graphql_client = GraphQLClient(api_key, base_api_url, base_ws_url)
+    result = await graphql_client.execute(GET_CHATS_QUERY)
+    click.echo(result)
+
+
+async def get_authenticated_user_task(
+    api_key: str,
+    base_api_url: str,
+    base_ws_url: str,
+) -> None:
+    graphql_client = GraphQLClient(api_key, base_api_url, base_ws_url)
+    async for it in graphql_client.subscribe(AUTHENTICATED_USER_SUBSCRIPTION):
+        click.echo(it)
+
+
+@config_cli.command(hidden=True)
+@use_settings
+def refresh_key(settings: Settings) -> None:
+    """Refresh your API key."""
+    if not settings.api_key:
+        redirect_to_login(settings)
+        return
+
+    click.echo("Refreshing API key...")
+    run_until_complete(
+        refresh_api_key_task(
+            api_key=settings.api_key,
+            base_api_url=settings.get_base_api_url(),
+            base_ws_url=settings.get_base_ws_url(),
+        )
+    )

exponent/commands/run_commands.py

@@ -0,0 +1,222 @@
+import asyncio
+import sys
+import time
+
+import click
+
+from exponent.commands.common import (
+    check_inside_git_repo,
+    check_running_from_home_directory,
+    check_ssl,
+    create_chat,
+    inside_ssh_session,
+    redirect_to_login,
+    start_client,
+)
+from exponent.commands.settings import use_settings
+from exponent.commands.types import exponent_cli_group
+from exponent.commands.utils import (
+    ConnectionTracker,
+    Spinner,
+    launch_exponent_browser,
+    print_exponent_message,
+)
+from exponent.core.config import Settings
+from exponent.core.remote_execution.client import (
+    REMOTE_EXECUTION_CLIENT_EXIT_INFO,
+    SwitchCLIChat,
+    WSDisconnected,
+)
+from exponent.core.remote_execution.types import ChatSource
+from exponent.core.remote_execution.utils import assert_unreachable
+from exponent.utils.version import check_exponent_version_and_upgrade
+
+try:
+    # this is an optional dependency for python <3.11
+    from async_timeout import timeout
+except ImportError:  # pragma: no cover
+    from asyncio import timeout
+
+
+@exponent_cli_group()
+def run_cli() -> None:
+    """Run AI-powered chat sessions."""
+    pass
+
+
+@run_cli.command()
+@click.option(
+    "--chat-id",
+    help="ID of an existing chat session to reconnect",
+    required=False,
+    prompt_required=False,
+    prompt="Enter the chat ID to reconnect to",
+)
+@click.option(
+    "--prompt",
+    help="Start a chat with a given prompt.",
+)
+@click.option(
+    "--workflow-id",
+    hidden=True,
+    required=False,
+)
+@click.option(
+    "--timeout-seconds",
+    type=int,
+    help="Number of seconds without receiving a request before shutting down",
+    envvar="INDENT_TIMEOUT_SECONDS",
+)
+@use_settings
+def run(
+    settings: Settings,
+    chat_id: str | None = None,
+    prompt: str | None = None,
+    workflow_id: str | None = None,
+    timeout_seconds: int | None = None,
+) -> None:
+    """[default] Start or reconnect to an Indent session."""
+    check_exponent_version_and_upgrade(settings)
+
+    if not settings.api_key:
+        redirect_to_login(settings)
+        return
+
+    loop = asyncio.get_event_loop()
+
+    check_running_from_home_directory()
+    check_ssl()
+    loop.run_until_complete(check_inside_git_repo(settings))
+
+    api_key = settings.api_key
+    base_url = settings.base_url
+    base_api_url = settings.get_base_api_url()
+    base_ws_url = settings.get_base_ws_url()
+
+    chat_uuid = chat_id or loop.run_until_complete(
+        create_chat(api_key, base_api_url, base_ws_url, ChatSource.CLI_RUN)
+    )
+
+    if isinstance(timeout_seconds, int) and timeout_seconds <= 0:
+        click.secho("Error: --timeout-seconds must be a positive integer", fg="red")
+        sys.exit(1)
+
+    if chat_uuid is None:
+        sys.exit(1)
+
+    if (
+        (not inside_ssh_session())
+        and (not workflow_id)
+        # If the user specified a chat ID, they probably don't want to re-launch the chat
+        and (not chat_id)
+    ):
+        # Open the chat in the browser
+        launch_exponent_browser(settings.environment, base_url, chat_uuid)
+
+    while True:
+        result = run_chat(
+            loop, api_key, chat_uuid, settings, prompt, workflow_id, timeout_seconds
+        )
+        if result is None or isinstance(result, WSDisconnected):
+            # NOTE: None here means that handle_connection_changes exited
+            # first. We should likely have a different message for this.
+            if result and result.error_message:
+                click.secho(f"Error: {result.error_message}", fg="red")
+                sys.exit(10)
+            else:
+                click.echo("Disconnected upon user request, shutting down...")
+                break
+        elif isinstance(result, SwitchCLIChat):
+            chat_uuid = result.new_chat_uuid
+            click.echo("\nSwitching chats...")
+        else:
+            assert_unreachable(result)
+
+
+def run_chat(
+    loop: asyncio.AbstractEventLoop,
+    api_key: str,
+    chat_uuid: str,
+    settings: Settings,
+    prompt: str | None,
+    workflow_id: str | None,
+    timeout_seconds: int | None,
+) -> REMOTE_EXECUTION_CLIENT_EXIT_INFO | None:
+    start_ts = time.time()
+    base_url = settings.base_url
+    base_api_url = settings.get_base_api_url()
+    base_ws_url = settings.get_base_ws_url()
+
+    print_exponent_message(base_url, chat_uuid)
+    click.echo()
+
+    connection_tracker = ConnectionTracker()
+
+    client_fut = loop.create_task(
+        start_client(
+            api_key,
+            base_url,
+            base_api_url,
+            base_ws_url,
+            chat_uuid,
+            prompt=prompt,
+            workflow_id=workflow_id,
+            connection_tracker=connection_tracker,
+            timeout_seconds=timeout_seconds,
+        )
+    )
+
+    conn_fut = loop.create_task(handle_connection_changes(connection_tracker, start_ts))
+
+    try:
+        done, _ = loop.run_until_complete(
+            asyncio.wait({client_fut, conn_fut}, return_when=asyncio.FIRST_COMPLETED)
+        )
+
+        if client_fut in done:
+            return client_fut.result()
+        else:
+            return None
+    finally:
+        for task in asyncio.all_tasks(loop):
+            task.cancel()
+
+        try:
+            loop.run_until_complete(asyncio.wait(asyncio.all_tasks(loop)))
+        except asyncio.CancelledError:
+            pass
+
+
+async def handle_connection_changes(
+    connection_tracker: ConnectionTracker, start_ts: float
+) -> None:
+    try:
+        async with timeout(5):
+            assert await connection_tracker.next_change()
+            click.echo(ready_message(start_ts))
+    except TimeoutError:
+        spinner = Spinner("Connecting...")
+        spinner.show()
+        assert await connection_tracker.next_change()
+        spinner.hide()
+        click.echo(ready_message(start_ts))
+
+    while True:
+        assert not await connection_tracker.next_change()
+
+        click.echo("Disconnected...", nl=False)
+        await asyncio.sleep(1)
+        spinner = Spinner("Reconnecting...")
+        spinner.show()
+        assert await connection_tracker.next_change()
+        spinner.hide()
+        click.echo("\x1b[1;32m✓ Reconnected", nl=False)
+        sys.stdout.flush()
+        await asyncio.sleep(1)
+        click.echo("\r\x1b[0m\x1b[2K", nl=False)
+        sys.stdout.flush()
+
+
+def ready_message(start_ts: float) -> str:
+    elapsed = round(time.time() - start_ts, 2)
+    return f"\x1b[32m✓\x1b[0m Ready in {elapsed}s"

exponent/commands/settings.py

@@ -0,0 +1,56 @@
+from collections.abc import Callable
+from functools import wraps
+from typing import Any
+
+import click
+
+from exponent.commands.utils import (
+    print_editable_install_forced_prod_warning,
+    print_editable_install_warning,
+)
+from exponent.core.config import (
+    Environment,
+    get_settings,
+    is_editable_install,
+)
+
+
+def use_settings(f: Callable[..., Any]) -> Callable[..., Any]:
+    @click.option(
+        "--prod",
+        is_flag=True,
+        hidden=True,
+        help="Use production URLs even if in editable mode",
+    )
+    @click.option(
+        "--staging",
+        is_flag=True,
+        hidden=True,
+        help="Use staging URLs even if in editable mode",
+    )
+    @wraps(f)
+    def decorated_function(*args: Any, **kwargs: Any) -> Any:
+        prod = kwargs.pop("prod", False)
+        staging = kwargs.pop("staging", False)
+        settings = get_settings(use_prod=prod, use_staging=staging)
+
+        if is_editable_install() and not (prod or staging):
+            assert settings.environment == Environment.development
+            print_editable_install_warning(settings)
+
+        return f(*args, settings=settings, **kwargs)
+
+    return decorated_function
+
+
+def use_prod_settings(f: Callable[..., Any]) -> Callable[..., Any]:
+    @wraps(f)
+    def decorated_function(*args: Any, **kwargs: Any) -> Any:
+        settings = get_settings(use_prod=True)
+
+        if is_editable_install():
+            print_editable_install_forced_prod_warning(settings)
+
+        return f(*args, settings=settings, **kwargs)
+
+    return decorated_function