Compare commits
9 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 6fa98dd655 | |||
| a7b3c44f1c | |||
| 840341d2dd | |||
| 8748fe53bc | |||
| 8d8d8fac65 | |||
| 3dd01e5313 | |||
| d2c2962ad1 | |||
| c5375b8a77 | |||
| e05438f0e3 |
@@ -0,0 +1,29 @@
|
||||
import urllib.request, json, threading, queue, sys
|
||||
BASE="http://192.168.88.184:8080"
|
||||
q=queue.Queue()
|
||||
def sse():
|
||||
try:
|
||||
r=urllib.request.urlopen(BASE+"/sse", timeout=30)
|
||||
for raw in r:
|
||||
s=raw.decode(errors="replace").strip()
|
||||
if s.startswith("data:"): q.put(s[5:].strip())
|
||||
except Exception as e: q.put("ERR:"+str(e))
|
||||
threading.Thread(target=sse,daemon=True).start()
|
||||
try:
|
||||
ep=q.get(timeout=10)
|
||||
if ep.startswith("ERR:"): print("stash unreachable:",ep); sys.exit(1)
|
||||
purl=BASE+ep if ep.startswith("/") else ep
|
||||
def post(o):
|
||||
urllib.request.urlopen(urllib.request.Request(purl,data=json.dumps(o).encode(),headers={"Content-Type":"application/json"}),timeout=15).read()
|
||||
post({"jsonrpc":"2.0","id":1,"method":"initialize","params":{"protocolVersion":"2024-11-05","capabilities":{},"clientInfo":{"name":"apropos","version":"1"}}})
|
||||
q.get(timeout=10)
|
||||
post({"jsonrpc":"2.0","method":"notifications/initialized"})
|
||||
query=sys.argv[1] if len(sys.argv)>1 else ""
|
||||
limit=int(sys.argv[2]) if len(sys.argv)>2 else 3
|
||||
post({"jsonrpc":"2.0","id":2,"method":"tools/call","params":{"name":"recall","arguments":{"query":query,"limit":limit}}})
|
||||
d=json.loads(q.get(timeout=25))
|
||||
txt=d.get("result",{}).get("content",[{}])[0].get("text","[]")
|
||||
facts=json.loads(txt)
|
||||
if not facts: print("(no memory found for: %s)"%query); sys.exit(0)
|
||||
for f in facts: print("- %s (score %.2f)" % (f.get("content","").strip(), f.get("score",0)))
|
||||
except Exception as e: print("recall error:",e); sys.exit(1)
|
||||
@@ -939,7 +939,7 @@ local function _check_auth(self, conn)
|
||||
if not self._auth_token then return true end
|
||||
if conn.method == "OPTIONS" then return true end
|
||||
local auth = conn.headers["authorization"] or ""
|
||||
local token = auth:match("^Bearer%s+(.+)$")
|
||||
local token = auth:match("^[Bb]earer%s+(.+)$")
|
||||
return token == self._auth_token
|
||||
end
|
||||
|
||||
|
||||
+114
-6
@@ -174,22 +174,39 @@ local function run(cmd, timeout_sec)
|
||||
end
|
||||
return output and output ~= "" and output or "(no output)"
|
||||
else
|
||||
-- POSIX: use shell backgrounding + wait with timeout
|
||||
-- sh -c '(cmd > out 2>&1; echo $? > done) &' then poll
|
||||
-- POSIX: run in its OWN session/process group (setsid) so a
|
||||
-- timeout or cancel can kill the WHOLE tree instead of orphaning
|
||||
-- backgrounded children (the classic "shell timed out, children
|
||||
-- kept thrashing" bug). $! is the setsid leader pid == pgid.
|
||||
local pid_file = base .. ".pid"
|
||||
local sh_cmd = string.format(
|
||||
"(%s) > '%s' 2>&1; echo $? > '%s'",
|
||||
cmd, out_file, done_file
|
||||
)
|
||||
os.execute("sh -c '" .. sh_cmd:gsub("'", "'\\''") .. "' &")
|
||||
os.execute("setsid sh -c '" .. sh_cmd:gsub("'", "'\\''")
|
||||
.. "' & echo $! > '" .. pid_file .. "'")
|
||||
local pgid = (read_file(pid_file) or ""):match("(%d+)")
|
||||
remove_silent(pid_file)
|
||||
|
||||
local completed = poll_loop()
|
||||
|
||||
-- Timeout or cancel -> kill the entire process group. No orphans.
|
||||
if not completed and pgid then
|
||||
os.execute("kill -TERM -" .. pgid .. " 2>/dev/null")
|
||||
sleep_ms(300)
|
||||
os.execute("kill -KILL -" .. pgid .. " 2>/dev/null")
|
||||
end
|
||||
|
||||
local output = read_file(out_file)
|
||||
remove_silent(out_file)
|
||||
remove_silent(done_file)
|
||||
|
||||
if not completed then
|
||||
if cancelled then return "(cancelled)" end
|
||||
return output or ("Error: command timed out after " .. timeout_sec .. "s")
|
||||
if cancelled then return "(cancelled -- process group killed)" end
|
||||
return (output and output ~= "" and (output .. "\n") or "")
|
||||
.. "Error: command timed out after " .. timeout_sec
|
||||
.. "s -- the process group was KILLED (nothing is still running). "
|
||||
.. "For a long-running command, re-run it with shell_bg."
|
||||
end
|
||||
return output and output ~= "" and output or "(no output)"
|
||||
end
|
||||
@@ -200,6 +217,13 @@ end
|
||||
local server_name = os.getenv("LMCP_NAME") or (WINDOWS and "windows-tools" or "linux-tools")
|
||||
local server = lmcp.new(server_name, {
|
||||
port = tonumber(os.getenv("LMCP_PORT") or arg[1]) or 8080,
|
||||
-- LMCP_HOST: bind interface (default 0.0.0.0). Hosts that need
|
||||
-- single-interface binding (hertz: 192.168.88.18 only) set this.
|
||||
host = os.getenv("LMCP_HOST"),
|
||||
-- LMCP_CONF: path to a conf file with bearer-token entries
|
||||
-- (e.g. /opt/herding/etc/hertz-tools.conf). Read by lmcp.lua's
|
||||
-- read_conf; the `.godparticle` entry becomes the bearer token.
|
||||
conf = os.getenv("LMCP_CONF"),
|
||||
})
|
||||
|
||||
-- ---- Tools ----
|
||||
@@ -276,7 +300,15 @@ server:tool("shell_bg",
|
||||
f:close()
|
||||
os.remove(pid_file)
|
||||
end
|
||||
return string.format("launched pid=%s log=%s", pid, log)
|
||||
-- register so list_jobs/kill_job can see and reap it (no more reboots)
|
||||
if pid ~= "?" then
|
||||
local reg = io.open("/tmp/lmcp-bg-jobs.tsv", "a")
|
||||
if reg then
|
||||
reg:write(pid.."\t"..log.."\t"..os.date("%Y-%m-%dT%H:%M:%S").."\t"..inner:gsub("[\t\n]"," ").."\n")
|
||||
reg:close()
|
||||
end
|
||||
end
|
||||
return string.format("launched pid=%s log=%s (kill with kill_job pid=%s)", pid, log, pid)
|
||||
end, {
|
||||
annotations = {
|
||||
title = "Run shell (background)",
|
||||
@@ -287,6 +319,42 @@ server:tool("shell_bg",
|
||||
},
|
||||
})
|
||||
|
||||
server:tool("kill_job",
|
||||
"Kill a runaway background job by PID. SIGKILLs the whole process group of a shell_bg/setsid job so no children survive. Use when a background job is thrashing a machine.",
|
||||
{ type = "object", properties = { pid = { type = "integer", description = "PID from shell_bg / list_jobs" } }, required = { "pid" } },
|
||||
function(a)
|
||||
if WINDOWS then return "Error: kill_job is Linux-only" end
|
||||
local pid = tostring(a.pid or ""):match("(%d+)")
|
||||
if not pid then return "Error: numeric pid required" end
|
||||
os.execute("kill -KILL -"..pid.." 2>/dev/null; kill -KILL "..pid.." 2>/dev/null")
|
||||
sleep_ms(200)
|
||||
local alive = os.execute("kill -0 "..pid.." 2>/dev/null")
|
||||
if alive == true or alive == 0 then return "pid "..pid.." may still be alive (uninterruptible?)" end
|
||||
return "killed pid "..pid.." (process group)"
|
||||
end,
|
||||
{ annotations = { title = "Kill background job", destructiveHint = true } })
|
||||
|
||||
server:tool("list_jobs",
|
||||
"List background jobs started via shell_bg and whether each is still running. Use to find runaway jobs to kill_job.",
|
||||
{ type = "object", properties = {} },
|
||||
function()
|
||||
if WINDOWS then return "Error: list_jobs is Linux-only" end
|
||||
local reg = io.open("/tmp/lmcp-bg-jobs.tsv", "r")
|
||||
if not reg then return "(no background jobs recorded)" end
|
||||
local out = {}
|
||||
for line in reg:lines() do
|
||||
local pid, log, ts, cmd = line:match("^(%d+)\t([^\t]*)\t([^\t]*)\t(.*)$")
|
||||
if pid then
|
||||
local alive = os.execute("kill -0 "..pid.." 2>/dev/null")
|
||||
local st = (alive == true or alive == 0) and "RUNNING" or "done"
|
||||
table.insert(out, string.format("pid=%s [%s] %s log=%s\n %s", pid, st, ts, log, (cmd or ""):sub(1,100)))
|
||||
end
|
||||
end
|
||||
reg:close()
|
||||
return #out>0 and table.concat(out, "\n") or "(no background jobs recorded)"
|
||||
end,
|
||||
{ annotations = { title = "List background jobs", readOnlyHint = true } })
|
||||
|
||||
server:tool("read_file", "Read a file.", {
|
||||
type = "object",
|
||||
properties = { path = { type = "string" } },
|
||||
@@ -1066,6 +1134,46 @@ if WINDOWS then
|
||||
})
|
||||
end
|
||||
|
||||
-- ---- host-local tool plugins (issue #22) ----
|
||||
-- Load every .lua file in LMCP_TOOLS_DIR (default /opt/lmcp/tools.d on POSIX,
|
||||
-- %ProgramData%\lmcp\tools.d on Windows). Each file is invoked as a function
|
||||
-- receiving the configured `server` instance and the `run` helper:
|
||||
--
|
||||
-- local server, run = ...
|
||||
-- server:tool("my_local_tool", "...", {...}, function(a) return run(...) end)
|
||||
--
|
||||
-- This is the standard plugin pattern (nginx conf.d/, systemd-tmpfiles.d, …).
|
||||
-- Hosts can ship their own tools alongside the packaged generics without
|
||||
-- forking the upstream server.lua.
|
||||
local plugin_dir = os.getenv("LMCP_TOOLS_DIR")
|
||||
or (WINDOWS and (os.getenv("ProgramData") or "C:\\ProgramData") .. "\\lmcp\\tools.d"
|
||||
or "/opt/lmcp/tools.d")
|
||||
local list_cmd = WINDOWS
|
||||
and ('dir /b "' .. plugin_dir .. '\\*.lua" 2>nul')
|
||||
or ('ls -1 "' .. plugin_dir .. '"/*.lua 2>/dev/null')
|
||||
local lh = io.popen(list_cmd)
|
||||
if lh then
|
||||
for path in lh:lines() do
|
||||
-- On Windows `dir /b` emits bare filenames; prefix the dir.
|
||||
local full = path:match("[/\\]") and path
|
||||
or (plugin_dir .. (WINDOWS and "\\" or "/") .. path)
|
||||
local chunk, err = loadfile(full)
|
||||
if chunk then
|
||||
local ok, perr = pcall(chunk, server, run)
|
||||
if ok then
|
||||
io.stderr:write("lmcp: loaded plugin " .. full .. "\n")
|
||||
else
|
||||
io.stderr:write("lmcp: plugin " .. full .. " errored: "
|
||||
.. tostring(perr) .. "\n")
|
||||
end
|
||||
else
|
||||
io.stderr:write("lmcp: plugin " .. full .. " load error: "
|
||||
.. tostring(err) .. "\n")
|
||||
end
|
||||
end
|
||||
lh:close()
|
||||
end
|
||||
|
||||
local transport = os.getenv("LMCP_TRANSPORT") or "http"
|
||||
if transport == "stdio" then
|
||||
if os.getenv("LMCP_PORT") then
|
||||
|
||||
@@ -0,0 +1,292 @@
|
||||
-- /opt/lmcp/tools.d/hertz.lua — hertz-specific tool registrations.
|
||||
--
|
||||
-- Invoked by the packaged server.lua's tools.d scan (lmcp ≥ v1.2.0, issue #22).
|
||||
-- Receives (server, run) — the configured lmcp instance and the
|
||||
-- coroutine-aware run() helper. We add hertz-only tools on top of the
|
||||
-- packaged generics (shell, read_file, write_file, edit_file, list_dir,
|
||||
-- search_files, fetch, web_search, shell_bg).
|
||||
--
|
||||
-- Previously these all lived in /opt/lmcp/server.lua (a copy-paste fork of
|
||||
-- the packaged server.lua). That pattern drifted on every release; now the
|
||||
-- packaged file stays canonical and only the genuine hertz-specifics live
|
||||
-- here.
|
||||
|
||||
local server, run = ...
|
||||
|
||||
-- Local helpers — single-host scoped, not worth lifting into the packaged lib.
|
||||
local function read_file_raw(path)
|
||||
local f = io.open(path, 'r')
|
||||
if not f then return nil end
|
||||
local c = f:read('*a'); f:close(); return c
|
||||
end
|
||||
|
||||
local function farad(cmd, timeout)
|
||||
return run("incus exec farad -- sh -c " .. string.format("%q", cmd),
|
||||
timeout or 15)
|
||||
end
|
||||
|
||||
-- ---- LXD tools ----
|
||||
|
||||
server:tool("incus_exec", "Execute a command inside an Incus container on hertz.", {
|
||||
type = "object",
|
||||
properties = {
|
||||
container = { type = "string", description = "Container name" },
|
||||
command = { type = "string", description = "Command to execute" },
|
||||
timeout = { type = "integer", default = 30 },
|
||||
},
|
||||
required = { "container", "command" },
|
||||
}, function(a)
|
||||
return run(string.format("incus exec %s -- sh -c %q",
|
||||
a.container:gsub("[^%w%-]", ""), a.command), a.timeout or 30)
|
||||
end)
|
||||
|
||||
server:tool("incus_list", "List all Incus containers on hertz.",
|
||||
{ type = "object" },
|
||||
function() return run("incus list -c ns4 -f csv", 10) end)
|
||||
|
||||
-- ---- Fritz!Box tools ----
|
||||
|
||||
server:tool("fritz", "Execute Fritz!Box TR-064 command (info, hosts, wan, "
|
||||
.. "wol <mac>, reconnect, reboot, reboot-repeater <ip>, routes, route-add, "
|
||||
.. "route-del, services, actions, call).",
|
||||
{
|
||||
type = "object",
|
||||
properties = {
|
||||
command = { type = "string", description = "fritz subcommand and args" },
|
||||
},
|
||||
required = { "command" },
|
||||
},
|
||||
function(a) return run("sudo /root/.local/bin/fritz " .. a.command, 15) end)
|
||||
|
||||
-- ---- Network tools ----
|
||||
|
||||
server:tool("ping_host", "Check if a host is reachable (1 ICMP ping, 2s timeout).", {
|
||||
type = "object",
|
||||
properties = { host = { type = "string" } },
|
||||
required = { "host" },
|
||||
}, function(a)
|
||||
local host = a.host:gsub("[^%w%.%-:]", "")
|
||||
return run("ping -c1 -W2 " .. host, 5)
|
||||
end)
|
||||
|
||||
server:tool("network_status",
|
||||
"Check reachability of all infrastructure hosts and MCP endpoints.",
|
||||
{ type = "object" },
|
||||
function()
|
||||
local script = [[
|
||||
hosts="hertz:localhost boltzmann:boltzmann tesla:tesla data:192.168.88.30 broglie:192.168.88.160 higgs:10.170.16.10"
|
||||
for entry in $hosts; do
|
||||
name="${entry%%:*}"
|
||||
ip="${entry#*:}"
|
||||
if ping -c1 -W2 "$ip" >/dev/null 2>&1; then
|
||||
status="UP"
|
||||
if [ "$name" != "data" ]; then
|
||||
if timeout 3 bash -c "echo >/dev/tcp/${ip}/8080" 2>/dev/null; then
|
||||
status="UP (MCP ok)"
|
||||
else
|
||||
status="UP (no MCP)"
|
||||
fi
|
||||
fi
|
||||
else
|
||||
status="DOWN"
|
||||
fi
|
||||
printf "%-12s %-20s %s\n" "$name" "$ip" "$status"
|
||||
done
|
||||
]]
|
||||
return run(script, 30)
|
||||
end)
|
||||
|
||||
server:tool("wol_and_wait",
|
||||
"Wake the data server via Fritz!Box WoL and wait until it (or broglie MCP) is reachable.",
|
||||
{
|
||||
type = "object",
|
||||
properties = {
|
||||
mac = { type = "string", default = "", description = "MAC address (auto-detected if empty)" },
|
||||
wait_for_mcp = { type = "boolean", default = true, description = "Wait for broglie MCP instead of just ping" },
|
||||
timeout = { type = "integer", default = 120 },
|
||||
},
|
||||
},
|
||||
function(a)
|
||||
local mac = a.mac
|
||||
if not mac or mac == "" then
|
||||
local hosts_out = run("sudo /root/.local/bin/fritz hosts 2>/dev/null | grep -i 'data\\|192.168.88.30'", 10)
|
||||
mac = hosts_out and hosts_out:match("(%x%x:%x%x:%x%x:%x%x:%x%x:%x%x)") or ""
|
||||
if mac == "" then return "Error: could not detect MAC for data. Provide it manually." end
|
||||
end
|
||||
run("sudo /root/.local/bin/fritz wol " .. mac:gsub("[^%x:]", ""), 10)
|
||||
|
||||
local timeout = a.timeout or 120
|
||||
local check
|
||||
if a.wait_for_mcp == false then
|
||||
check = "ping -c1 -W2 192.168.88.30"
|
||||
else
|
||||
check = "timeout 3 bash -c 'echo >/dev/tcp/192.168.88.160/8080'"
|
||||
end
|
||||
local target = (a.wait_for_mcp == false) and "data" or "broglie:8080"
|
||||
|
||||
local elapsed = 0
|
||||
while elapsed < timeout do
|
||||
os.execute("sleep 5")
|
||||
elapsed = elapsed + 5
|
||||
local rc = os.execute(check)
|
||||
if rc == true or rc == 0 then
|
||||
return string.format("WoL sent to %s. %s reachable after %ds.",
|
||||
mac, target, elapsed)
|
||||
end
|
||||
end
|
||||
return string.format("WoL sent to %s but %s not reachable after %ds.",
|
||||
mac, target, timeout)
|
||||
end)
|
||||
|
||||
-- ---- Proxmox fallback ----
|
||||
|
||||
server:tool("pct_exec",
|
||||
"Execute a command in a Proxmox CT on data (SSH fallback when broglie is down).",
|
||||
{
|
||||
type = "object",
|
||||
properties = {
|
||||
ctid = { type = "string", description = "Container ID (e.g. 108, 165)" },
|
||||
command = { type = "string" },
|
||||
timeout = { type = "integer", default = 30 },
|
||||
},
|
||||
required = { "ctid", "command" },
|
||||
},
|
||||
function(a)
|
||||
local ctid = a.ctid:gsub("[^%d]", "")
|
||||
return run(string.format("ssh -o ConnectTimeout=5 root@data 'pct exec %s -- sh -c %q'",
|
||||
ctid, a.command), a.timeout or 30)
|
||||
end)
|
||||
|
||||
-- ---- Home Assistant ----
|
||||
|
||||
server:tool("ha_cli",
|
||||
"Run Home Assistant CLI command inside farad (e.g. 'core restart', 'backups list', 'info').",
|
||||
{
|
||||
type = "object",
|
||||
properties = {
|
||||
command = { type = "string", description = "ha subcommand, e.g. 'core restart', 'supervisor info', 'backups list'" },
|
||||
raw_json = { type = "boolean", default = false, description = "Return raw JSON output" },
|
||||
},
|
||||
required = { "command" },
|
||||
},
|
||||
function(a)
|
||||
local flags = a.raw_json and " --raw-json" or ""
|
||||
return farad("ha " .. a.command .. flags)
|
||||
end)
|
||||
|
||||
server:tool("ha_api",
|
||||
"Call Home Assistant Core REST API. Requires token in /opt/lmcp/ha_token on hertz.",
|
||||
{
|
||||
type = "object",
|
||||
properties = {
|
||||
method = { type = "string", default = "GET" },
|
||||
endpoint = { type = "string", description = "/api/states, /api/services/climate/set_temperature, etc." },
|
||||
body = { type = "string", description = "JSON body for POST" },
|
||||
},
|
||||
required = { "endpoint" },
|
||||
},
|
||||
function(a)
|
||||
local token = read_file_raw("/opt/lmcp/ha_token")
|
||||
if not token or token == "" then
|
||||
return "Error: no HA token. Create a long-lived token in HA UI → Profile → "
|
||||
.. "Long-Lived Access Tokens, then: echo '<token>' | sudo tee /opt/lmcp/ha_token"
|
||||
end
|
||||
token = token:gsub("%s+", "")
|
||||
local method = (a.method or "GET"):upper()
|
||||
local cmd = string.format(
|
||||
"curl -sf -X %s -H 'Authorization: Bearer %s' -H 'Content-Type: application/json'",
|
||||
method, token)
|
||||
if a.body and a.body ~= "" then
|
||||
cmd = cmd .. " -d " .. string.format("'%s'", a.body:gsub("'", "'\\''"))
|
||||
end
|
||||
cmd = cmd .. " http://localhost:8123" .. a.endpoint
|
||||
return farad(cmd, 15)
|
||||
end)
|
||||
|
||||
-- ---- MQTT (Eurotronic thermostats + general) ----
|
||||
|
||||
server:tool("mqtt_pub", "Publish an MQTT message (via Mosquitto in farad).", {
|
||||
type = "object",
|
||||
properties = {
|
||||
topic = { type = "string" },
|
||||
message = { type = "string" },
|
||||
retain = { type = "boolean", default = false },
|
||||
},
|
||||
required = { "topic", "message" },
|
||||
}, function(a)
|
||||
local retain = a.retain and "-r " or ""
|
||||
return farad(string.format("mosquitto_pub %s-t '%s' -m '%s'",
|
||||
retain, a.topic:gsub("'", "'\\''"), a.message:gsub("'", "'\\''")))
|
||||
end)
|
||||
|
||||
server:tool("mqtt_sub",
|
||||
"Subscribe to MQTT topics and collect messages (via Mosquitto in farad).",
|
||||
{
|
||||
type = "object",
|
||||
properties = {
|
||||
topic = { type = "string", description = "Topic pattern, e.g. 'eurotronic/#' or '#'" },
|
||||
count = { type = "integer", default = 10, description = "Number of messages to collect" },
|
||||
timeout = { type = "integer", default = 5, description = "Seconds to wait" },
|
||||
verbose = { type = "boolean", default = true, description = "Show topics with messages" },
|
||||
},
|
||||
required = { "topic" },
|
||||
},
|
||||
function(a)
|
||||
local v = a.verbose ~= false and "-v " or ""
|
||||
return farad(string.format("mosquitto_sub %s-t '%s' -C %d -W %d",
|
||||
v, a.topic:gsub("'", "'\\''"), a.count or 10, a.timeout or 5),
|
||||
(a.timeout or 5) + 5)
|
||||
end)
|
||||
|
||||
-- ---- Mediagrab (kids show downloader) ----
|
||||
|
||||
server:tool("mediagrab",
|
||||
"Manage kids show downloads in doppler container. Commands: list, weekly, "
|
||||
.. "archive, test <url>, add '<json>'",
|
||||
{
|
||||
type = "object",
|
||||
properties = {
|
||||
command = { type = "string", description = "Command: list, weekly, archive, 'test <url>', 'add <json>'" },
|
||||
},
|
||||
required = { "command" },
|
||||
},
|
||||
function(a)
|
||||
return run("incus exec doppler -- python3 /opt/mediagrab/mediagrab.py "
|
||||
.. a.command, 120)
|
||||
end)
|
||||
|
||||
-- ---- Fleet wake (pipi: wake-only, no power-off) ----
|
||||
server:tool("wake_fleet",
|
||||
"Wake a fleet NUC (pve1..pve4) via Fritz!Box Wake-on-LAN. Powers a node ON only; it cannot power anything off. Node boots in ~30-60s.",
|
||||
{
|
||||
type = "object",
|
||||
properties = {
|
||||
node = { type = "string", description = "Node to wake: '1'..'4' or 'pve1'..'pve4'" },
|
||||
},
|
||||
required = { "node" },
|
||||
},
|
||||
function(a)
|
||||
local node = tostring(a.node or ""):gsub("[^%w]", "")
|
||||
if not node:match("^p?v?e?[1-4]$") then
|
||||
return "Error: node must be 1-4 or pve1-pve4 (got: " .. tostring(a.node) .. ")"
|
||||
end
|
||||
return run("sudo /root/.local/bin/wake-pve " .. node .. " 2>&1", 15)
|
||||
end)
|
||||
|
||||
-- ---- apropos: lean facade for stash recall (read-only memory) ----
|
||||
server:tool("apropos",
|
||||
"Search shared fleet memory (stash) for facts about the fleet, projects, decisions, and preferences. query = 2-6 words on the topic; limit = max results (default 3). Read-only.",
|
||||
{
|
||||
type = "object",
|
||||
properties = {
|
||||
query = { type = "string", description = "2-6 words describing what to recall" },
|
||||
limit = { type = "integer", description = "max results, default 3" },
|
||||
},
|
||||
required = { "query" },
|
||||
},
|
||||
function(a)
|
||||
local q = tostring(a.query or ""):gsub("[^%w%s%-%.]", " "):gsub("%s+", " ")
|
||||
if q:gsub("%s","") == "" then return "Error: query required" end
|
||||
local lim = tonumber(a.limit) or 3
|
||||
return run("python3 /opt/lmcp/helpers/stash_recall.py '" .. q .. "' " .. lim, 30)
|
||||
end)
|
||||
@@ -0,0 +1,103 @@
|
||||
-- /opt/lmcp/tools.d/nash.lua -- nash shared memory tools
|
||||
--
|
||||
-- Provides add/search/list/delete tools for the nash memory service.
|
||||
-- Requires NASH_URL env var (e.g. http://192.168.88.143:8000).
|
||||
-- When unset, tools return an error guiding the user to set it.
|
||||
--
|
||||
-- SPDX-License-Identifier: MIT
|
||||
|
||||
local server, run = ...
|
||||
|
||||
local function nash_url()
|
||||
local url = os.getenv("NASH_URL")
|
||||
if not url or url == "" then return nil end
|
||||
return url:gsub("/+$", "")
|
||||
end
|
||||
|
||||
local function curl_json(url, method, body, timeout)
|
||||
timeout = timeout or 10
|
||||
local out = "/tmp/lmcp-nash-" .. os.time() .. "-" .. math.random(10000, 99999) .. ".json"
|
||||
local fmt = "http_code=%{http_code}"
|
||||
local cmd
|
||||
if body then
|
||||
local tmp = out .. ".req"
|
||||
local f = io.open(tmp, "w")
|
||||
if f then f:write(body); f:close() end
|
||||
cmd = string.format(
|
||||
"curl -sS -X %s --max-time %d -H 'Content-Type: application/json' --data-binary '@%s' -o '%s' -w '%s' '%s'",
|
||||
method, timeout, tmp, out, fmt, url
|
||||
)
|
||||
os.execute("rm -f '" .. tmp .. "' 2>/dev/null")
|
||||
else
|
||||
cmd = string.format(
|
||||
"curl -sS -X %s --max-time %d -o '%s' -w '%s' '%s'",
|
||||
method, timeout, out, fmt, url
|
||||
)
|
||||
end
|
||||
local raw = run(cmd, timeout + 5) or ""
|
||||
local http_code = tonumber(raw:match("(%d+)$")) or 0
|
||||
local body_out = ""
|
||||
local f = io.open(out, "r")
|
||||
if f then body_out = f:read("*a") or ""; f:close() end
|
||||
os.execute("rm -f '" .. out .. "' 2>/dev/null")
|
||||
return body_out, http_code
|
||||
end
|
||||
|
||||
server:tool("nash_add", "Store a text entry in shared nash memory.", {
|
||||
type = "object",
|
||||
properties = {
|
||||
text = { type = "string", description = "Text content to remember" },
|
||||
},
|
||||
required = { "text" },
|
||||
}, function(a)
|
||||
local base = nash_url()
|
||||
if not base then return "Error: set NASH_URL env var (e.g. http://192.168.88.143:8000)" end
|
||||
local json, code = curl_json(base .. "/add", "POST", '{"text":' .. require("json").encode(a.text) .. '}')
|
||||
if code ~= 200 then return "Error: nash API HTTP " .. code .. ": " .. json end
|
||||
return json
|
||||
end)
|
||||
|
||||
server:tool("nash_search", "Semantic search across shared nash memory.", {
|
||||
type = "object",
|
||||
properties = {
|
||||
query = { type = "string", description = "Search query" },
|
||||
limit = { type = "integer", description = "Max results (default 5)", default = 5 },
|
||||
},
|
||||
required = { "query" },
|
||||
}, function(a)
|
||||
local base = nash_url()
|
||||
if not base then return "Error: set NASH_URL env var (e.g. http://192.168.88.143:8000)" end
|
||||
local body = '{"query":' .. require("json").encode(a.query) .. ',"limit":' .. (a.limit or 5) .. '}'
|
||||
local json, code = curl_json(base .. "/search", "POST", body)
|
||||
if code ~= 200 then return "Error: nash API HTTP " .. code .. ": " .. json end
|
||||
return json
|
||||
end)
|
||||
|
||||
server:tool("nash_list", "List all entries in shared nash memory.", {
|
||||
type = "object",
|
||||
properties = {
|
||||
limit = { type = "integer", description = "Max results (default 100)", default = 100 },
|
||||
},
|
||||
}, function(a)
|
||||
local base = nash_url()
|
||||
if not base then return "Error: set NASH_URL env var (e.g. http://192.168.88.143:8000)" end
|
||||
local body = '{"limit":' .. (a.limit or 100) .. '}'
|
||||
local json, code = curl_json(base .. "/scroll", "POST", body)
|
||||
if code ~= 200 then return "Error: nash API HTTP " .. code .. ": " .. json end
|
||||
return json
|
||||
end)
|
||||
|
||||
server:tool("nash_delete", "Delete an entry from shared nash memory by ID.", {
|
||||
type = "object",
|
||||
properties = {
|
||||
id = { type = "string", description = "Entry ID to delete" },
|
||||
},
|
||||
required = { "id" },
|
||||
}, function(a)
|
||||
local base = nash_url()
|
||||
if not base then return "Error: set NASH_URL env var (e.g. http://192.168.88.143:8000)" end
|
||||
local body = '{"id":' .. require("json").encode(a.id) .. '}'
|
||||
local json, code = curl_json(base .. "/delete", "POST", body)
|
||||
if code ~= 200 then return "Error: nash API HTTP " .. code .. ": " .. json end
|
||||
return json
|
||||
end)
|
||||
Reference in New Issue
Block a user