Initial Commit

tests/try-tool.sh

@@ -0,0 +1,60 @@
+#!/usr/bin/env bash
+# Call one tool by hand, the way Open WebUI would.
+#
+#   ./tests/try-tool.sh                     is the server up?
+#   ./tests/try-tool.sh web_search '{"query":"systemd socket activation"}'
+#   ./tests/try-tool.sh subnet_calc '{"network":"192.168.1.0/24"}'
+#
+# WHY THIS IS NOT `curl http://localhost:8000/...`
+#
+# Because that does not work, and worse, it can appear to. NOTHING in
+# this stack publishes a port to the host: the app container has no
+# default route and the gateway is only on Open WebUI's docker network.
+# A curl to localhost:8000 reaches whatever else happens to be listening
+# there - on the machine this was written on, something answered "OK" and
+# looked for all the world like a healthy tool server.
+#
+# So the call is made from inside the api container, which is where the
+# token already is. Same code path a real tool call takes, minus nginx.
+set -euo pipefail
+cd "$(dirname "$0")/.."
+
+TOOL="${1:-}"
+BODY="${2:-{\}}"
+
+if [ -z "$TOOL" ]; then
+  docker compose exec -T api python -c \
+    "import urllib.request;print(urllib.request.urlopen('http://127.0.0.1:8000/health').read().decode())"
+  exit 0
+fi
+
+docker compose exec -T -e TOOL="$TOOL" -e BODY="$BODY" api python - <<'PY'
+import json, os, sys, urllib.error, urllib.request
+
+tool, body = os.environ["TOOL"], os.environ["BODY"]
+try:
+    json.loads(body)
+except ValueError as exc:
+    sys.exit(f"the second argument is not valid JSON: {exc}")
+
+request = urllib.request.Request(
+    f"http://127.0.0.1:8000/tools/{tool}",
+    data=body.encode(),
+    headers={"Authorization": "Bearer " + os.environ["OWUITOOLS_AUTH_TOKEN"],
+             "Content-Type": "application/json"},
+)
+try:
+    answer = urllib.request.urlopen(request, timeout=120).read().decode()
+except urllib.error.HTTPError as exc:
+    if exc.code == 404:
+        sys.exit(f"there is no tool called {tool!r}. It may be switched off in "
+                 "configs/tools.toml - see docs/tools.md.")
+    sys.exit(f"HTTP {exc.code}: {exc.read().decode()[:400]}")
+
+# Tools return a JSON string; print what the model would actually read.
+try:
+    answer = json.loads(answer)
+except ValueError:
+    pass
+print(answer if isinstance(answer, str) else json.dumps(answer, indent=2))
+PY