Timothy Zhang
74 lines
3.1 KiB
PowerShell
74 lines
3.1 KiB
PowerShell
#!/usr/bin/env pwsh
|
|
# Wrapper script for the Hive CLI (Windows).
|
|
# Uses uv to run the hive command in the project's virtual environment.
|
|
#
|
|
# On Windows, User-level environment variables (set via quickstart.ps1) are
|
|
# stored in the registry but may not be loaded into the current terminal
|
|
# session (VS Code terminals, Windows Terminal tabs, etc.). This script
|
|
# explicitly loads them before running the agent — the Windows equivalent
|
|
# of Linux shells sourcing ~/.bashrc.
|
|
|
|
$ErrorActionPreference = "Stop"
|
|
$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Definition
|
|
|
|
# ── Validate project directory ──────────────────────────────────────
|
|
|
|
if ((Get-Location).Path -ne $ScriptDir) {
|
|
Write-Error "hive must be run from the project directory.`nCurrent directory: $(Get-Location)`nExpected directory: $ScriptDir`n`nRun: cd $ScriptDir"
|
|
exit 1
|
|
}
|
|
|
|
if (-not (Test-Path (Join-Path $ScriptDir "pyproject.toml")) -or -not (Test-Path (Join-Path $ScriptDir "core"))) {
|
|
Write-Error "Not a valid Hive project directory: $ScriptDir"
|
|
exit 1
|
|
}
|
|
|
|
if (-not (Test-Path (Join-Path $ScriptDir ".venv"))) {
|
|
Write-Error "Virtual environment not found. Run .\quickstart.ps1 first to set up the project."
|
|
exit 1
|
|
}
|
|
|
|
# ── Ensure uv is available ──────────────────────────────────────────
|
|
|
|
if (-not (Get-Command uv -ErrorAction SilentlyContinue)) {
|
|
# Check default install location before giving up
|
|
$uvExe = Join-Path $env:USERPROFILE ".local\bin\uv.exe"
|
|
if (Test-Path $uvExe) {
|
|
$env:Path = (Split-Path $uvExe) + ";" + $env:Path
|
|
} else {
|
|
Write-Error "uv is not installed. Run .\quickstart.ps1 first."
|
|
exit 1
|
|
}
|
|
}
|
|
|
|
# ── Load environment variables from Windows Registry ────────────────
|
|
# Windows stores User-level env vars in the registry. New terminal
|
|
# sessions may not have them (especially VS Code integrated terminals).
|
|
# Load them explicitly so agents can find their API keys.
|
|
|
|
$configPath = Join-Path (Join-Path $env:USERPROFILE ".hive") "configuration.json"
|
|
if (Test-Path $configPath) {
|
|
try {
|
|
$config = Get-Content $configPath -Raw | ConvertFrom-Json
|
|
$envVarName = $config.llm.api_key_env_var
|
|
if ($envVarName) {
|
|
$val = [System.Environment]::GetEnvironmentVariable($envVarName, "User")
|
|
if ($val -and -not (Test-Path "Env:\$envVarName" -ErrorAction SilentlyContinue)) {
|
|
Set-Item -Path "Env:\$envVarName" -Value $val
|
|
}
|
|
}
|
|
} catch {
|
|
# Non-fatal: agent may still work if env vars are already set
|
|
}
|
|
}
|
|
|
|
# Load HIVE_CREDENTIAL_KEY for encrypted credential store
|
|
$credKey = [System.Environment]::GetEnvironmentVariable("HIVE_CREDENTIAL_KEY", "User")
|
|
if ($credKey -and -not $env:HIVE_CREDENTIAL_KEY) {
|
|
$env:HIVE_CREDENTIAL_KEY = $credKey
|
|
}
|
|
|
|
# ── Run the Hive CLI ────────────────────────────────────────────────
|
|
|
|
& uv run hive @args
|