bonecode 1.0.0 → 1.2.0

package/bone/output/session/src/extensions.ts

@@ -1,215 +1,215 @@
-// Generated by BoneScript compiler.
-// Extension points: implement the functions below.
-// Code between sentinel comments is preserved on recompile.
-// DO NOT remove the sentinel comments.
-
-import { query, queryOne, pool } from "./db";
-import { eventBus } from "./events";
-import { logger } from "./logger";
-import { counter } from "./metrics";
-import { broadcastToChannel } from "./websocket";
-
-// ─── LLM helpers ─────────────────────────────────────────────────────────────
-
-function llmFetch(body: object): Promise<Response> {
-  const baseUrl = process.env.OPENAI_BASE_URL || "https://api.openai.com/v1";
-  const apiKey = process.env.OPENAI_API_KEY || "not-needed";
-  return fetch(`${baseUrl}/chat/completions`, {
-    method: "POST",
-    headers: { "Content-Type": "application/json", Authorization: `Bearer ${apiKey}` },
-    body: JSON.stringify(body),
-  });
-}
-
-/**
- * Extension point: activate_session
- * STABLE: implementation required.
- */
-export async function activate_session(session_id: string): Promise<boolean> {
-  // <bonescript:ext:activate_session:begin>
-  const row = await queryOne<{ state: string }>("SELECT state FROM sessions WHERE id = $1", [session_id]);
-  if (!row) return false;
-  if (row.state === "active" || row.state === "busy") return true;
-  if (!["created", "compacting", "archived"].includes(row.state)) return false;
-  await query("UPDATE sessions SET state = 'active', updated_at = NOW() WHERE id = $1", [session_id]);
-  return true;
-  // <bonescript:ext:activate_session:end>
-}
-
-/**
- * Extension point: deactivate_session
- * Optional.
- */
-export async function deactivate_session(session_id: string): Promise<void> {
-  // <bonescript:ext:deactivate_session:begin>
-  await query("UPDATE sessions SET state = 'active', updated_at = NOW() WHERE id = $1 AND state = 'busy'", [session_id]);
-  // <bonescript:ext:deactivate_session:end>
-}
-
-/**
- * Extension point: send_prompt_to_llm
- * STABLE: implementation required.
- */
-export async function send_prompt_to_llm(session_id: string, message_id: string): Promise<Record<string, unknown>> {
-  // <bonescript:ext:send_prompt_to_llm:begin>
-  // Load agent config
-  const agentRow = await queryOne<any>(
-    "SELECT * FROM agent_instances WHERE session_id = $1 AND state IN ('running','waiting_for_tool') ORDER BY created_at DESC LIMIT 1",
-    [session_id]
-  );
-  if (!agentRow) throw new Error(`No active agent for session ${session_id}`);
-
-  // Load message history with parts
-  const messages = await query<any>(
-    `SELECT m.id, m.role, m.model_id, m.provider_id,
-            json_agg(p.* ORDER BY p.order_index) FILTER (WHERE p.id IS NOT NULL) AS parts
-     FROM messages m
-     LEFT JOIN parts p ON p.message_id = m.id
-     WHERE m.session_id = $1
-     GROUP BY m.id ORDER BY m.created_at ASC`,
-    [session_id]
-  );
-
-  // Build AI SDK message array
-  const aiMessages = messages.map((m: any) => {
-    const parts = m.parts || [];
-    if (m.role === "user") {
-      return { role: "user" as const, content: parts.filter((p: any) => p.part_type === "text").map((p: any) => ({ type: "text", text: p.data?.text || "" })) };
-    }
-    if (m.role === "assistant") {
-      const content: any[] = [];
-      for (const p of parts) {
-        if (p.part_type === "text") content.push({ type: "text", text: p.data?.text || "" });
-        else if (p.part_type === "tool_invocation") content.push({ type: "tool-call", toolCallId: p.data?.tool_call_id, toolName: p.data?.tool_name, args: p.data?.args });
-      }
-      return { role: "assistant" as const, content };
-    }
-    if (m.role === "tool") {
-      return { role: "tool" as const, content: parts.filter((p: any) => p.part_type === "tool_result").map((p: any) => ({ type: "tool-result", toolCallId: p.data?.tool_call_id, toolName: p.data?.tool_name, result: p.data?.result })) };
-    }
-    return { role: m.role, content: "" };
-  });
-
-  // Inject codebase context using no-embedding strategies
-  // (FTS, git recency, session history, import graph, instruction files)
-  let systemPrompt = agentRow.config?.system_prompt || agentRow.system_prompt || "";
-  try {
-    const lastUserMsg = messages.filter((m: any) => m.role === "user").slice(-1)[0];
-    const userText = (lastUserMsg?.parts || []).filter((p: any) => p.part_type === "text").map((p: any) => p.data?.text || "").join(" ");
-
-    if (userText) {
-      // Get project + worktree for this session
-      const sessionRow = await queryOne<any>(
-        `SELECT s.project_id, p.worktree FROM sessions s JOIN projects p ON p.id = s.project_id WHERE s.id = $1`,
-        [session_id]
-      );
-
-      if (sessionRow?.worktree) {
-        const { buildContext, formatContextForPrompt } = await import("../../../../src/context_builder");
-        const ctxResult = await buildContext({
-          session_id,
-          project_id: sessionRow.project_id,
-          worktree: sessionRow.worktree,
-          query: userText,
-          max_chunks: 10,
-          max_chars: 10_000,
-        });
-
-        const ctxText = formatContextForPrompt(ctxResult, sessionRow.worktree);
-        if (ctxText) {
-          systemPrompt += `\n\n${ctxText}`;
-          logger.info("context_injected", { event: "context", metadata: { summary: ctxResult.summary } });
-        }
-      }
-    }
-  } catch (e: any) {
-    logger.error("context_build_failed", { event: "context", metadata: { error: e.message } });
-  }
-
-  // Call the LLM
-  const model = agentRow.model_id || process.env.DEFAULT_MODEL || "local-model";
-  const r = await llmFetch({
-    model,
-    messages: aiMessages,
-    system: systemPrompt || undefined,
-    temperature: agentRow.temperature ?? 0.7,
-    max_tokens: agentRow.max_tokens ?? 2048,
-  });
-  if (!r.ok) throw new Error(`LLM API error ${r.status}: ${await r.text()}`);
-  const data = await r.json() as any;
-
-  const msg = data.choices?.[0]?.message;
-  const text = msg?.content || msg?.reasoning_content || "";
-  const toolCalls = msg?.tool_calls || [];
-  const usage = data.usage || {};
-
-  // Update agent token/cost counters
-  await query(
-    "UPDATE agent_instances SET total_tokens_in = total_tokens_in + $2, total_tokens_out = total_tokens_out + $3, context_window_used = context_window_used + $4, updated_at = NOW() WHERE id = $1",
-    [agentRow.id, usage.prompt_tokens || 0, usage.completion_tokens || 0, (usage.prompt_tokens || 0) + (usage.completion_tokens || 0)]
-  );
-
-  // Broadcast text delta to WebSocket part_stream channel
-  if (text) {
-    broadcastToChannel("part_stream", { type: "part.delta", session_id, message_id, delta: { type: "text", text } });
-  }
-
-  counter("llm.request.completed", { model, finish_reason: data.choices?.[0]?.finish_reason || "unknown" });
-
-  return {
-    text,
-    tool_calls: toolCalls,
-    usage: { input_tokens: usage.prompt_tokens || 0, output_tokens: usage.completion_tokens || 0 },
-    finish_reason: data.choices?.[0]?.finish_reason,
-  };
-  // <bonescript:ext:send_prompt_to_llm:end>
-}
-
-/**
- * Extension point: cancel_llm_request
- * Optional.
- */
-export async function cancel_llm_request(session_id: string): Promise<void> {
-  // <bonescript:ext:cancel_llm_request:begin>
-  // In-flight requests tracked by AbortController — signal abort
-  const controller = activeRequests.get(session_id);
-  if (controller) { controller.abort(); activeRequests.delete(session_id); }
-  // <bonescript:ext:cancel_llm_request:end>
-}
-
-const activeRequests = new Map<string, AbortController>();
-
-/**
- * Extension point: execute_tool_calls
- * STABLE: implementation required.
- */
-export async function execute_tool_calls(
-  session_id: string,
-  tool_calls: Array<{ id: string; tool_name: string; tool_input: Record<string, unknown> }>
-): Promise<Array<{ id: string; result: { success: boolean; output?: string; error?: string } }>> {
-  // <bonescript:ext:execute_tool_calls:begin>
-  const { execute_tool_calls: execTools } = await import("../../../../extensions/tool_executor");
-  return execTools(session_id, tool_calls);
-  // <bonescript:ext:execute_tool_calls:end>
-}
-
-/**
- * Extension point: revert_tool_effects
- * Optional.
- */
-export async function revert_tool_effects(session_id: string): Promise<void> {
-  // <bonescript:ext:revert_tool_effects:begin>
-  const snapshotRow = await queryOne<any>(
-    "SELECT git_hash FROM snapshots WHERE session_id = $1 AND state = 'committed' ORDER BY created_at DESC LIMIT 1",
-    [session_id]
-  );
-  if (!snapshotRow) return;
-  const wsRow = await queryOne<any>(
-    "SELECT w.id FROM workspaces w JOIN sessions s ON s.project_id = w.project_id WHERE s.id = $1 LIMIT 1",
-    [session_id]
-  );
-  if (!wsRow) return;
-  const { git_revert } = await import("../../../../extensions/snapshot");
-  await git_revert(wsRow.id, snapshotRow.git_hash);
-  // <bonescript:ext:revert_tool_effects:end>
-}
+// Generated by BoneScript compiler.
+// Extension points: implement the functions below.
+// Code between sentinel comments is preserved on recompile.
+// DO NOT remove the sentinel comments.
+
+import { query, queryOne, pool } from "./db";
+import { eventBus } from "./events";
+import { logger } from "./logger";
+import { counter } from "./metrics";
+import { broadcastToChannel } from "./websocket";
+
+// ─── LLM helpers ─────────────────────────────────────────────────────────────
+
+function llmFetch(body: object): Promise<Response> {
+  const baseUrl = process.env.OPENAI_BASE_URL || "https://api.openai.com/v1";
+  const apiKey = process.env.OPENAI_API_KEY || "not-needed";
+  return fetch(`${baseUrl}/chat/completions`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json", Authorization: `Bearer ${apiKey}` },
+    body: JSON.stringify(body),
+  });
+}
+
+/**
+ * Extension point: activate_session
+ * STABLE: implementation required.
+ */
+export async function activate_session(session_id: string): Promise<boolean> {
+  // <bonescript:ext:activate_session:begin>
+  const row = await queryOne<{ state: string }>("SELECT state FROM sessions WHERE id = $1", [session_id]);
+  if (!row) return false;
+  if (row.state === "active" || row.state === "busy") return true;
+  if (!["created", "compacting", "archived"].includes(row.state)) return false;
+  await query("UPDATE sessions SET state = 'active', updated_at = NOW() WHERE id = $1", [session_id]);
+  return true;
+  // <bonescript:ext:activate_session:end>
+}
+
+/**
+ * Extension point: deactivate_session
+ * Optional.
+ */
+export async function deactivate_session(session_id: string): Promise<void> {
+  // <bonescript:ext:deactivate_session:begin>
+  await query("UPDATE sessions SET state = 'active', updated_at = NOW() WHERE id = $1 AND state = 'busy'", [session_id]);
+  // <bonescript:ext:deactivate_session:end>
+}
+
+/**
+ * Extension point: send_prompt_to_llm
+ * STABLE: implementation required.
+ */
+export async function send_prompt_to_llm(session_id: string, message_id: string): Promise<Record<string, unknown>> {
+  // <bonescript:ext:send_prompt_to_llm:begin>
+  // Load agent config
+  const agentRow = await queryOne<any>(
+    "SELECT * FROM agent_instances WHERE session_id = $1 AND state IN ('running','waiting_for_tool') ORDER BY created_at DESC LIMIT 1",
+    [session_id]
+  );
+  if (!agentRow) throw new Error(`No active agent for session ${session_id}`);
+
+  // Load message history with parts
+  const messages = await query<any>(
+    `SELECT m.id, m.role, m.model_id, m.provider_id,
+            json_agg(p.* ORDER BY p.order_index) FILTER (WHERE p.id IS NOT NULL) AS parts
+     FROM messages m
+     LEFT JOIN parts p ON p.message_id = m.id
+     WHERE m.session_id = $1
+     GROUP BY m.id ORDER BY m.created_at ASC`,
+    [session_id]
+  );
+
+  // Build AI SDK message array
+  const aiMessages = messages.map((m: any) => {
+    const parts = m.parts || [];
+    if (m.role === "user") {
+      return { role: "user" as const, content: parts.filter((p: any) => p.part_type === "text").map((p: any) => ({ type: "text", text: p.data?.text || "" })) };
+    }
+    if (m.role === "assistant") {
+      const content: any[] = [];
+      for (const p of parts) {
+        if (p.part_type === "text") content.push({ type: "text", text: p.data?.text || "" });
+        else if (p.part_type === "tool_invocation") content.push({ type: "tool-call", toolCallId: p.data?.tool_call_id, toolName: p.data?.tool_name, args: p.data?.args });
+      }
+      return { role: "assistant" as const, content };
+    }
+    if (m.role === "tool") {
+      return { role: "tool" as const, content: parts.filter((p: any) => p.part_type === "tool_result").map((p: any) => ({ type: "tool-result", toolCallId: p.data?.tool_call_id, toolName: p.data?.tool_name, result: p.data?.result })) };
+    }
+    return { role: m.role, content: "" };
+  });
+
+  // Inject codebase context using no-embedding strategies
+  // (FTS, git recency, session history, import graph, instruction files)
+  let systemPrompt = agentRow.config?.system_prompt || agentRow.system_prompt || "";
+  try {
+    const lastUserMsg = messages.filter((m: any) => m.role === "user").slice(-1)[0];
+    const userText = (lastUserMsg?.parts || []).filter((p: any) => p.part_type === "text").map((p: any) => p.data?.text || "").join(" ");
+
+    if (userText) {
+      // Get project + worktree for this session
+      const sessionRow = await queryOne<any>(
+        `SELECT s.project_id, p.worktree FROM sessions s JOIN projects p ON p.id = s.project_id WHERE s.id = $1`,
+        [session_id]
+      );
+
+      if (sessionRow?.worktree) {
+        const { buildContext, formatContextForPrompt } = await import("../../../../src/context_builder");
+        const ctxResult = await buildContext({
+          session_id,
+          project_id: sessionRow.project_id,
+          worktree: sessionRow.worktree,
+          query: userText,
+          max_chunks: 10,
+          max_chars: 10_000,
+        });
+
+        const ctxText = formatContextForPrompt(ctxResult, sessionRow.worktree);
+        if (ctxText) {
+          systemPrompt += `\n\n${ctxText}`;
+          logger.info("context_injected", { event: "context", metadata: { summary: ctxResult.summary } });
+        }
+      }
+    }
+  } catch (e: any) {
+    logger.error("context_build_failed", { event: "context", metadata: { error: e.message } });
+  }
+
+  // Call the LLM
+  const model = agentRow.model_id || process.env.DEFAULT_MODEL || "local-model";
+  const r = await llmFetch({
+    model,
+    messages: aiMessages,
+    system: systemPrompt || undefined,
+    temperature: agentRow.temperature ?? 0.7,
+    max_tokens: agentRow.max_tokens ?? 2048,
+  });
+  if (!r.ok) throw new Error(`LLM API error ${r.status}: ${await r.text()}`);
+  const data = await r.json() as any;
+
+  const msg = data.choices?.[0]?.message;
+  const text = msg?.content || msg?.reasoning_content || "";
+  const toolCalls = msg?.tool_calls || [];
+  const usage = data.usage || {};
+
+  // Update agent token/cost counters
+  await query(
+    "UPDATE agent_instances SET total_tokens_in = total_tokens_in + $2, total_tokens_out = total_tokens_out + $3, context_window_used = context_window_used + $4, updated_at = NOW() WHERE id = $1",
+    [agentRow.id, usage.prompt_tokens || 0, usage.completion_tokens || 0, (usage.prompt_tokens || 0) + (usage.completion_tokens || 0)]
+  );
+
+  // Broadcast text delta to WebSocket part_stream channel
+  if (text) {
+    broadcastToChannel("part_stream", { type: "part.delta", session_id, message_id, delta: { type: "text", text } });
+  }
+
+  counter("llm.request.completed", { model, finish_reason: data.choices?.[0]?.finish_reason || "unknown" });
+
+  return {
+    text,
+    tool_calls: toolCalls,
+    usage: { input_tokens: usage.prompt_tokens || 0, output_tokens: usage.completion_tokens || 0 },
+    finish_reason: data.choices?.[0]?.finish_reason,
+  };
+  // <bonescript:ext:send_prompt_to_llm:end>
+}
+
+/**
+ * Extension point: cancel_llm_request
+ * Optional.
+ */
+export async function cancel_llm_request(session_id: string): Promise<void> {
+  // <bonescript:ext:cancel_llm_request:begin>
+  // In-flight requests tracked by AbortController — signal abort
+  const controller = activeRequests.get(session_id);
+  if (controller) { controller.abort(); activeRequests.delete(session_id); }
+  // <bonescript:ext:cancel_llm_request:end>
+}
+
+const activeRequests = new Map<string, AbortController>();
+
+/**
+ * Extension point: execute_tool_calls
+ * STABLE: implementation required.
+ */
+export async function execute_tool_calls(
+  session_id: string,
+  tool_calls: Array<{ id: string; tool_name: string; tool_input: Record<string, unknown> }>
+): Promise<Array<{ id: string; result: { success: boolean; output?: string; error?: string } }>> {
+  // <bonescript:ext:execute_tool_calls:begin>
+  const { execute_tool_calls: execTools } = await import("../../../../extensions/tool_executor");
+  return execTools(session_id, tool_calls);
+  // <bonescript:ext:execute_tool_calls:end>
+}
+
+/**
+ * Extension point: revert_tool_effects
+ * Optional.
+ */
+export async function revert_tool_effects(session_id: string): Promise<void> {
+  // <bonescript:ext:revert_tool_effects:begin>
+  const snapshotRow = await queryOne<any>(
+    "SELECT git_hash FROM snapshots WHERE session_id = $1 AND state = 'committed' ORDER BY created_at DESC LIMIT 1",
+    [session_id]
+  );
+  if (!snapshotRow) return;
+  const wsRow = await queryOne<any>(
+    "SELECT w.id FROM workspaces w JOIN sessions s ON s.project_id = w.project_id WHERE s.id = $1 LIMIT 1",
+    [session_id]
+  );
+  if (!wsRow) return;
+  const { git_revert } = await import("../../../../extensions/snapshot");
+  await git_revert(wsRow.id, snapshotRow.git_hash);
+  // <bonescript:ext:revert_tool_effects:end>
+}