bonecode 1.1.0 → 1.2.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +145 -9
- package/bin/bonecode +47 -42
- package/compat/opencode_adapter.ts +188 -17
- package/dist/bone/output/agent/src/algorithms.d.ts +1 -0
- package/dist/bone/output/agent/src/algorithms.js +3 -0
- package/dist/bone/output/agent/src/algorithms.js.map +1 -0
- package/dist/bone/output/agent/src/audit.d.ts +3 -0
- package/dist/bone/output/agent/src/audit.js +40 -0
- package/dist/bone/output/agent/src/audit.js.map +1 -0
- package/dist/bone/output/agent/src/auth.d.ts +8 -0
- package/dist/bone/output/agent/src/auth.js +56 -0
- package/dist/bone/output/agent/src/auth.js.map +1 -0
- package/dist/bone/output/agent/src/db.d.ts +6 -0
- package/dist/bone/output/agent/src/db.js +63 -0
- package/dist/bone/output/agent/src/db.js.map +1 -0
- package/dist/bone/output/agent/src/events.d.ts +25 -0
- package/dist/bone/output/agent/src/events.js +184 -0
- package/dist/bone/output/agent/src/events.js.map +1 -0
- package/dist/bone/output/agent/src/logger.d.ts +28 -0
- package/dist/bone/output/agent/src/logger.js +45 -0
- package/dist/bone/output/agent/src/logger.js.map +1 -0
- package/dist/bone/output/agent/src/metrics.d.ts +5 -0
- package/dist/bone/output/agent/src/metrics.js +60 -0
- package/dist/bone/output/agent/src/metrics.js.map +1 -0
- package/dist/bone/output/agent/src/routes/agent_instance.d.ts +1 -0
- package/dist/bone/output/agent/src/routes/agent_instance.js +253 -0
- package/dist/bone/output/agent/src/routes/agent_instance.js.map +1 -0
- package/dist/bone/output/agent/src/routes/build_step.d.ts +1 -0
- package/dist/bone/output/agent/src/routes/build_step.js +133 -0
- package/dist/bone/output/agent/src/routes/build_step.js.map +1 -0
- package/dist/bone/output/agent/src/routes/plan.d.ts +1 -0
- package/dist/bone/output/agent/src/routes/plan.js +119 -0
- package/dist/bone/output/agent/src/routes/plan.js.map +1 -0
- package/dist/bone/output/agent/src/routes/task.d.ts +1 -0
- package/dist/bone/output/agent/src/routes/task.js +133 -0
- package/dist/bone/output/agent/src/routes/task.js.map +1 -0
- package/dist/bone/output/agent/src/routes/tool_call.d.ts +1 -0
- package/dist/bone/output/agent/src/routes/tool_call.js +190 -0
- package/dist/bone/output/agent/src/routes/tool_call.js.map +1 -0
- package/dist/bone/output/agent/src/state_machines/agent_instance.d.ts +9 -0
- package/dist/bone/output/agent/src/state_machines/agent_instance.js +22 -0
- package/dist/bone/output/agent/src/state_machines/agent_instance.js.map +1 -0
- package/dist/bone/output/agent/src/state_machines/build_step.d.ts +9 -0
- package/dist/bone/output/agent/src/state_machines/build_step.js +20 -0
- package/dist/bone/output/agent/src/state_machines/build_step.js.map +1 -0
- package/dist/bone/output/agent/src/state_machines/plan.d.ts +9 -0
- package/dist/bone/output/agent/src/state_machines/plan.js +20 -0
- package/dist/bone/output/agent/src/state_machines/plan.js.map +1 -0
- package/dist/bone/output/agent/src/state_machines/task.d.ts +9 -0
- package/dist/bone/output/agent/src/state_machines/task.js +20 -0
- package/dist/bone/output/agent/src/state_machines/task.js.map +1 -0
- package/dist/bone/output/agent/src/state_machines/tool_call.d.ts +9 -0
- package/dist/bone/output/agent/src/state_machines/tool_call.js +20 -0
- package/dist/bone/output/agent/src/state_machines/tool_call.js.map +1 -0
- package/dist/bone/output/rag/src/algorithms.d.ts +1 -0
- package/dist/bone/output/rag/src/algorithms.js +3 -0
- package/dist/bone/output/rag/src/algorithms.js.map +1 -0
- package/dist/bone/output/rag/src/auth.d.ts +8 -0
- package/dist/bone/output/rag/src/auth.js +56 -0
- package/dist/bone/output/rag/src/auth.js.map +1 -0
- package/dist/bone/output/rag/src/db.d.ts +6 -0
- package/dist/bone/output/rag/src/db.js +63 -0
- package/dist/bone/output/rag/src/db.js.map +1 -0
- package/dist/bone/output/rag/src/events.d.ts +25 -0
- package/dist/bone/output/rag/src/events.js +184 -0
- package/dist/bone/output/rag/src/events.js.map +1 -0
- package/dist/bone/output/rag/src/extensions.d.ts +83 -0
- package/dist/bone/output/rag/src/extensions.js +329 -0
- package/dist/bone/output/rag/src/extensions.js.map +1 -0
- package/dist/bone/output/rag/src/flows.d.ts +24 -0
- package/dist/bone/output/rag/src/flows.js +236 -0
- package/dist/bone/output/rag/src/flows.js.map +1 -0
- package/dist/bone/output/rag/src/logger.d.ts +28 -0
- package/dist/bone/output/rag/src/logger.js +45 -0
- package/dist/bone/output/rag/src/logger.js.map +1 -0
- package/dist/bone/output/rag/src/metrics.d.ts +5 -0
- package/dist/bone/output/rag/src/metrics.js +60 -0
- package/dist/bone/output/rag/src/metrics.js.map +1 -0
- package/dist/bone/output/rag/src/routes/code_chunk.d.ts +1 -0
- package/dist/bone/output/rag/src/routes/code_chunk.js +100 -0
- package/dist/bone/output/rag/src/routes/code_chunk.js.map +1 -0
- package/dist/bone/output/rag/src/routes/code_file.d.ts +1 -0
- package/dist/bone/output/rag/src/routes/code_file.js +127 -0
- package/dist/bone/output/rag/src/routes/code_file.js.map +1 -0
- package/dist/bone/output/rag/src/routes/indexing_job.d.ts +1 -0
- package/dist/bone/output/rag/src/routes/indexing_job.js +113 -0
- package/dist/bone/output/rag/src/routes/indexing_job.js.map +1 -0
- package/dist/bone/output/rag/src/routes/knowledge_base.d.ts +1 -0
- package/dist/bone/output/rag/src/routes/knowledge_base.js +242 -0
- package/dist/bone/output/rag/src/routes/knowledge_base.js.map +1 -0
- package/dist/bone/output/rag/src/routes/memory_entry.d.ts +1 -0
- package/dist/bone/output/rag/src/routes/memory_entry.js +113 -0
- package/dist/bone/output/rag/src/routes/memory_entry.js.map +1 -0
- package/dist/bone/output/rag/src/state_machines/code_file.d.ts +9 -0
- package/dist/bone/output/rag/src/state_machines/code_file.js +21 -0
- package/dist/bone/output/rag/src/state_machines/code_file.js.map +1 -0
- package/dist/bone/output/rag/src/state_machines/indexing_job.d.ts +9 -0
- package/dist/bone/output/rag/src/state_machines/indexing_job.js +20 -0
- package/dist/bone/output/rag/src/state_machines/indexing_job.js.map +1 -0
- package/dist/bone/output/rag/src/state_machines/knowledge_base.d.ts +9 -0
- package/dist/bone/output/rag/src/state_machines/knowledge_base.js +21 -0
- package/dist/bone/output/rag/src/state_machines/knowledge_base.js.map +1 -0
- package/dist/bone/output/rag/src/state_machines/memory_entry.d.ts +9 -0
- package/dist/bone/output/rag/src/state_machines/memory_entry.js +18 -0
- package/dist/bone/output/rag/src/state_machines/memory_entry.js.map +1 -0
- package/dist/bone/output/session/src/algorithms.d.ts +1 -0
- package/dist/bone/output/session/src/algorithms.js +3 -0
- package/dist/bone/output/session/src/algorithms.js.map +1 -0
- package/dist/bone/output/session/src/audit.d.ts +3 -0
- package/dist/bone/output/session/src/audit.js +40 -0
- package/dist/bone/output/session/src/audit.js.map +1 -0
- package/dist/bone/output/session/src/auth.d.ts +8 -0
- package/dist/bone/output/session/src/auth.js +56 -0
- package/dist/bone/output/session/src/auth.js.map +1 -0
- package/dist/bone/output/session/src/db.d.ts +6 -0
- package/dist/bone/output/session/src/db.js +63 -0
- package/dist/bone/output/session/src/db.js.map +1 -0
- package/dist/bone/output/session/src/events.d.ts +26 -0
- package/dist/bone/output/session/src/events.js +212 -0
- package/dist/bone/output/session/src/events.js.map +1 -0
- package/dist/bone/output/session/src/extensions.d.ts +41 -0
- package/dist/bone/output/session/src/extensions.js +217 -0
- package/dist/bone/output/session/src/extensions.js.map +1 -0
- package/dist/bone/output/session/src/logger.d.ts +28 -0
- package/dist/bone/output/session/src/logger.js +44 -0
- package/dist/bone/output/session/src/logger.js.map +1 -0
- package/dist/bone/output/session/src/metrics.d.ts +5 -0
- package/dist/bone/output/session/src/metrics.js +60 -0
- package/dist/bone/output/session/src/metrics.js.map +1 -0
- package/dist/bone/output/session/src/routes/message.d.ts +1 -0
- package/dist/bone/output/session/src/routes/message.js +120 -0
- package/dist/bone/output/session/src/routes/message.js.map +1 -0
- package/dist/bone/output/session/src/routes/part.d.ts +1 -0
- package/dist/bone/output/session/src/routes/part.js +106 -0
- package/dist/bone/output/session/src/routes/part.js.map +1 -0
- package/dist/bone/output/session/src/routes/permission.d.ts +1 -0
- package/dist/bone/output/session/src/routes/permission.js +106 -0
- package/dist/bone/output/session/src/routes/permission.js.map +1 -0
- package/dist/bone/output/session/src/routes/project.d.ts +1 -0
- package/dist/bone/output/session/src/routes/project.js +106 -0
- package/dist/bone/output/session/src/routes/project.js.map +1 -0
- package/dist/bone/output/session/src/routes/session.d.ts +1 -0
- package/dist/bone/output/session/src/routes/session.js +308 -0
- package/dist/bone/output/session/src/routes/session.js.map +1 -0
- package/dist/bone/output/session/src/state_machines/session.d.ts +9 -0
- package/dist/bone/output/session/src/state_machines/session.js +21 -0
- package/dist/bone/output/session/src/state_machines/session.js.map +1 -0
- package/dist/bone/output/session/src/websocket.d.ts +15 -0
- package/dist/bone/output/session/src/websocket.js +215 -0
- package/dist/bone/output/session/src/websocket.js.map +1 -0
- package/dist/bone/output/workspace/src/algorithms.d.ts +1 -0
- package/dist/bone/output/workspace/src/algorithms.js +3 -0
- package/dist/bone/output/workspace/src/algorithms.js.map +1 -0
- package/dist/bone/output/workspace/src/auth.d.ts +8 -0
- package/dist/bone/output/workspace/src/auth.js +56 -0
- package/dist/bone/output/workspace/src/auth.js.map +1 -0
- package/dist/bone/output/workspace/src/db.d.ts +6 -0
- package/dist/bone/output/workspace/src/db.js +63 -0
- package/dist/bone/output/workspace/src/db.js.map +1 -0
- package/dist/bone/output/workspace/src/events.d.ts +25 -0
- package/dist/bone/output/workspace/src/events.js +184 -0
- package/dist/bone/output/workspace/src/events.js.map +1 -0
- package/dist/bone/output/workspace/src/logger.d.ts +28 -0
- package/dist/bone/output/workspace/src/logger.js +45 -0
- package/dist/bone/output/workspace/src/logger.js.map +1 -0
- package/dist/bone/output/workspace/src/metrics.d.ts +5 -0
- package/dist/bone/output/workspace/src/metrics.js +60 -0
- package/dist/bone/output/workspace/src/metrics.js.map +1 -0
- package/dist/bone/output/workspace/src/routes/codebase.d.ts +1 -0
- package/dist/bone/output/workspace/src/routes/codebase.js +113 -0
- package/dist/bone/output/workspace/src/routes/codebase.js.map +1 -0
- package/dist/bone/output/workspace/src/routes/snapshot.d.ts +1 -0
- package/dist/bone/output/workspace/src/routes/snapshot.js +151 -0
- package/dist/bone/output/workspace/src/routes/snapshot.js.map +1 -0
- package/dist/bone/output/workspace/src/routes/workspace.d.ts +1 -0
- package/dist/bone/output/workspace/src/routes/workspace.js +209 -0
- package/dist/bone/output/workspace/src/routes/workspace.js.map +1 -0
- package/dist/bone/output/workspace/src/state_machines/codebase.d.ts +9 -0
- package/dist/bone/output/workspace/src/state_machines/codebase.js +19 -0
- package/dist/bone/output/workspace/src/state_machines/codebase.js.map +1 -0
- package/dist/bone/output/workspace/src/state_machines/snapshot.d.ts +9 -0
- package/dist/bone/output/workspace/src/state_machines/snapshot.js +18 -0
- package/dist/bone/output/workspace/src/state_machines/snapshot.js.map +1 -0
- package/dist/bone/output/workspace/src/state_machines/workspace.d.ts +9 -0
- package/dist/bone/output/workspace/src/state_machines/workspace.js +19 -0
- package/dist/bone/output/workspace/src/state_machines/workspace.js.map +1 -0
- package/dist/compat/opencode_adapter.d.ts +25 -0
- package/dist/compat/opencode_adapter.js +599 -0
- package/dist/compat/opencode_adapter.js.map +1 -0
- package/dist/extensions/chunker.d.ts +24 -0
- package/dist/extensions/chunker.js +360 -0
- package/dist/extensions/chunker.js.map +1 -0
- package/dist/extensions/embedding_provider.d.ts +18 -0
- package/dist/extensions/embedding_provider.js +150 -0
- package/dist/extensions/embedding_provider.js.map +1 -0
- package/dist/extensions/llm_provider.d.ts +33 -0
- package/dist/extensions/llm_provider.js +338 -0
- package/dist/extensions/llm_provider.js.map +1 -0
- package/dist/extensions/mcp_bridge.d.ts +44 -0
- package/dist/extensions/mcp_bridge.js +151 -0
- package/dist/extensions/mcp_bridge.js.map +1 -0
- package/dist/extensions/rag_search.d.ts +38 -0
- package/dist/extensions/rag_search.js +242 -0
- package/dist/extensions/rag_search.js.map +1 -0
- package/dist/extensions/snapshot.d.ts +14 -0
- package/dist/extensions/snapshot.js +158 -0
- package/dist/extensions/snapshot.js.map +1 -0
- package/dist/extensions/tool_executor.d.ts +28 -0
- package/dist/extensions/tool_executor.js +268 -0
- package/dist/extensions/tool_executor.js.map +1 -0
- package/dist/src/cli.d.ts +15 -0
- package/dist/src/cli.js +687 -0
- package/dist/src/cli.js.map +1 -0
- package/dist/src/config.d.ts +44 -0
- package/dist/src/config.js +165 -0
- package/dist/src/config.js.map +1 -0
- package/dist/src/context_builder.d.ts +51 -0
- package/dist/src/context_builder.js +558 -0
- package/dist/src/context_builder.js.map +1 -0
- package/dist/src/db_adapter.d.ts +24 -0
- package/dist/src/db_adapter.js +341 -0
- package/dist/src/db_adapter.js.map +1 -0
- package/dist/src/engine/session/compaction_logic.d.ts +11 -0
- package/dist/src/engine/session/compaction_logic.js +113 -0
- package/dist/src/engine/session/compaction_logic.js.map +1 -0
- package/dist/src/engine/session/instruction_loader.d.ts +5 -0
- package/dist/src/engine/session/instruction_loader.js +78 -0
- package/dist/src/engine/session/instruction_loader.js.map +1 -0
- package/dist/src/engine/session/overflow_check.d.ts +14 -0
- package/dist/src/engine/session/overflow_check.js +45 -0
- package/dist/src/engine/session/overflow_check.js.map +1 -0
- package/dist/src/engine/session/prompt.d.ts +45 -0
- package/dist/src/engine/session/prompt.js +584 -0
- package/dist/src/engine/session/prompt.js.map +1 -0
- package/dist/src/engine/session/provider_transform.d.ts +59 -0
- package/dist/src/engine/session/provider_transform.js +193 -0
- package/dist/src/engine/session/provider_transform.js.map +1 -0
- package/dist/src/engine/session/retry_logic.d.ts +12 -0
- package/dist/src/engine/session/retry_logic.js +72 -0
- package/dist/src/engine/session/retry_logic.js.map +1 -0
- package/dist/src/engine/session/system_prompt.d.ts +9 -0
- package/dist/src/engine/session/system_prompt.js +96 -0
- package/dist/src/engine/session/system_prompt.js.map +1 -0
- package/dist/src/engine/session/tool_registry.d.ts +5 -0
- package/dist/src/engine/session/tool_registry.js +117 -0
- package/dist/src/engine/session/tool_registry.js.map +1 -0
- package/dist/src/export.d.ts +13 -0
- package/dist/src/export.js +103 -0
- package/dist/src/export.js.map +1 -0
- package/dist/src/mdns.d.ts +7 -0
- package/dist/src/mdns.js +60 -0
- package/dist/src/mdns.js.map +1 -0
- package/dist/src/rag_worker.d.ts +38 -0
- package/dist/src/rag_worker.js +435 -0
- package/dist/src/rag_worker.js.map +1 -0
- package/dist/src/server.d.ts +11 -0
- package/dist/src/server.js +214 -0
- package/dist/src/server.js.map +1 -0
- package/dist/src/stats.d.ts +45 -0
- package/dist/src/stats.js +233 -0
- package/dist/src/stats.js.map +1 -0
- package/dist/src/tui.d.ts +29 -0
- package/dist/src/tui.js +1053 -0
- package/dist/src/tui.js.map +1 -0
- package/package.json +7 -4
- package/src/cli.ts +247 -5
- package/src/export.ts +122 -0
- package/src/mdns.ts +53 -0
- package/src/server.ts +32 -0
- package/src/stats.ts +290 -0
- package/src/tui.ts +749 -248
|
@@ -0,0 +1,341 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* BoneCode Database Adapter
|
|
4
|
+
*
|
|
5
|
+
* Provides a unified pool.query() interface that works with either:
|
|
6
|
+
* - PostgreSQL (full experience — RAG, pgvector, durable events)
|
|
7
|
+
* - SQLite (zero-setup fallback — sessions and agent turns only, no RAG)
|
|
8
|
+
*
|
|
9
|
+
* Detection order:
|
|
10
|
+
* 1. If DATABASE_URL is set and Postgres is reachable → use Postgres
|
|
11
|
+
* 2. If Docker is available → offer to auto-start the bonecode-postgres container
|
|
12
|
+
* 3. Otherwise → fall back to SQLite at ~/.bonecode/bonecode.db
|
|
13
|
+
*
|
|
14
|
+
* The SQLite adapter translates the pg Pool.query() interface so all existing
|
|
15
|
+
* code that calls pool.query(sql, params) works without changes.
|
|
16
|
+
* Postgres-specific syntax (RETURNING, ON CONFLICT, jsonb, vector) is
|
|
17
|
+
* translated or stubbed where needed.
|
|
18
|
+
*/
|
|
19
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
20
|
+
if (k2 === undefined) k2 = k;
|
|
21
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
22
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
23
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
24
|
+
}
|
|
25
|
+
Object.defineProperty(o, k2, desc);
|
|
26
|
+
}) : (function(o, m, k, k2) {
|
|
27
|
+
if (k2 === undefined) k2 = k;
|
|
28
|
+
o[k2] = m[k];
|
|
29
|
+
}));
|
|
30
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
31
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
32
|
+
}) : function(o, v) {
|
|
33
|
+
o["default"] = v;
|
|
34
|
+
});
|
|
35
|
+
var __importStar = (this && this.__importStar) || function (mod) {
|
|
36
|
+
if (mod && mod.__esModule) return mod;
|
|
37
|
+
var result = {};
|
|
38
|
+
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
|
|
39
|
+
__setModuleDefault(result, mod);
|
|
40
|
+
return result;
|
|
41
|
+
};
|
|
42
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
43
|
+
exports.initDatabase = exports.getDBMode = void 0;
|
|
44
|
+
const fs = __importStar(require("fs"));
|
|
45
|
+
const path = __importStar(require("path"));
|
|
46
|
+
const os = __importStar(require("os"));
|
|
47
|
+
let _mode = null;
|
|
48
|
+
function getDBMode() {
|
|
49
|
+
return _mode || "sqlite";
|
|
50
|
+
}
|
|
51
|
+
exports.getDBMode = getDBMode;
|
|
52
|
+
// ─── SQLite pool shim ─────────────────────────────────────────────────────────
|
|
53
|
+
// Wraps better-sqlite3 in a pg Pool-compatible interface.
|
|
54
|
+
function createSQLitePool(dbPath) {
|
|
55
|
+
const Database = require("better-sqlite3");
|
|
56
|
+
fs.mkdirSync(path.dirname(dbPath), { recursive: true });
|
|
57
|
+
const db = new Database(dbPath);
|
|
58
|
+
// Enable WAL mode for better concurrent read performance
|
|
59
|
+
db.pragma("journal_mode = WAL");
|
|
60
|
+
db.pragma("foreign_keys = ON");
|
|
61
|
+
// Translate common Postgres-isms to SQLite
|
|
62
|
+
function translateSQL(sql) {
|
|
63
|
+
return sql
|
|
64
|
+
// $1, $2 → ? placeholders
|
|
65
|
+
.replace(/\$\d+/g, "?")
|
|
66
|
+
// RETURNING * / RETURNING id → handled below
|
|
67
|
+
// NOW() → datetime('now')
|
|
68
|
+
.replace(/\bNOW\(\)/gi, "datetime('now')")
|
|
69
|
+
// ILIKE → LIKE (SQLite is case-insensitive for ASCII by default)
|
|
70
|
+
.replace(/\bILIKE\b/gi, "LIKE")
|
|
71
|
+
// ::text, ::jsonb, ::vector casts → remove
|
|
72
|
+
.replace(/::[a-z_[\]()0-9]+/gi, "")
|
|
73
|
+
// gen_random_uuid() → (not needed — we use uuid() from JS)
|
|
74
|
+
.replace(/gen_random_uuid\(\)/gi, "lower(hex(randomblob(4)))||'-'||lower(hex(randomblob(2)))||'-4'||substr(lower(hex(randomblob(2))),2)||'-'||substr('89ab',abs(random())%4+1,1)||substr(lower(hex(randomblob(2))),2)||'-'||lower(hex(randomblob(6)))")
|
|
75
|
+
// JSONB operators → not supported, strip
|
|
76
|
+
.replace(/->>?\s*'[^']*'/g, "")
|
|
77
|
+
// json_agg(...) FILTER (...) → NULL (SQLite doesn't support this)
|
|
78
|
+
.replace(/json_agg\([^)]+\)\s*FILTER\s*\([^)]+\)/gi, "NULL")
|
|
79
|
+
// to_tsvector / to_tsquery / ts_rank → stub (FTS not available in SQLite fallback)
|
|
80
|
+
.replace(/to_tsvector\([^)]+\)/gi, "''")
|
|
81
|
+
.replace(/to_tsquery\([^)]+\)/gi, "''")
|
|
82
|
+
.replace(/ts_rank\([^)]+\)/gi, "0");
|
|
83
|
+
}
|
|
84
|
+
const pool = {
|
|
85
|
+
query: async (sql, params) => {
|
|
86
|
+
try {
|
|
87
|
+
const translated = translateSQL(sql);
|
|
88
|
+
const flatParams = (params || []).map((p) => typeof p === "object" && p !== null ? JSON.stringify(p) : p);
|
|
89
|
+
// Detect RETURNING clause — SQLite doesn't support it
|
|
90
|
+
const hasReturning = /\bRETURNING\b/i.test(sql);
|
|
91
|
+
const returningMatch = sql.match(/RETURNING\s+(.+)$/i);
|
|
92
|
+
// Strip RETURNING from the translated SQL
|
|
93
|
+
const execSQL = translated.replace(/\s+RETURNING\s+.+$/i, "");
|
|
94
|
+
const isSelect = /^\s*(SELECT|WITH|PRAGMA)/i.test(execSQL);
|
|
95
|
+
if (isSelect) {
|
|
96
|
+
const stmt = db.prepare(execSQL);
|
|
97
|
+
const rows = stmt.all(...flatParams);
|
|
98
|
+
return { rows, rowCount: rows.length };
|
|
99
|
+
}
|
|
100
|
+
else {
|
|
101
|
+
const stmt = db.prepare(execSQL);
|
|
102
|
+
const info = stmt.run(...flatParams);
|
|
103
|
+
// If RETURNING was requested, fetch the last inserted/updated row
|
|
104
|
+
if (hasReturning && info.lastInsertRowid) {
|
|
105
|
+
// Try to get the table name from the SQL
|
|
106
|
+
const tableMatch = sql.match(/(?:INSERT INTO|UPDATE)\s+(\w+)/i);
|
|
107
|
+
if (tableMatch) {
|
|
108
|
+
try {
|
|
109
|
+
const row = db.prepare(`SELECT * FROM ${tableMatch[1]} WHERE rowid = ?`).get(info.lastInsertRowid);
|
|
110
|
+
return { rows: row ? [row] : [], rowCount: info.changes };
|
|
111
|
+
}
|
|
112
|
+
catch { /* table might not have rowid */ }
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
return { rows: [], rowCount: info.changes };
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
catch (e) {
|
|
119
|
+
// Silently ignore errors for Postgres-specific features not in SQLite
|
|
120
|
+
if (e.message?.includes("no such function") ||
|
|
121
|
+
e.message?.includes("no such column") ||
|
|
122
|
+
e.message?.includes("syntax error")) {
|
|
123
|
+
return { rows: [], rowCount: 0 };
|
|
124
|
+
}
|
|
125
|
+
throw e;
|
|
126
|
+
}
|
|
127
|
+
},
|
|
128
|
+
connect: async () => {
|
|
129
|
+
// Return a fake client for transaction support
|
|
130
|
+
return {
|
|
131
|
+
query: async (sql, params) => pool.query(sql, params),
|
|
132
|
+
release: () => { },
|
|
133
|
+
};
|
|
134
|
+
},
|
|
135
|
+
end: async () => {
|
|
136
|
+
db.close();
|
|
137
|
+
},
|
|
138
|
+
on: (_event, _handler) => { },
|
|
139
|
+
};
|
|
140
|
+
return pool;
|
|
141
|
+
}
|
|
142
|
+
// ─── SQLite schema ────────────────────────────────────────────────────────────
|
|
143
|
+
// Minimal schema for sessions, messages, parts, projects — enough for the
|
|
144
|
+
// core agent loop without RAG.
|
|
145
|
+
const SQLITE_SCHEMA = `
|
|
146
|
+
CREATE TABLE IF NOT EXISTS projects (
|
|
147
|
+
id TEXT PRIMARY KEY,
|
|
148
|
+
root_commit_hash TEXT,
|
|
149
|
+
worktree TEXT,
|
|
150
|
+
vcs TEXT DEFAULT 'git',
|
|
151
|
+
name TEXT,
|
|
152
|
+
sandboxes TEXT DEFAULT '[]',
|
|
153
|
+
created_at TEXT DEFAULT (datetime('now')),
|
|
154
|
+
updated_at TEXT DEFAULT (datetime('now'))
|
|
155
|
+
);
|
|
156
|
+
|
|
157
|
+
CREATE TABLE IF NOT EXISTS sessions (
|
|
158
|
+
id TEXT PRIMARY KEY,
|
|
159
|
+
slug TEXT,
|
|
160
|
+
project_id TEXT,
|
|
161
|
+
parent_id TEXT,
|
|
162
|
+
directory TEXT,
|
|
163
|
+
title TEXT,
|
|
164
|
+
version TEXT DEFAULT '1.0.0',
|
|
165
|
+
state TEXT DEFAULT 'created',
|
|
166
|
+
summary_files INTEGER,
|
|
167
|
+
summary_additions INTEGER,
|
|
168
|
+
summary_deletions INTEGER,
|
|
169
|
+
share_url TEXT,
|
|
170
|
+
workspace_id TEXT,
|
|
171
|
+
created_at TEXT DEFAULT (datetime('now')),
|
|
172
|
+
updated_at TEXT DEFAULT (datetime('now'))
|
|
173
|
+
);
|
|
174
|
+
|
|
175
|
+
CREATE TABLE IF NOT EXISTS messages (
|
|
176
|
+
id TEXT PRIMARY KEY,
|
|
177
|
+
session_id TEXT,
|
|
178
|
+
role TEXT,
|
|
179
|
+
model_id TEXT,
|
|
180
|
+
provider_id TEXT,
|
|
181
|
+
tokens_input INTEGER DEFAULT 0,
|
|
182
|
+
tokens_output INTEGER DEFAULT 0,
|
|
183
|
+
cost_usd REAL DEFAULT 0,
|
|
184
|
+
created_at TEXT DEFAULT (datetime('now')),
|
|
185
|
+
updated_at TEXT DEFAULT (datetime('now'))
|
|
186
|
+
);
|
|
187
|
+
|
|
188
|
+
CREATE TABLE IF NOT EXISTS parts (
|
|
189
|
+
id TEXT PRIMARY KEY,
|
|
190
|
+
message_id TEXT,
|
|
191
|
+
session_id TEXT,
|
|
192
|
+
part_type TEXT,
|
|
193
|
+
data TEXT DEFAULT '{}',
|
|
194
|
+
order_index INTEGER DEFAULT 0,
|
|
195
|
+
created_at TEXT DEFAULT (datetime('now')),
|
|
196
|
+
updated_at TEXT DEFAULT (datetime('now'))
|
|
197
|
+
);
|
|
198
|
+
|
|
199
|
+
CREATE TABLE IF NOT EXISTS agent_instances (
|
|
200
|
+
id TEXT PRIMARY KEY,
|
|
201
|
+
session_id TEXT,
|
|
202
|
+
model_id TEXT,
|
|
203
|
+
provider_id TEXT,
|
|
204
|
+
context_window_used INTEGER DEFAULT 0,
|
|
205
|
+
context_window_max INTEGER DEFAULT 200000,
|
|
206
|
+
total_cost_usd REAL DEFAULT 0,
|
|
207
|
+
total_tokens_in INTEGER DEFAULT 0,
|
|
208
|
+
total_tokens_out INTEGER DEFAULT 0,
|
|
209
|
+
config TEXT DEFAULT '{}',
|
|
210
|
+
state TEXT DEFAULT 'running',
|
|
211
|
+
created_at TEXT DEFAULT (datetime('now')),
|
|
212
|
+
updated_at TEXT DEFAULT (datetime('now'))
|
|
213
|
+
);
|
|
214
|
+
|
|
215
|
+
CREATE TABLE IF NOT EXISTS tool_calls (
|
|
216
|
+
id TEXT PRIMARY KEY,
|
|
217
|
+
session_id TEXT,
|
|
218
|
+
agent_id TEXT,
|
|
219
|
+
tool_name TEXT,
|
|
220
|
+
tool_input TEXT DEFAULT '{}',
|
|
221
|
+
tool_output TEXT,
|
|
222
|
+
state TEXT DEFAULT 'running',
|
|
223
|
+
created_at TEXT DEFAULT (datetime('now')),
|
|
224
|
+
updated_at TEXT DEFAULT (datetime('now'))
|
|
225
|
+
);
|
|
226
|
+
|
|
227
|
+
CREATE TABLE IF NOT EXISTS permissions (
|
|
228
|
+
id TEXT PRIMARY KEY,
|
|
229
|
+
session_id TEXT,
|
|
230
|
+
tool_name TEXT,
|
|
231
|
+
decision TEXT,
|
|
232
|
+
created_at TEXT DEFAULT (datetime('now'))
|
|
233
|
+
);
|
|
234
|
+
|
|
235
|
+
CREATE INDEX IF NOT EXISTS idx_sessions_project ON sessions(project_id);
|
|
236
|
+
CREATE INDEX IF NOT EXISTS idx_messages_session ON messages(session_id);
|
|
237
|
+
CREATE INDEX IF NOT EXISTS idx_parts_message ON parts(message_id);
|
|
238
|
+
CREATE INDEX IF NOT EXISTS idx_parts_session ON parts(session_id);
|
|
239
|
+
CREATE INDEX IF NOT EXISTS idx_tool_calls_session ON tool_calls(session_id);
|
|
240
|
+
`;
|
|
241
|
+
// ─── Postgres availability check ──────────────────────────────────────────────
|
|
242
|
+
async function checkPostgres(url) {
|
|
243
|
+
try {
|
|
244
|
+
const { Pool } = require("pg");
|
|
245
|
+
const testPool = new Pool({ connectionString: url, connectionTimeoutMillis: 3000, max: 1 });
|
|
246
|
+
await testPool.query("SELECT 1");
|
|
247
|
+
await testPool.end();
|
|
248
|
+
return true;
|
|
249
|
+
}
|
|
250
|
+
catch {
|
|
251
|
+
return false;
|
|
252
|
+
}
|
|
253
|
+
}
|
|
254
|
+
// ─── Auto-start Docker container ─────────────────────────────────────────────
|
|
255
|
+
async function tryStartDockerPostgres() {
|
|
256
|
+
try {
|
|
257
|
+
const { execSync } = require("child_process");
|
|
258
|
+
// Check if docker is available
|
|
259
|
+
execSync("docker --version", { stdio: "ignore" });
|
|
260
|
+
// Check if container already exists
|
|
261
|
+
try {
|
|
262
|
+
const status = execSync("docker inspect --format={{.State.Running}} bonecode-postgres", { stdio: "pipe" }).toString().trim();
|
|
263
|
+
if (status === "true")
|
|
264
|
+
return true; // already running
|
|
265
|
+
// Exists but stopped — start it
|
|
266
|
+
execSync("docker start bonecode-postgres", { stdio: "ignore" });
|
|
267
|
+
await new Promise(r => setTimeout(r, 2000));
|
|
268
|
+
return true;
|
|
269
|
+
}
|
|
270
|
+
catch {
|
|
271
|
+
// Container doesn't exist — create it
|
|
272
|
+
execSync("docker run -d --name bonecode-postgres -e POSTGRES_DB=bonecode -e POSTGRES_USER=postgres -e POSTGRES_PASSWORD=postgres -p 5434:5432 pgvector/pgvector:pg16", { stdio: "ignore" });
|
|
273
|
+
// Wait for Postgres to be ready
|
|
274
|
+
for (let i = 0; i < 15; i++) {
|
|
275
|
+
await new Promise(r => setTimeout(r, 1000));
|
|
276
|
+
if (await checkPostgres("postgresql://postgres:postgres@localhost:5434/bonecode")) {
|
|
277
|
+
return true;
|
|
278
|
+
}
|
|
279
|
+
}
|
|
280
|
+
return false;
|
|
281
|
+
}
|
|
282
|
+
}
|
|
283
|
+
catch {
|
|
284
|
+
return false; // Docker not available
|
|
285
|
+
}
|
|
286
|
+
}
|
|
287
|
+
// ─── Main init ────────────────────────────────────────────────────────────────
|
|
288
|
+
async function initDatabase(silent = false) {
|
|
289
|
+
const log = (msg) => { if (!silent)
|
|
290
|
+
process.stdout.write(msg + "\n"); };
|
|
291
|
+
// 1. Try configured DATABASE_URL
|
|
292
|
+
const dbUrl = process.env.DATABASE_URL;
|
|
293
|
+
if (dbUrl && !dbUrl.startsWith("sqlite")) {
|
|
294
|
+
if (await checkPostgres(dbUrl)) {
|
|
295
|
+
_mode = "postgres";
|
|
296
|
+
const { Pool } = require("pg");
|
|
297
|
+
const pgPool = new Pool({ connectionString: dbUrl, max: 20 });
|
|
298
|
+
pgPool.on("error", (err) => {
|
|
299
|
+
if (process.env.NODE_ENV !== "test") {
|
|
300
|
+
process.stderr.write(`[DB] Pool error: ${err.message}\n`);
|
|
301
|
+
}
|
|
302
|
+
});
|
|
303
|
+
return { pool: pgPool, mode: "postgres", pgAvailable: true };
|
|
304
|
+
}
|
|
305
|
+
log(`\x1b[33m[BoneCode] Postgres not reachable at ${dbUrl.replace(/:[^:@]+@/, ":***@")} — trying Docker...\x1b[0m`);
|
|
306
|
+
}
|
|
307
|
+
// 2. Try auto-starting Docker
|
|
308
|
+
const dockerStarted = await tryStartDockerPostgres();
|
|
309
|
+
if (dockerStarted) {
|
|
310
|
+
// Use port 5434 (our standard bonecode-postgres container)
|
|
311
|
+
const dockerUrl = "postgresql://postgres:postgres@localhost:5434/bonecode";
|
|
312
|
+
if (await checkPostgres(dockerUrl)) {
|
|
313
|
+
process.env.DATABASE_URL = dockerUrl;
|
|
314
|
+
_mode = "postgres";
|
|
315
|
+
const { Pool } = require("pg");
|
|
316
|
+
const pgPool = new Pool({ connectionString: dockerUrl, max: 20 });
|
|
317
|
+
pgPool.on("error", (err) => {
|
|
318
|
+
if (process.env.NODE_ENV !== "test") {
|
|
319
|
+
process.stderr.write(`[DB] Pool error: ${err.message}\n`);
|
|
320
|
+
}
|
|
321
|
+
});
|
|
322
|
+
log(`\x1b[32m[BoneCode] PostgreSQL ready (Docker) — full RAG enabled\x1b[0m`);
|
|
323
|
+
return { pool: pgPool, mode: "postgres", pgAvailable: true };
|
|
324
|
+
}
|
|
325
|
+
}
|
|
326
|
+
// 3. Fall back to SQLite
|
|
327
|
+
const dbDir = path.join(os.homedir(), ".bonecode");
|
|
328
|
+
const dbPath = path.join(dbDir, "bonecode.db");
|
|
329
|
+
_mode = "sqlite";
|
|
330
|
+
log(`\x1b[33m[BoneCode] Using SQLite (${dbPath})\x1b[0m`);
|
|
331
|
+
log(`\x1b[2m For full RAG support, install Docker: https://docs.docker.com/get-docker/\x1b[0m`);
|
|
332
|
+
const sqlitePool = createSQLitePool(dbPath);
|
|
333
|
+
// Run schema
|
|
334
|
+
const Database = require("better-sqlite3");
|
|
335
|
+
const db = new Database(dbPath);
|
|
336
|
+
db.exec(SQLITE_SCHEMA);
|
|
337
|
+
db.close();
|
|
338
|
+
return { pool: sqlitePool, mode: "sqlite", pgAvailable: false };
|
|
339
|
+
}
|
|
340
|
+
exports.initDatabase = initDatabase;
|
|
341
|
+
//# sourceMappingURL=db_adapter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"db_adapter.js","sourceRoot":"","sources":["../../src/db_adapter.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;GAgBG;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,uCAAyB;AACzB,2CAA6B;AAC7B,uCAAyB;AAIzB,IAAI,KAAK,GAAkB,IAAI,CAAC;AAEhC,SAAgB,SAAS;IACvB,OAAO,KAAK,IAAI,QAAQ,CAAC;AAC3B,CAAC;AAFD,8BAEC;AAED,iFAAiF;AACjF,0DAA0D;AAE1D,SAAS,gBAAgB,CAAC,MAAc;IACtC,MAAM,QAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC3C,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;IAEhC,yDAAyD;IACzD,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;IAChC,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAE/B,2CAA2C;IAC3C,SAAS,YAAY,CAAC,GAAW;QAC/B,OAAO,GAAG;YACR,0BAA0B;aACzB,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC;YACvB,6CAA6C;YAC7C,0BAA0B;aACzB,OAAO,CAAC,aAAa,EAAE,iBAAiB,CAAC;YAC1C,iEAAiE;aAChE,OAAO,CAAC,aAAa,EAAE,MAAM,CAAC;YAC/B,2CAA2C;aAC1C,OAAO,CAAC,qBAAqB,EAAE,EAAE,CAAC;YACnC,2DAA2D;aAC1D,OAAO,CAAC,uBAAuB,EAAE,oNAAoN,CAAC;YACvP,yCAAyC;aACxC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC;YAC/B,kEAAkE;aACjE,OAAO,CAAC,0CAA0C,EAAE,MAAM,CAAC;YAC5D,mFAAmF;aAClF,OAAO,CAAC,wBAAwB,EAAE,IAAI,CAAC;aACvC,OAAO,CAAC,uBAAuB,EAAE,IAAI,CAAC;aACtC,OAAO,CAAC,oBAAoB,EAAE,GAAG,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,IAAI,GAAG;QACX,KAAK,EAAE,KAAK,EAAE,GAAW,EAAE,MAAc,EAA8C,EAAE;YACvF,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;gBACrC,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAC/C,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAC5D,CAAC;gBAEF,sDAAsD;gBACtD,MAAM,YAAY,GAAG,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAChD,MAAM,cAAc,GAAG,GAAG,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;gBAEvD,0CAA0C;gBAC1C,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;gBAE9D,MAAM,QAAQ,GAAG,2BAA2B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAE3D,IAAI,QAAQ,EAAE,CAAC;oBACb,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBACjC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;oBACrC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACzC,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBACjC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;oBAErC,kEAAkE;oBAClE,IAAI,YAAY,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;wBACzC,yCAAyC;wBACzC,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;wBAChE,IAAI,UAAU,EAAE,CAAC;4BACf,IAAI,CAAC;gCACH,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,iBAAiB,UAAU,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;gCACnG,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC;4BAC5D,CAAC;4BAAC,MAAM,CAAC,CAAC,gCAAgC,CAAC,CAAC;wBAC9C,CAAC;oBACH,CAAC;oBAED,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC9C,CAAC;YACH,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,sEAAsE;gBACtE,IACE,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,kBAAkB,CAAC;oBACvC,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,gBAAgB,CAAC;oBACrC,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,cAAc,CAAC,EACnC,CAAC;oBACD,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;gBACnC,CAAC;gBACD,MAAM,CAAC,CAAC;YACV,CAAC;QACH,CAAC;QAED,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,+CAA+C;YAC/C,OAAO;gBACL,KAAK,EAAE,KAAK,EAAE,GAAW,EAAE,MAAc,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC;gBACrE,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC;aAClB,CAAC;QACJ,CAAC;QAED,GAAG,EAAE,KAAK,IAAI,EAAE;YACd,EAAE,CAAC,KAAK,EAAE,CAAC;QACb,CAAC;QAED,EAAE,EAAE,CAAC,MAAc,EAAE,QAAa,EAAE,EAAE,GAAE,CAAC;KAC1C,CAAC;IAEF,OAAO,IAAI,CAAC;AACd,CAAC;AAED,iFAAiF;AACjF,0EAA0E;AAC1E,+BAA+B;AAE/B,MAAM,aAAa,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA+FrB,CAAC;AAEF,iFAAiF;AAEjF,KAAK,UAAU,aAAa,CAAC,GAAW;IACtC,IAAI,CAAC;QACH,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAC/B,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,EAAE,gBAAgB,EAAE,GAAG,EAAE,uBAAuB,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QAC5F,MAAM,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACjC,MAAM,QAAQ,CAAC,GAAG,EAAE,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,gFAAgF;AAEhF,KAAK,UAAU,sBAAsB;IACnC,IAAI,CAAC;QACH,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;QAC9C,+BAA+B;QAC/B,QAAQ,CAAC,kBAAkB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAElD,oCAAoC;QACpC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,QAAQ,CACrB,8DAA8D,EAC9D,EAAE,KAAK,EAAE,MAAM,EAAE,CAClB,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;YACpB,IAAI,MAAM,KAAK,MAAM;gBAAE,OAAO,IAAI,CAAC,CAAC,kBAAkB;YACtD,gCAAgC;YAChC,QAAQ,CAAC,gCAAgC,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAChE,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YAC5C,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,sCAAsC;YACtC,QAAQ,CACN,4JAA4J,EAC5J,EAAE,KAAK,EAAE,QAAQ,EAAE,CACpB,CAAC;YACF,gCAAgC;YAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC5C,IAAI,MAAM,aAAa,CAAC,wDAAwD,CAAC,EAAE,CAAC;oBAClF,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC,CAAC,uBAAuB;IACvC,CAAC;AACH,CAAC;AAED,iFAAiF;AAE1E,KAAK,UAAU,YAAY,CAAC,MAAM,GAAG,KAAK;IAK/C,MAAM,GAAG,GAAG,CAAC,GAAW,EAAE,EAAE,GAAG,IAAI,CAAC,MAAM;QAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAEhF,iCAAiC;IACjC,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IACvC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzC,IAAI,MAAM,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,KAAK,GAAG,UAAU,CAAC;YACnB,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,EAAE,gBAAgB,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9D,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;gBAChC,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;oBACpC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,oBAAoB,GAAG,CAAC,OAAO,IAAI,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QAC/D,CAAC;QACD,GAAG,CAAC,gDAAgD,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,4BAA4B,CAAC,CAAC;IACtH,CAAC;IAED,8BAA8B;IAC9B,MAAM,aAAa,GAAG,MAAM,sBAAsB,EAAE,CAAC;IACrD,IAAI,aAAa,EAAE,CAAC;QAClB,2DAA2D;QAC3D,MAAM,SAAS,GAAG,wDAAwD,CAAC;QAC3E,IAAI,MAAM,aAAa,CAAC,SAAS,CAAC,EAAE,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC;YACrC,KAAK,GAAG,UAAU,CAAC;YACnB,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,EAAE,gBAAgB,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;YAClE,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;gBAChC,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;oBACpC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,oBAAoB,GAAG,CAAC,OAAO,IAAI,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC,CAAC,CAAC;YACH,GAAG,CAAC,wEAAwE,CAAC,CAAC;YAC9E,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,WAAW,CAAC,CAAC;IACnD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IAC/C,KAAK,GAAG,QAAQ,CAAC;IAEjB,GAAG,CAAC,oCAAoC,MAAM,UAAU,CAAC,CAAC;IAC1D,GAAG,CAAC,2FAA2F,CAAC,CAAC;IAEjG,MAAM,UAAU,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAE5C,aAAa;IACb,MAAM,QAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC3C,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;IAChC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACvB,EAAE,CAAC,KAAK,EAAE,CAAC;IAEX,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;AAClE,CAAC;AA7DD,oCA6DC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Context compaction — ported from OpenCode's session/compaction.ts
|
|
3
|
+
* Same SUMMARY_TEMPLATE and algorithm, no Effect.js.
|
|
4
|
+
*/
|
|
5
|
+
export interface CompactionInput {
|
|
6
|
+
messages: any[];
|
|
7
|
+
model_id: string;
|
|
8
|
+
provider_id: string;
|
|
9
|
+
session_id: string;
|
|
10
|
+
}
|
|
11
|
+
export declare function buildCompactionSummary(input: CompactionInput): Promise<string | null>;
|
|
@@ -0,0 +1,113 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Context compaction — ported from OpenCode's session/compaction.ts
|
|
4
|
+
* Same SUMMARY_TEMPLATE and algorithm, no Effect.js.
|
|
5
|
+
*/
|
|
6
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
7
|
+
exports.buildCompactionSummary = void 0;
|
|
8
|
+
const openai_1 = require("@ai-sdk/openai");
|
|
9
|
+
const anthropic_1 = require("@ai-sdk/anthropic");
|
|
10
|
+
const ai_1 = require("ai");
|
|
11
|
+
// Verbatim from OpenCode — this template is battle-tested
|
|
12
|
+
const SUMMARY_TEMPLATE = `Output exactly the Markdown structure shown inside <template> and keep the section order unchanged. Do not include the <template> tags in your response.
|
|
13
|
+
<template>
|
|
14
|
+
## Goal
|
|
15
|
+
- [single-sentence task summary]
|
|
16
|
+
|
|
17
|
+
## Constraints & Preferences
|
|
18
|
+
- [user constraints, preferences, specs, or "(none)"]
|
|
19
|
+
|
|
20
|
+
## Progress
|
|
21
|
+
### Done
|
|
22
|
+
- [completed work or "(none)"]
|
|
23
|
+
|
|
24
|
+
### In Progress
|
|
25
|
+
- [current work or "(none)"]
|
|
26
|
+
|
|
27
|
+
### Blocked
|
|
28
|
+
- [blockers or "(none)"]
|
|
29
|
+
|
|
30
|
+
## Key Decisions
|
|
31
|
+
- [decision and why, or "(none)"]
|
|
32
|
+
|
|
33
|
+
## Next Steps
|
|
34
|
+
- [ordered next actions or "(none)"]
|
|
35
|
+
|
|
36
|
+
## Critical Context
|
|
37
|
+
- [important technical facts, errors, open questions, or "(none)"]
|
|
38
|
+
|
|
39
|
+
## Relevant Files
|
|
40
|
+
- [file or directory path: why it matters, or "(none)"]
|
|
41
|
+
</template>
|
|
42
|
+
|
|
43
|
+
Rules:
|
|
44
|
+
- Keep every section, even when empty.
|
|
45
|
+
- Use terse bullets, not prose paragraphs.
|
|
46
|
+
- Preserve exact file paths, commands, error strings, and identifiers when known.
|
|
47
|
+
- Do not mention the summary process or that context was compacted.`;
|
|
48
|
+
async function buildCompactionSummary(input) {
|
|
49
|
+
const { messages, model_id, provider_id } = input;
|
|
50
|
+
// Build conversation text for summarization
|
|
51
|
+
// Strip tool outputs > 2000 chars (same as OpenCode's TOOL_OUTPUT_MAX_CHARS)
|
|
52
|
+
const TOOL_OUTPUT_MAX = 2000;
|
|
53
|
+
const conversationLines = [];
|
|
54
|
+
for (const msg of messages) {
|
|
55
|
+
const parts = msg.parts || [];
|
|
56
|
+
if (msg.role === "user") {
|
|
57
|
+
const text = parts.filter((p) => p.part_type === "text").map((p) => p.data?.text || "").join("\n");
|
|
58
|
+
if (text.trim())
|
|
59
|
+
conversationLines.push(`User: ${text.trim()}`);
|
|
60
|
+
}
|
|
61
|
+
else if (msg.role === "assistant") {
|
|
62
|
+
for (const p of parts) {
|
|
63
|
+
if (p.part_type === "text" && p.data?.text && !p.data?.is_compaction_summary) {
|
|
64
|
+
conversationLines.push(`Assistant: ${p.data.text.trim()}`);
|
|
65
|
+
}
|
|
66
|
+
else if (p.part_type === "tool_invocation") {
|
|
67
|
+
conversationLines.push(`Tool call: ${p.data?.tool_name}(${JSON.stringify(p.data?.args || {}).slice(0, 200)})`);
|
|
68
|
+
}
|
|
69
|
+
else if (p.part_type === "tool_result") {
|
|
70
|
+
const output = String(p.data?.result || "").slice(0, TOOL_OUTPUT_MAX);
|
|
71
|
+
conversationLines.push(`Tool result: ${output}`);
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
if (conversationLines.length === 0)
|
|
77
|
+
return null;
|
|
78
|
+
const prompt = [
|
|
79
|
+
"Create a new anchored summary from the conversation history above.",
|
|
80
|
+
SUMMARY_TEMPLATE,
|
|
81
|
+
"",
|
|
82
|
+
"<conversation>",
|
|
83
|
+
conversationLines.join("\n"),
|
|
84
|
+
"</conversation>",
|
|
85
|
+
].join("\n\n");
|
|
86
|
+
try {
|
|
87
|
+
const model = getCompactionModel(provider_id, model_id);
|
|
88
|
+
const { text } = await (0, ai_1.generateText)({
|
|
89
|
+
model,
|
|
90
|
+
messages: [{ role: "user", content: prompt }],
|
|
91
|
+
maxTokens: 4096,
|
|
92
|
+
temperature: 0,
|
|
93
|
+
});
|
|
94
|
+
return text.replace(/<think>[\s\S]*?<\/think>\s*/g, "").trim();
|
|
95
|
+
}
|
|
96
|
+
catch (e) {
|
|
97
|
+
return null;
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
exports.buildCompactionSummary = buildCompactionSummary;
|
|
101
|
+
function getCompactionModel(provider_id, model_id) {
|
|
102
|
+
const apiKey = process.env[`${provider_id.toUpperCase()}_API_KEY`] || process.env.OPENAI_API_KEY || "not-needed";
|
|
103
|
+
const baseUrl = process.env[`${provider_id.toUpperCase()}_BASE_URL`] || process.env.OPENAI_BASE_URL;
|
|
104
|
+
// Use a smaller/faster model for compaction if available
|
|
105
|
+
const compactionModel = process.env.COMPACTION_MODEL || model_id;
|
|
106
|
+
switch (provider_id.toLowerCase()) {
|
|
107
|
+
case "anthropic":
|
|
108
|
+
return (0, anthropic_1.createAnthropic)({ apiKey, baseURL: baseUrl })(compactionModel);
|
|
109
|
+
default:
|
|
110
|
+
return (0, openai_1.createOpenAI)({ apiKey, baseURL: baseUrl })(compactionModel);
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
//# sourceMappingURL=compaction_logic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"compaction_logic.js","sourceRoot":"","sources":["../../../../src/engine/session/compaction_logic.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,2CAA8C;AAC9C,iDAAoD;AACpD,2BAAkC;AAElC,0DAA0D;AAC1D,MAAM,gBAAgB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;oEAmC2C,CAAC;AAS9D,KAAK,UAAU,sBAAsB,CAAC,KAAsB;IACjE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAElD,4CAA4C;IAC5C,6EAA6E;IAC7E,MAAM,eAAe,GAAG,IAAK,CAAC;IAC9B,MAAM,iBAAiB,GAAa,EAAE,CAAC;IAEvC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;QAC9B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACxB,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7G,IAAI,IAAI,CAAC,IAAI,EAAE;gBAAE,iBAAiB,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAClE,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACpC,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;gBACtB,IAAI,CAAC,CAAC,SAAS,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,qBAAqB,EAAE,CAAC;oBAC7E,iBAAiB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC7D,CAAC;qBAAM,IAAI,CAAC,CAAC,SAAS,KAAK,iBAAiB,EAAE,CAAC;oBAC7C,iBAAiB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,IAAI,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;gBACjH,CAAC;qBAAM,IAAI,CAAC,CAAC,SAAS,KAAK,aAAa,EAAE,CAAC;oBACzC,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;oBACtE,iBAAiB,CAAC,IAAI,CAAC,gBAAgB,MAAM,EAAE,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAEhD,MAAM,MAAM,GAAG;QACb,oEAAoE;QACpE,gBAAgB;QAChB,EAAE;QACF,gBAAgB;QAChB,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC;QAC5B,iBAAiB;KAClB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAEf,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,kBAAkB,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QACxD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAA,iBAAY,EAAC;YAClC,KAAK;YACL,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAC7C,SAAS,EAAE,IAAI;YACf,WAAW,EAAE,CAAC;SACf,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,OAAO,CAAC,8BAA8B,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACjE,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAlDD,wDAkDC;AAED,SAAS,kBAAkB,CAAC,WAAmB,EAAE,QAAgB;IAC/D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,YAAY,CAAC;IACjH,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC,WAAW,EAAE,WAAW,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAEpG,yDAAyD;IACzD,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,QAAQ,CAAC;IAEjE,QAAQ,WAAW,CAAC,WAAW,EAAE,EAAE,CAAC;QAClC,KAAK,WAAW;YACd,OAAO,IAAA,2BAAe,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC;QACxE;YACE,OAAO,IAAA,qBAAY,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC;IACvE,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Instruction file loader — ported from OpenCode's session/instruction.ts
|
|
4
|
+
* Scans for AGENTS.md, CLAUDE.md, OPENCODE.md walking up from the working directory.
|
|
5
|
+
*/
|
|
6
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
7
|
+
if (k2 === undefined) k2 = k;
|
|
8
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
9
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
10
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
11
|
+
}
|
|
12
|
+
Object.defineProperty(o, k2, desc);
|
|
13
|
+
}) : (function(o, m, k, k2) {
|
|
14
|
+
if (k2 === undefined) k2 = k;
|
|
15
|
+
o[k2] = m[k];
|
|
16
|
+
}));
|
|
17
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
18
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
19
|
+
}) : function(o, v) {
|
|
20
|
+
o["default"] = v;
|
|
21
|
+
});
|
|
22
|
+
var __importStar = (this && this.__importStar) || function (mod) {
|
|
23
|
+
if (mod && mod.__esModule) return mod;
|
|
24
|
+
var result = {};
|
|
25
|
+
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
|
|
26
|
+
__setModuleDefault(result, mod);
|
|
27
|
+
return result;
|
|
28
|
+
};
|
|
29
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
30
|
+
exports.loadInstructionFiles = void 0;
|
|
31
|
+
const fs = __importStar(require("fs/promises"));
|
|
32
|
+
const path = __importStar(require("path"));
|
|
33
|
+
const INSTRUCTION_FILES = ["AGENTS.md", "CLAUDE.md", "OPENCODE.md", ".opencode/instructions.md"];
|
|
34
|
+
async function loadInstructionFiles(worktree) {
|
|
35
|
+
const found = [];
|
|
36
|
+
// Walk up from worktree looking for instruction files
|
|
37
|
+
let dir = worktree;
|
|
38
|
+
const visited = new Set();
|
|
39
|
+
while (dir && !visited.has(dir)) {
|
|
40
|
+
visited.add(dir);
|
|
41
|
+
for (const filename of INSTRUCTION_FILES) {
|
|
42
|
+
const filePath = path.join(dir, filename);
|
|
43
|
+
try {
|
|
44
|
+
const content = await fs.readFile(filePath, "utf-8");
|
|
45
|
+
if (content.trim()) {
|
|
46
|
+
found.push(`<!-- Instructions from ${filePath} -->\n${content.trim()}`);
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
catch {
|
|
50
|
+
// File doesn't exist — skip
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
const parent = path.dirname(dir);
|
|
54
|
+
if (parent === dir)
|
|
55
|
+
break;
|
|
56
|
+
dir = parent;
|
|
57
|
+
}
|
|
58
|
+
// Also check global config dirs
|
|
59
|
+
const globalDirs = [
|
|
60
|
+
path.join(process.env.HOME || process.env.USERPROFILE || "", ".config", "opencode"),
|
|
61
|
+
path.join(process.env.HOME || process.env.USERPROFILE || "", ".opencode"),
|
|
62
|
+
];
|
|
63
|
+
for (const globalDir of globalDirs) {
|
|
64
|
+
for (const filename of INSTRUCTION_FILES) {
|
|
65
|
+
const filePath = path.join(globalDir, filename);
|
|
66
|
+
try {
|
|
67
|
+
const content = await fs.readFile(filePath, "utf-8");
|
|
68
|
+
if (content.trim()) {
|
|
69
|
+
found.push(`<!-- Global instructions from ${filePath} -->\n${content.trim()}`);
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
catch { }
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
return found.join("\n\n");
|
|
76
|
+
}
|
|
77
|
+
exports.loadInstructionFiles = loadInstructionFiles;
|
|
78
|
+
//# sourceMappingURL=instruction_loader.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"instruction_loader.js","sourceRoot":"","sources":["../../../../src/engine/session/instruction_loader.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,gDAAkC;AAClC,2CAA6B;AAE7B,MAAM,iBAAiB,GAAG,CAAC,WAAW,EAAE,WAAW,EAAE,aAAa,EAAE,2BAA2B,CAAC,CAAC;AAE1F,KAAK,UAAU,oBAAoB,CAAC,QAAgB;IACzD,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,sDAAsD;IACtD,IAAI,GAAG,GAAG,QAAQ,CAAC;IACnB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAElC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACjB,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;YACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAC1C,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACrD,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;oBACnB,KAAK,CAAC,IAAI,CAAC,0BAA0B,QAAQ,SAAS,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC1E,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,4BAA4B;YAC9B,CAAC;QACH,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,MAAM,KAAK,GAAG;YAAE,MAAM;QAC1B,GAAG,GAAG,MAAM,CAAC;IACf,CAAC;IAED,gCAAgC;IAChC,MAAM,UAAU,GAAG;QACjB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE,SAAS,EAAE,UAAU,CAAC;QACnF,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE,WAAW,CAAC;KAC1E,CAAC;IACF,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;YACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAChD,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACrD,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;oBACnB,KAAK,CAAC,IAAI,CAAC,iCAAiC,QAAQ,SAAS,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACjF,CAAC;YACH,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC5B,CAAC;AA3CD,oDA2CC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Context overflow detection — ported from OpenCode's session/overflow.ts
|
|
3
|
+
* Uses model context limits from env config since we don't have the full
|
|
4
|
+
* Provider service. Falls back to conservative 80% threshold.
|
|
5
|
+
*/
|
|
6
|
+
export declare function isOverflow(tokens: {
|
|
7
|
+
input?: number;
|
|
8
|
+
output?: number;
|
|
9
|
+
total?: number;
|
|
10
|
+
cache?: {
|
|
11
|
+
read?: number;
|
|
12
|
+
write?: number;
|
|
13
|
+
};
|
|
14
|
+
}, model_id: string): boolean;
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Context overflow detection — ported from OpenCode's session/overflow.ts
|
|
4
|
+
* Uses model context limits from env config since we don't have the full
|
|
5
|
+
* Provider service. Falls back to conservative 80% threshold.
|
|
6
|
+
*/
|
|
7
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
8
|
+
exports.isOverflow = void 0;
|
|
9
|
+
const COMPACTION_BUFFER = 20000;
|
|
10
|
+
// Model context window limits (tokens). Extend as needed.
|
|
11
|
+
const MODEL_CONTEXT_LIMITS = {
|
|
12
|
+
"claude-opus-4-5": 200000,
|
|
13
|
+
"claude-sonnet-4-5": 200000,
|
|
14
|
+
"claude-haiku-3-5": 200000,
|
|
15
|
+
"gpt-4o": 128000,
|
|
16
|
+
"gpt-4o-mini": 128000,
|
|
17
|
+
"o1": 200000,
|
|
18
|
+
"o3-mini": 200000,
|
|
19
|
+
"gemini-2.0-flash": 1000000,
|
|
20
|
+
"gemini-1.5-pro": 2000000,
|
|
21
|
+
};
|
|
22
|
+
function getContextLimit(model_id) {
|
|
23
|
+
// Exact match
|
|
24
|
+
if (MODEL_CONTEXT_LIMITS[model_id])
|
|
25
|
+
return MODEL_CONTEXT_LIMITS[model_id];
|
|
26
|
+
// Prefix match
|
|
27
|
+
for (const [key, limit] of Object.entries(MODEL_CONTEXT_LIMITS)) {
|
|
28
|
+
if (model_id.includes(key) || key.includes(model_id))
|
|
29
|
+
return limit;
|
|
30
|
+
}
|
|
31
|
+
// Default: 128k
|
|
32
|
+
return 128000;
|
|
33
|
+
}
|
|
34
|
+
function isOverflow(tokens, model_id) {
|
|
35
|
+
const limit = getContextLimit(model_id);
|
|
36
|
+
if (limit === 0)
|
|
37
|
+
return false;
|
|
38
|
+
const count = tokens.total ||
|
|
39
|
+
(tokens.input || 0) + (tokens.output || 0) +
|
|
40
|
+
(tokens.cache?.read || 0) + (tokens.cache?.write || 0);
|
|
41
|
+
const usable = Math.max(0, limit - COMPACTION_BUFFER - 32000); // reserve output tokens
|
|
42
|
+
return count >= usable;
|
|
43
|
+
}
|
|
44
|
+
exports.isOverflow = isOverflow;
|
|
45
|
+
//# sourceMappingURL=overflow_check.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"overflow_check.js","sourceRoot":"","sources":["../../../../src/engine/session/overflow_check.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAEH,MAAM,iBAAiB,GAAG,KAAM,CAAC;AAEjC,0DAA0D;AAC1D,MAAM,oBAAoB,GAA2B;IACnD,iBAAiB,EAAE,MAAO;IAC1B,mBAAmB,EAAE,MAAO;IAC5B,kBAAkB,EAAE,MAAO;IAC3B,QAAQ,EAAE,MAAO;IACjB,aAAa,EAAE,MAAO;IACtB,IAAI,EAAE,MAAO;IACb,SAAS,EAAE,MAAO;IAClB,kBAAkB,EAAE,OAAS;IAC7B,gBAAgB,EAAE,OAAS;CAC5B,CAAC;AAEF,SAAS,eAAe,CAAC,QAAgB;IACvC,cAAc;IACd,IAAI,oBAAoB,CAAC,QAAQ,CAAC;QAAE,OAAO,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IAC1E,eAAe;IACf,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAChE,IAAI,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,KAAK,CAAC;IACrE,CAAC;IACD,gBAAgB;IAChB,OAAO,MAAO,CAAC;AACjB,CAAC;AAED,SAAgB,UAAU,CACxB,MAAsG,EACtG,QAAgB;IAEhB,MAAM,KAAK,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;IACxC,IAAI,KAAK,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAE9B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK;QACxB,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC;YAC1C,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;IAEzD,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,iBAAiB,GAAG,KAAM,CAAC,CAAC,CAAC,wBAAwB;IACxF,OAAO,KAAK,IAAI,MAAM,CAAC;AACzB,CAAC;AAbD,gCAaC"}
|