| 1 | |
| 2 | ZION_PM_DIR="/shared/projects/zion-pm" |
| 3 | SECRETS="/shared/projects/.env.secrets" |
| 4 | |
| 5 | if [ -f "$SECRETS" ]; then |
| 6 | set -a |
| 7 | source <(tr -d '\r' < "$SECRETS") |
| 8 | set +a |
| 9 | fi |
| 10 | |
| 11 | PM_BRAIN="${PM_BRAIN:-glm}" |
| 12 | |
| 13 | case "$PM_BRAIN" in |
| 14 | glm) |
| 15 | export ANTHROPIC_BASE_URL="https://api.z.ai/api/anthropic" |
| 16 | export ANTHROPIC_AUTH_TOKEN="${ZAI_API_KEY:?ZAI_API_KEY missing from .env.secrets}" |
| 17 | export ANTHROPIC_MODEL="${PM_MODEL:-glm-4.6}" |
| 18 | export ANTHROPIC_SMALL_FAST_MODEL="glm-4.5-air" |
| 19 | ;; |
| 20 | local) |
| 21 | export ANTHROPIC_BASE_URL="http://REDACTED-IP:11434" |
| 22 | export ANTHROPIC_AUTH_TOKEN="ollama" |
| 23 | export ANTHROPIC_MODEL="${PM_MODEL:-mistral-nemo:12b}" |
| 24 | export ANTHROPIC_SMALL_FAST_MODEL="mistral-nemo:12b" |
| 25 | ;; |
| 26 | max) |
| 27 | unset ANTHROPIC_BASE_URL ANTHROPIC_AUTH_TOKEN |
| 28 | export ANTHROPIC_MODEL="${PM_MODEL:-opus}" |
| 29 | ;; |
| 30 | *) |
| 31 | echo "unknown PM_BRAIN=$PM_BRAIN (use glm|local|max)" >&2 |
| 32 | ;; |
| 33 | esac |
| 34 | |
| 35 | [ "$PM_BRAIN" != "max" ] && unset ANTHROPIC_API_KEY |
| 36 | |
| 37 | export ZION_PM_DIR |
| 38 | echo "[pm] brain=$PM_BRAIN model=${ANTHROPIC_MODEL} base=${ANTHROPIC_BASE_URL:-<anthropic-default>}" >&2 |