1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
#!/bin/bash
#
# Copyright (C) 2026 Danilo M. <danix@danix.xyz>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# The one runnable check for ai-state.sh. It puts stub curl, pgrep,
# llamachat and imggen on PATH and a stub assistant.sh beside them, then runs
# the real script, so nothing here touches the live stack. T_* variables pick
# the state each stub reports.
#
# Usage: ./test-ai-state.sh (exit 0 = all passed)
set -u
here="$(cd "$(dirname "$0")" && pwd)"
stub="$(mktemp -d)"
trap 'rm -rf "$stub"' EXIT
# The last argument is the URL. /slots refuses a probe without
# autoload=false, because the real server would load the model for it.
cat > "$stub/curl" <<'STUB'
#!/bin/bash
url="${!#}"
case "$url" in
*/models)
case "${T_LLAMA:-down}" in
down) exit 7 ;;
unloaded) echo '{"data":[{"id":"A","status":{"value":"unloaded"}}]}' ;;
*) echo '{"data":[{"id":"A","status":{"value":"unloaded"}},{"id":"Gemma","status":{"value":"loaded"}}]}' ;;
esac ;;
*/slots)
[[ " $* " == *" autoload=false "* ]] || exit 99
[[ " $* " == *" model=Gemma "* ]] || exit 22
if [[ "$T_LLAMA" == busy ]]; then
echo '[{"id":0,"is_processing":false},{"id":1,"is_processing":true}]'
else
echo '[{"id":0,"is_processing":false}]'
fi ;;
*) exit 6 ;;
esac
STUB
cat > "$stub/pgrep" <<'STUB'
#!/bin/bash
case "$*" in
*sd-cli*)
[[ -n "${T_CLI:-}" ]] || exit 1
echo "16618 sd-cli --diffusion-model /data/SD/Krea-2-Turbo-Q6_K.gguf --vae /data/SD/vae/wan_2.1_vae.safetensors -p a prompt -m ignored" ;;
*sd-server*)
[[ -n "${T_SD:-}" ]] || exit 1
echo "4242 sd-server --listen-port 7860 --diffusion-model /data/SD/z_image_turbo-Q8_0.gguf --vae /data/SD/vae/flux1-ae.safetensors" ;;
*fanfictioner*)
[[ -n "${T_FANFIC:-}" ]] || exit 1
echo 31337 ;;
*) exit 1 ;;
esac
STUB
cat > "$stub/llamachat" <<'STUB'
#!/bin/bash
[[ "$1" == --ping && -n "${T_CHAT:-}" ]]
STUB
# The real imggen exits 0 either way and says which on stdout.
cat > "$stub/imggen" <<'STUB'
#!/bin/bash
[[ "$1" == status ]] || exit 1
if [[ -n "${T_IMG:-}" ]]; then echo '{"model": "realvis", "ready": true}'; else echo "daemon down"; fi
STUB
cat > "$stub/assistant.sh" <<'STUB'
#!/bin/bash
[[ "$1" == status && -n "${T_ASSIST:-}" ]]
STUB
chmod +x "$stub"/*
log="$stub/sd-server.log"
: > "$log"
pass=0
fail=0
check() {
local name="$1" want="$2" got="$3"
if [[ "$want" == "$got" ]]; then
pass=$((pass + 1))
else
fail=$((fail + 1))
printf 'FAIL: %s\n want: %q\n got: %q\n' "$name" "$want" "$got"
fi
}
# One line of the protocol, tab joined.
l() { local IFS=$'\t'; printf '%s\n' "$*"; }
run() {
env PATH="$stub:$PATH" AI_ASSISTANT="$stub/assistant.sh" AI_SD_LOG="$log" "$@" \
bash "$here/ai-state.sh"
}
apps_down="$(l app assistant stopped -; l app chat stopped -; l app imggen stopped -; l app fanfic stopped -)"
want="$(l engine llama down idle -; l engine sd down idle -)
$apps_down"
check "everything down" "$want" "$(run)"
run >/dev/null
check "everything down exits zero" 0 $?
got="$(run T_LLAMA=unloaded | head -n1)"
check "llama up, no model loaded" "$(l engine llama up idle -)" "$got"
got="$(run T_LLAMA=idle | head -n1)"
check "llama loaded, idle" "$(l engine llama up idle Gemma)" "$got"
got="$(run T_LLAMA=busy | head -n1)"
check "llama loaded, generating" "$(l engine llama up busy Gemma)" "$got"
touch "$log"
got="$(run T_SD=1 | sed -n 2p)"
check "sd running, log just written" "$(l engine sd up busy z_image_turbo-Q8_0.gguf)" "$got"
touch -d '1 minute ago' "$log"
got="$(run T_SD=1 | sed -n 2p)"
check "sd running, log quiet" "$(l engine sd up idle z_image_turbo-Q8_0.gguf)" "$got"
# A one-shot sd-cli (fanfictioner's) holds VRAM and is generating for as
# long as it runs, whatever the web UI is doing. Its prompt is free text, so
# a "-m" inside it must not be read as the model.
got="$(run T_CLI=1 | sed -n 2p)"
check "sd-cli running" "$(l engine sd up busy "Krea-2-Turbo-Q6_K.gguf (sd-cli)")" "$got"
got="$(run T_CLI=1 T_SD=1 | sed -n 2p)"
check "sd-cli wins over an idle web UI" "$(l engine sd up busy "Krea-2-Turbo-Q6_K.gguf (sd-cli)")" "$got"
want="$(l app assistant running -; l app chat running -; l app imggen running realvis; l app fanfic running 31337)"
got="$(run T_ASSIST=1 T_CHAT=1 T_IMG=1 T_FANFIC=1 | tail -n4)"
check "every app running" "$want" "$got"
printf '\n%d passed, %d failed\n' "$pass" "$fail"
[[ "$fail" -eq 0 ]]
|
