grok-dev 1.0.0-rc1

package/dist/grok/tools.js

@@ -0,0 +1,230 @@
+import { generateText, tool } from "ai";
+import { z } from "zod";
+import { editFile, readFile, writeFile } from "../tools/file";
+const RESPONSES_SEARCH_MODEL = "grok-4-fast-non-reasoning";
+export function createTools(bash, provider, mode = "agent", options = {}) {
+    const cwd = () => bash.getCwd();
+    const runResponsesSearch = async (query, toolName) => {
+        try {
+            const { text } = await generateText({
+                model: provider.responses(RESPONSES_SEARCH_MODEL),
+                maxOutputTokens: 4096,
+                prompt: query,
+                tools: {
+                    ...(toolName === "web_search" ? { web_search: provider.tools.webSearch() } : {}),
+                    ...(toolName === "x_search" ? { x_search: provider.tools.xSearch() } : {}),
+                },
+            });
+            return {
+                success: true,
+                output: text || "No search results found.",
+            };
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            const label = toolName === "web_search" ? "Web search" : "X search";
+            return { success: false, output: `${label} failed: ${msg}` };
+        }
+    };
+    const base = {
+        bash: tool({
+            description: "Execute a bash command. Use for searching (grep, rg, find), git, build tools, package managers, running tests, and any other shell command. Set background=true for long-running processes like dev servers, watchers, or anything that should keep running while you continue working. For file read/write/edit, prefer the dedicated file tools instead.",
+            inputSchema: z.object({
+                command: z.string().describe("The bash command to execute"),
+                timeout: z
+                    .number()
+                    .optional()
+                    .describe("Timeout in milliseconds (default: 30000). Use higher values for long-running commands. Ignored when background=true."),
+                background: z
+                    .boolean()
+                    .optional()
+                    .describe("Run the command as a background process. Returns immediately with a process ID. Use process_logs/process_stop/process_list to manage it."),
+            }),
+            execute: async ({ command, timeout, background }) => {
+                if (background) {
+                    return bash.startBackground(command);
+                }
+                const result = await bash.execute(command, timeout);
+                return {
+                    success: result.success,
+                    output: result.success
+                        ? result.output || "Command executed successfully (no output)"
+                        : result.error || "Command failed",
+                };
+            },
+        }),
+        process_logs: tool({
+            description: "View recent output (stdout + stderr) from a background process by its ID. Returns the last N lines of the log.",
+            inputSchema: z.object({
+                id: z.number().describe("The background process ID"),
+                tail: z.number().optional().describe("Number of lines to return from the end (default: 50)"),
+            }),
+            execute: async ({ id, tail }) => {
+                return bash.getProcessLogs(id, tail);
+            },
+        }),
+        process_stop: tool({
+            description: "Stop a running background process by its ID. Sends SIGTERM, then SIGKILL after 3 seconds if still alive.",
+            inputSchema: z.object({
+                id: z.number().describe("The background process ID to stop"),
+            }),
+            execute: async ({ id }) => {
+                return bash.stopProcess(id);
+            },
+        }),
+        process_list: tool({
+            description: "List all background processes with their ID, status, PID, age, and command.",
+            inputSchema: z.object({}),
+            execute: async () => {
+                return bash.listProcesses();
+            },
+        }),
+        read_file: tool({
+            description: "Read the contents of a file. Returns numbered lines with a header showing the range and total line count. Use start_line/end_line to read specific sections of large files iteratively.",
+            inputSchema: z.object({
+                path: z.string().describe("File path (relative to cwd or absolute)"),
+                start_line: z.number().optional().describe("First line to read (1-indexed, default: 1)"),
+                end_line: z.number().optional().describe("Last line to read (inclusive, default: end of file)"),
+            }),
+            execute: async ({ path, start_line, end_line }) => {
+                return readFile(path, cwd(), start_line, end_line);
+            },
+        }),
+        search_web: tool({
+            description: "Search the web for current information, documentation, APIs, tutorials, news, or any real-time data. Returns summarized results with sources.",
+            inputSchema: z.object({
+                query: z.string().describe("The search query"),
+            }),
+            execute: async ({ query }) => {
+                return runResponsesSearch(query, "web_search");
+            },
+        }),
+        search_x: tool({
+            description: "Search X (Twitter) for real-time posts, discussions, opinions, and trends. Returns relevant posts with authors and engagement data.",
+            inputSchema: z.object({
+                query: z.string().describe("The search query"),
+            }),
+            execute: async ({ query }) => {
+                return runResponsesSearch(query, "x_search");
+            },
+        }),
+    };
+    const tools = { ...base };
+    if (mode === "agent") {
+        tools.write_file = tool({
+            description: "Create or overwrite a file with the given content. Use for creating new files or completely rewriting existing ones. Returns a diff of the changes.",
+            inputSchema: z.object({
+                path: z.string().describe("File path (relative to cwd or absolute)"),
+                content: z.string().describe("The full file content to write"),
+            }),
+            execute: async ({ path, content }) => {
+                return writeFile(path, content, cwd());
+            },
+        });
+        tools.edit_file = tool({
+            description: "Edit a file by replacing a unique string with new content. The old_string must appear exactly once in the file. Include enough surrounding context lines in old_string to make it unique. Returns a diff of the changes.",
+            inputSchema: z.object({
+                path: z.string().describe("File path (relative to cwd or absolute)"),
+                old_string: z.string().describe("The exact text to find (must be unique in the file)"),
+                new_string: z.string().describe("The replacement text"),
+            }),
+            execute: async ({ path, old_string, new_string }) => {
+                return editFile(path, old_string, new_string, cwd());
+            },
+        });
+        if (options.runTask) {
+            tools.task = tool({
+                description: "Delegate a focused foreground task to a sub-agent. Prefer this proactively for review, research, investigation, and code quality work instead of waiting for the user to request a sub-agent. Use `general` for multi-step execution and `explore` for fast read-only investigation. Provide a short description plus a detailed prompt for the child agent.",
+                inputSchema: z.object({
+                    agent: z.enum(["general", "explore"]).default("general").describe("Which sub-agent to use"),
+                    description: z
+                        .string()
+                        .describe("A short label for the delegated task, such as 'Deep code quality analysis'"),
+                    prompt: z.string().describe("Detailed instructions for the sub-agent to complete"),
+                }),
+                execute: async ({ agent, description, prompt }, { abortSignal }) => {
+                    return options.runTask({ agent, description, prompt }, abortSignal);
+                },
+            });
+        }
+        if (options.runDelegation) {
+            tools.delegate = tool({
+                description: "Launch a read-only background agent that can keep researching while you continue working. Use this only for `explore` tasks that do not edit files or make shell changes. You will be notified when it completes.",
+                inputSchema: z.object({
+                    agent: z
+                        .enum(["explore"])
+                        .default("explore")
+                        .describe("Background delegations currently support only the read-only explore agent"),
+                    description: z.string().describe("A short label for the delegation, such as 'OAuth callback research'"),
+                    prompt: z.string().describe("Detailed instructions for the background agent to complete"),
+                }),
+                execute: async ({ agent, description, prompt }, { abortSignal }) => {
+                    return options.runDelegation({ agent, description, prompt }, abortSignal);
+                },
+            });
+        }
+        if (options.readDelegation) {
+            tools.delegation_read = tool({
+                description: "Read the saved output of a background delegation by ID. Use this after a completion notice or when revisiting prior research.",
+                inputSchema: z.object({
+                    id: z.string().describe("The delegation ID, such as 'calm-blue-fox'"),
+                }),
+                execute: async ({ id }) => {
+                    return options.readDelegation(id);
+                },
+            });
+        }
+        if (options.listDelegations) {
+            tools.delegation_list = tool({
+                description: "List recent background delegations for the current project. Use sparingly to discover IDs or review prior results, not for repeated polling.",
+                inputSchema: z.object({}),
+                execute: async () => {
+                    return options.listDelegations();
+                },
+            });
+        }
+    }
+    if (mode !== "plan")
+        return tools;
+    tools.generate_plan = tool({
+        description: "Generate an interactive implementation plan with steps and optional questions for the user. The plan is displayed in a structured UI where the user can review steps and answer questions. Always use this tool when creating plans.",
+        inputSchema: z.object({
+            title: z.string().describe("Plan title"),
+            summary: z.string().describe("Brief summary of what the plan accomplishes"),
+            steps: z
+                .array(z.object({
+                title: z.string().describe("Step title"),
+                description: z.string().describe("Detailed description of what this step involves"),
+                filePaths: z.array(z.string()).optional().describe("Files affected by this step"),
+            }))
+                .describe("Ordered list of implementation steps"),
+            questions: z
+                .array(z.object({
+                id: z.string().describe("Unique question identifier"),
+                question: z.string().describe("The question to ask the user"),
+                header: z.string().optional().describe("Single-word tab label (e.g. 'Format', 'Storage', 'Testing')"),
+                type: z
+                    .enum(["select", "multiselect", "text"])
+                    .describe("Question type: select (pick one), multiselect (pick many), or text (free-form)"),
+                options: z
+                    .array(z.object({
+                    id: z.string().describe("Option identifier"),
+                    label: z.string().describe("Option display text"),
+                }))
+                    .optional()
+                    .describe("Options for select/multiselect questions"),
+            }))
+                .optional()
+                .describe("Questions for the user to answer before proceeding"),
+        }),
+        execute: async ({ title, summary, steps, questions }) => {
+            return {
+                success: true,
+                output: `Plan "${title}" generated with ${steps.length} steps`,
+                plan: { title, summary, steps, questions },
+            };
+        },
+    });
+    return tools;
+}
+//# sourceMappingURL=tools.js.map

package/dist/grok/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/grok/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAgB,IAAI,EAAE,MAAM,IAAI,CAAC;AACtD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAI9D,MAAM,sBAAsB,GAAG,2BAA2B,CAAC;AAS3D,MAAM,UAAU,WAAW,CACzB,IAAc,EACd,QAAqB,EACrB,OAAkB,OAAO,EACzB,UAA8B,EAAE;IAEhC,MAAM,GAAG,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;IAEhC,MAAM,kBAAkB,GAAG,KAAK,EAC9B,KAAa,EACb,QAAmC,EACY,EAAE;QACjD,IAAI,CAAC;YACH,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,YAAY,CAAC;gBAClC,KAAK,EAAE,QAAQ,CAAC,SAAS,CAAC,sBAAsB,CAAC;gBACjD,eAAe,EAAE,IAAI;gBACrB,MAAM,EAAE,KAAK;gBACb,KAAK,EAAE;oBACL,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,QAAQ,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAChF,GAAG,CAAC,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC3E;aACF,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,IAAI,IAAI,0BAA0B;aAC3C,CAAC;QACJ,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,MAAM,KAAK,GAAG,QAAQ,KAAK,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,UAAU,CAAC;YACpE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,KAAK,YAAY,GAAG,EAAE,EAAE,CAAC;QAC/D,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,IAAI,GAAG;QACX,IAAI,EAAE,IAAI,CAAC;YACT,WAAW,EACT,4VAA4V;YAC9V,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;gBAC3D,OAAO,EAAE,CAAC;qBACP,MAAM,EAAE;qBACR,QAAQ,EAAE;qBACV,QAAQ,CACP,sHAAsH,CACvH;gBACH,UAAU,EAAE,CAAC;qBACV,OAAO,EAAE;qBACT,QAAQ,EAAE;qBACV,QAAQ,CACP,0IAA0I,CAC3I;aACJ,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE;gBAClD,IAAI,UAAU,EAAE,CAAC;oBACf,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACvC,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBACpD,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,MAAM,EAAE,MAAM,CAAC,OAAO;wBACpB,CAAC,CAAC,MAAM,CAAC,MAAM,IAAI,2CAA2C;wBAC9D,CAAC,CAAC,MAAM,CAAC,KAAK,IAAI,gBAAgB;iBACrC,CAAC;YACJ,CAAC;SACF,CAAC;QAEF,YAAY,EAAE,IAAI,CAAC;YACjB,WAAW,EACT,gHAAgH;YAClH,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;gBACpD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sDAAsD,CAAC;aAC7F,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;gBAC9B,OAAO,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACvC,CAAC;SACF,CAAC;QAEF,YAAY,EAAE,IAAI,CAAC;YACjB,WAAW,EACT,0GAA0G;YAC5G,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;aAC7D,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE;gBACxB,OAAO,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAC9B,CAAC;SACF,CAAC;QAEF,YAAY,EAAE,IAAI,CAAC;YACjB,WAAW,EAAE,6EAA6E;YAC1F,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,KAAK,IAAI,EAAE;gBAClB,OAAO,IAAI,CAAC,aAAa,EAAE,CAAC;YAC9B,CAAC;SACF,CAAC;QAEF,SAAS,EAAE,IAAI,CAAC;YACd,WAAW,EACT,yLAAyL;YAC3L,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;gBACpE,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;gBACxF,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qDAAqD,CAAC;aAChG,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE;gBAChD,OAAO,QAAQ,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;YACrD,CAAC;SACF,CAAC;QAEF,UAAU,EAAE,IAAI,CAAC;YACf,WAAW,EACT,+IAA+I;YACjJ,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;aAC/C,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC3B,OAAO,kBAAkB,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;YACjD,CAAC;SACF,CAAC;QAEF,QAAQ,EAAE,IAAI,CAAC;YACb,WAAW,EACT,qIAAqI;YACvI,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;aAC/C,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC3B,OAAO,kBAAkB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAC/C,CAAC;SACF,CAAC;KACH,CAAC;IAEF,MAAM,KAAK,GAAY,EAAE,GAAG,IAAI,EAAE,CAAC;IAEnC,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;QACrB,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC;YACtB,WAAW,EACT,qJAAqJ;YACvJ,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;gBACpE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;aAC/D,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE;gBACnC,OAAO,SAAS,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;YACzC,CAAC;SACF,CAAC,CAAC;QAEH,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;YACrB,WAAW,EACT,0NAA0N;YAC5N,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;gBACpE,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qDAAqD,CAAC;gBACtF,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,sBAAsB,CAAC;aACxD,CAAC;YACF,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,EAAE;gBAClD,OAAO,QAAQ,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;YACvD,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;gBAChB,WAAW,EACT,8VAA8V;gBAChW,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;oBACpB,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,wBAAwB,CAAC;oBAC3F,WAAW,EAAE,CAAC;yBACX,MAAM,EAAE;yBACR,QAAQ,CAAC,4EAA4E,CAAC;oBACzF,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qDAAqD,CAAC;iBACnF,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;oBACjE,OAAO,OAAO,CAAC,OAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,WAAW,CAAC,CAAC;gBACvE,CAAC;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;YAC1B,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC;gBACpB,WAAW,EACT,mNAAmN;gBACrN,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;oBACpB,KAAK,EAAE,CAAC;yBACL,IAAI,CAAC,CAAC,SAAS,CAAC,CAAC;yBACjB,OAAO,CAAC,SAAS,CAAC;yBAClB,QAAQ,CAAC,2EAA2E,CAAC;oBACxF,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qEAAqE,CAAC;oBACvG,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4DAA4D,CAAC;iBAC1F,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;oBACjE,OAAO,OAAO,CAAC,aAAc,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,WAAW,CAAC,CAAC;gBAC7E,CAAC;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC;gBAC3B,WAAW,EACT,+HAA+H;gBACjI,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;oBACpB,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;iBACtE,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE;oBACxB,OAAO,OAAO,CAAC,cAAe,CAAC,EAAE,CAAC,CAAC;gBACrC,CAAC;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;YAC5B,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC;gBAC3B,WAAW,EACT,8IAA8I;gBAChJ,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzB,OAAO,EAAE,KAAK,IAAI,EAAE;oBAClB,OAAO,OAAO,CAAC,eAAgB,EAAE,CAAC;gBACpC,CAAC;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,IAAI,KAAK,MAAM;QAAE,OAAO,KAAK,CAAC;IAElC,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;QACzB,WAAW,EACT,sOAAsO;QACxO,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;YACxC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6CAA6C,CAAC;YAC3E,KAAK,EAAE,CAAC;iBACL,KAAK,CACJ,CAAC,CAAC,MAAM,CAAC;gBACP,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;gBACxC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iDAAiD,CAAC;gBACnF,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;aAClF,CAAC,CACH;iBACA,QAAQ,CAAC,sCAAsC,CAAC;YACnD,SAAS,EAAE,CAAC;iBACT,KAAK,CACJ,CAAC,CAAC,MAAM,CAAC;gBACP,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC;gBACrD,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;gBAC7D,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6DAA6D,CAAC;gBACrG,IAAI,EAAE,CAAC;qBACJ,IAAI,CAAC,CAAC,QAAQ,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;qBACvC,QAAQ,CAAC,gFAAgF,CAAC;gBAC7F,OAAO,EAAE,CAAC;qBACP,KAAK,CACJ,CAAC,CAAC,MAAM,CAAC;oBACP,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;oBAC5C,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;iBAClD,CAAC,CACH;qBACA,QAAQ,EAAE;qBACV,QAAQ,CAAC,0CAA0C,CAAC;aACxD,CAAC,CACH;iBACA,QAAQ,EAAE;iBACV,QAAQ,CAAC,oDAAoD,CAAC;SAClE,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE;YACtD,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,SAAS,KAAK,oBAAoB,KAAK,CAAC,MAAM,QAAQ;gBAC9D,IAAI,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE;aAC3C,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env bun
+export {};

package/dist/index.js

@@ -0,0 +1,192 @@
+#!/usr/bin/env bun
+import { program } from "commander";
+import * as dotenv from "dotenv";
+import { Agent } from "./agent/agent";
+import { completeDelegation, failDelegation, loadDelegation } from "./agent/delegations";
+import { MODELS } from "./grok/models";
+import { getApiKey, getBaseURL, getCurrentModel, saveUserSettings } from "./utils/settings";
+dotenv.config();
+process.on("SIGTERM", () => {
+    process.exit(0);
+});
+process.on("uncaughtException", (err) => {
+    console.error("Fatal:", err.message);
+    process.exit(1);
+});
+process.on("unhandledRejection", (reason) => {
+    console.error("Unhandled rejection:", reason);
+    process.exit(1);
+});
+async function startInteractive(apiKey, baseURL, model, maxToolRounds, session, initialMessage) {
+    const agent = new Agent(apiKey, baseURL, model, maxToolRounds, { session });
+    const { createCliRenderer } = await import("@opentui/core");
+    const { createRoot } = await import("@opentui/react");
+    const { createElement } = await import("react");
+    const { App } = await import("./ui/app");
+    const renderer = await createCliRenderer({
+        exitOnCtrlC: false,
+        // Lets terminals (Kitty, iTerm2, WezTerm, …) report Command as `super` on KeyEvent — needed for ⌘C in the TUI.
+        useKittyKeyboard: {
+            disambiguate: true,
+            alternateKeys: true,
+        },
+    });
+    const onExit = () => {
+        renderer.destroy();
+        process.exit(0);
+    };
+    createRoot(renderer).render(createElement(App, {
+        agent,
+        startupConfig: {
+            apiKey,
+            baseURL,
+            model,
+            maxToolRounds,
+        },
+        initialMessage,
+        onExit,
+    }));
+}
+async function runHeadless(prompt, apiKey, baseURL, model, maxToolRounds, session) {
+    const agent = new Agent(apiKey, baseURL, model, maxToolRounds, { session });
+    process.stdout.write(`\x1b[36m⏳ Processing...\x1b[0m\n`);
+    if (agent.getSessionId()) {
+        process.stderr.write(`\x1b[2mSession: ${agent.getSessionId()}\x1b[0m\n`);
+    }
+    for await (const chunk of agent.processMessage(prompt)) {
+        switch (chunk.type) {
+            case "content":
+                if (chunk.content)
+                    process.stdout.write(chunk.content);
+                break;
+            case "tool_calls":
+                if (chunk.toolCalls) {
+                    for (const tc of chunk.toolCalls) {
+                        process.stderr.write(`\x1b[33m⚙ ${tc.function.name}\x1b[0m\n`);
+                    }
+                }
+                break;
+            case "tool_result":
+                if (chunk.toolResult) {
+                    const icon = chunk.toolResult.success ? "✓" : "✗";
+                    const color = chunk.toolResult.success ? "\x1b[32m" : "\x1b[31m";
+                    process.stderr.write(`${color}${icon} ${chunk.toolCall?.function.name || "tool"}\x1b[0m\n`);
+                }
+                break;
+            case "error":
+                process.stderr.write(`\x1b[31m${chunk.content}\x1b[0m\n`);
+                break;
+            case "done":
+                process.stdout.write("\n");
+                break;
+        }
+    }
+}
+async function runBackgroundDelegation(jobPath, options) {
+    let output = "";
+    try {
+        const delegation = await loadDelegation(jobPath);
+        const apiKey = options.apiKey || getApiKey();
+        if (!apiKey) {
+            throw new Error("API key required. Set GROK_API_KEY, use --api-key, or save it to ~/.grok/user-settings.json.");
+        }
+        const baseURL = options.baseUrl || getBaseURL();
+        const model = options.model || delegation.model || getCurrentModel();
+        const maxToolRounds = parseInt(options.maxToolRounds || String(delegation.maxToolRounds), 10) || delegation.maxToolRounds;
+        const agent = new Agent(apiKey, baseURL, model, maxToolRounds, { persistSession: false });
+        const result = await agent.runTaskRequest({
+            agent: delegation.agent,
+            description: delegation.description,
+            prompt: delegation.prompt,
+        });
+        output = (result.output || "").trim();
+        if (!result.success) {
+            await failDelegation(jobPath, result.output || result.error || "Background delegation failed.", output);
+            return;
+        }
+        await completeDelegation(jobPath, output, result.task?.summary);
+    }
+    catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        try {
+            await failDelegation(jobPath, msg, output);
+        }
+        catch {
+            // Best effort — background tasks should fail silently if persistence is unavailable.
+        }
+        process.exit(1);
+    }
+}
+function resolveConfig(options) {
+    const apiKey = options.apiKey || getApiKey();
+    const baseURL = options.baseUrl || getBaseURL();
+    const model = options.model || getCurrentModel();
+    const maxToolRounds = parseInt(options.maxToolRounds || "400", 10) || 400;
+    if (options.apiKey)
+        saveUserSettings({ apiKey: options.apiKey });
+    if (options.model)
+        saveUserSettings({ defaultModel: options.model });
+    return { apiKey, baseURL, model, maxToolRounds };
+}
+function requireApiKey(apiKey) {
+    if (!apiKey) {
+        console.error("Error: API key required. Set GROK_API_KEY env var, use --api-key, or save to ~/.grok/user-settings.json");
+        process.exit(1);
+    }
+    return apiKey;
+}
+program
+    .name("grok")
+    .description("AI coding agent powered by Grok — built with Bun and OpenTUI")
+    .version("1.0.0")
+    .argument("[message...]", "Initial message to send")
+    .option("-k, --api-key <key>", "Grok API key")
+    .option("-u, --base-url <url>", "API base URL")
+    .option("-m, --model <model>", "Model to use")
+    .option("-d, --directory <dir>", "Working directory", process.cwd())
+    .option("-p, --prompt <prompt>", "Run a single prompt headlessly")
+    .option("-s, --session <id>", "Continue a saved session by id, or use 'latest'")
+    .option("--background-task-file <path>", "Run a persisted background delegation")
+    .option("--max-tool-rounds <n>", "Max tool execution rounds", "400")
+    .action(async (message, options) => {
+    if (options.directory) {
+        try {
+            process.chdir(options.directory);
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            console.error(`Cannot change to directory ${options.directory}: ${msg}`);
+            process.exit(1);
+        }
+    }
+    if (options.backgroundTaskFile) {
+        await runBackgroundDelegation(options.backgroundTaskFile, options);
+        return;
+    }
+    const config = resolveConfig(options);
+    if (options.prompt) {
+        await runHeadless(options.prompt, requireApiKey(config.apiKey), config.baseURL, config.model, config.maxToolRounds, options.session);
+        return;
+    }
+    const initialMessage = message.length > 0 ? message.join(" ") : undefined;
+    await startInteractive(config.apiKey, config.baseURL, config.model, config.maxToolRounds, options.session, initialMessage);
+});
+program
+    .command("models")
+    .description("List available Grok models")
+    .action(() => {
+    console.log("\nAvailable Grok Models:\n");
+    for (const m of MODELS) {
+        const reasoning = m.reasoning ? " (reasoning)" : "";
+        console.log(`  \x1b[36m${m.id}\x1b[0m — ${m.name}${reasoning}`);
+        console.log(`    ${m.description} | ${formatContext(m.contextWindow)} context | $${m.inputPrice}/$${m.outputPrice} per 1M tokens`);
+    }
+    console.log();
+});
+program.parse();
+function formatContext(tokens) {
+    if (tokens >= 1_000_000)
+        return `${tokens / 1_000_000}M`;
+    return `${tokens / 1_000}K`;
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACtC,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACzF,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACvC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAE5F,MAAM,CAAC,MAAM,EAAE,CAAC;AAEhB,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;IACzB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,GAAG,EAAE,EAAE;IACtC,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;IACrC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,EAAE;IAC1C,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,MAAM,CAAC,CAAC;IAC9C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,KAAK,UAAU,gBAAgB,CAC7B,MAA0B,EAC1B,OAAe,EACf,KAAa,EACb,aAAqB,EACrB,OAAgB,EAChB,cAAuB;IAEvB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5E,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,CAAC;IAC5D,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;IACtD,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,EAAE,GAAG,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;IAEzC,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CAAC;QACvC,WAAW,EAAE,KAAK;QAClB,+GAA+G;QAC/G,gBAAgB,EAAE;YAChB,YAAY,EAAE,IAAI;YAClB,aAAa,EAAE,IAAI;SACpB;KACF,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,GAAG,EAAE;QAClB,QAAQ,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC;IAEF,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CACzB,aAAa,CAAC,GAAG,EAAE;QACjB,KAAK;QACL,aAAa,EAAE;YACb,MAAM;YACN,OAAO;YACP,KAAK;YACL,aAAa;SACd;QACD,cAAc;QACd,MAAM;KACP,CAAC,CACH,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,WAAW,CACxB,MAAc,EACd,MAAc,EACd,OAAe,EACf,KAAa,EACb,aAAqB,EACrB,OAAgB;IAEhB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAE5E,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACzD,IAAI,KAAK,CAAC,YAAY,EAAE,EAAE,CAAC;QACzB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,KAAK,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;IAC3E,CAAC;IAED,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC;QACvD,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,SAAS;gBACZ,IAAI,KAAK,CAAC,OAAO;oBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACvD,MAAM;YACR,KAAK,YAAY;gBACf,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;oBACpB,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;wBACjC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC,QAAQ,CAAC,IAAI,WAAW,CAAC,CAAC;oBACjE,CAAC;gBACH,CAAC;gBACD,MAAM;YACR,KAAK,aAAa;gBAChB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;oBAClD,MAAM,KAAK,GAAG,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC;oBACjE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,KAAK,GAAG,IAAI,IAAI,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,IAAI,MAAM,WAAW,CAAC,CAAC;gBAC9F,CAAC;gBACD,MAAM;YACR,KAAK,OAAO;gBACV,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,KAAK,CAAC,OAAO,WAAW,CAAC,CAAC;gBAC1D,MAAM;YACR,KAAK,MAAM;gBACT,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC3B,MAAM;QACV,CAAC;IACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,uBAAuB,CAAC,OAAe,EAAE,OAA2C;IACjG,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;QAC7C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,8FAA8F,CAAC,CAAC;QAClH,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,UAAU,EAAE,CAAC;QAChD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,UAAU,CAAC,KAAK,IAAI,eAAe,EAAE,CAAC;QACrE,MAAM,aAAa,GACjB,QAAQ,CAAC,OAAO,CAAC,aAAa,IAAI,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,EAAE,CAAC,IAAI,UAAU,CAAC,aAAa,CAAC;QACtG,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1F,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC;YACxC,KAAK,EAAE,UAAU,CAAC,KAAK;YACvB,WAAW,EAAE,UAAU,CAAC,WAAW;YACnC,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QAEH,MAAM,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAEtC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,KAAK,IAAI,+BAA+B,EAAE,MAAM,CAAC,CAAC;YACxG,OAAO;QACT,CAAC;QAED,MAAM,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAClE,CAAC;IAAC,OAAO,GAAY,EAAE,CAAC;QACtB,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7D,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QAC7C,CAAC;QAAC,MAAM,CAAC;YACP,qFAAqF;QACvF,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,OAA2C;IAChE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;IAC7C,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,UAAU,EAAE,CAAC;IAChD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,eAAe,EAAE,CAAC;IACjD,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,aAAa,IAAI,KAAK,EAAE,EAAE,CAAC,IAAI,GAAG,CAAC;IAE1E,IAAI,OAAO,CAAC,MAAM;QAAE,gBAAgB,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IACjE,IAAI,OAAO,CAAC,KAAK;QAAE,gBAAgB,CAAC,EAAE,YAAY,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;IAErE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;AACnD,CAAC;AAED,SAAS,aAAa,CAAC,MAA0B;IAC/C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,KAAK,CACX,yGAAyG,CAC1G,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,OAAO;KACJ,IAAI,CAAC,MAAM,CAAC;KACZ,WAAW,CAAC,8DAA8D,CAAC;KAC3E,OAAO,CAAC,OAAO,CAAC;KAChB,QAAQ,CAAC,cAAc,EAAE,yBAAyB,CAAC;KACnD,MAAM,CAAC,qBAAqB,EAAE,cAAc,CAAC;KAC7C,MAAM,CAAC,sBAAsB,EAAE,cAAc,CAAC;KAC9C,MAAM,CAAC,qBAAqB,EAAE,cAAc,CAAC;KAC7C,MAAM,CAAC,uBAAuB,EAAE,mBAAmB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC;KACnE,MAAM,CAAC,uBAAuB,EAAE,gCAAgC,CAAC;KACjE,MAAM,CAAC,oBAAoB,EAAE,iDAAiD,CAAC;KAC/E,MAAM,CAAC,+BAA+B,EAAE,uCAAuC,CAAC;KAChF,MAAM,CAAC,uBAAuB,EAAE,2BAA2B,EAAE,KAAK,CAAC;KACnE,MAAM,CAAC,KAAK,EAAE,OAAiB,EAAE,OAAO,EAAE,EAAE;IAC3C,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QACtB,IAAI,CAAC;YACH,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,OAAO,CAAC,KAAK,CAAC,8BAA8B,OAAO,CAAC,SAAS,KAAK,GAAG,EAAE,CAAC,CAAC;YACzE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED,IAAI,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAC/B,MAAM,uBAAuB,CAAC,OAAO,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC;QACnE,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IAEtC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,MAAM,WAAW,CACf,OAAO,CAAC,MAAM,EACd,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,EAC5B,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,aAAa,EACpB,OAAO,CAAC,OAAO,CAChB,CAAC;QACF,OAAO;IACT,CAAC;IAED,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1E,MAAM,gBAAgB,CACpB,MAAM,CAAC,MAAM,EACb,MAAM,CAAC,OAAO,EACd,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,aAAa,EACpB,OAAO,CAAC,OAAO,EACf,cAAc,CACf,CAAC;AACJ,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,QAAQ,CAAC;KACjB,WAAW,CAAC,4BAA4B,CAAC;KACzC,MAAM,CAAC,GAAG,EAAE;IACX,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;IAC1C,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;QACpD,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,IAAI,GAAG,SAAS,EAAE,CAAC,CAAC;QAChE,OAAO,CAAC,GAAG,CACT,OAAO,CAAC,CAAC,WAAW,MAAM,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC,UAAU,KAAK,CAAC,CAAC,WAAW,gBAAgB,CACtH,CAAC;IACJ,CAAC;IACD,OAAO,CAAC,GAAG,EAAE,CAAC;AAChB,CAAC,CAAC,CAAC;AAEL,OAAO,CAAC,KAAK,EAAE,CAAC;AAEhB,SAAS,aAAa,CAAC,MAAc;IACnC,IAAI,MAAM,IAAI,SAAS;QAAE,OAAO,GAAG,MAAM,GAAG,SAAS,GAAG,CAAC;IACzD,OAAO,GAAG,MAAM,GAAG,KAAK,GAAG,CAAC;AAC9B,CAAC"}

package/dist/storage/db.d.ts

@@ -0,0 +1,18 @@
+export interface SQLiteStatement {
+    run(...params: unknown[]): unknown;
+    get(...params: unknown[]): unknown;
+    all(...params: unknown[]): unknown[];
+}
+export interface SQLiteDatabase {
+    exec(sql: string): void;
+    prepare(sql: string): SQLiteStatement;
+    pragma(query: string, options?: {
+        simple?: boolean;
+    }): unknown;
+    transaction<T>(fn: () => T): () => T;
+    close(): void;
+}
+export declare function getDatabasePath(): string;
+export declare function getDatabase(): SQLiteDatabase;
+export declare function withTransaction<T>(fn: (database: SQLiteDatabase) => T): T;
+export declare function closeDatabase(): void;

package/dist/storage/db.js

@@ -0,0 +1,71 @@
+import { Database } from "bun:sqlite";
+import fs from "fs";
+import os from "os";
+import path from "path";
+import { applyMigrations } from "./migrations";
+let db = null;
+export function getDatabasePath() {
+    const dir = path.join(os.homedir(), ".grok");
+    fs.mkdirSync(dir, { recursive: true, mode: 0o700 });
+    return path.join(dir, "grok.db");
+}
+export function getDatabase() {
+    if (db)
+        return db;
+    const database = new BunSqliteDatabase(getDatabasePath());
+    database.pragma("journal_mode = WAL");
+    database.pragma("foreign_keys = ON");
+    database.pragma("busy_timeout = 5000");
+    database.pragma("synchronous = NORMAL");
+    applyMigrations(database);
+    db = database;
+    return database;
+}
+export function withTransaction(fn) {
+    const database = getDatabase();
+    return database.transaction(() => fn(database))();
+}
+export function closeDatabase() {
+    db?.close();
+    db = null;
+}
+class BunSqliteDatabase {
+    db;
+    constructor(filename) {
+        this.db = new Database(filename, { create: true, strict: true });
+    }
+    exec(sql) {
+        this.db.exec(sql);
+    }
+    prepare(sql) {
+        return {
+            run: (...params) => this.db.run(sql, normalizeBinding(params)),
+            get: (...params) => this.db.query(sql).get(normalizeBinding(params)),
+            all: (...params) => this.db.query(sql).all(normalizeBinding(params)),
+        };
+    }
+    pragma(query, options) {
+        if (query.includes("=")) {
+            this.db.exec(`PRAGMA ${query}`);
+            return undefined;
+        }
+        const row = this.db.query(`PRAGMA ${query}`).get();
+        if (!options?.simple)
+            return row;
+        if (!row)
+            return undefined;
+        return Object.values(row)[0];
+    }
+    transaction(fn) {
+        return this.db.transaction(fn);
+    }
+    close() {
+        this.db.close();
+    }
+}
+function normalizeBinding(params) {
+    if (params.length === 0)
+        return undefined;
+    return params.length === 1 ? params[0] : params;
+}
+//# sourceMappingURL=db.js.map

package/dist/storage/db.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"db.js","sourceRoot":"","sources":["../../src/storage/db.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAgB/C,IAAI,EAAE,GAA0B,IAAI,CAAC;AAErC,MAAM,UAAU,eAAe;IAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;IAC7C,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IACpD,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,WAAW;IACzB,IAAI,EAAE;QAAE,OAAO,EAAE,CAAC;IAElB,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,eAAe,EAAE,CAAC,CAAC;IAC1D,QAAQ,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;IACtC,QAAQ,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;IACrC,QAAQ,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;IACvC,QAAQ,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC;IACxC,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC1B,EAAE,GAAG,QAAQ,CAAC;IACd,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,eAAe,CAAI,EAAmC;IACpE,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAC;IAC/B,OAAO,QAAQ,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,aAAa;IAC3B,EAAE,EAAE,KAAK,EAAE,CAAC;IACZ,EAAE,GAAG,IAAI,CAAC;AACZ,CAAC;AAED,MAAM,iBAAiB;IACJ,EAAE,CAAW;IAE9B,YAAY,QAAgB;QAC1B,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,CAAC,GAAW;QACd,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACpB,CAAC;IAED,OAAO,CAAC,GAAW;QACjB,OAAO;YACL,GAAG,EAAE,CAAC,GAAG,MAAiB,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;YACzE,GAAG,EAAE,CAAC,GAAG,MAAiB,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC/E,GAAG,EAAE,CAAC,GAAG,MAAiB,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;SAChF,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAa,EAAE,OAA8B;QAClD,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACxB,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,KAAK,EAAE,CAAC,CAAC;YAChC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,KAAK,EAAE,CAAC,CAAC,GAAG,EAAyC,CAAC;QAC1F,IAAI,CAAC,OAAO,EAAE,MAAM;YAAE,OAAO,GAAG,CAAC;QACjC,IAAI,CAAC,GAAG;YAAE,OAAO,SAAS,CAAC;QAC3B,OAAO,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/B,CAAC;IAED,WAAW,CAAI,EAAW;QACxB,OAAO,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,MAAiB;IACzC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC1C,OAAO,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;AAClD,CAAC"}

package/dist/storage/index.d.ts

@@ -0,0 +1,4 @@
+export { getDatabasePath } from "./db";
+export { SessionStore } from "./sessions";
+export { appendMessages, appendSystemMessage, buildChatEntries, loadTranscript } from "./transcript";
+export { getSessionTotalTokens, listSessionUsage, recordUsageEvent, type TokenUsageLike } from "./usage";

package/dist/storage/index.js

@@ -0,0 +1,5 @@
+export { getDatabasePath } from "./db";
+export { SessionStore } from "./sessions";
+export { appendMessages, appendSystemMessage, buildChatEntries, loadTranscript } from "./transcript";
+export { getSessionTotalTokens, listSessionUsage, recordUsageEvent } from "./usage";
+//# sourceMappingURL=index.js.map

package/dist/storage/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,MAAM,CAAC;AACvC,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC1C,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AACrG,OAAO,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,gBAAgB,EAAuB,MAAM,SAAS,CAAC"}

package/dist/storage/migrations.d.ts

@@ -0,0 +1,2 @@
+import type { SQLiteDatabase } from "./db";
+export declare function applyMigrations(db: SQLiteDatabase): void;