bone-agent 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Vincent Miranda
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,19 @@
+# bone-agent
+
+A Rust+Lua terminal AI coding assistant.
+
+## Install
+
+```bash
+npm install -g bone-agent
+```
+
+## Usage
+
+```bash
+bone
+```
+
+## Config
+
+User config lives in `~/.bone-rust/`. Lua files are extracted there on first install and persist across npm updates.

package/bin/bone.js

@@ -0,0 +1,57 @@
+#!/usr/bin/env node
+/**
+ * bone-rs - npm wrapper for the Rust binary
+ * Finds the correct platform binary and executes it.
+ */
+
+const { spawnSync } = require('child_process');
+const os = require('os');
+const path = require('path');
+const fs = require('fs');
+
+// Resolve package directory from package.json location
+let packageDir;
+try {
+  packageDir = path.dirname(require.resolve('../package.json'));
+} catch {
+  // Fallback: walk up from this file
+  packageDir = __dirname;
+  while (packageDir !== path.dirname(packageDir)) {
+    if (fs.existsSync(path.join(packageDir, 'package.json'))) break;
+    packageDir = path.dirname(packageDir);
+  }
+}
+
+function getBinaryPath() {
+  const plat = os.platform();
+  const arch = os.arch();
+  const ext = plat === 'win32' ? '.exe' : '';
+  const key = `${plat}-${arch}`;
+
+  const binPath = path.join(packageDir, 'bin', `bone-${key}${ext}`);
+  if (fs.existsSync(binPath)) return binPath;
+
+  // Fallback for arm64 on darwin (Apple Silicon) — try x64 Rosetta binary
+  if (plat === 'darwin' && arch === 'arm64') {
+    const x64Path = path.join(packageDir, 'bin', `bone-darwin-x64${ext}`);
+    if (fs.existsSync(x64Path)) return x64Path;
+  }
+
+  return null;
+}
+
+const binPath = getBinaryPath();
+if (!binPath) {
+  console.error(
+    `bone-rs: no binary found for ${os.platform()}-${os.arch()}. ` +
+    `Supported: linux-x64, darwin-x64, darwin-arm64, win-x64`
+  );
+  process.exit(1);
+}
+
+const result = spawnSync(binPath, process.argv.slice(2), {
+  stdio: 'inherit',
+  cwd: process.cwd(),
+});
+
+process.exit(result.status ?? 1);

package/defaults/lua/commands/compact.lua

@@ -0,0 +1,360 @@
+-- /compact — manual context compaction and automatic before-turn reduction.
+--
+-- Implemented entirely in Lua. Remove or edit this file to disable or
+-- customize compaction behaviour.
+--
+-- Requires: ctx.conversation.history(), ctx.agent.run(), ctx.usage.snapshot(),
+--           action = "conversation.replace", bone.on("before_turn", ...)
+
+-- ---------------------------------------------------------------------------
+-- Configuration — read from config/general.yaml.
+-- ---------------------------------------------------------------------------
+
+local function config_int(ctx, key)
+    if not ctx.config or not ctx.config.get then
+        return nil
+    end
+
+    local value = ctx.config.get("general", key)
+    if value == nil then
+        return nil
+    end
+    if type(value) == "string" then
+        value = value:gsub("^%s+", ""):gsub("%s+$", "")
+        if value == "" then
+            return nil
+        end
+    end
+
+    local number = tonumber(value)
+    if not number or number < 1 or number ~= math.floor(number) then
+        return nil
+    end
+    return number
+end
+
+local function compact_config(ctx)
+    return {
+        auto_tokens = config_int(ctx, "auto_compact_tokens"),
+        keep_messages = config_int(ctx, "auto_compact_keep_messages"),
+    }
+end
+
+-- ---------------------------------------------------------------------------
+-- Helpers
+-- ---------------------------------------------------------------------------
+
+local function truncate_utf8(s, max_bytes)
+    if #s <= max_bytes then
+        return s
+    end
+    for cut = max_bytes, math.max(max_bytes - 4, 1), -1 do
+        local chunk = s:sub(1, cut)
+        local ok, len = pcall(utf8.len, chunk)
+        if ok and len then
+            return chunk .. "..."
+        end
+    end
+    return "..."
+end
+
+--- Build a summary prompt for the model to condense older messages.
+local function summarization_prompt(older, recent_count)
+    local parts = {
+        "You are a context summarizer. Summarize the conversation below into a compact description.",
+        "",
+        "Instructions:",
+        "- Capture key facts, decisions, and user preferences.",
+        "- Include file paths, code changes, and errors when relevant.",
+        "- Write a concise summary in plain prose, no markdown headings.",
+        "",
+        "The last " .. recent_count .. " user/assistant messages, plus any matching tool results, are preserved verbatim and will follow this summary.",
+        "",
+        "--- Conversation to summarize ---",
+    }
+
+    for _, msg in ipairs(older) do
+        local role = msg.role or "unknown"
+        local content = truncate_utf8(msg.content or "", 1997)
+        parts[#parts + 1] = string.format("[%s] %s", role, content)
+    end
+
+    return table.concat(parts, "\n")
+end
+
+--- Count the approximate token count of a string (chars / 4).
+local function estimate_tokens(s)
+    return math.ceil(#s / 4)
+end
+
+-- ---------------------------------------------------------------------------
+-- Core compaction logic
+-- ---------------------------------------------------------------------------
+
+local function sanitize_tool_chains(messages)
+    -- Pass 1: collect tool_call_ids that have results.
+    local result_ids = {}
+    for _, msg in ipairs(messages) do
+        if msg.role == "tool" and msg.tool_call_id then
+            result_ids[msg.tool_call_id] = true
+        end
+    end
+
+    -- Pass 2: filter assistant tool_calls; collect which ids are kept.
+    local kept_call_ids = {}
+    local filtered = {}
+    for _, msg in ipairs(messages) do
+        if msg.role == "assistant" and msg.tool_calls then
+            local calls = {}
+            for _, call in ipairs(msg.tool_calls) do
+                if call.id and result_ids[call.id] then
+                    calls[#calls + 1] = call
+                    kept_call_ids[call.id] = true
+                end
+            end
+            if #calls > 0 then
+                local copy = {}
+                for k, v in pairs(msg) do copy[k] = v end
+                copy.tool_calls = calls
+                filtered[#filtered + 1] = copy
+            elseif msg.content and msg.content ~= "" then
+                local copy = {}
+                for k, v in pairs(msg) do copy[k] = v end
+                copy.tool_calls = nil
+                filtered[#filtered + 1] = copy
+            end
+        else
+            filtered[#filtered + 1] = msg
+        end
+    end
+
+    -- Pass 3: filter tool results to only those whose call id was kept.
+    local result = {}
+    for _, msg in ipairs(filtered) do
+        if msg.role == "tool" then
+            if msg.tool_call_id and kept_call_ids[msg.tool_call_id] then
+                result[#result + 1] = msg
+            end
+        else
+            result[#result + 1] = msg
+        end
+    end
+
+    return result
+end
+
+--- Run compaction on the current transcript. Returns the replacement messages
+--- table, or nil on failure / when history is already small enough.
+local function compact(history, ctx, keep_messages)
+    if not history or #history == 0 then
+        return nil
+    end
+
+    -- Filter to user+assistant for the keep window; tool messages between
+    -- user/assistant pairs are fragile to reorder, so for v1 we drop them
+    -- from the replacement and let the model see only user/assistant.
+    local keep = {}
+    local older = {}
+
+    -- Pass 1: walk backward to find which user/assistant messages are in the
+    -- keep window, and collect tool_call_ids from kept assistants so we can
+    -- correctly route tool results (a tool result should be kept only if its
+    -- matching assistant is in keep).
+    local keep_indices = {}
+    local kept_call_ids = {}
+    local kept = 0
+    for i = #history, 1, -1 do
+        local msg = history[i]
+        if msg.role == "user" or msg.role == "assistant" then
+            kept = kept + 1
+            if kept <= keep_messages then
+                keep_indices[i] = true
+                if msg.tool_calls then
+                    for _, call in ipairs(msg.tool_calls) do
+                        if call.id then
+                            kept_call_ids[call.id] = true
+                        end
+                    end
+                end
+            end
+        end
+    end
+
+    -- Pass 2: assign messages to keep or older using the collected data.
+    for i = #history, 1, -1 do
+        local msg = history[i]
+        if keep_indices[i] then
+            keep[#keep + 1] = msg
+        elseif msg.role == "tool" and msg.tool_call_id and kept_call_ids[msg.tool_call_id] then
+            -- This tool result belongs to an assistant in keep. Keep it
+            -- regardless of its position (it may trail the last kept user msg).
+            keep[#keep + 1] = msg
+        else
+            table.insert(older, 1, msg)
+        end
+    end
+
+    -- If nothing to compact, skip.
+    if #older == 0 then
+        return nil
+    end
+
+    -- Build the summary via ctx.agent.run().
+    local prompt = summarization_prompt(older, keep_messages)
+    local run_result = ctx.agent.run(prompt, { timeout_ms = 120000 })
+    if not run_result.ok then
+        ctx.ui.notify("compact: summarization failed: " .. (run_result.error or "unknown"), "warn")
+        return nil
+    end
+
+    local summary = (run_result.content or ""):gsub("^%s+", ""):gsub("%s+$", "")
+    if #summary == 0 then
+        ctx.ui.notify("compact: empty summary, skipping", "warn")
+        return nil
+    end
+
+    -- Build replacement messages: synthetic user summary + preserved messages
+    -- (the keep array was built backward, so reverse it).
+    local messages = {}
+    messages[#messages + 1] = {
+        role = "user",
+        content = "[Context summary]\n" .. summary,
+    }
+    for i = #keep, 1, -1 do
+        messages[#messages + 1] = keep[i]
+    end
+
+    return sanitize_tool_chains(messages)
+end
+
+-- ---------------------------------------------------------------------------
+-- Auto-compaction: before_turn hook
+-- ---------------------------------------------------------------------------
+
+local last_auto_context = {}
+
+bone.on("before_turn", function(event, ctx)
+    -- Safety: skip if usage or conversation APIs are unavailable.
+    if not ctx.usage or not ctx.usage.snapshot then
+        return nil
+    end
+    if not ctx.conversation or not ctx.conversation.history then
+        return nil
+    end
+
+    -- Check that the compact command is enabled (respects /config toggle).
+    local compact_enabled = ctx.config.get("commands", "compact")
+    if compact_enabled ~= true then
+        return nil
+    end
+
+    local config = compact_config(ctx)
+    if not config.auto_tokens or not config.keep_messages then
+        return nil
+    end
+
+    local snapshot = ctx.usage.snapshot()
+    if not snapshot then
+        return nil
+    end
+
+    local context_length = snapshot.context_length or 0
+    if context_length < config.auto_tokens then
+        return nil
+    end
+
+    local conv = ctx.conversation.current and ctx.conversation.current() or nil
+    local context_key = conv and conv.id or "default"
+    local previous_context = last_auto_context[context_key]
+    -- Avoid repeated runs when context_length hasn't changed meaningfully.
+    if previous_context and math.abs(context_length - previous_context) < 50 then
+        return nil
+    end
+
+    local history = ctx.conversation.history()
+    if not history then
+        return nil
+    end
+
+    local messages = compact(history, ctx, config.keep_messages)
+    if not messages then
+        return nil
+    end
+
+    local compacted_tokens = estimate_tokens(cjson.encode(messages))
+    last_auto_context[context_key] = compacted_tokens
+
+    ctx.ui.notify(
+        string.format(
+            "compacting: %d messages → %d (context: %d → ~%d tokens)",
+            #history, #messages, context_length, compacted_tokens
+        ),
+        "info"
+    )
+
+    return {
+        action = "conversation.replace",
+        messages = messages,
+    }
+end)
+
+-- ---------------------------------------------------------------------------
+-- Manual /compact command
+-- ---------------------------------------------------------------------------
+
+bone.register_command("compact", {
+    description = "Manually compact conversation context by summarizing older messages",
+    handler = function(_, ctx)
+        if not ctx.conversation or not ctx.conversation.history then
+            return {
+                display = "Conversation history not available in this context.",
+                submit = false,
+            }
+        end
+
+        local config = compact_config(ctx)
+        if not config.keep_messages then
+            return {
+                display = "Compaction requires auto_compact_keep_messages in general config.",
+                submit = false,
+            }
+        end
+
+        local history = ctx.conversation.history()
+        if not history or #history == 0 then
+            return { display = "Nothing to compact.", submit = false }
+        end
+
+        -- Check if there's enough to compact: need more than configured keep messages.
+        local user_assistant_count = 0
+        for _, msg in ipairs(history) do
+            if msg.role == "user" or msg.role == "assistant" then
+                user_assistant_count = user_assistant_count + 1
+            end
+        end
+        if user_assistant_count <= config.keep_messages then
+            return {
+                display = string.format(
+                    "History is already small (%d user+assistant messages; threshold: %d).",
+                    user_assistant_count, config.keep_messages
+                ),
+                submit = false,
+            }
+        end
+
+        local messages = compact(history, ctx, config.keep_messages)
+        if not messages then
+            return { display = "Compaction produced no changes.", submit = false }
+        end
+
+        return {
+            display = string.format(
+                "Compacted: %d messages → %d (~%d tokens).",
+                #history, #messages, estimate_tokens(cjson.encode(messages))
+            ),
+            action = "conversation.replace",
+            messages = messages,
+            submit = false,
+        }
+    end,
+})

package/defaults/lua/commands/customize.lua

@@ -0,0 +1,143 @@
+-- /customize — quick-start guide for asking bone to customize itself.
+
+local guide = [[
+  Customize bone
+  ══════════════
+
+  Ask for the outcome you want in plain language. Bone can inspect the
+  config, explain the options, make the change, and tell you how to reload.
+
+  ── Configs ──────────────────────────────────────────────────────────
+
+  YAML files that control bone's behavior. Stored in your config
+  directory (~/.bone-rust/).
+
+  providers.yaml       — LLM providers and models. Change which
+                         provider to use, switch models, set defaults.
+                         Example: swap "openai" for "anthropic", or
+                         change the default model from gpt-4o to claude-sonnet-4-20250514.
+
+  command-policy.yaml  — Shell command safety tiers. Controls which
+                         commands auto-run vs. require approval.
+                         Example: mark git commands as "safe" to skip
+                         approval, or make all file writes require "danger" clearance.
+
+  config/*.yaml        — Feature toggles and thresholds.
+                         Example: set auto-compaction token limits, adjust
+                         memory update frequency, or change the max tool
+                         nesting depth.
+
+  ── Commands ─────────────────────────────────────────────────────────
+
+  Lua scripts in lua/commands/ that add slash commands like /compact
+  or /memory. Run on demand from the chat.
+
+  What you can change:
+  • Rename or remove bundled commands (e.g. drop /compact entirely).
+  • Change command behavior — tighten compaction thresholds, alter
+    what /memory summarizes, or change output formatting.
+  • Add new commands — a /release checklist, a /git-status summary,
+    a /find-dead-code helper.
+
+  Commands get a full ctx with shell access, file I/O, agent spawning,
+  and session history.
+
+  ── Tools ────────────────────────────────────────────────────────────
+
+  Lua scripts in lua/tools/ that extend what the LLM can do. Each
+  tool has a name, description, typed parameters, and an execute
+  function.
+
+  What you can change:
+  • Modify existing tools — tighten web_search result limits, add
+    filtering to task_list, change cron's default timeout.
+  • Add new tools — a GitHub issue search tool, a database query
+    wrapper, a project-specific linter runner.
+  • Change safety level — mark your custom tool "safe" for auto-run
+    or "danger" for approval-gated execution.
+  • Control TUI display — show/hide panes, customize what args and
+    results appear in the interface.
+
+  Tools are the LLM's primary interface to the outside world: shell,
+  filesystem, other tools, subagents, and more.
+
+  ── Lua (init.lua) ──────────────────────────────────────────────────
+
+  A startup script in the config directory that runs once when bone
+  launches. Use it to register custom tools, subagents, commands, and
+  event hooks.
+
+  What you can do:
+  • Register subagents — declare a researcher, reviewer, or test
+    verifier with its own system prompt, provider, and model.
+  • Register event hooks — run code before each turn, after errors,
+    or on other lifecycle events.
+  • Set up one-time initialization — create config files, seed
+    templates, or log startup diagnostics.
+
+  Errors in init.lua are non-fatal — bone logs a warning and continues
+  without Lua support.
+
+  ── ctx (the context object) ────────────────────────────────────────
+
+  Passed to every tool and command handler. Gives your Lua code access
+  to bone's internals. Not all fields are available everywhere.
+
+  Key capabilities:
+  • ctx.config — read values from YAML config files (read-only).
+  • ctx.fs, ctx.read_file, ctx.write_file — filesystem operations.
+  • ctx.shell, ctx.shell_streaming — run commands through the approval
+    pipeline.
+  • ctx.tools.call — invoke other registered tools by name.
+  • ctx.agent.run, ctx.agent.spawn — create and manage subagents.
+  • ctx.state — session-scoped key-value store for persisting data
+    across tool calls.
+  • ctx.conversation — read the active chat transcript.
+  • ctx.usage.snapshot — check token counts and costs.
+  • ctx.ui.notify, ctx.ui.status — send messages to the user.
+
+  Context availability varies:
+  • Tools get the full ctx (shell, files, tools, agents, etc.).
+  • Commands get most of the same, but no live event emission.
+  • Event hooks get a minimal ctx — only config_dir, ui.notify, and
+    config.dir. They cannot run shell commands or read files.
+
+  ── Prompt examples ─────────────────────────────────────────────────
+
+  Make reviews stricter about security and race conditions.
+  Add a command that prepares a release checklist.
+  Create a tool that searches my issue tracker.
+  Use a quieter, more direct assistant style.
+  Ask before running commands that modify files.
+  Show me the config files involved before changing anything.
+  Add a subagent for test verification.
+  Remember that I prefer small targeted fixes.
+
+  Helpful phrases:
+
+  Explain the current behavior first, then change it.
+  Keep this project-agnostic.
+  Make the smallest change that works.
+  Remove anything unused after the change.
+  Verify it with the right command when done.
+
+  Common areas to customize:
+
+  Providers and models
+  Tools and command approval
+  Slash commands
+  Subagents
+  Memory and assistant style
+  Status, usage, and UI settings
+
+  If you are unsure what to ask, start with:
+
+  Look at my bone config and suggest practical customizations for how I work.
+]]
+
+bone.register_command("customize", {
+  description = "Quick-start guide to customizing bone",
+  handler = function()
+    return { display = guide, submit = false }
+  end,
+})