#!/bin/bash
C=$(docker ps --format "{{.Names}}" | grep "friend-deploy" | head -1)
FT=$(docker exec postgres_db psql -U knowledge -d yunwoai -t -A -c "SELECT friend_token FROM user_friend_agents WHERE instance_id='friend-abc1d454';")

echo "=== clear-cooldown ==="
docker exec $C sh -c "wget -qO- --timeout=10 --header=\"Authorization: Bearer $FT\" --method=POST http://127.0.0.1:8000/api/llm/models/1/clear-cooldown 2>/dev/null"; echo
docker exec $C sh -c "wget -qO- --timeout=10 --header=\"Authorization: Bearer $FT\" --method=POST http://127.0.0.1:8000/api/llm/models/2/clear-cooldown 2>/dev/null"; echo

add_model() {
  local mk="$1" dn="$2"
  echo "--- add $mk ---"
  docker exec $C sh -c "wget -qO- --timeout=15 --header=\"Authorization: Bearer $FT\" --header=\"Content-Type: application/json\" --body-data='{\"model_key\":\"$mk\",\"upstream_model_id\":\"$mk\",\"display_name\":\"$dn\",\"enabled\":true}' --method=POST http://127.0.0.1:8000/api/llm/channels/1/models 2>&1" | head -c 250
  echo
}

add_model "mistral-medium-latest" "mistral-medium-latest"
add_model "nvidia/nemotron-3-super-120b-a12b:free" "nemotron-3-super-120b"
add_model "openrouter/free" "openrouter-free"

echo "=== 渠道下全部模型 ==="
docker exec $C sh -c "wget -qO- --header=\"Authorization: Bearer $FT\" http://127.0.0.1:8000/api/llm/channels/1/models 2>/dev/null" | python3 -c "
import sys,json
try:
    for m in json.load(sys.stdin):
        r=m.get(\"runtime\") or {}
        print(m.get(\"id\"), m.get(\"model_key\"), \"state:\", r.get(\"state\"))
except Exception as e: print(\"err:\", e)
"