From f4a551ef4959bd76c97f15f80b5bec0940f21670 Mon Sep 17 00:00:00 2001 From: Haitao Pan Date: Sat, 13 Jun 2026 07:44:15 +0800 Subject: [PATCH] fix: fallback litellm model counting --- api/metric_probe.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/api/metric_probe.go b/api/metric_probe.go index 30f8f9e..8c8d9a9 100644 --- a/api/metric_probe.go +++ b/api/metric_probe.go @@ -52,6 +52,10 @@ func countProcesses(names ...string) int { func countLiteLLMModels() int { out := commandOutput("sh", "-lc", "curl -fsS --max-time 1 http://127.0.0.1:4000/v1/models 2>/dev/null | python3 -c 'import json,sys; print(len(json.load(sys.stdin).get(\"data\", [])))' 2>/dev/null") + if count := parseIntOrZero(out); count > 0 { + return count + } + out = commandOutput("sh", "-lc", "grep -R \"^[[:space:]]*- model_name:\" \"$HOME/.local/share/xworkspace/litellm-config.yaml\" /etc/litellm*.yaml 2>/dev/null | wc -l") return parseIntOrZero(out) }