ZacharyZhang-NYandClaude Fable 5
e2a473481d
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01QKxGW1raAWaqeU8WdHsMsp
77 lines
2.7 KiB
Bash
Executable File
77 lines
2.7 KiB
Bash
Executable File
#!/bin/bash
|
||
# omarchy:summary=Interactive wizard: pick harness, provider, model, store key, test
|
||
# omarchy:examples=omarchycn ai setup
|
||
|
||
set -euo pipefail
|
||
|
||
source "$OMARCHY_PATH/cn/lib/ai.sh"
|
||
|
||
if [[ ! -t 0 ]]; then
|
||
echo "omarchycn ai setup 需要交互终端；非交互场景用 omarchycn ai profile create" >&2
|
||
exit 1
|
||
fi
|
||
|
||
harness=$(cn_ai_harness_ids | gum choose --header "选择 Harness")
|
||
|
||
# Only combos whose adapter actually renders at launch
|
||
mapfile -t providers < <(jq -r --arg h "$harness" \
|
||
'.combos[$h] | to_entries[] | select(.value.adapter) | .key' "$CN_AI_COMPAT")
|
||
if (( ${#providers[@]} == 0 )); then
|
||
echo "兼容矩阵中没有 $harness 的可用 Provider 组合" >&2
|
||
exit 1
|
||
fi
|
||
provider=$(printf '%s\n' "${providers[@]}" | gum choose --header "选择 Provider（兼容级别见 docs）")
|
||
|
||
harness_proto=$(cn_ai_harness_field "$harness" protocol)
|
||
allowlist=$(cn_ai_combo_models "$harness" "$provider")
|
||
if [[ -n $allowlist ]]; then
|
||
model=$(gum choose --header "选择模型" <<<"$allowlist")
|
||
else
|
||
# Hide models bound to a different wire protocol than the harness speaks
|
||
model=$(jq -r --arg p "$provider" --arg hp "$harness_proto" \
|
||
'.providers[$p].models[] | select((.protocol // $hp) == $hp) | .id' \
|
||
"$CN_AI_PROVIDERS" | gum choose --header "选择模型")
|
||
fi
|
||
|
||
default_name="$harness-$provider"
|
||
name=$(gum input --header "Profile 名称" --value "$default_name")
|
||
|
||
proto=$(cn_ai_harness_field "$harness" protocol)
|
||
if [[ $proto == "native" ]]; then
|
||
omarchy-cn-ai-profile-create "$name" "$harness" "$provider" "$model"
|
||
omarchy-cn-ai-profile-use "$name"
|
||
echo "$harness 自管鉴权：首次启动时按其官方流程登录/配置"
|
||
echo "完成。启动: omarchycn ai launch"
|
||
exit 0
|
||
fi
|
||
|
||
if ! omarchy-cn-ai-secret get "$provider" > /dev/null 2>&1; then
|
||
key=$(gum input --password --header "$provider API Key（获取: $(jq -r --arg p "$provider" '.providers[$p].key_url' "$CN_AI_PROVIDERS")）")
|
||
printf '%s' "$key" | omarchy-cn-ai-secret set "$provider"
|
||
fi
|
||
|
||
# Keep any existing profile intact until the new config proves itself
|
||
profile_file="$CN_AI_PROFILE_DIR/$name.toml"
|
||
if [[ -f $profile_file ]]; then
|
||
cp "$profile_file" "$profile_file.omarchycn-prev"
|
||
fi
|
||
|
||
omarchy-cn-ai-profile-create "$name" "$harness" "$provider" "$model"
|
||
|
||
if gum confirm "运行连接测试？"; then
|
||
if ! omarchy-cn-ai-test "$name"; then
|
||
if [[ -f $profile_file.omarchycn-prev ]]; then
|
||
mv "$profile_file.omarchycn-prev" "$profile_file"
|
||
echo "测试未通过，已恢复原 Profile $name" >&2
|
||
else
|
||
rm -f "$profile_file"
|
||
echo "测试未通过，Profile $name 未保留" >&2
|
||
fi
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
rm -f "$profile_file.omarchycn-prev"
|
||
omarchy-cn-ai-profile-use "$name"
|
||
echo "完成。启动: omarchycn ai launch"
|