init

scripts/llm_fallback_check.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+import os
+import subprocess
+import sys
+
+
+def main() -> None:
+    code = r"""
+import asyncio
+from services.llm import LLMService
+
+async def run():
+    svc = LLMService()
+    text, source = await svc.reply_with_meta("测试降级")
+    print("source:", source)
+    print("reply:", text)
+
+asyncio.run(run())
+"""
+    env = os.environ.copy()
+    env["LLM_BASE_URL"] = "http://127.0.0.1:9"
+    env["LLM_TIMEOUT"] = "1"
+    proc = subprocess.run([sys.executable, "-c", code], env=env, capture_output=True, text=True)
+    print(proc.stdout.strip())
+    if proc.returncode != 0:
+        print(proc.stderr.strip())
+        raise SystemExit(proc.returncode)
+    if "source: fallback" not in proc.stdout:
+        raise SystemExit("fallback check failed: source is not fallback")
+    print("fallback check: ok")
+
+
+if __name__ == "__main__":
+    main()