tianshu-harness 3.23.0

package/completions/rivet.ps1

@@ -0,0 +1,230 @@
+# PowerShell completion for Rivet (rivet)
+#
+# Install: dot-source this file from your $PROFILE so the completer loads in every session:
+#   Add-Content $PROFILE ". C:\path\to\rivet.ps1"
+# Then restart PowerShell (or run `. .\rivet.ps1` once).
+#
+# Works on Windows PowerShell 5.1 and PowerShell 7+.
+
+function Get-RivetProviders {
+    $configFile = Join-Path $env:USERPROFILE ".rivet\config.json"
+    if (-not (Test-Path $configFile)) { return @() }
+    try {
+        $cfg = Get-Content $configFile -Raw | ConvertFrom-Json
+        if ($cfg.provider -and $cfg.provider.providers) {
+            return @($cfg.provider.providers.PSObject.Properties.Name)
+        }
+    } catch { }
+    return @()
+}
+
+# The completer is self-contained (all tables defined inside) so it does not depend
+# on outer-scope variable resolution when invoked by the PowerShell engine.
+$rivetCompleter = {
+    param($wordToComplete, $commandAst, $cursorPosition)
+
+    # NOTE: both helpers are always consumed via @(...) at the call sites. PowerShell
+    # unwraps a single-element array into a scalar, and `CompletionResult + array`
+    # throws op_Addition, so the @() cast around every call is load-bearing.
+    function Add-Match($map, $type, $prefix) {
+        $out = @()
+        foreach ($k in $map.Keys) {
+            if ($k -like "$prefix*") {
+                $out += [System.Management.Automation.CompletionResult]::new($k, $k, $type, $map[$k])
+            }
+        }
+        return $out
+    }
+
+    function Add-List($items, $type, $prefix) {
+        $out = @()
+        foreach ($k in $items) {
+            if ($k -like "$prefix*") {
+                $out += [System.Management.Automation.CompletionResult]::new($k, $k, $type, $k)
+            }
+        }
+        return $out
+    }
+
+    $topCommands = [ordered]@{
+        'config'   = 'Manage providers, API keys and models'
+        'serve'    = 'Start the sidecar HTTP/SSE runtime server'
+        'sessions' = 'Print the session list and exit'
+        'browser'  = 'Check or install the Chromium used by browser tools'
+        'logs'     = 'Show where sessions, caches and logs are written'
+    }
+
+    $topFlags = [ordered]@{
+        '-p'                             = 'Run a single prompt headlessly and print the result'
+        '--print'                        = 'Run a single prompt headlessly and print the result'
+        '--json'                         = 'Emit a single JSON result object'
+        '--stream-json'                  = 'Emit the run as an NDJSON event stream'
+        '--goal'                         = 'Run headless autonomous goal mode'
+        '--budget'                       = 'Turn budget for goal mode (default 100)'
+        '--model'                        = 'Override the model for this session'
+        '--provider'                     = 'Override the provider for this session'
+        '-c'                             = 'Resume the most recent session for this directory'
+        '--continue'                     = 'Resume the most recent session for this directory'
+        '-r'                             = 'Resume a session by id or prefix'
+        '--resume'                       = 'Resume a session by id or prefix'
+        '--new'                          = 'Force a brand-new session'
+        '--list'                         = 'Print the session list and exit'
+        '--dangerously-skip-permissions' = 'Skip all tool approvals'
+        '--screen-reader'                = 'Screen-reader friendly output'
+        '--skip-welcome'                 = 'Skip the welcome screen'
+        '--stream-events'                = 'Mirror the run as an NDJSON SessionEvent file'
+    }
+
+    $configCommands = [ordered]@{
+        'show'         = 'Print the resolved configuration as JSON'
+        'providers'    = 'List configured providers and key status'
+        'setup'        = 'Configure a provider in one shot'
+        'set-url'      = 'Set the base URL for a provider'
+        'set-model'    = 'Set the active model for a provider'
+        'set-key'      = 'Set the API key for a provider'
+        'set-key-env'  = 'Read the API key from an environment variable'
+        'set-default'  = 'Set the default provider'
+        'set-approval' = 'Set the tool approval mode'
+        'add-model'    = 'Add a model to a provider'
+        'remove-model' = 'Remove a model from a provider'
+        'mcp'          = 'Manage MCP servers'
+    }
+
+    $mcpCommands = [ordered]@{
+        'list'      = 'List configured MCP servers'
+        'add-stdio' = 'Add a stdio MCP server'
+        'add-sse'   = 'Add an SSE MCP server'
+        'remove'    = 'Remove an MCP server'
+        'enable'    = 'Enable an MCP server'
+        'disable'   = 'Disable an MCP server'
+    }
+
+    $approvalModes = [ordered]@{
+        'auto-safe'                      = 'Auto-approve read-only tools, ask for the rest'
+        'manual'                         = 'Ask before every tool call'
+        'auto-accept'                    = 'Auto-approve everything except destructive actions'
+        'dangerously-skip-permissions'   = 'Skip all approvals'
+    }
+
+    $setupFlags = [ordered]@{
+        '--key'            = 'API key literal'
+        '--key-env'        = 'Environment variable holding the key'
+        '--url'            = 'Base URL'
+        '--model'          = 'Model id'
+        '--context-window' = 'Context window size in tokens'
+        '--max-tokens'     = 'Max output tokens'
+        '--alias'          = 'Model alias'
+        '--default'        = 'Make this the default provider'
+    }
+
+    $browserCommands = [ordered]@{
+        'status'  = 'Report whether Chromium is ready'
+        'check'   = 'Alias for status'
+        'install' = 'Download and install Chromium'
+        'help'    = 'Show browser command help'
+    }
+
+    $logsFlags = [ordered]@{
+        '--session' = 'Restrict the report to one session'
+        '--json'    = 'Emit a structured manifest'
+    }
+
+    $providerVerbs = @('setup', 'set-url', 'set-model', 'set-key', 'set-key-env', 'set-default', 'add-model', 'remove-model')
+
+    # --- Tokenise the line, excluding the word currently being typed ---
+    $all = @(($commandAst.ToString().Trim()) -split '\s+' | Where-Object { $_ })
+    $tokens = @()
+    if ($all.Count -gt 1) { $tokens = @($all[1..($all.Count - 1)]) }
+    if ($wordToComplete -and $tokens.Count -gt 0 -and $tokens[-1] -eq $wordToComplete) {
+        if ($tokens.Count -eq 1) { $tokens = @() }
+        else { $tokens = @($tokens[0..($tokens.Count - 2)]) }
+    }
+
+    # A value-taking flag immediately before the cursor wins over positional logic.
+    if ($tokens.Count -gt 0) {
+        switch ($tokens[-1]) {
+            '--provider' { return @(Add-List (Get-RivetProviders) 'ParameterValue' $wordToComplete) }
+            '--model'    { return @() }
+            '--budget'   { return @() }
+            '--goal'     { return @() }
+            '--port'     { return @() }
+            '--session'  { return @() }
+        }
+    }
+
+    # Positional tokens only (flags filtered out), so interleaved flags do not shift positions.
+    $positional = @($tokens | Where-Object { $_ -notlike '-*' })
+
+    if ($positional.Count -eq 0) {
+        return @(Add-Match $topCommands 'Command' $wordToComplete) +
+               @(Add-Match $topFlags 'ParameterName' $wordToComplete)
+    }
+
+    switch ($positional[0]) {
+        'config' {
+            if ($positional.Count -eq 1) {
+                return @(Add-Match $configCommands 'Command' $wordToComplete)
+            }
+            $sub = $positional[1]
+            switch ($sub) {
+                'mcp' {
+                    if ($positional.Count -eq 2) {
+                        return @(Add-Match $mcpCommands 'Command' $wordToComplete)
+                    }
+                    return @()
+                }
+                'set-approval' {
+                    if ($positional.Count -eq 2) {
+                        return @(Add-Match $approvalModes 'ParameterValue' $wordToComplete)
+                    }
+                    return @()
+                }
+                'setup' {
+                    if ($positional.Count -eq 2) {
+                        return @(Add-List (Get-RivetProviders) 'ParameterValue' $wordToComplete) +
+                               @(Add-Match $setupFlags 'ParameterName' $wordToComplete)
+                    }
+                    return @(Add-Match $setupFlags 'ParameterName' $wordToComplete)
+                }
+            }
+            if ($positional.Count -eq 2 -and $providerVerbs -contains $sub) {
+                return @(Add-List (Get-RivetProviders) 'ParameterValue' $wordToComplete)
+            }
+            return @()
+        }
+        'serve' {
+            return @(Add-Match ([ordered]@{ '--port' = 'Port to listen on' }) 'ParameterName' $wordToComplete)
+        }
+        'browser' {
+            if ($positional.Count -eq 1) {
+                return @(Add-Match $browserCommands 'Command' $wordToComplete)
+            }
+            if ($positional[1] -eq 'install') {
+                return @(Add-Match ([ordered]@{ '--no-mirror' = 'Skip the download mirror' }) 'ParameterName' $wordToComplete)
+            }
+            return @()
+        }
+        'logs' {
+            if ($positional.Count -eq 1) {
+                return @(Add-Match ([ordered]@{ 'open' = 'Open the log location' }) 'Command' $wordToComplete) +
+                       @(Add-Match $logsFlags 'ParameterName' $wordToComplete)
+            }
+            if ($positional[1] -eq 'open' -and $positional.Count -eq 2) {
+                return @(Add-Match ([ordered]@{ 'desktop' = 'Open the desktop log directory' }) 'Command' $wordToComplete) +
+                       @(Add-Match $logsFlags 'ParameterName' $wordToComplete)
+            }
+            return @(Add-Match $logsFlags 'ParameterName' $wordToComplete)
+        }
+        'sessions' { return @() }
+    }
+
+    return @()
+}
+
+# -Native is required for external commands on PowerShell 7+, but is not a valid
+# switch on Windows PowerShell 5.1 (where native completion is implicit).
+$registerParams = @{ CommandName = 'rivet'; ScriptBlock = $rivetCompleter }
+if ((Get-Command Register-ArgumentCompleter).Parameters.ContainsKey('Native')) {
+    $registerParams['Native'] = $true
+}
+Register-ArgumentCompleter @registerParams

package/completions/rivet.zsh

@@ -0,0 +1,187 @@
+#compdef rivet
+#
+# Zsh completion for Rivet (rivet)
+#
+# Install: put this file somewhere on your $fpath as `_rivet`, e.g.
+#   mkdir -p ~/.zsh/completions && cp rivet.zsh ~/.zsh/completions/_rivet
+#   echo 'fpath=(~/.zsh/completions $fpath)' >> ~/.zshrc
+
+# Provider names are read from ~/.rivet/config.json when available.
+_rivet_providers() {
+  local config_file="$HOME/.rivet/config.json"
+  [[ -f $config_file ]] || return 0
+  local -a providers
+  providers=(${(f)"$(node -e '
+    try {
+      const c = JSON.parse(require("fs").readFileSync(process.argv[1], "utf-8"))
+      console.log(Object.keys((c.provider && c.provider.providers) || {}).join("\n"))
+    } catch {}
+  ' "$config_file" 2>/dev/null)"})
+  (( ${#providers} )) && _describe -t providers 'provider' providers
+}
+
+_rivet_config() {
+  local -a config_commands mcp_commands approval_modes
+
+  config_commands=(
+    'show:Print the resolved configuration as JSON'
+    'providers:List configured providers and key status'
+    'setup:Configure a provider in one shot'
+    'set-url:Set the base URL for a provider'
+    'set-model:Set the active model for a provider'
+    'set-key:Set the API key for a provider'
+    'set-key-env:Read the API key from an environment variable'
+    'set-default:Set the default provider'
+    'set-approval:Set the tool approval mode'
+    'add-model:Add a model to a provider'
+    'remove-model:Remove a model from a provider'
+    'mcp:Manage MCP servers'
+  )
+
+  mcp_commands=(
+    'list:List configured MCP servers'
+    'add-stdio:Add a stdio MCP server'
+    'add-sse:Add an SSE MCP server'
+    'remove:Remove an MCP server'
+    'enable:Enable an MCP server'
+    'disable:Disable an MCP server'
+  )
+
+  approval_modes=(
+    'auto-safe:Auto-approve read-only tools, ask for the rest'
+    'manual:Ask before every tool call'
+    'auto-accept:Auto-approve everything except destructive actions'
+    'dangerously-skip-permissions:Skip all approvals'
+  )
+
+  _arguments -C \
+    '1:config command:->config_cmd' \
+    '*::config arg:->config_args'
+
+  case $state in
+    config_cmd)
+      _describe -t commands 'config command' config_commands
+      ;;
+    config_args)
+      case $words[1] in
+        mcp)
+          if (( CURRENT == 2 )); then
+            _describe -t commands 'mcp command' mcp_commands
+          fi
+          ;;
+        set-approval)
+          if (( CURRENT == 2 )); then
+            _describe -t modes 'approval mode' approval_modes
+          fi
+          ;;
+        setup)
+          # `setup` may target a provider that does not exist yet, so the flags are
+          # offered alongside the known provider names at the first argument slot.
+          if (( CURRENT == 2 )); then
+            _rivet_providers
+            _arguments \
+              '--key[API key literal]:key:' \
+              '--key-env[Environment variable holding the key]:env var:' \
+              '--url[Base URL]:url:' \
+              '--model[Model id]:model:' \
+              '--context-window[Context window size]:tokens:' \
+              '--max-tokens[Max output tokens]:tokens:' \
+              '--alias[Model alias]:alias:' \
+              '--default[Make this the default provider]'
+          else
+            _arguments \
+              '--key[API key literal]:key:' \
+              '--key-env[Environment variable holding the key]:env var:' \
+              '--url[Base URL]:url:' \
+              '--model[Model id]:model:' \
+              '--context-window[Context window size]:tokens:' \
+              '--max-tokens[Max output tokens]:tokens:' \
+              '--alias[Model alias]:alias:' \
+              '--default[Make this the default provider]'
+          fi
+          ;;
+        set-url|set-model|set-key|set-key-env|set-default|add-model|remove-model)
+          if (( CURRENT == 2 )); then
+            _rivet_providers
+          fi
+          ;;
+      esac
+      ;;
+  esac
+}
+
+_rivet() {
+  local -a commands
+  commands=(
+    'config:Manage providers, API keys and models'
+    'serve:Start the sidecar HTTP/SSE runtime server'
+    'sessions:Print the session list and exit'
+    'browser:Check or install the Chromium used by browser tools'
+    'logs:Show where sessions, caches and logs are written'
+  )
+
+  _arguments -C \
+    '(-p --print)'{-p,--print}'[Run a single prompt headlessly and print the result]:prompt:' \
+    '--json[Emit a single JSON result object]' \
+    '--stream-json[Emit the run as an NDJSON event stream]' \
+    '--goal[Run headless autonomous goal mode]:goal:' \
+    '--budget[Turn budget for goal mode (default 100)]:turns:' \
+    '--model[Override the model for this session]:model:' \
+    '--provider[Override the provider for this session]:provider:_rivet_providers' \
+    '(-c --continue)'{-c,--continue}'[Resume the most recent session for this directory]' \
+    '(-r --resume)'{-r,--resume}'[Resume a session by id or prefix]:session id:' \
+    '--new[Force a brand-new session]' \
+    '--list[Print the session list and exit]' \
+    '--dangerously-skip-permissions[Skip all tool approvals]' \
+    '--screen-reader[Screen-reader friendly output]' \
+    '--skip-welcome[Skip the welcome screen]' \
+    '--stream-events[Mirror the run as an NDJSON SessionEvent file]:path:_files' \
+    '1:command:->command' \
+    '*::arg:->args'
+
+  case $state in
+    command)
+      _describe -t commands 'command' commands
+      ;;
+    args)
+      case $words[1] in
+        config)
+          _rivet_config
+          ;;
+        serve)
+          _arguments '--port[Port to listen on]:port:'
+          ;;
+        browser)
+          if (( CURRENT == 2 )); then
+            local -a browser_commands
+            browser_commands=(
+              'status:Report whether Chromium is ready'
+              'check:Alias for status'
+              'install:Download and install Chromium'
+              'help:Show browser command help'
+            )
+            _describe -t commands 'browser command' browser_commands
+          elif [[ $words[2] == install ]]; then
+            _arguments '--no-mirror[Skip the download mirror]'
+          fi
+          ;;
+        logs)
+          if (( CURRENT == 2 )); then
+            local -a logs_commands
+            logs_commands=('open:Open the log location')
+            _describe -t commands 'logs command' logs_commands
+            _arguments '--session[Restrict to one session]:session id:' '--json[Emit a structured manifest]'
+          elif [[ $words[2] == open ]] && (( CURRENT == 3 )); then
+            local -a open_targets
+            open_targets=('desktop:Open the desktop log directory')
+            _describe -t targets 'target' open_targets
+          else
+            _arguments '--session[Restrict to one session]:session id:' '--json[Emit a structured manifest]'
+          fi
+          ;;
+      esac
+      ;;
+  esac
+}
+
+_rivet "$@"

package/dist/account-OAEKORZE.js

@@ -0,0 +1,47 @@
+#!/usr/bin/env -S node --expose-gc --max-old-space-size=4096
+import { createRequire as __rivetCreateRequire } from 'node:module'; const require = __rivetCreateRequire(import.meta.url);
+import {
+  ACCOUNT_IDENTITY_TTL_MS,
+  accountApiBase,
+  accountIdentityUrl,
+  accountStore,
+  cachedAccountIdentity,
+  checkDeviceOnce,
+  deviceAuthorizeUrl,
+  fetchAccountProfile,
+  fetchStellarIdentity,
+  isAccountIdentityStale,
+  isTerminalPollStatus,
+  jwtSubject,
+  parseDeviceCreate,
+  parseDevicePoll,
+  parseStellarIdentity,
+  pollForAccountToken,
+  requestDeviceCode,
+  saveAccountIdentity,
+  saveAccountToken
+} from "./chunk-AEWHC4WM.js";
+import "./chunk-2VYRLULP.js";
+import "./chunk-P5KJURUR.js";
+import "./chunk-5GGOGLSN.js";
+export {
+  ACCOUNT_IDENTITY_TTL_MS,
+  accountApiBase,
+  accountIdentityUrl,
+  accountStore,
+  cachedAccountIdentity,
+  checkDeviceOnce,
+  deviceAuthorizeUrl,
+  fetchAccountProfile,
+  fetchStellarIdentity,
+  isAccountIdentityStale,
+  isTerminalPollStatus,
+  jwtSubject,
+  parseDeviceCreate,
+  parseDevicePoll,
+  parseStellarIdentity,
+  pollForAccountToken,
+  requestDeviceCode,
+  saveAccountIdentity,
+  saveAccountToken
+};

package/dist/account-login-V6FOLHJR.js

@@ -0,0 +1,65 @@
+#!/usr/bin/env -S node --expose-gc --max-old-space-size=4096
+import { createRequire as __rivetCreateRequire } from 'node:module'; const require = __rivetCreateRequire(import.meta.url);
+import {
+  init_esm_shims
+} from "./chunk-5GGOGLSN.js";
+
+// src/tui/account-login.ts
+init_esm_shims();
+import { hostname } from "os";
+async function handleAccountLogin(app) {
+  const {
+    requestDeviceCode,
+    pollForAccountToken,
+    accountStore,
+    saveAccountToken,
+    isTerminalPollStatus
+  } = await import("./account-OAEKORZE.js");
+  const { openInBrowser } = await import("./login-flow-CD4ZGXOC.js");
+  const { rivetHome } = await import("./paths-LVV5IMQN.js");
+  app.commitStatic("\u6B63\u5728\u5411\u5929\u67A2\u7533\u8BF7\u8BBE\u5907\u6388\u6743\u2026");
+  let created;
+  try {
+    created = await requestDeviceCode({ deviceName: hostname() });
+  } catch (e) {
+    app.commitStatic(`\u26A0\uFE0F \u65E0\u6CD5\u8FDE\u63A5\u5929\u67A2\u8D26\u53F7\u670D\u52A1\uFF1A${e.message}`);
+    return true;
+  }
+  openInBrowser(created.verifyUrl);
+  app.commitStatic(
+    `\u8BF7\u5728\u6D4F\u89C8\u5668\u5B8C\u6210\u6388\u6743\uFF08\u5DF2\u5C1D\u8BD5\u81EA\u52A8\u6253\u5F00\uFF09\uFF1A
+${created.verifyUrl}
+
+\u8BBE\u5907\u7801\uFF1A${created.userCode} \xB7 ${Math.round(created.expiresIn / 60)} \u5206\u949F\u5185\u6709\u6548
+\u7B49\u5F85\u6388\u6743\u4E2D\u2026\uFF08Ctrl+C \u53EF\u53D6\u6D88\uFF09`
+  );
+  const result = await pollForAccountToken(created.deviceCode, {
+    intervalSeconds: created.pollInterval,
+    timeoutMs: created.expiresIn * 1e3
+  });
+  if (result.status === "approved" && result.accessToken) {
+    const token = saveAccountToken(accountStore(rivetHome()), result);
+    const { primeStellarIdentity } = await import("./account-status-KHNPPAFL.js");
+    const identityLine = await primeStellarIdentity(token.accessToken);
+    app.commitStatic("\u2705 \u5DF2\u767B\u5F55\u5929\u67A2\u8D26\u53F7\u2014\u2014\u684C\u9762\u7AEF\u4E0E\u7F51\u9875\u7AEF\u73B0\u5728\u770B\u5230\u540C\u4E00\u4EFD\u8BBE\u5907\u4E0E\u8BA2\u9605\u3002");
+    if (identityLine) app.commitStatic(`\u661F\u7C4D\uFF1A${identityLine}`);
+    return true;
+  }
+  const hint = result.status === "denied" ? "\u4F60\u5728\u6D4F\u89C8\u5668\u91CC\u62D2\u7EDD\u4E86\u672C\u6B21\u6388\u6743\u3002" : result.status === "expired" ? "\u6388\u6743\u7801\u5DF2\u8FC7\u671F\uFF085 \u5206\u949F\u6709\u6548\uFF09\uFF0C\u8BF7\u91CD\u65B0\u6267\u884C /login account\u3002" : `\u672C\u6B21\u6388\u6743\u672A\u80FD\u5B8C\u6210\uFF08${result.status}\uFF09\u3002`;
+  void isTerminalPollStatus;
+  app.commitStatic(`\u26A0\uFE0F ${hint}`);
+  return true;
+}
+async function handleAccountLogout(app) {
+  const { accountStore } = await import("./account-OAEKORZE.js");
+  const { rivetHome } = await import("./paths-LVV5IMQN.js");
+  const store = accountStore(rivetHome());
+  const had = store.load() !== null;
+  store.clear();
+  app.commitStatic(had ? "\u2705 \u5DF2\u767B\u51FA\u5929\u67A2\u8D26\u53F7\u3002" : "\u5F53\u524D\u672A\u767B\u5F55\u5929\u67A2\u8D26\u53F7\u3002");
+  return true;
+}
+export {
+  handleAccountLogin,
+  handleAccountLogout
+};

package/dist/account-status-KHNPPAFL.js

@@ -0,0 +1,23 @@
+#!/usr/bin/env -S node --expose-gc --max-old-space-size=4096
+import { createRequire as __rivetCreateRequire } from 'node:module'; const require = __rivetCreateRequire(import.meta.url);
+import {
+  accountIdentityLines,
+  commitWelcomeStellarIdentityLine,
+  formatStellarIdentityLine,
+  primeStellarIdentity
+} from "./chunk-AOVQ7VIF.js";
+import "./chunk-AEWHC4WM.js";
+import "./chunk-CGOADM2X.js";
+import "./chunk-ADIVZ2D5.js";
+import "./chunk-2VYRLULP.js";
+import "./chunk-CGFZGR7G.js";
+import "./chunk-P5KJURUR.js";
+import "./chunk-WMUWEO2C.js";
+import "./chunk-BCQXEZPJ.js";
+import "./chunk-5GGOGLSN.js";
+export {
+  accountIdentityLines,
+  commitWelcomeStellarIdentityLine,
+  formatStellarIdentityLine,
+  primeStellarIdentity
+};