agent-block 0.5.0

Lua-first Agent Runtime built on AgentMesh
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

-- Embedded Lua source loaded by src/bridge/kv.rs.
-- Defines std.kv.register_tools(opts?) — LLM-facing tool registration helper.
--
-- opts (all optional):
--   allowed : array of op names   (default: {"get","set","delete","list"})
--   prefix  : tool name prefix    (default: "kv_")
--   ns_lock : string              (locks the ns arg; LLM cannot choose namespace)
--
-- Returns: array of registered tool names.

std.kv.register_tools = function(opts)
    opts = opts or {}
    local allowed = opts.allowed or { "get", "set", "delete", "list" }
    local prefix = opts.prefix or "kv_"
    local ns_lock = opts.ns_lock

    local function build_schema(extra_props, extra_required)
        local props = {}
        local required = {}
        if not ns_lock then
            props.ns = { type = "string", description = "Namespace (logical group)" }
            table.insert(required, "ns")
        end
        for k, v in pairs(extra_props) do props[k] = v end
        for _, r in ipairs(extra_required) do table.insert(required, r) end
        local schema = { type = "object", properties = props }
        -- Empty Lua tables serialize as JSON objects, but Anthropic expects
        -- `required` to be a JSON array. Omit the field when empty.
        if #required > 0 then schema.required = required end
        return schema
    end

    local defs = {
        get = {
            description = "Get a value from the agent's local key-value store (JSON-file backed, persists across runs, agent-private). Returns { value = ... } (nil if the key is missing).",
            input_schema = build_schema({ key = { type = "string" } }, { "key" }),
            handler = function(input)
                local ns = ns_lock or input.ns
                return { value = std.kv.get(ns, input.key) }
            end,
        },
        set = {
            description = "Store a value in the agent's local key-value store (JSON-file backed, persists across runs, agent-private). Overwrites any existing value at the same key.",
            input_schema = build_schema({
                key   = { type = "string" },
                value = { description = "Value to store (string / number / bool / table)" },
            }, { "key", "value" }),
            handler = function(input)
                local ns = ns_lock or input.ns
                std.kv.set(ns, input.key, input.value)
                return { ok = true }
            end,
        },
        delete = {
            description = "Delete a key from the agent's local key-value store. Returns { deleted = bool } (false if the key did not exist).",
            input_schema = build_schema({ key = { type = "string" } }, { "key" }),
            handler = function(input)
                local ns = ns_lock or input.ns
                return { deleted = std.kv.delete(ns, input.key) }
            end,
        },
        list = {
            description = "List keys in a namespace of the agent's local key-value store, optionally filtered by prefix. Returns { keys = [...] }.",
            input_schema = build_schema(
                { prefix = { type = "string", description = "Optional key prefix filter" } },
                {}
            ),
            handler = function(input)
                local ns = ns_lock or input.ns
                return { keys = std.kv.list(ns, input.prefix) }
            end,
        },
    }

    local registered = {}
    for _, op in ipairs(allowed) do
        local d = defs[op]
        if d then
            local name = prefix .. op
            tool.register(name, { description = d.description, input_schema = d.input_schema }, d.handler)
            table.insert(registered, name)
        end
    end
    return registered
end