diff --git a/pyproject.toml b/pyproject.toml index e36c1e6..0d7151a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -10,6 +10,7 @@ readme = "README.md" requires-python = ">=3.12" dependencies = [ "databricks-sql-connector>=3.6.0", + "pyyaml>=6.0", "questionary>=2.0.0", "tomlkit>=0.13.0", "typer>=0.12.0", diff --git a/src/ucode/agents/__init__.py b/src/ucode/agents/__init__.py index 14cfd1b..eba3a18 100644 --- a/src/ucode/agents/__init__.py +++ b/src/ucode/agents/__init__.py @@ -33,12 +33,13 @@ spinner, ) -from . import claude, codex, copilot, gemini, opencode, pi +from . import claude, codex, copilot, gemini, goose, opencode, pi _MODULES = { "codex": codex, "claude": claude, "gemini": gemini, + "goose": goose, "opencode": opencode, "copilot": copilot, "pi": pi, @@ -52,6 +53,7 @@ "claude-code": "claude", "gemini": "gemini", "gemini-cli": "gemini", + "goose": "goose", "opencode": "opencode", "copilot": "copilot", "pi": "pi", @@ -65,7 +67,7 @@ def normalize_tool(tool: str) -> str: normalized = TOOL_ALIASES.get(tool.strip().lower()) if not normalized: raise RuntimeError( - f"Unsupported tool '{tool}'. Use one of: codex, claude, gemini, opencode, copilot, pi." + f"Unsupported tool '{tool}'. Use one of: codex, claude, gemini, goose, opencode, copilot, pi." ) return normalized @@ -135,6 +137,16 @@ def install_tool_binary(tool: str, *, strict: bool = True, update_existing: bool raise RuntimeError(version_error) return True + if not package: + message = ( + f"`{binary}` is not installed. " + f"Install {spec['display']} and ensure `{binary}` is on your PATH." + ) + if strict: + raise RuntimeError(message) + print_warning(message) + return False + if not shutil.which("npm"): message = f"`{binary}` is not installed and npm is not available to install it." if strict: @@ -208,6 +220,8 @@ def configure_tool(tool: str, state: dict, model: str | None = None) -> dict: result = claude.write_tool_config(state, model) elif tool == "gemini": result = gemini.write_tool_config(state, model) + elif tool == "goose": + result = goose.write_tool_config(state, model) elif tool == "copilot": result = copilot.write_tool_config(state, model) elif tool == "pi": @@ -236,6 +250,8 @@ def check_gateway_endpoint(state: dict, tool: str) -> bool: return bool(state.get("gemini_models")) if tool == "copilot": return bool(state.get("claude_models")) or bool(state.get("codex_models")) + if tool == "goose": + return bool(state.get("claude_models")) or bool(state.get("gemini_models")) if tool == "pi": return ( bool(state.get("claude_models")) @@ -247,6 +263,7 @@ def check_gateway_endpoint(state: dict, tool: str) -> bool: _TOOL_DISCOVERY_SOURCES: dict[str, tuple[str, ...]] = { "claude": ("claude",), + "goose": ("claude", "gemini"), "opencode": ("claude", "gemini"), "codex": ("codex",), "gemini": ("gemini",), diff --git a/src/ucode/agents/goose.py b/src/ucode/agents/goose.py new file mode 100644 index 0000000..e40ed14 --- /dev/null +++ b/src/ucode/agents/goose.py @@ -0,0 +1,239 @@ +"""Goose agent: merges Databricks settings into ~/.config/goose/config.yaml. + +Goose has a built-in Databricks provider that reads DATABRICKS_HOST from the +config file and DATABRICKS_TOKEN from the environment (env var takes precedence +over keyring). We merge only the three keys we own into the existing config so +that user-defined extensions, preferences, and other settings are preserved. + +The token is injected as DATABRICKS_TOKEN at launch and refreshed every 30 +minutes so long-running sessions stay authenticated. + +Install goose from https://github.com/aaif-goose/goose — it ships as a native +binary (not an npm package), typically installed to ~/.local/bin via: + curl -fsSL https://github.com/aaif-goose/goose/releases/download/stable/download_cli.sh | bash +""" + +from __future__ import annotations + +import os +import signal +import subprocess +import threading +from pathlib import Path + +from ucode.config_io import ( + APP_DIR, + ToolSpec, + backup_existing_file, + deep_merge_dict, + read_yaml_safe, + write_yaml_file, +) +from ucode.databricks import ( + TOKEN_REFRESH_INTERVAL_SECONDS, + get_databricks_token, +) +from ucode.state import mark_tool_managed, save_state + +GOOSE_CONFIG_DIR = Path.home() / ".config" / "goose" +GOOSE_CONFIG_PATH = GOOSE_CONFIG_DIR / "config.yaml" +GOOSE_BACKUP_PATH = APP_DIR / "goose-config.backup.yaml" + +SPEC: ToolSpec = { + "binary": "goose", + "package": "", # not an npm package; install from https://github.com/aaif-goose/goose + "display": "Goose", + "config_path": GOOSE_CONFIG_PATH, + "backup_path": GOOSE_BACKUP_PATH, +} + +MANAGED_KEYS: list[str] = [ + "DATABRICKS_HOST", + "GOOSE_PROVIDER", + "GOOSE_MODEL", + "OAUTH_TOKEN", +] + +GOOSE_MCP_AUTH_ENV_KEY = "OAUTH_TOKEN" + + +def is_update_available() -> tuple[str, str] | None: + return None # no npm update check for native binary + + +def default_model(state: dict) -> str | None: + """Prefer Claude sonnet, then opus, then haiku; fall back to gemini.""" + claude_models = state.get("claude_models") or {} + for family in ("sonnet", "opus", "haiku"): + if claude_models.get(family): + return claude_models[family] + gemini_models = state.get("gemini_models") or [] + if gemini_models: + return gemini_models[0] + return None + + +def render_overlay(workspace: str, model: str) -> dict: + """Return only the keys ucode manages — merged into the existing config.""" + return { + "DATABRICKS_HOST": workspace, + "GOOSE_PROVIDER": "databricks", + "GOOSE_MODEL": model, + "extensions": { + "skills": { + "enabled": True, + "type": "platform", + "name": "skills", + "description": "Load and use skills from .claude/skills or .goose/skills directories", + "bundled": True, + "available_tools": [], + } + }, + } + + +def build_runtime_env(workspace: str, token: str) -> dict[str, str]: + env = os.environ.copy() + env["DATABRICKS_HOST"] = workspace + env["DATABRICKS_TOKEN"] = token + env["OAUTH_TOKEN"] = token + return env + + +def _mcp_slug(name: str) -> str: + return name.lower().replace("-", "_") + + +def build_mcp_server_entry(name: str, url: str, token: str = "") -> dict: + return { + "enabled": True, + "type": "streamable_http", + "name": name, + "description": f"Databricks MCP server: {name}", + "uri": url, + "envs": {GOOSE_MCP_AUTH_ENV_KEY: token}, + "env_keys": [], + "headers": {"Authorization": f"Bearer ${{{GOOSE_MCP_AUTH_ENV_KEY}}}"}, + "timeout": 300, + "bundled": None, + "available_tools": [], + } + + +def write_mcp_server_config(name: str, url: str, token: str = "") -> bool: + backup_existing_file(GOOSE_CONFIG_PATH, GOOSE_BACKUP_PATH) + existing = read_yaml_safe(GOOSE_CONFIG_PATH) + extensions = existing.get("extensions") + if not isinstance(extensions, dict): + extensions = {} + slug = _mcp_slug(name) + removed = slug in extensions + extensions[slug] = build_mcp_server_entry(name, url, token) + existing["extensions"] = extensions + write_yaml_file(GOOSE_CONFIG_PATH, existing) + return removed + + +def remove_mcp_server_config(name: str) -> bool: + existing = read_yaml_safe(GOOSE_CONFIG_PATH) + extensions = existing.get("extensions") + if not isinstance(extensions, dict): + return False + slug = _mcp_slug(name) + if slug not in extensions: + return False + extensions.pop(slug) + existing["extensions"] = extensions + write_yaml_file(GOOSE_CONFIG_PATH, existing) + return True + + +def write_tool_config( + state: dict, + model: str, + token: str | None = None, + *, + force_refresh: bool = False, +) -> tuple[dict, str]: + backup_existing_file(GOOSE_CONFIG_PATH, GOOSE_BACKUP_PATH) + if token is None: + token = get_databricks_token(state["workspace"], force_refresh=force_refresh) + overlay = render_overlay(state["workspace"], model) + existing = read_yaml_safe(GOOSE_CONFIG_PATH) + deep_merge_dict(existing, overlay) + extensions = existing.get("extensions") + if isinstance(extensions, dict): + for ext in extensions.values(): + if isinstance(ext, dict) and ext.get("type") == "streamable_http": + envs = ext.get("envs") + if isinstance(envs, dict): + envs[GOOSE_MCP_AUTH_ENV_KEY] = token + else: + ext["envs"] = {GOOSE_MCP_AUTH_ENV_KEY: token} + write_yaml_file(GOOSE_CONFIG_PATH, existing) + state = mark_tool_managed(state, "goose", MANAGED_KEYS) + save_state(state) + return state, token + + +def _refresh_token_once(state: dict, *, force_refresh: bool = False) -> tuple[str, str]: + model = default_model(state) + if not model: + raise RuntimeError("No Goose model is available on this workspace.") + _, token = write_tool_config(state, model, force_refresh=force_refresh) + return model, token + + +def _refresh_forever(state: dict, stop_event: threading.Event) -> None: + while not stop_event.wait(TOKEN_REFRESH_INTERVAL_SECONDS): + try: + _refresh_token_once(state, force_refresh=True) + except RuntimeError: + continue + + +def launch(state: dict, tool_args: list[str]) -> None: + model, token = _refresh_token_once(state) + env = build_runtime_env(state["workspace"], token) + + stop_event = threading.Event() + refresher = threading.Thread( + target=_refresh_forever, + args=(state, stop_event), + daemon=True, + ) + refresher.start() + + proc = subprocess.Popen(["goose", "session", *tool_args], env=env) + try: + returncode = proc.wait() + except KeyboardInterrupt: + proc.send_signal(signal.SIGINT) + returncode = proc.wait() + finally: + stop_event.set() + refresher.join(timeout=1) + + raise SystemExit(returncode) + + +def validate_cmd(binary: str) -> list[str]: + return [ + binary, + "run", + "--text", + "say hi in 5 words or less", + "--no-session", + "--max-turns", + "1", + ] + + +def validate_env(state: dict) -> dict[str, str]: + workspace = state.get("workspace") + if not workspace: + raise RuntimeError("No workspace configured.") + if not default_model(state): + raise RuntimeError("No Goose model is available on this workspace.") + token = get_databricks_token(workspace) + return build_runtime_env(workspace, token) diff --git a/src/ucode/cli.py b/src/ucode/cli.py index 1530609..b85dac6 100644 --- a/src/ucode/cli.py +++ b/src/ucode/cli.py @@ -59,9 +59,9 @@ from ucode.usage import usage as usage_report _DISCOVERY_CONSUMERS: dict[str, tuple[str, ...]] = { - "claude": ("claude", "opencode", "copilot", "pi"), + "claude": ("claude", "goose", "opencode", "copilot", "pi"), "codex": ("codex", "copilot", "pi"), - "gemini": ("gemini", "opencode", "pi"), + "gemini": ("gemini", "goose", "opencode", "pi"), } @@ -166,9 +166,16 @@ def configure_shared_state( print_success("Unity AI Gateway detected") want_claude = ( - fetch_all or "claude" in tools or "opencode" in tools or "copilot" in tools or "pi" in tools + fetch_all + or "claude" in tools + or "goose" in tools + or "opencode" in tools + or "copilot" in tools + or "pi" in tools + ) + want_gemini = ( + fetch_all or "gemini" in tools or "goose" in tools or "opencode" in tools or "pi" in tools ) - want_gemini = fetch_all or "gemini" in tools or "opencode" in tools or "pi" in tools want_codex = fetch_all or "codex" in tools or "copilot" in tools or "pi" in tools claude_reason: str | None = None @@ -496,7 +503,7 @@ def _launch_tool(tool_name: str, ctx: typer.Context) -> None: print_section(f"ucode with {TOOL_SPECS[tool]['display']}") if resolved_model: print_kv("Model", resolved_model) - if tool in ("gemini", "opencode", "copilot", "pi"): + if tool in ("gemini", "goose", "opencode", "copilot", "pi"): print_note( f"{TOOL_SPECS[tool]['display']} token refresh is managed automatically " f"every 30 minutes while the session is running." @@ -537,6 +544,12 @@ def opencode_cmd(ctx: typer.Context) -> None: _launch_tool("opencode", ctx) +@app.command("goose", context_settings={"allow_extra_args": True, "ignore_unknown_options": True}) +def goose_cmd(ctx: typer.Context) -> None: + """Launch Goose via Databricks.""" + _launch_tool("goose", ctx) + + @app.command("copilot", context_settings={"allow_extra_args": True, "ignore_unknown_options": True}) def copilot_cmd(ctx: typer.Context) -> None: """Launch GitHub Copilot CLI via Databricks.""" @@ -559,7 +572,7 @@ def configure( str | None, typer.Option( "--agent", - help="Configure only the named agent (e.g. claude, codex, gemini, opencode, copilot, pi).", + help="Configure only the named agent (e.g. claude, codex, gemini, goose, opencode, copilot, pi).", ), ] = None, agents: Annotated[ diff --git a/src/ucode/config_io.py b/src/ucode/config_io.py index 06446b9..c15f425 100644 --- a/src/ucode/config_io.py +++ b/src/ucode/config_io.py @@ -8,6 +8,7 @@ import tomlkit import tomlkit.exceptions +import yaml from ucode.ui import console @@ -171,3 +172,25 @@ def parse_dotenv(path: Path) -> dict[str, str]: def write_dotenv(path: Path, env: dict[str, str]) -> None: content = "".join(f'{key}="{val}"\n' for key, val in env.items()) write_text_file(path, content) + + +def read_yaml_safe(path: Path) -> dict: + if not path.exists(): + return {} + try: + data = yaml.safe_load(path.read_text(encoding="utf-8")) + except (OSError, yaml.YAMLError): + return {} + return data if isinstance(data, dict) else {} + + +def write_yaml_file(path: Path, data: dict) -> None: + content = yaml.dump(data, default_flow_style=False, allow_unicode=True, sort_keys=False) + if _dry_run: + console.print(f"\n[bold]\\[dry run] {path}[/bold]\n{content}") + return + ensure_parent_dir(path) + try: + path.write_text(content, encoding="utf-8") + except OSError as exc: + raise RuntimeError(f"Failed to write config file: {path}") from exc diff --git a/src/ucode/mcp.py b/src/ucode/mcp.py index 6cd6a63..643972f 100644 --- a/src/ucode/mcp.py +++ b/src/ucode/mcp.py @@ -22,10 +22,11 @@ from questionary.question import Question from questionary.styles import merge_styles_default -from ucode.agents import copilot, opencode +from ucode.agents import copilot, goose, opencode from ucode.config_io import restore_file from ucode.databricks import ( ensure_databricks_auth, + get_databricks_token, list_databricks_apps, list_databricks_connections, list_genie_spaces, @@ -63,6 +64,11 @@ "display": "OpenCode", "list_command": "opencode mcp list", }, + "goose": { + "binary": "goose", + "display": "Goose", + "list_command": "goose session --help", + }, "copilot": { "binary": "copilot", "display": "GitHub Copilot CLI", @@ -234,7 +240,9 @@ def configured_mcp_clients(state: dict, installed_clients: list[str]) -> list[st ] -def configure_client_mcp_server(client: str, name: str, url: str, entry: dict) -> list[str]: +def configure_client_mcp_server( + client: str, name: str, url: str, entry: dict, state: dict | None = None +) -> list[str]: if client == "claude": removed_scopes = [ scope for scope in MCP_CLEANUP_SCOPES if remove_claude_mcp_server(name, scope) @@ -255,6 +263,16 @@ def configure_client_mcp_server(client: str, name: str, url: str, entry: dict) - if client == "copilot": removed = copilot.write_mcp_server_config(name, url) return [MCP_USER_SCOPE] if removed else [] + if client == "goose": + token = "" + workspace = (state or {}).get("workspace") or "" + if workspace: + try: + token = get_databricks_token(workspace) + except RuntimeError: + pass + removed = goose.write_mcp_server_config(name, url, token=token) + return [MCP_USER_SCOPE] if removed else [] raise RuntimeError(f"Unsupported MCP client '{client}'.") @@ -269,6 +287,8 @@ def remove_client_mcp_server(client: str, name: str) -> list[str]: return [MCP_USER_SCOPE] if opencode.remove_mcp_server_config(name) else [] if client == "copilot": return [MCP_USER_SCOPE] if copilot.remove_mcp_server_config(name) else [] + if client == "goose": + return [MCP_USER_SCOPE] if goose.remove_mcp_server_config(name) else [] raise RuntimeError(f"Unsupported MCP client '{client}'.") @@ -717,6 +737,7 @@ def apply_mcp_server_changes( original_servers: list[dict], working_servers: list[dict], clients: list[str], + state: dict | None = None, ) -> bool: original_by_name = _servers_by_name(original_servers) working_by_name = _servers_by_name(working_servers) @@ -737,7 +758,7 @@ def apply_mcp_server_changes( continue entry = build_mcp_http_entry(url) for client in clients: - configure_client_mcp_server(client, name, url, entry) + configure_client_mcp_server(client, name, url, entry, state=state) changed = True return changed @@ -752,14 +773,14 @@ def configure_mcp_command() -> int: installed_clients = available_mcp_clients() if not installed_clients: raise RuntimeError( - "No supported MCP clients are installed. Install Claude, Codex, Gemini, OpenCode, " + "No supported MCP clients are installed. Install Claude, Codex, Gemini, Goose, OpenCode, " "or GitHub Copilot CLI." ) clients = configured_mcp_clients(state, installed_clients) if not clients: raise RuntimeError( "No configured MCP-capable coding agents are installed. Run `ucode configure` " - "for Codex, Claude, Gemini, OpenCode, or GitHub Copilot CLI first." + "for Codex, Claude, Gemini, Goose, OpenCode, or GitHub Copilot CLI first." ) configured_tools = set(state.get("available_tools") or []) missing_clients = [ @@ -836,7 +857,9 @@ def configure_mcp_command() -> int: ) working_names.add(entry_name) - changed = apply_mcp_server_changes(original_mcp_servers, working_mcp_servers, clients) + changed = apply_mcp_server_changes( + original_mcp_servers, working_mcp_servers, clients, state=state + ) if changed or original_mcp_servers != working_mcp_servers: state["mcp_servers"] = working_mcp_servers save_state(state) diff --git a/tests/test_agent_goose.py b/tests/test_agent_goose.py new file mode 100644 index 0000000..6338b93 --- /dev/null +++ b/tests/test_agent_goose.py @@ -0,0 +1,482 @@ +"""Tests for agents/goose.py.""" + +from __future__ import annotations + +from ucode.agents import goose + +WS = "https://example.databricks.com" + + +class TestGooseSpec: + def test_binary(self): + assert goose.SPEC["binary"] == "goose" + + def test_package_is_empty(self): + # Goose is a native binary, not an npm package. + assert goose.SPEC["package"] == "" + + def test_display(self): + assert goose.SPEC["display"] == "Goose" + + def test_config_path_is_yaml(self): + assert goose.SPEC["config_path"].name == "config.yaml" + + def test_config_path_under_dot_config_goose(self): + assert ".config/goose" in str(goose.SPEC["config_path"]) + + +class TestDefaultModel: + def test_prefers_claude_sonnet(self): + state = {"claude_models": {"sonnet": "s4", "opus": "o4", "haiku": "h4"}} + assert goose.default_model(state) == "s4" + + def test_falls_back_to_opus(self): + state = {"claude_models": {"opus": "o4", "haiku": "h4"}} + assert goose.default_model(state) == "o4" + + def test_falls_back_to_haiku(self): + state = {"claude_models": {"haiku": "h4"}} + assert goose.default_model(state) == "h4" + + def test_returns_none_when_no_claude_models(self): + assert goose.default_model({}) is None + + def test_falls_back_to_gemini(self): + state = {"gemini_models": ["gemini-pro"]} + assert goose.default_model(state) == "gemini-pro" + + def test_ignores_codex_models(self): + state = {"codex_models": ["gpt-5"]} + assert goose.default_model(state) is None + + +class TestRenderOverlay: + def test_sets_databricks_host(self): + overlay = goose.render_overlay(WS, "databricks-claude-sonnet-4-6") + assert overlay["DATABRICKS_HOST"] == WS + + def test_sets_goose_provider(self): + overlay = goose.render_overlay(WS, "databricks-claude-sonnet-4-6") + assert overlay["GOOSE_PROVIDER"] == "databricks" + + def test_sets_goose_model(self): + overlay = goose.render_overlay(WS, "databricks-claude-sonnet-4-6") + assert overlay["GOOSE_MODEL"] == "databricks-claude-sonnet-4-6" + + def test_contains_top_level_managed_keys(self): + overlay = goose.render_overlay(WS, "m") + assert {"DATABRICKS_HOST", "GOOSE_PROVIDER", "GOOSE_MODEL", "extensions"} <= set(overlay) + + def test_enables_skills_extension(self): + overlay = goose.render_overlay(WS, "m") + assert overlay["extensions"]["skills"]["enabled"] is True + + def test_skills_is_platform_type(self): + overlay = goose.render_overlay(WS, "m") + assert overlay["extensions"]["skills"]["type"] == "platform" + + +class TestBuildRuntimeEnv: + def test_inherits_path(self): + env = goose.build_runtime_env(WS, "tok") + assert "PATH" in env + + def test_sets_databricks_host(self): + env = goose.build_runtime_env(WS, "tok") + assert env["DATABRICKS_HOST"] == WS + + def test_sets_databricks_token(self): + env = goose.build_runtime_env(WS, "tok123") + assert env["DATABRICKS_TOKEN"] == "tok123" + + def test_sets_oauth_token(self): + env = goose.build_runtime_env(WS, "tok123") + assert env["OAUTH_TOKEN"] == "tok123" + + +class TestIsUpdateAvailable: + def test_returns_none(self): + assert goose.is_update_available() is None + + +class TestManagedKeys: + def test_includes_databricks_host(self): + assert "DATABRICKS_HOST" in goose.MANAGED_KEYS + + def test_includes_goose_provider(self): + assert "GOOSE_PROVIDER" in goose.MANAGED_KEYS + + def test_includes_goose_model(self): + assert "GOOSE_MODEL" in goose.MANAGED_KEYS + + def test_includes_oauth_token(self): + assert "OAUTH_TOKEN" in goose.MANAGED_KEYS + + +class TestValidateCmd: + def test_starts_with_binary(self): + cmd = goose.validate_cmd("goose") + assert cmd[0] == "goose" + + def test_uses_run_subcommand(self): + cmd = goose.validate_cmd("goose") + assert cmd[1] == "run" + + def test_has_text_flag(self): + cmd = goose.validate_cmd("goose") + assert "--text" in cmd + + def test_text_prompt_is_non_empty(self): + cmd = goose.validate_cmd("goose") + idx = cmd.index("--text") + assert cmd[idx + 1].strip() + + def test_has_no_session_flag(self): + cmd = goose.validate_cmd("goose") + assert "--no-session" in cmd + + def test_has_max_turns_1(self): + cmd = goose.validate_cmd("goose") + idx = cmd.index("--max-turns") + assert cmd[idx + 1] == "1" + + +class TestValidateEnv: + def test_raises_when_no_workspace(self): + import pytest + + with pytest.raises(RuntimeError, match="No workspace"): + goose.validate_env({}) + + def test_raises_when_no_models(self): + import pytest + + with pytest.raises(RuntimeError, match="No Goose model"): + goose.validate_env({"workspace": WS}) + + def test_returns_env_with_token(self, monkeypatch): + import ucode.agents.goose as goose_mod + + monkeypatch.setattr(goose_mod, "get_databricks_token", lambda ws: "tok-from-cli") + state = {"workspace": WS, "claude_models": {"sonnet": "databricks-claude-sonnet-4-6"}} + env = goose.validate_env(state) + assert env["DATABRICKS_TOKEN"] == "tok-from-cli" + assert env["DATABRICKS_HOST"] == WS + assert env["OAUTH_TOKEN"] == "tok-from-cli" + + +class TestBuildMcpServerEntry: + def test_is_streamable_http(self): + entry = goose.build_mcp_server_entry("databricks-sql", f"{WS}/api/2.0/mcp/sql") + assert entry["type"] == "streamable_http" + + def test_uses_url_as_uri(self): + url = f"{WS}/api/2.0/mcp/sql" + entry = goose.build_mcp_server_entry("databricks-sql", url) + assert entry["uri"] == url + + def test_is_enabled(self): + entry = goose.build_mcp_server_entry("databricks-sql", f"{WS}/api/2.0/mcp/sql") + assert entry["enabled"] is True + + def test_has_oauth_token_auth_header(self): + entry = goose.build_mcp_server_entry("databricks-sql", f"{WS}/api/2.0/mcp/sql") + assert "OAUTH_TOKEN" in entry["headers"]["Authorization"] + + def test_token_stored_in_envs(self): + entry = goose.build_mcp_server_entry("databricks-sql", f"{WS}/api/2.0/mcp/sql", "tok123") + assert entry["envs"]["OAUTH_TOKEN"] == "tok123" + + def test_env_keys_is_empty(self): + entry = goose.build_mcp_server_entry("databricks-sql", f"{WS}/api/2.0/mcp/sql") + assert entry["env_keys"] == [] + + +class TestMcpSlug: + def test_lowercases(self): + assert goose._mcp_slug("GitHub-MCP") == "github_mcp" + + def test_replaces_dashes_with_underscores(self): + assert goose._mcp_slug("databricks-sql") == "databricks_sql" + + +class TestWriteMcpServerConfig: + def test_writes_extension_to_config(self, tmp_path, monkeypatch): + import yaml + + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + + goose_mod.write_mcp_server_config("databricks-sql", f"{WS}/api/2.0/mcp/sql") + + written = yaml.safe_load(config_path.read_text()) + assert "databricks_sql" in written["extensions"] + assert written["extensions"]["databricks_sql"]["uri"] == f"{WS}/api/2.0/mcp/sql" + assert written["extensions"]["databricks_sql"]["type"] == "streamable_http" + + def test_returns_false_when_new_entry(self, tmp_path, monkeypatch): + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + + removed = goose_mod.write_mcp_server_config("databricks-sql", f"{WS}/api/2.0/mcp/sql") + assert removed is False + + def test_returns_true_when_replacing_existing(self, tmp_path, monkeypatch): + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + + goose_mod.write_mcp_server_config("databricks-sql", f"{WS}/api/2.0/mcp/sql") + removed = goose_mod.write_mcp_server_config("databricks-sql", f"{WS}/api/2.0/mcp/sql") + assert removed is True + + def test_preserves_existing_extensions(self, tmp_path, monkeypatch): + import yaml + + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + + config_path.write_text( + yaml.dump({"extensions": {"developer": {"enabled": True, "type": "builtin"}}}), + encoding="utf-8", + ) + goose_mod.write_mcp_server_config("databricks-sql", f"{WS}/api/2.0/mcp/sql") + + written = yaml.safe_load(config_path.read_text()) + assert written["extensions"]["developer"]["enabled"] is True + assert "databricks_sql" in written["extensions"] + + +class TestRemoveMcpServerConfig: + def test_removes_extension(self, tmp_path, monkeypatch): + import yaml + + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + + goose_mod.write_mcp_server_config("databricks-sql", f"{WS}/api/2.0/mcp/sql") + result = goose_mod.remove_mcp_server_config("databricks-sql") + + assert result is True + written = yaml.safe_load(config_path.read_text()) + assert "databricks_sql" not in (written.get("extensions") or {}) + + def test_returns_false_when_not_present(self, tmp_path, monkeypatch): + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + + result = goose_mod.remove_mcp_server_config("nonexistent") + assert result is False + + def test_preserves_other_extensions(self, tmp_path, monkeypatch): + import yaml + + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + + config_path.write_text( + yaml.dump( + { + "extensions": { + "databricks_sql": {"type": "streamable_http", "enabled": True}, + "developer": {"type": "builtin", "enabled": True}, + } + } + ), + encoding="utf-8", + ) + goose_mod.remove_mcp_server_config("databricks-sql") + + written = yaml.safe_load(config_path.read_text()) + assert written["extensions"]["developer"]["enabled"] is True + assert "databricks_sql" not in written["extensions"] + + +class TestWriteToolConfig: + def test_writes_yaml_config(self, tmp_path, monkeypatch): + import yaml + + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + import ucode.state as state_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + monkeypatch.setattr(state_mod, "STATE_PATH", tmp_path / "state.json") + monkeypatch.setattr(goose_mod, "get_databricks_token", lambda *a, **kw: "test-token") + + state = {"workspace": WS, "claude_models": {"sonnet": "databricks-claude-sonnet-4-6"}} + returned_state, token = goose_mod.write_tool_config(state, "databricks-claude-sonnet-4-6") + + assert config_path.exists() + written = yaml.safe_load(config_path.read_text()) + assert written["DATABRICKS_HOST"] == WS + assert written["GOOSE_PROVIDER"] == "databricks" + assert written["GOOSE_MODEL"] == "databricks-claude-sonnet-4-6" + assert token == "test-token" + assert "goose" in (returned_state.get("managed_configs") or {}) + assert written["extensions"]["skills"]["enabled"] is True + + def test_uses_explicit_token_when_provided(self, tmp_path, monkeypatch): + import yaml + + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + import ucode.state as state_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + monkeypatch.setattr(state_mod, "STATE_PATH", tmp_path / "state.json") + # get_databricks_token should NOT be called when token is passed explicitly + monkeypatch.setattr( + goose_mod, + "get_databricks_token", + lambda *a, **kw: (_ for _ in ()).throw( + AssertionError("should not call get_databricks_token") + ), + ) + + state = {"workspace": WS, "claude_models": {"sonnet": "databricks-claude-sonnet-4-6"}} + _, token = goose_mod.write_tool_config( + state, "databricks-claude-sonnet-4-6", token="explicit-tok" + ) + + assert token == "explicit-tok" + written = yaml.safe_load(config_path.read_text()) + assert written["GOOSE_MODEL"] == "databricks-claude-sonnet-4-6" + + def test_updates_model_on_reconfigure(self, tmp_path, monkeypatch): + import yaml + + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + import ucode.state as state_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + monkeypatch.setattr(state_mod, "STATE_PATH", tmp_path / "state.json") + monkeypatch.setattr(goose_mod, "get_databricks_token", lambda *a, **kw: "tok") + + state = {"workspace": WS, "claude_models": {"sonnet": "databricks-claude-sonnet-4-6"}} + goose_mod.write_tool_config(state, "databricks-claude-sonnet-4-6") + + # Reconfigure with a different model (e.g., workspace now has a newer endpoint). + goose_mod.write_tool_config(state, "databricks-claude-opus-4-7") + + written = yaml.safe_load(config_path.read_text()) + assert written["GOOSE_MODEL"] == "databricks-claude-opus-4-7" + + def test_preserves_existing_config_keys(self, tmp_path, monkeypatch): + import yaml + + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + import ucode.state as state_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + monkeypatch.setattr(state_mod, "STATE_PATH", tmp_path / "state.json") + monkeypatch.setattr(goose_mod, "get_databricks_token", lambda *a, **kw: "tok") + + # Pre-populate with user settings that should be preserved. + config_path.write_text( + yaml.dump({"GOOSE_MAX_TOKENS": 16000, "extensions": {"developer": {"enabled": True}}}), + encoding="utf-8", + ) + + state = {"workspace": WS, "claude_models": {"sonnet": "databricks-claude-sonnet-4-6"}} + goose_mod.write_tool_config(state, "databricks-claude-sonnet-4-6") + + written = yaml.safe_load(config_path.read_text()) + assert written["GOOSE_MAX_TOKENS"] == 16000 + assert written["extensions"]["developer"]["enabled"] is True + assert written["GOOSE_PROVIDER"] == "databricks" + + def test_refreshes_token_in_streamable_http_extension_envs(self, tmp_path, monkeypatch): + import yaml + + import ucode.agents.goose as goose_mod + import ucode.config_io as config_io_mod + import ucode.state as state_mod + + monkeypatch.setattr(config_io_mod, "APP_DIR", tmp_path) + config_path = tmp_path / "config.yaml" + backup_path = tmp_path / "goose-backup.yaml" + monkeypatch.setattr(goose_mod, "GOOSE_CONFIG_PATH", config_path) + monkeypatch.setattr(goose_mod, "GOOSE_BACKUP_PATH", backup_path) + monkeypatch.setattr(state_mod, "STATE_PATH", tmp_path / "state.json") + monkeypatch.setattr(goose_mod, "get_databricks_token", lambda *a, **kw: "new-token") + + config_path.write_text( + yaml.dump( + { + "extensions": { + "my_mcp": { + "type": "streamable_http", + "envs": {"OAUTH_TOKEN": "old-token"}, + "headers": {"Authorization": "Bearer ${OAUTH_TOKEN}"}, + } + } + } + ), + encoding="utf-8", + ) + + state = {"workspace": WS, "claude_models": {"sonnet": "databricks-claude-sonnet-4-6"}} + goose_mod.write_tool_config(state, "databricks-claude-sonnet-4-6") + + written = yaml.safe_load(config_path.read_text()) + assert written["extensions"]["my_mcp"]["envs"]["OAUTH_TOKEN"] == "new-token" diff --git a/tests/test_agents_init.py b/tests/test_agents_init.py index 15d1e36..06d264d 100644 --- a/tests/test_agents_init.py +++ b/tests/test_agents_init.py @@ -22,7 +22,15 @@ class TestToolSpecs: def test_all_tools_present(self): - assert set(TOOL_SPECS) == {"codex", "claude", "gemini", "opencode", "copilot", "pi"} + assert set(TOOL_SPECS) == { + "codex", + "claude", + "gemini", + "goose", + "opencode", + "copilot", + "pi", + } def test_each_spec_has_required_keys(self): required = {"binary", "package", "display", "config_path", "backup_path"} @@ -47,6 +55,7 @@ class TestNormalizeTool: ("claude-code", "claude"), ("gemini", "gemini"), ("gemini-cli", "gemini"), + ("goose", "goose"), ("opencode", "opencode"), ("copilot", "copilot"), ("pi", "pi"), @@ -70,6 +79,15 @@ def test_claude_unavailable_when_no_models(self): assert check_gateway_endpoint({"claude_models": {}}, "claude") is False assert check_gateway_endpoint({}, "claude") is False + def test_goose_available_with_claude(self): + assert check_gateway_endpoint({"claude_models": {"sonnet": "s4"}}, "goose") is True + + def test_goose_unavailable_without_models(self): + assert check_gateway_endpoint({"codex_models": ["m"]}, "goose") is False + + def test_goose_available_with_gemini(self): + assert check_gateway_endpoint({"gemini_models": ["gemini-pro"]}, "goose") is True + def test_codex_available(self): assert check_gateway_endpoint({"codex_models": ["model-a"]}, "codex") is True diff --git a/tests/test_cli.py b/tests/test_cli.py index 6b335a8..b44dbd0 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -510,7 +510,9 @@ def test_agents_flag_rejects_unknown(self): result = runner.invoke(app, ["configure", "--agents", "claude,bogus"]) assert result.exit_code != 0 assert "Unsupported tool 'bogus'" in result.output - assert "codex, claude, gemini, opencode, copilot, pi" in " ".join(result.output.split()) + assert "codex, claude, gemini, goose, opencode, copilot, pi" in " ".join( + result.output.split() + ) mock_cfg.assert_not_called() def test_agents_flag_rejects_empty_list(self): diff --git a/tests/test_mcp.py b/tests/test_mcp.py index 3e5f73a..b89e74c 100644 --- a/tests/test_mcp.py +++ b/tests/test_mcp.py @@ -10,7 +10,7 @@ WS = "https://example.databricks.com" CLAUDE_STATE = {"workspace": WS, "available_tools": ["claude"]} -ALL_MCP_CLIENTS = ["claude", "codex", "gemini", "opencode", "copilot"] +ALL_MCP_CLIENTS = ["claude", "codex", "gemini", "opencode", "copilot", "goose"] class TestBuildMcpHttpEntry: @@ -223,6 +223,40 @@ def test_configures_copilot_mcp_server(self, monkeypatch): assert calls == [("github", f"{WS}/api/2.0/mcp/external/github")] +class TestGooseMcpConfig: + def test_configure_writes_goose_extension(self, monkeypatch): + written: list[tuple[str, str]] = [] + monkeypatch.setattr( + mcp.goose, + "write_mcp_server_config", + lambda name, url, token="": written.append((name, url)) or False, + ) + mcp.configure_client_mcp_server("goose", "databricks-sql", f"{WS}/api/2.0/mcp/sql", {}) + assert written == [("databricks-sql", f"{WS}/api/2.0/mcp/sql")] + + def test_configure_returns_user_scope_when_replacing(self, monkeypatch): + monkeypatch.setattr(mcp.goose, "write_mcp_server_config", lambda name, url, token="": True) + scopes = mcp.configure_client_mcp_server( + "goose", "databricks-sql", f"{WS}/api/2.0/mcp/sql", {} + ) + assert scopes == ["user"] + + def test_configure_returns_empty_when_new(self, monkeypatch): + monkeypatch.setattr(mcp.goose, "write_mcp_server_config", lambda name, url, token="": False) + scopes = mcp.configure_client_mcp_server("goose", "new-server", f"{WS}/api/2.0/mcp/new", {}) + assert scopes == [] + + def test_remove_returns_user_scope_when_found(self, monkeypatch): + monkeypatch.setattr(mcp.goose, "remove_mcp_server_config", lambda name: True) + scopes = mcp.remove_client_mcp_server("goose", "databricks-sql") + assert scopes == ["user"] + + def test_remove_returns_empty_when_not_found(self, monkeypatch): + monkeypatch.setattr(mcp.goose, "remove_mcp_server_config", lambda name: False) + scopes = mcp.remove_client_mcp_server("goose", "nonexistent") + assert scopes == [] + + class TestMcpPicker: def test_prompt_uses_scrolling_checkbox_selector(self, monkeypatch): checkbox_calls: list[dict] = [] @@ -445,7 +479,7 @@ def test_registers_discovered_external_server(self, monkeypatch): monkeypatch.setattr(mcp, "discover_app_mcp_servers", lambda workspace, profile=None: []) _patch_mcp_choices(monkeypatch, f"{mcp.MCP_ADD_PREFIX}external:github-mcp") - def fake_configure_client_mcp_server(client, name, url, entry): + def fake_configure_client_mcp_server(client, name, url, entry, state=None): configured.append((client, name, url, entry)) return [] @@ -469,6 +503,7 @@ def fake_configure_client_mcp_server(client, name, url, entry): ("codex", "github-mcp", f"{WS}/api/2.0/mcp/external/github-mcp", expected_entry), ("gemini", "github-mcp", f"{WS}/api/2.0/mcp/external/github-mcp", expected_entry), ("opencode", "github-mcp", f"{WS}/api/2.0/mcp/external/github-mcp", expected_entry), + ("goose", "github-mcp", f"{WS}/api/2.0/mcp/external/github-mcp", expected_entry), ("copilot", "github-mcp", f"{WS}/api/2.0/mcp/external/github-mcp", expected_entry), ] assert saved_states[-1]["mcp_servers"] == [ @@ -476,7 +511,7 @@ def fake_configure_client_mcp_server(client, name, url, entry): "name": "github-mcp", "url": f"{WS}/api/2.0/mcp/external/github-mcp", "auth": "env:OAUTH_TOKEN", - "clients": ["claude", "codex", "gemini", "opencode", "copilot"], + "clients": ["claude", "codex", "gemini", "opencode", "goose", "copilot"], } ] @@ -507,7 +542,9 @@ def test_registers_discovered_genie_space_server(self, monkeypatch): monkeypatch.setattr( mcp, "configure_client_mcp_server", - lambda client, name, url, entry: configured.append((client, name, url, entry)) or [], + lambda client, name, url, entry, state=None: ( + configured.append((client, name, url, entry)) or [] + ), ) monkeypatch.setattr(mcp, "save_state", lambda state: saved_states.append(state.copy())) @@ -561,7 +598,9 @@ def test_registers_discovered_app_mcp_server(self, monkeypatch): monkeypatch.setattr( mcp, "configure_client_mcp_server", - lambda client, name, url, entry: configured.append((client, name, url, entry)) or [], + lambda client, name, url, entry, state=None: ( + configured.append((client, name, url, entry)) or [] + ), ) monkeypatch.setattr(mcp, "save_state", lambda state: saved_states.append(state.copy())) @@ -699,7 +738,9 @@ def test_continues_when_optional_discovery_fails(self, monkeypatch, capsys): monkeypatch.setattr( mcp, "configure_client_mcp_server", - lambda client, name, url, entry: configured.append((client, name, url, entry)) or [], + lambda client, name, url, entry, state=None: ( + configured.append((client, name, url, entry)) or [] + ), ) monkeypatch.setattr(mcp, "save_state", lambda state: saved_states.append(state.copy())) @@ -770,7 +811,9 @@ def test_configures_only_ucode_configured_clients(self, monkeypatch, capsys): monkeypatch.setattr( mcp, "configure_client_mcp_server", - lambda client, name, url, entry: configured.append((client, name, url, entry)) or [], + lambda client, name, url, entry, state=None: ( + configured.append((client, name, url, entry)) or [] + ), ) monkeypatch.setattr(mcp, "save_state", lambda state: saved_states.append(state.copy())) @@ -804,7 +847,9 @@ def test_registers_databricks_sql_server(self, monkeypatch): monkeypatch.setattr( mcp, "configure_client_mcp_server", - lambda client, name, url, entry: configured.append((client, name, url, entry)) or [], + lambda client, name, url, entry, state=None: ( + configured.append((client, name, url, entry)) or [] + ), ) monkeypatch.setattr(mcp, "save_state", lambda state: saved_states.append(state.copy())) diff --git a/uv.lock b/uv.lock index 15fd025..cbaf058 100644 --- a/uv.lock +++ b/uv.lock @@ -451,6 +451,52 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/10/99/781fe0c827be2742bcc775efefccb3b048a3a9c6ce9aec0cbf4a101677e5/pytz-2026.1.post1-py2.py3-none-any.whl", hash = "sha256:f2fd16142fda348286a75e1a524be810bb05d444e5a081f37f7affc635035f7a", size = 510489, upload-time = "2026-03-03T07:47:49.167Z" }, ] +[[package]] +name = "pyyaml" +version = "6.0.3" +source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/05/8e/961c0007c59b8dd7729d542c61a4d537767a59645b82a0b521206e1e25c2/pyyaml-6.0.3.tar.gz", hash = "sha256:d76623373421df22fb4cf8817020cbb7ef15c725b9d5e45f17e189bfc384190f", size = 130960, upload-time = "2025-09-25T21:33:16.546Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/d1/33/422b98d2195232ca1826284a76852ad5a86fe23e31b009c9886b2d0fb8b2/pyyaml-6.0.3-cp312-cp312-macosx_10_13_x86_64.whl", hash = "sha256:7f047e29dcae44602496db43be01ad42fc6f1cc0d8cd6c83d342306c32270196", size = 182063, upload-time = "2025-09-25T21:32:11.445Z" }, + { url = "https://files.pythonhosted.org/packages/89/a0/6cf41a19a1f2f3feab0e9c0b74134aa2ce6849093d5517a0c550fe37a648/pyyaml-6.0.3-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:fc09d0aa354569bc501d4e787133afc08552722d3ab34836a80547331bb5d4a0", size = 173973, upload-time = "2025-09-25T21:32:12.492Z" }, + { url = "https://files.pythonhosted.org/packages/ed/23/7a778b6bd0b9a8039df8b1b1d80e2e2ad78aa04171592c8a5c43a56a6af4/pyyaml-6.0.3-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:9149cad251584d5fb4981be1ecde53a1ca46c891a79788c0df828d2f166bda28", size = 775116, upload-time = "2025-09-25T21:32:13.652Z" }, + { url = "https://files.pythonhosted.org/packages/65/30/d7353c338e12baef4ecc1b09e877c1970bd3382789c159b4f89d6a70dc09/pyyaml-6.0.3-cp312-cp312-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:5fdec68f91a0c6739b380c83b951e2c72ac0197ace422360e6d5a959d8d97b2c", size = 844011, upload-time = "2025-09-25T21:32:15.21Z" }, + { url = "https://files.pythonhosted.org/packages/8b/9d/b3589d3877982d4f2329302ef98a8026e7f4443c765c46cfecc8858c6b4b/pyyaml-6.0.3-cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:ba1cc08a7ccde2d2ec775841541641e4548226580ab850948cbfda66a1befcdc", size = 807870, upload-time = "2025-09-25T21:32:16.431Z" }, + { url = "https://files.pythonhosted.org/packages/05/c0/b3be26a015601b822b97d9149ff8cb5ead58c66f981e04fedf4e762f4bd4/pyyaml-6.0.3-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:8dc52c23056b9ddd46818a57b78404882310fb473d63f17b07d5c40421e47f8e", size = 761089, upload-time = "2025-09-25T21:32:17.56Z" }, + { url = "https://files.pythonhosted.org/packages/be/8e/98435a21d1d4b46590d5459a22d88128103f8da4c2d4cb8f14f2a96504e1/pyyaml-6.0.3-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:41715c910c881bc081f1e8872880d3c650acf13dfa8214bad49ed4cede7c34ea", size = 790181, upload-time = "2025-09-25T21:32:18.834Z" }, + { url = "https://files.pythonhosted.org/packages/74/93/7baea19427dcfbe1e5a372d81473250b379f04b1bd3c4c5ff825e2327202/pyyaml-6.0.3-cp312-cp312-win32.whl", hash = "sha256:96b533f0e99f6579b3d4d4995707cf36df9100d67e0c8303a0c55b27b5f99bc5", size = 137658, upload-time = "2025-09-25T21:32:20.209Z" }, + { url = "https://files.pythonhosted.org/packages/86/bf/899e81e4cce32febab4fb42bb97dcdf66bc135272882d1987881a4b519e9/pyyaml-6.0.3-cp312-cp312-win_amd64.whl", hash = "sha256:5fcd34e47f6e0b794d17de1b4ff496c00986e1c83f7ab2fb8fcfe9616ff7477b", size = 154003, upload-time = "2025-09-25T21:32:21.167Z" }, + { url = "https://files.pythonhosted.org/packages/1a/08/67bd04656199bbb51dbed1439b7f27601dfb576fb864099c7ef0c3e55531/pyyaml-6.0.3-cp312-cp312-win_arm64.whl", hash = "sha256:64386e5e707d03a7e172c0701abfb7e10f0fb753ee1d773128192742712a98fd", size = 140344, upload-time = "2025-09-25T21:32:22.617Z" }, + { url = "https://files.pythonhosted.org/packages/d1/11/0fd08f8192109f7169db964b5707a2f1e8b745d4e239b784a5a1dd80d1db/pyyaml-6.0.3-cp313-cp313-macosx_10_13_x86_64.whl", hash = "sha256:8da9669d359f02c0b91ccc01cac4a67f16afec0dac22c2ad09f46bee0697eba8", size = 181669, upload-time = "2025-09-25T21:32:23.673Z" }, + { url = "https://files.pythonhosted.org/packages/b1/16/95309993f1d3748cd644e02e38b75d50cbc0d9561d21f390a76242ce073f/pyyaml-6.0.3-cp313-cp313-macosx_11_0_arm64.whl", hash = "sha256:2283a07e2c21a2aa78d9c4442724ec1eb15f5e42a723b99cb3d822d48f5f7ad1", size = 173252, upload-time = "2025-09-25T21:32:25.149Z" }, + { url = "https://files.pythonhosted.org/packages/50/31/b20f376d3f810b9b2371e72ef5adb33879b25edb7a6d072cb7ca0c486398/pyyaml-6.0.3-cp313-cp313-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:ee2922902c45ae8ccada2c5b501ab86c36525b883eff4255313a253a3160861c", size = 767081, upload-time = "2025-09-25T21:32:26.575Z" }, + { url = "https://files.pythonhosted.org/packages/49/1e/a55ca81e949270d5d4432fbbd19dfea5321eda7c41a849d443dc92fd1ff7/pyyaml-6.0.3-cp313-cp313-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:a33284e20b78bd4a18c8c2282d549d10bc8408a2a7ff57653c0cf0b9be0afce5", size = 841159, upload-time = "2025-09-25T21:32:27.727Z" }, + { url = "https://files.pythonhosted.org/packages/74/27/e5b8f34d02d9995b80abcef563ea1f8b56d20134d8f4e5e81733b1feceb2/pyyaml-6.0.3-cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:0f29edc409a6392443abf94b9cf89ce99889a1dd5376d94316ae5145dfedd5d6", size = 801626, upload-time = "2025-09-25T21:32:28.878Z" }, + { url = "https://files.pythonhosted.org/packages/f9/11/ba845c23988798f40e52ba45f34849aa8a1f2d4af4b798588010792ebad6/pyyaml-6.0.3-cp313-cp313-musllinux_1_2_aarch64.whl", hash = "sha256:f7057c9a337546edc7973c0d3ba84ddcdf0daa14533c2065749c9075001090e6", size = 753613, upload-time = "2025-09-25T21:32:30.178Z" }, + { url = "https://files.pythonhosted.org/packages/3d/e0/7966e1a7bfc0a45bf0a7fb6b98ea03fc9b8d84fa7f2229e9659680b69ee3/pyyaml-6.0.3-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:eda16858a3cab07b80edaf74336ece1f986ba330fdb8ee0d6c0d68fe82bc96be", size = 794115, upload-time = "2025-09-25T21:32:31.353Z" }, + { url = "https://files.pythonhosted.org/packages/de/94/980b50a6531b3019e45ddeada0626d45fa85cbe22300844a7983285bed3b/pyyaml-6.0.3-cp313-cp313-win32.whl", hash = "sha256:d0eae10f8159e8fdad514efdc92d74fd8d682c933a6dd088030f3834bc8e6b26", size = 137427, upload-time = "2025-09-25T21:32:32.58Z" }, + { url = "https://files.pythonhosted.org/packages/97/c9/39d5b874e8b28845e4ec2202b5da735d0199dbe5b8fb85f91398814a9a46/pyyaml-6.0.3-cp313-cp313-win_amd64.whl", hash = "sha256:79005a0d97d5ddabfeeea4cf676af11e647e41d81c9a7722a193022accdb6b7c", size = 154090, upload-time = "2025-09-25T21:32:33.659Z" }, + { url = "https://files.pythonhosted.org/packages/73/e8/2bdf3ca2090f68bb3d75b44da7bbc71843b19c9f2b9cb9b0f4ab7a5a4329/pyyaml-6.0.3-cp313-cp313-win_arm64.whl", hash = "sha256:5498cd1645aa724a7c71c8f378eb29ebe23da2fc0d7a08071d89469bf1d2defb", size = 140246, upload-time = "2025-09-25T21:32:34.663Z" }, + { url = "https://files.pythonhosted.org/packages/9d/8c/f4bd7f6465179953d3ac9bc44ac1a8a3e6122cf8ada906b4f96c60172d43/pyyaml-6.0.3-cp314-cp314-macosx_10_13_x86_64.whl", hash = "sha256:8d1fab6bb153a416f9aeb4b8763bc0f22a5586065f86f7664fc23339fc1c1fac", size = 181814, upload-time = "2025-09-25T21:32:35.712Z" }, + { url = "https://files.pythonhosted.org/packages/bd/9c/4d95bb87eb2063d20db7b60faa3840c1b18025517ae857371c4dd55a6b3a/pyyaml-6.0.3-cp314-cp314-macosx_11_0_arm64.whl", hash = "sha256:34d5fcd24b8445fadc33f9cf348c1047101756fd760b4dacb5c3e99755703310", size = 173809, upload-time = "2025-09-25T21:32:36.789Z" }, + { url = "https://files.pythonhosted.org/packages/92/b5/47e807c2623074914e29dabd16cbbdd4bf5e9b2db9f8090fa64411fc5382/pyyaml-6.0.3-cp314-cp314-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:501a031947e3a9025ed4405a168e6ef5ae3126c59f90ce0cd6f2bfc477be31b7", size = 766454, upload-time = "2025-09-25T21:32:37.966Z" }, + { url = "https://files.pythonhosted.org/packages/02/9e/e5e9b168be58564121efb3de6859c452fccde0ab093d8438905899a3a483/pyyaml-6.0.3-cp314-cp314-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:b3bc83488de33889877a0f2543ade9f70c67d66d9ebb4ac959502e12de895788", size = 836355, upload-time = "2025-09-25T21:32:39.178Z" }, + { url = "https://files.pythonhosted.org/packages/88/f9/16491d7ed2a919954993e48aa941b200f38040928474c9e85ea9e64222c3/pyyaml-6.0.3-cp314-cp314-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:c458b6d084f9b935061bc36216e8a69a7e293a2f1e68bf956dcd9e6cbcd143f5", size = 794175, upload-time = "2025-09-25T21:32:40.865Z" }, + { url = "https://files.pythonhosted.org/packages/dd/3f/5989debef34dc6397317802b527dbbafb2b4760878a53d4166579111411e/pyyaml-6.0.3-cp314-cp314-musllinux_1_2_aarch64.whl", hash = "sha256:7c6610def4f163542a622a73fb39f534f8c101d690126992300bf3207eab9764", size = 755228, upload-time = "2025-09-25T21:32:42.084Z" }, + { url = "https://files.pythonhosted.org/packages/d7/ce/af88a49043cd2e265be63d083fc75b27b6ed062f5f9fd6cdc223ad62f03e/pyyaml-6.0.3-cp314-cp314-musllinux_1_2_x86_64.whl", hash = "sha256:5190d403f121660ce8d1d2c1bb2ef1bd05b5f68533fc5c2ea899bd15f4399b35", size = 789194, upload-time = "2025-09-25T21:32:43.362Z" }, + { url = "https://files.pythonhosted.org/packages/23/20/bb6982b26a40bb43951265ba29d4c246ef0ff59c9fdcdf0ed04e0687de4d/pyyaml-6.0.3-cp314-cp314-win_amd64.whl", hash = "sha256:4a2e8cebe2ff6ab7d1050ecd59c25d4c8bd7e6f400f5f82b96557ac0abafd0ac", size = 156429, upload-time = "2025-09-25T21:32:57.844Z" }, + { url = "https://files.pythonhosted.org/packages/f4/f4/a4541072bb9422c8a883ab55255f918fa378ecf083f5b85e87fc2b4eda1b/pyyaml-6.0.3-cp314-cp314-win_arm64.whl", hash = "sha256:93dda82c9c22deb0a405ea4dc5f2d0cda384168e466364dec6255b293923b2f3", size = 143912, upload-time = "2025-09-25T21:32:59.247Z" }, + { url = "https://files.pythonhosted.org/packages/7c/f9/07dd09ae774e4616edf6cda684ee78f97777bdd15847253637a6f052a62f/pyyaml-6.0.3-cp314-cp314t-macosx_10_13_x86_64.whl", hash = "sha256:02893d100e99e03eda1c8fd5c441d8c60103fd175728e23e431db1b589cf5ab3", size = 189108, upload-time = "2025-09-25T21:32:44.377Z" }, + { url = "https://files.pythonhosted.org/packages/4e/78/8d08c9fb7ce09ad8c38ad533c1191cf27f7ae1effe5bb9400a46d9437fcf/pyyaml-6.0.3-cp314-cp314t-macosx_11_0_arm64.whl", hash = "sha256:c1ff362665ae507275af2853520967820d9124984e0f7466736aea23d8611fba", size = 183641, upload-time = "2025-09-25T21:32:45.407Z" }, + { url = "https://files.pythonhosted.org/packages/7b/5b/3babb19104a46945cf816d047db2788bcaf8c94527a805610b0289a01c6b/pyyaml-6.0.3-cp314-cp314t-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl", hash = "sha256:6adc77889b628398debc7b65c073bcb99c4a0237b248cacaf3fe8a557563ef6c", size = 831901, upload-time = "2025-09-25T21:32:48.83Z" }, + { url = "https://files.pythonhosted.org/packages/8b/cc/dff0684d8dc44da4d22a13f35f073d558c268780ce3c6ba1b87055bb0b87/pyyaml-6.0.3-cp314-cp314t-manylinux2014_s390x.manylinux_2_17_s390x.manylinux_2_28_s390x.whl", hash = "sha256:a80cb027f6b349846a3bf6d73b5e95e782175e52f22108cfa17876aaeff93702", size = 861132, upload-time = "2025-09-25T21:32:50.149Z" }, + { url = "https://files.pythonhosted.org/packages/b1/5e/f77dc6b9036943e285ba76b49e118d9ea929885becb0a29ba8a7c75e29fe/pyyaml-6.0.3-cp314-cp314t-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl", hash = "sha256:00c4bdeba853cc34e7dd471f16b4114f4162dc03e6b7afcc2128711f0eca823c", size = 839261, upload-time = "2025-09-25T21:32:51.808Z" }, + { url = "https://files.pythonhosted.org/packages/ce/88/a9db1376aa2a228197c58b37302f284b5617f56a5d959fd1763fb1675ce6/pyyaml-6.0.3-cp314-cp314t-musllinux_1_2_aarch64.whl", hash = "sha256:66e1674c3ef6f541c35191caae2d429b967b99e02040f5ba928632d9a7f0f065", size = 805272, upload-time = "2025-09-25T21:32:52.941Z" }, + { url = "https://files.pythonhosted.org/packages/da/92/1446574745d74df0c92e6aa4a7b0b3130706a4142b2d1a5869f2eaa423c6/pyyaml-6.0.3-cp314-cp314t-musllinux_1_2_x86_64.whl", hash = "sha256:16249ee61e95f858e83976573de0f5b2893b3677ba71c9dd36b9cf8be9ac6d65", size = 829923, upload-time = "2025-09-25T21:32:54.537Z" }, + { url = "https://files.pythonhosted.org/packages/f0/7a/1c7270340330e575b92f397352af856a8c06f230aa3e76f86b39d01b416a/pyyaml-6.0.3-cp314-cp314t-win_amd64.whl", hash = "sha256:4ad1906908f2f5ae4e5a8ddfce73c320c2a1429ec52eafd27138b7f1cbe341c9", size = 174062, upload-time = "2025-09-25T21:32:55.767Z" }, + { url = "https://files.pythonhosted.org/packages/f1/12/de94a39c2ef588c7e6455cfbe7343d3b2dc9d6b6b2f40c4c6565744c873d/pyyaml-6.0.3-cp314-cp314t-win_arm64.whl", hash = "sha256:ebc55a14a21cb14062aa4162f906cd962b28e2e9ea38f9b4391244cd8de4ae0b", size = 149341, upload-time = "2025-09-25T21:32:56.828Z" }, +] + [[package]] name = "questionary" version = "2.1.1" @@ -606,6 +652,7 @@ version = "0.1.0" source = { editable = "." } dependencies = [ { name = "databricks-sql-connector" }, + { name = "pyyaml" }, { name = "questionary" }, { name = "tomlkit" }, { name = "typer" }, @@ -621,6 +668,7 @@ dev = [ [package.metadata] requires-dist = [ { name = "databricks-sql-connector", specifier = ">=3.6.0" }, + { name = "pyyaml", specifier = ">=6.0" }, { name = "questionary", specifier = ">=2.0.0" }, { name = "tomlkit", specifier = ">=0.13.0" }, { name = "typer", specifier = ">=0.12.0" },