Add AI profiles and harness adapters (claude-code, codex, opencode, native)

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01QKxGW1raAWaqeU8WdHsMsp
2026-08-24 20:29:48 -04:00
parent 8c3ef3d884
commit 02d0d5c991
5 changed files with 283 additions and 0 deletions
@@ -0,0 +1,126 @@
+# shellcheck shell=bash
+# Sourced helpers for omarchy-cn-ai-* commands
+
+CN_AI_PROVIDERS="$OMARCHY_PATH/cn/registry/ai-providers.json"
+CN_AI_HARNESSES="$OMARCHY_PATH/cn/registry/ai-harnesses.json"
+CN_AI_COMPAT="$OMARCHY_PATH/cn/registry/ai-compatibility.json"
+CN_AI_PROFILE_DIR="$HOME/.config/omarchycn/ai/profiles"
+CN_AI_CURRENT="$HOME/.config/omarchycn/ai/current"
+
+cn_ai_provider_ids() { jq -r '.providers | keys[]' "$CN_AI_PROVIDERS"; }
+cn_ai_harness_ids() { jq -r '.harnesses | keys[]' "$CN_AI_HARNESSES"; }
+
+cn_ai_endpoint() {
+  jq -re --arg p "$1" --arg proto "$2" '.providers[$p].endpoints[$proto]' "$CN_AI_PROVIDERS"
+}
+
+cn_ai_models() {
+  jq -r --arg p "$1" '.providers[$p].models[].id' "$CN_AI_PROVIDERS"
+}
+
+cn_ai_model_by_alias() {
+  jq -re --arg p "$1" --arg a "$2" \
+    '.providers[$p].models[] | select(.aliases // [] | index($a)) | .id' "$CN_AI_PROVIDERS"
+}
+
+cn_ai_fast_model() {
+  cn_ai_model_by_alias "$1" fast 2>/dev/null || cn_ai_model_by_alias "$1" default-coding
+}
+
+cn_ai_harness_field() {
+  jq -re --arg h "$1" --arg f "$2" '.harnesses[$h][$f]' "$CN_AI_HARNESSES"
+}
+
+cn_ai_combo_level() {
+  jq -r --arg h "$1" --arg p "$2" '.combos[$h][$p].level // "unsupported"' "$CN_AI_COMPAT"
+}
+
+# Profile files are flat key="value" lines
+cn_ai_profile_field() {
+  local file="$CN_AI_PROFILE_DIR/$1.toml" key="$2"
+  grep -E "^$key = " "$file" | head -1 | cut -d\" -f2
+}
+
+cn_ai_current_profile() {
+  if [[ ! -f $CN_AI_CURRENT ]]; then
+    echo "No default AI profile set (run: omarchycn ai profile use <name>)" >&2
+    return 1
+  fi
+  cat "$CN_AI_CURRENT"
+}
+
+# Print export statements for the profile's harness x provider combo.
+# Secrets are resolved at launch time only; nothing is written to disk.
+cn_ai_render_env() {
+  local harness="$1" provider="$2" model="$3" key="$4"
+  local base fast
+
+  case "$harness" in
+  claude-code)
+    base=$(cn_ai_endpoint "$provider" anthropic)
+    fast=$(cn_ai_fast_model "$provider")
+    printf 'export ANTHROPIC_BASE_URL=%q\n' "$base"
+    printf 'export ANTHROPIC_AUTH_TOKEN=%q\n' "$key"
+    printf 'export ANTHROPIC_MODEL=%q\n' "$model"
+    printf 'export ANTHROPIC_DEFAULT_SONNET_MODEL=%q\n' "$model"
+    printf 'export ANTHROPIC_DEFAULT_OPUS_MODEL=%q\n' "$model"
+    printf 'export ANTHROPIC_DEFAULT_HAIKU_MODEL=%q\n' "$fast"
+    ;;
+  opencode)
+    if [[ $provider != "deepseek" ]]; then
+      echo "opencode adapter renders only the deepseek combo (native provider)" >&2
+      return 1
+    fi
+    printf 'export DEEPSEEK_API_KEY=%q\n' "$key"
+    ;;
+  *)
+    echo "No env renderer for harness: $harness" >&2
+    return 1
+    ;;
+  esac
+}
+
+# Write ~/.codex/config.toml provider block (official DeepSeek codex format)
+cn_ai_render_codex_config() {
+  local provider="$1" model="$2" key="$3"
+  local base cfg="$HOME/.codex/config.toml"
+
+  if [[ $provider != "deepseek" ]]; then
+    echo "codex adapter renders only the deepseek combo (official integration)" >&2
+    return 1
+  fi
+  base="https://api.deepseek.com/"
+
+  mkdir -p "${cfg%/*}"
+  touch "$cfg"
+  if [[ -s $cfg ]]; then
+    cp "$cfg" "$cfg.omarchycn-bak-$(date +%Y%m%d-%H%M%S)"
+  fi
+
+  # Top-level keys must precede the first [section]; provider table goes last.
+  # Strip our old blocks and any pre-section model keys, keep the rest intact.
+  awk '
+    /^# OmarchyCN ai (model|provider) begin$/ { skip = 1; next }
+    /^# OmarchyCN ai (model|provider) end$/ { skip = 0; next }
+    skip { next }
+    /^\[/ { in_section = 1 }
+    !in_section && /^(model|model_provider) = / { next }
+    { print }
+  ' "$cfg" > "$cfg.omarchycn-tmp"
+
+  {
+    echo "# OmarchyCN ai model begin"
+    echo "model = \"$model\""
+    echo "model_provider = \"$provider\""
+    echo "# OmarchyCN ai model end"
+    cat "$cfg.omarchycn-tmp"
+    echo "# OmarchyCN ai provider begin"
+    echo "[model_providers.$provider]"
+    echo "name = \"$provider\""
+    echo "base_url = \"$base\""
+    echo "experimental_bearer_token = \"$key\""
+    echo "# OmarchyCN ai provider end"
+  } > "$cfg"
+  rm -f "$cfg.omarchycn-tmp"
+  chmod 600 "$cfg"
+}