@goondocks/myco 0.1.0
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/.claude-plugin/marketplace.json +22 -0
- package/.claude-plugin/plugin.json +18 -0
- package/CONTRIBUTING.md +143 -0
- package/LICENSE +21 -0
- package/README.md +74 -0
- package/commands/init.md +231 -0
- package/commands/setup-llm.md +89 -0
- package/commands/status.md +112 -0
- package/dist/src/agents/adapter.d.ts +76 -0
- package/dist/src/agents/adapter.d.ts.map +1 -0
- package/dist/src/agents/adapter.js +124 -0
- package/dist/src/agents/adapter.js.map +1 -0
- package/dist/src/agents/claude-code.d.ts +3 -0
- package/dist/src/agents/claude-code.d.ts.map +1 -0
- package/dist/src/agents/claude-code.js +22 -0
- package/dist/src/agents/claude-code.js.map +1 -0
- package/dist/src/agents/cursor.d.ts +3 -0
- package/dist/src/agents/cursor.d.ts.map +1 -0
- package/dist/src/agents/cursor.js +154 -0
- package/dist/src/agents/cursor.js.map +1 -0
- package/dist/src/agents/index.d.ts +6 -0
- package/dist/src/agents/index.d.ts.map +1 -0
- package/dist/src/agents/index.js +5 -0
- package/dist/src/agents/index.js.map +1 -0
- package/dist/src/agents/registry.d.ts +34 -0
- package/dist/src/agents/registry.d.ts.map +1 -0
- package/dist/src/agents/registry.js +95 -0
- package/dist/src/agents/registry.js.map +1 -0
- package/dist/src/artifacts/candidates.d.ts +20 -0
- package/dist/src/artifacts/candidates.d.ts.map +1 -0
- package/dist/src/artifacts/candidates.js +84 -0
- package/dist/src/artifacts/candidates.js.map +1 -0
- package/dist/src/artifacts/slugify.d.ts +2 -0
- package/dist/src/artifacts/slugify.d.ts.map +1 -0
- package/dist/src/artifacts/slugify.js +22 -0
- package/dist/src/artifacts/slugify.js.map +1 -0
- package/dist/src/capture/artifact-watcher.d.ts +19 -0
- package/dist/src/capture/artifact-watcher.d.ts.map +1 -0
- package/dist/src/capture/artifact-watcher.js +37 -0
- package/dist/src/capture/artifact-watcher.js.map +1 -0
- package/dist/src/capture/buffer.d.ts +20 -0
- package/dist/src/capture/buffer.d.ts.map +1 -0
- package/dist/src/capture/buffer.js +55 -0
- package/dist/src/capture/buffer.js.map +1 -0
- package/dist/src/capture/plan-detector.d.ts +15 -0
- package/dist/src/capture/plan-detector.d.ts.map +1 -0
- package/dist/src/capture/plan-detector.js +34 -0
- package/dist/src/capture/plan-detector.js.map +1 -0
- package/dist/src/capture/processor.d.ts +2 -0
- package/dist/src/capture/processor.d.ts.map +1 -0
- package/dist/src/capture/processor.js +3 -0
- package/dist/src/capture/processor.js.map +1 -0
- package/dist/src/capture/prompts/classify.md +28 -0
- package/dist/src/capture/prompts/debugging.md +91 -0
- package/dist/src/capture/prompts/exploration.md +88 -0
- package/dist/src/capture/prompts/extraction.md +93 -0
- package/dist/src/capture/prompts/implementation.md +90 -0
- package/dist/src/capture/prompts/prompts/classify.md +28 -0
- package/dist/src/capture/prompts/prompts/debugging.md +91 -0
- package/dist/src/capture/prompts/prompts/exploration.md +88 -0
- package/dist/src/capture/prompts/prompts/extraction.md +93 -0
- package/dist/src/capture/prompts/prompts/implementation.md +90 -0
- package/dist/src/capture/prompts/prompts/schema.yaml +97 -0
- package/dist/src/capture/prompts/prompts/session-summary.md +65 -0
- package/dist/src/capture/prompts/prompts/session-title.md +46 -0
- package/dist/src/capture/prompts/schema.yaml +97 -0
- package/dist/src/capture/prompts/session-summary.md +65 -0
- package/dist/src/capture/prompts/session-title.md +46 -0
- package/dist/src/capture/prompts.d.ts +77 -0
- package/dist/src/capture/prompts.d.ts.map +1 -0
- package/dist/src/capture/prompts.js +255 -0
- package/dist/src/capture/prompts.js.map +1 -0
- package/dist/src/capture/transcript-miner.d.ts +31 -0
- package/dist/src/capture/transcript-miner.d.ts.map +1 -0
- package/dist/src/capture/transcript-miner.js +61 -0
- package/dist/src/capture/transcript-miner.js.map +1 -0
- package/dist/src/cli.d.ts +3 -0
- package/dist/src/cli.d.ts.map +1 -0
- package/dist/src/cli.js +584 -0
- package/dist/src/cli.js.map +1 -0
- package/dist/src/config/loader.d.ts +4 -0
- package/dist/src/config/loader.d.ts.map +1 -0
- package/dist/src/config/loader.js +32 -0
- package/dist/src/config/loader.js.map +1 -0
- package/dist/src/config/schema.d.ts +83 -0
- package/dist/src/config/schema.d.ts.map +1 -0
- package/dist/src/config/schema.js +55 -0
- package/dist/src/config/schema.js.map +1 -0
- package/dist/src/constants.d.ts +73 -0
- package/dist/src/constants.d.ts.map +1 -0
- package/dist/src/constants.js +86 -0
- package/dist/src/constants.js.map +1 -0
- package/dist/src/context/injector.d.ts +18 -0
- package/dist/src/context/injector.d.ts.map +1 -0
- package/dist/src/context/injector.js +71 -0
- package/dist/src/context/injector.js.map +1 -0
- package/dist/src/context/relevance.d.ts +13 -0
- package/dist/src/context/relevance.d.ts.map +1 -0
- package/dist/src/context/relevance.js +44 -0
- package/dist/src/context/relevance.js.map +1 -0
- package/dist/src/daemon/batch.d.ts +22 -0
- package/dist/src/daemon/batch.d.ts.map +1 -0
- package/dist/src/daemon/batch.js +38 -0
- package/dist/src/daemon/batch.js.map +1 -0
- package/dist/src/daemon/lifecycle.d.ts +27 -0
- package/dist/src/daemon/lifecycle.d.ts.map +1 -0
- package/dist/src/daemon/lifecycle.js +50 -0
- package/dist/src/daemon/lifecycle.js.map +1 -0
- package/dist/src/daemon/lineage.d.ts +42 -0
- package/dist/src/daemon/lineage.d.ts.map +1 -0
- package/dist/src/daemon/lineage.js +116 -0
- package/dist/src/daemon/lineage.js.map +1 -0
- package/dist/src/daemon/logger.d.ts +33 -0
- package/dist/src/daemon/logger.d.ts.map +1 -0
- package/dist/src/daemon/logger.js +88 -0
- package/dist/src/daemon/logger.js.map +1 -0
- package/dist/src/daemon/main.d.ts +2 -0
- package/dist/src/daemon/main.d.ts.map +1 -0
- package/dist/src/daemon/main.js +738 -0
- package/dist/src/daemon/main.js.map +1 -0
- package/dist/src/daemon/processor.d.ts +44 -0
- package/dist/src/daemon/processor.d.ts.map +1 -0
- package/dist/src/daemon/processor.js +142 -0
- package/dist/src/daemon/processor.js.map +1 -0
- package/dist/src/daemon/server.d.ts +24 -0
- package/dist/src/daemon/server.d.ts.map +1 -0
- package/dist/src/daemon/server.js +117 -0
- package/dist/src/daemon/server.js.map +1 -0
- package/dist/src/daemon/watcher.d.ts +29 -0
- package/dist/src/daemon/watcher.d.ts.map +1 -0
- package/dist/src/daemon/watcher.js +67 -0
- package/dist/src/daemon/watcher.js.map +1 -0
- package/dist/src/hooks/client.d.ts +20 -0
- package/dist/src/hooks/client.d.ts.map +1 -0
- package/dist/src/hooks/client.js +111 -0
- package/dist/src/hooks/client.js.map +1 -0
- package/dist/src/hooks/post-tool-use.d.ts +2 -0
- package/dist/src/hooks/post-tool-use.d.ts.map +1 -0
- package/dist/src/hooks/post-tool-use.js +40 -0
- package/dist/src/hooks/post-tool-use.js.map +1 -0
- package/dist/src/hooks/read-stdin.d.ts +2 -0
- package/dist/src/hooks/read-stdin.d.ts.map +1 -0
- package/dist/src/hooks/read-stdin.js +10 -0
- package/dist/src/hooks/read-stdin.js.map +1 -0
- package/dist/src/hooks/session-end.d.ts +2 -0
- package/dist/src/hooks/session-end.d.ts.map +1 -0
- package/dist/src/hooks/session-end.js +23 -0
- package/dist/src/hooks/session-end.js.map +1 -0
- package/dist/src/hooks/session-start.d.ts +2 -0
- package/dist/src/hooks/session-start.d.ts.map +1 -0
- package/dist/src/hooks/session-start.js +49 -0
- package/dist/src/hooks/session-start.js.map +1 -0
- package/dist/src/hooks/stop.d.ts +2 -0
- package/dist/src/hooks/stop.d.ts.map +1 -0
- package/dist/src/hooks/stop.js +34 -0
- package/dist/src/hooks/stop.js.map +1 -0
- package/dist/src/hooks/user-prompt-submit.d.ts +2 -0
- package/dist/src/hooks/user-prompt-submit.d.ts.map +1 -0
- package/dist/src/hooks/user-prompt-submit.js +46 -0
- package/dist/src/hooks/user-prompt-submit.js.map +1 -0
- package/dist/src/index/fts.d.ts +16 -0
- package/dist/src/index/fts.d.ts.map +1 -0
- package/dist/src/index/fts.js +53 -0
- package/dist/src/index/fts.js.map +1 -0
- package/dist/src/index/rebuild.d.ts +4 -0
- package/dist/src/index/rebuild.d.ts.map +1 -0
- package/dist/src/index/rebuild.js +40 -0
- package/dist/src/index/rebuild.js.map +1 -0
- package/dist/src/index/sqlite.d.ts +33 -0
- package/dist/src/index/sqlite.d.ts.map +1 -0
- package/dist/src/index/sqlite.js +99 -0
- package/dist/src/index/sqlite.js.map +1 -0
- package/dist/src/index/vectors.d.ts +24 -0
- package/dist/src/index/vectors.d.ts.map +1 -0
- package/dist/src/index/vectors.js +97 -0
- package/dist/src/index/vectors.js.map +1 -0
- package/dist/src/intelligence/anthropic.d.ts +17 -0
- package/dist/src/intelligence/anthropic.d.ts.map +1 -0
- package/dist/src/intelligence/anthropic.js +36 -0
- package/dist/src/intelligence/anthropic.js.map +1 -0
- package/dist/src/intelligence/embeddings.d.ts +3 -0
- package/dist/src/intelligence/embeddings.d.ts.map +1 -0
- package/dist/src/intelligence/embeddings.js +15 -0
- package/dist/src/intelligence/embeddings.js.map +1 -0
- package/dist/src/intelligence/haiku.d.ts +17 -0
- package/dist/src/intelligence/haiku.d.ts.map +1 -0
- package/dist/src/intelligence/haiku.js +35 -0
- package/dist/src/intelligence/haiku.js.map +1 -0
- package/dist/src/intelligence/llm.d.ts +33 -0
- package/dist/src/intelligence/llm.d.ts.map +1 -0
- package/dist/src/intelligence/llm.js +26 -0
- package/dist/src/intelligence/llm.js.map +1 -0
- package/dist/src/intelligence/lm-studio.d.ts +20 -0
- package/dist/src/intelligence/lm-studio.d.ts.map +1 -0
- package/dist/src/intelligence/lm-studio.js +59 -0
- package/dist/src/intelligence/lm-studio.js.map +1 -0
- package/dist/src/intelligence/ollama.d.ts +22 -0
- package/dist/src/intelligence/ollama.d.ts.map +1 -0
- package/dist/src/intelligence/ollama.js +64 -0
- package/dist/src/intelligence/ollama.js.map +1 -0
- package/dist/src/intelligence/response.d.ts +29 -0
- package/dist/src/intelligence/response.d.ts.map +1 -0
- package/dist/src/intelligence/response.js +71 -0
- package/dist/src/intelligence/response.js.map +1 -0
- package/dist/src/intelligence/service.d.ts +18 -0
- package/dist/src/intelligence/service.d.ts.map +1 -0
- package/dist/src/intelligence/service.js +66 -0
- package/dist/src/intelligence/service.js.map +1 -0
- package/dist/src/logs/format.d.ts +6 -0
- package/dist/src/logs/format.d.ts.map +1 -0
- package/dist/src/logs/format.js +46 -0
- package/dist/src/logs/format.js.map +1 -0
- package/dist/src/logs/reader.d.ts +28 -0
- package/dist/src/logs/reader.d.ts.map +1 -0
- package/dist/src/logs/reader.js +106 -0
- package/dist/src/logs/reader.js.map +1 -0
- package/dist/src/mcp/server.d.ts +16 -0
- package/dist/src/mcp/server.d.ts.map +1 -0
- package/dist/src/mcp/server.js +305 -0
- package/dist/src/mcp/server.js.map +1 -0
- package/dist/src/mcp/tools/consolidate.d.ts +15 -0
- package/dist/src/mcp/tools/consolidate.d.ts.map +1 -0
- package/dist/src/mcp/tools/consolidate.js +49 -0
- package/dist/src/mcp/tools/consolidate.js.map +1 -0
- package/dist/src/mcp/tools/graph.d.ts +30 -0
- package/dist/src/mcp/tools/graph.d.ts.map +1 -0
- package/dist/src/mcp/tools/graph.js +106 -0
- package/dist/src/mcp/tools/graph.js.map +1 -0
- package/dist/src/mcp/tools/logs.d.ts +3 -0
- package/dist/src/mcp/tools/logs.d.ts.map +1 -0
- package/dist/src/mcp/tools/logs.js +7 -0
- package/dist/src/mcp/tools/logs.js.map +1 -0
- package/dist/src/mcp/tools/plans.d.ts +23 -0
- package/dist/src/mcp/tools/plans.d.ts.map +1 -0
- package/dist/src/mcp/tools/plans.js +63 -0
- package/dist/src/mcp/tools/plans.js.map +1 -0
- package/dist/src/mcp/tools/recall.d.ts +30 -0
- package/dist/src/mcp/tools/recall.d.ts.map +1 -0
- package/dist/src/mcp/tools/recall.js +34 -0
- package/dist/src/mcp/tools/recall.js.map +1 -0
- package/dist/src/mcp/tools/remember.d.ts +15 -0
- package/dist/src/mcp/tools/remember.d.ts.map +1 -0
- package/dist/src/mcp/tools/remember.js +18 -0
- package/dist/src/mcp/tools/remember.js.map +1 -0
- package/dist/src/mcp/tools/search.d.ts +19 -0
- package/dist/src/mcp/tools/search.d.ts.map +1 -0
- package/dist/src/mcp/tools/search.js +59 -0
- package/dist/src/mcp/tools/search.js.map +1 -0
- package/dist/src/mcp/tools/sessions.d.ts +21 -0
- package/dist/src/mcp/tools/sessions.d.ts.map +1 -0
- package/dist/src/mcp/tools/sessions.js +36 -0
- package/dist/src/mcp/tools/sessions.js.map +1 -0
- package/dist/src/mcp/tools/supersede.d.ts +14 -0
- package/dist/src/mcp/tools/supersede.d.ts.map +1 -0
- package/dist/src/mcp/tools/supersede.js +30 -0
- package/dist/src/mcp/tools/supersede.js.map +1 -0
- package/dist/src/mcp/tools/team.d.ts +16 -0
- package/dist/src/mcp/tools/team.d.ts.map +1 -0
- package/dist/src/mcp/tools/team.js +32 -0
- package/dist/src/mcp/tools/team.js.map +1 -0
- package/dist/src/obsidian/formatter.d.ts +80 -0
- package/dist/src/obsidian/formatter.d.ts.map +1 -0
- package/dist/src/obsidian/formatter.js +227 -0
- package/dist/src/obsidian/formatter.js.map +1 -0
- package/dist/src/prompts/classification.md +42 -0
- package/dist/src/prompts/extraction.md +45 -0
- package/dist/src/prompts/index.d.ts +13 -0
- package/dist/src/prompts/index.d.ts.map +1 -0
- package/dist/src/prompts/index.js +75 -0
- package/dist/src/prompts/index.js.map +1 -0
- package/dist/src/prompts/session-similarity.md +24 -0
- package/dist/src/prompts/summary.md +9 -0
- package/dist/src/prompts/title.md +8 -0
- package/dist/src/vault/frontmatter.d.ts +6 -0
- package/dist/src/vault/frontmatter.d.ts.map +1 -0
- package/dist/src/vault/frontmatter.js +10 -0
- package/dist/src/vault/frontmatter.js.map +1 -0
- package/dist/src/vault/observations.d.ts +10 -0
- package/dist/src/vault/observations.d.ts.map +1 -0
- package/dist/src/vault/observations.js +33 -0
- package/dist/src/vault/observations.js.map +1 -0
- package/dist/src/vault/reader.d.ts +10 -0
- package/dist/src/vault/reader.d.ts.map +1 -0
- package/dist/src/vault/reader.js +48 -0
- package/dist/src/vault/reader.js.map +1 -0
- package/dist/src/vault/resolve.d.ts +18 -0
- package/dist/src/vault/resolve.d.ts.map +1 -0
- package/dist/src/vault/resolve.js +51 -0
- package/dist/src/vault/resolve.js.map +1 -0
- package/dist/src/vault/session-id.d.ts +16 -0
- package/dist/src/vault/session-id.d.ts.map +1 -0
- package/dist/src/vault/session-id.js +29 -0
- package/dist/src/vault/session-id.js.map +1 -0
- package/dist/src/vault/types.d.ts +88 -0
- package/dist/src/vault/types.d.ts.map +1 -0
- package/dist/src/vault/types.js +94 -0
- package/dist/src/vault/types.js.map +1 -0
- package/dist/src/vault/writer.d.ts +66 -0
- package/dist/src/vault/writer.d.ts.map +1 -0
- package/dist/src/vault/writer.js +217 -0
- package/dist/src/vault/writer.js.map +1 -0
- package/hooks/hooks.json +60 -0
- package/package.json +52 -0
- package/skills/myco/SKILL.md +206 -0
- package/skills/myco/references/wisdom.md +61 -0
- package/skills/rules/SKILL.md +185 -0
- package/skills/rules/references/rules-bad-example.md +106 -0
- package/skills/rules/references/rules-good-example.md +90 -0
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
import Database from 'better-sqlite3';
|
|
2
|
+
export class MycoIndex {
|
|
3
|
+
db;
|
|
4
|
+
constructor(dbPath) {
|
|
5
|
+
this.db = new Database(dbPath);
|
|
6
|
+
this.db.pragma('journal_mode = WAL');
|
|
7
|
+
this.db.pragma('foreign_keys = ON');
|
|
8
|
+
this.init();
|
|
9
|
+
}
|
|
10
|
+
init() {
|
|
11
|
+
this.db.exec(`
|
|
12
|
+
CREATE TABLE IF NOT EXISTS notes (
|
|
13
|
+
path TEXT PRIMARY KEY,
|
|
14
|
+
type TEXT NOT NULL,
|
|
15
|
+
id TEXT NOT NULL,
|
|
16
|
+
title TEXT NOT NULL DEFAULT '',
|
|
17
|
+
content TEXT NOT NULL DEFAULT '',
|
|
18
|
+
frontmatter TEXT NOT NULL DEFAULT '{}',
|
|
19
|
+
created TEXT NOT NULL,
|
|
20
|
+
updated_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
21
|
+
);
|
|
22
|
+
|
|
23
|
+
CREATE INDEX IF NOT EXISTS idx_notes_type ON notes(type);
|
|
24
|
+
CREATE INDEX IF NOT EXISTS idx_notes_id ON notes(id);
|
|
25
|
+
CREATE INDEX IF NOT EXISTS idx_notes_created ON notes(created);
|
|
26
|
+
`);
|
|
27
|
+
}
|
|
28
|
+
getPragma(name) {
|
|
29
|
+
return this.db.pragma(`${name}`, { simple: true });
|
|
30
|
+
}
|
|
31
|
+
getDb() {
|
|
32
|
+
return this.db;
|
|
33
|
+
}
|
|
34
|
+
upsertNote(note) {
|
|
35
|
+
const stmt = this.db.prepare(`
|
|
36
|
+
INSERT INTO notes (path, type, id, title, content, frontmatter, created, updated_at)
|
|
37
|
+
VALUES (?, ?, ?, ?, ?, ?, ?, datetime('now'))
|
|
38
|
+
ON CONFLICT(path) DO UPDATE SET
|
|
39
|
+
type = excluded.type,
|
|
40
|
+
id = excluded.id,
|
|
41
|
+
title = excluded.title,
|
|
42
|
+
content = excluded.content,
|
|
43
|
+
frontmatter = excluded.frontmatter,
|
|
44
|
+
created = excluded.created,
|
|
45
|
+
updated_at = datetime('now')
|
|
46
|
+
`);
|
|
47
|
+
stmt.run(note.path, note.type, note.id, note.title, note.content, JSON.stringify(note.frontmatter), note.created);
|
|
48
|
+
}
|
|
49
|
+
getNoteByPath(notePath) {
|
|
50
|
+
const row = this.db.prepare('SELECT * FROM notes WHERE path = ?').get(notePath);
|
|
51
|
+
if (!row)
|
|
52
|
+
return null;
|
|
53
|
+
return { ...row, frontmatter: JSON.parse(row.frontmatter) };
|
|
54
|
+
}
|
|
55
|
+
deleteNote(notePath) {
|
|
56
|
+
this.db.prepare('DELETE FROM notes WHERE path = ?').run(notePath);
|
|
57
|
+
}
|
|
58
|
+
query(options) {
|
|
59
|
+
const conditions = [];
|
|
60
|
+
const params = [];
|
|
61
|
+
if (options.type) {
|
|
62
|
+
conditions.push('type = ?');
|
|
63
|
+
params.push(options.type);
|
|
64
|
+
}
|
|
65
|
+
if (options.id) {
|
|
66
|
+
conditions.push('id = ?');
|
|
67
|
+
params.push(options.id);
|
|
68
|
+
}
|
|
69
|
+
if (options.since) {
|
|
70
|
+
conditions.push('created >= ?');
|
|
71
|
+
params.push(options.since);
|
|
72
|
+
}
|
|
73
|
+
if (options.frontmatter) {
|
|
74
|
+
for (const [key, value] of Object.entries(options.frontmatter)) {
|
|
75
|
+
conditions.push(`json_extract(frontmatter, '$.' || ?) = ?`);
|
|
76
|
+
params.push(key, value);
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
const where = conditions.length > 0 ? `WHERE ${conditions.join(' AND ')}` : '';
|
|
80
|
+
const limitClause = options.limit ? 'LIMIT ?' : '';
|
|
81
|
+
if (options.limit)
|
|
82
|
+
params.push(options.limit);
|
|
83
|
+
const sql = `SELECT * FROM notes ${where} ORDER BY created DESC ${limitClause}`;
|
|
84
|
+
const rows = this.db.prepare(sql).all(...params);
|
|
85
|
+
return rows.map((row) => ({ ...row, frontmatter: JSON.parse(row.frontmatter) }));
|
|
86
|
+
}
|
|
87
|
+
queryByIds(ids) {
|
|
88
|
+
if (ids.length === 0)
|
|
89
|
+
return [];
|
|
90
|
+
const placeholders = ids.map(() => '?').join(',');
|
|
91
|
+
const sql = `SELECT * FROM notes WHERE id IN (${placeholders})`;
|
|
92
|
+
const rows = this.db.prepare(sql).all(...ids);
|
|
93
|
+
return rows.map((row) => ({ ...row, frontmatter: JSON.parse(row.frontmatter) }));
|
|
94
|
+
}
|
|
95
|
+
close() {
|
|
96
|
+
this.db.close();
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
//# sourceMappingURL=sqlite.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sqlite.js","sourceRoot":"","sources":["../../../src/index/sqlite.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAsBtC,MAAM,OAAO,SAAS;IACZ,EAAE,CAAoB;IAE9B,YAAY,MAAc;QACxB,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC/B,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QACrC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAEO,IAAI;QACV,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;KAeZ,CAAC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACrD,CAAC;IAED,KAAK;QACH,OAAO,IAAI,CAAC,EAAE,CAAC;IACjB,CAAC;IAED,UAAU,CAAC,IAAqC;QAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;;;;;;KAW5B,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CACN,IAAI,CAAC,IAAI,EACT,IAAI,CAAC,IAAI,EACT,IAAI,CAAC,EAAE,EACP,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,EAChC,IAAI,CAAC,OAAO,CACb,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,QAAgB;QAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAQ,CAAC;QACvF,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QACtB,OAAO,EAAE,GAAG,GAAG,EAAE,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;IAC9D,CAAC;IAED,UAAU,CAAC,QAAgB;QACzB,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,OAAqB;QACzB,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,MAAM,GAAc,EAAE,CAAC;QAE7B,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACjB,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QACD,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;YACf,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC;QACD,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACxB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC/D,UAAU,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;gBAC5D,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/E,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;QACnD,IAAI,OAAO,CAAC,KAAK;YAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,GAAG,GAAG,uBAAuB,KAAK,0BAA0B,WAAW,EAAE,CAAC;QAEhF,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;QAC1D,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC;IACnF,CAAC;IAED,UAAU,CAAC,GAAa;QACtB,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAChC,MAAM,YAAY,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,GAAG,GAAG,oCAAoC,YAAY,GAAG,CAAC;QAChE,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAU,CAAC;QACvD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC;IACnF,CAAC;IAED,KAAK;QACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;CACF"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
export interface VectorSearchResult {
|
|
2
|
+
id: string;
|
|
3
|
+
similarity: number;
|
|
4
|
+
metadata: Record<string, string>;
|
|
5
|
+
}
|
|
6
|
+
export interface VectorSearchOptions {
|
|
7
|
+
limit?: number;
|
|
8
|
+
/** Drop results below this fraction of the top result's score (0-1). Default 0.5. */
|
|
9
|
+
relativeThreshold?: number;
|
|
10
|
+
type?: string;
|
|
11
|
+
importance?: string;
|
|
12
|
+
}
|
|
13
|
+
export declare class VectorIndex {
|
|
14
|
+
private db;
|
|
15
|
+
private dimensions;
|
|
16
|
+
constructor(dbPath: string, dimensions: number);
|
|
17
|
+
private init;
|
|
18
|
+
upsert(id: string, embedding: number[], metadata?: Record<string, string>): void;
|
|
19
|
+
search(query: number[], options?: VectorSearchOptions): VectorSearchResult[];
|
|
20
|
+
delete(id: string): void;
|
|
21
|
+
count(): number;
|
|
22
|
+
close(): void;
|
|
23
|
+
}
|
|
24
|
+
//# sourceMappingURL=vectors.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"vectors.d.ts","sourceRoot":"","sources":["../../../src/index/vectors.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAClC;AAED,MAAM,WAAW,mBAAmB;IAClC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,qFAAqF;IACrF,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,EAAE,CAAoB;IAC9B,OAAO,CAAC,UAAU,CAAS;gBAEf,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM;IAO9C,OAAO,CAAC,IAAI;IAkBZ,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAM,GAAG,IAAI;IAWpF,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,OAAO,GAAE,mBAAwB,GAAG,kBAAkB,EAAE;IAwDhF,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAKxB,KAAK,IAAI,MAAM;IAIf,KAAK,IAAI,IAAI;CACd"}
|
|
@@ -0,0 +1,97 @@
|
|
|
1
|
+
import Database from 'better-sqlite3';
|
|
2
|
+
import * as sqliteVec from 'sqlite-vec';
|
|
3
|
+
export class VectorIndex {
|
|
4
|
+
db;
|
|
5
|
+
dimensions;
|
|
6
|
+
constructor(dbPath, dimensions) {
|
|
7
|
+
this.dimensions = dimensions;
|
|
8
|
+
this.db = new Database(dbPath);
|
|
9
|
+
sqliteVec.load(this.db);
|
|
10
|
+
this.init();
|
|
11
|
+
}
|
|
12
|
+
init() {
|
|
13
|
+
this.db.exec(`
|
|
14
|
+
CREATE TABLE IF NOT EXISTS vec_metadata (
|
|
15
|
+
id TEXT PRIMARY KEY,
|
|
16
|
+
type TEXT NOT NULL DEFAULT '',
|
|
17
|
+
importance TEXT NOT NULL DEFAULT '',
|
|
18
|
+
session_id TEXT NOT NULL DEFAULT '',
|
|
19
|
+
file_path TEXT NOT NULL DEFAULT '',
|
|
20
|
+
branch TEXT NOT NULL DEFAULT '',
|
|
21
|
+
created TEXT NOT NULL DEFAULT (datetime('now'))
|
|
22
|
+
);
|
|
23
|
+
CREATE VIRTUAL TABLE IF NOT EXISTS vec_embeddings USING vec0(
|
|
24
|
+
id TEXT PRIMARY KEY,
|
|
25
|
+
embedding float[${this.dimensions}]
|
|
26
|
+
);
|
|
27
|
+
`);
|
|
28
|
+
}
|
|
29
|
+
upsert(id, embedding, metadata = {}) {
|
|
30
|
+
this.db.prepare('DELETE FROM vec_metadata WHERE id = ?').run(id);
|
|
31
|
+
this.db.prepare('DELETE FROM vec_embeddings WHERE id = ?').run(id);
|
|
32
|
+
this.db.prepare('INSERT INTO vec_metadata (id, type, importance, session_id, file_path, branch) VALUES (?, ?, ?, ?, ?, ?)').run(id, metadata.type ?? '', metadata.importance ?? '', metadata.session_id ?? '', metadata.file_path ?? '', metadata.branch ?? '');
|
|
33
|
+
this.db.prepare('INSERT INTO vec_embeddings (id, embedding) VALUES (?, ?)').run(id, new Float32Array(embedding));
|
|
34
|
+
}
|
|
35
|
+
search(query, options = {}) {
|
|
36
|
+
const limit = options.limit ?? 10;
|
|
37
|
+
// vec0 KNN queries require LIMIT to be a direct constraint on the virtual table.
|
|
38
|
+
// JOINs with additional WHERE filters confuse the query planner, so we use a
|
|
39
|
+
// subquery to get KNN candidates first, then filter by metadata in the outer query.
|
|
40
|
+
const knnParams = [new Float32Array(query), limit * 4];
|
|
41
|
+
const knnRows = this.db.prepare(`
|
|
42
|
+
SELECT id, distance
|
|
43
|
+
FROM vec_embeddings
|
|
44
|
+
WHERE embedding MATCH ?
|
|
45
|
+
ORDER BY distance
|
|
46
|
+
LIMIT ?
|
|
47
|
+
`).all(...knnParams);
|
|
48
|
+
if (knnRows.length === 0)
|
|
49
|
+
return [];
|
|
50
|
+
// Now fetch metadata for the candidates and apply optional filters
|
|
51
|
+
const metaConditions = ['id IN (' + knnRows.map(() => '?').join(',') + ')'];
|
|
52
|
+
const metaParams = knnRows.map((r) => r.id);
|
|
53
|
+
if (options.type) {
|
|
54
|
+
metaConditions.push('type = ?');
|
|
55
|
+
metaParams.push(options.type);
|
|
56
|
+
}
|
|
57
|
+
if (options.importance) {
|
|
58
|
+
metaConditions.push('importance = ?');
|
|
59
|
+
metaParams.push(options.importance);
|
|
60
|
+
}
|
|
61
|
+
const metaRows = this.db.prepare(`
|
|
62
|
+
SELECT id, type, importance, session_id, file_path, branch
|
|
63
|
+
FROM vec_metadata
|
|
64
|
+
WHERE ${metaConditions.join(' AND ')}
|
|
65
|
+
`).all(...metaParams);
|
|
66
|
+
const metaMap = new Map(metaRows.map((m) => [m.id, m]));
|
|
67
|
+
const scored = knnRows
|
|
68
|
+
.filter((r) => metaMap.has(r.id))
|
|
69
|
+
.map((r) => {
|
|
70
|
+
const m = metaMap.get(r.id);
|
|
71
|
+
return {
|
|
72
|
+
id: r.id,
|
|
73
|
+
similarity: 1 - r.distance,
|
|
74
|
+
metadata: { type: m.type, importance: m.importance, session_id: m.session_id, file_path: m.file_path, branch: m.branch },
|
|
75
|
+
};
|
|
76
|
+
});
|
|
77
|
+
if (scored.length === 0)
|
|
78
|
+
return [];
|
|
79
|
+
// Relative threshold: drop results below a fraction of the best score.
|
|
80
|
+
// Adapts automatically to any embedding model's score distribution.
|
|
81
|
+
const topScore = scored[0].similarity;
|
|
82
|
+
const threshold = options.relativeThreshold ?? 0.5;
|
|
83
|
+
const floor = topScore * threshold;
|
|
84
|
+
return scored
|
|
85
|
+
.filter((r) => r.similarity >= floor)
|
|
86
|
+
.slice(0, limit);
|
|
87
|
+
}
|
|
88
|
+
delete(id) {
|
|
89
|
+
this.db.prepare('DELETE FROM vec_metadata WHERE id = ?').run(id);
|
|
90
|
+
this.db.prepare('DELETE FROM vec_embeddings WHERE id = ?').run(id);
|
|
91
|
+
}
|
|
92
|
+
count() {
|
|
93
|
+
return this.db.prepare('SELECT COUNT(*) as c FROM vec_metadata').get().c;
|
|
94
|
+
}
|
|
95
|
+
close() { this.db.close(); }
|
|
96
|
+
}
|
|
97
|
+
//# sourceMappingURL=vectors.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"vectors.js","sourceRoot":"","sources":["../../../src/index/vectors.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,KAAK,SAAS,MAAM,YAAY,CAAC;AAgBxC,MAAM,OAAO,WAAW;IACd,EAAE,CAAoB;IACtB,UAAU,CAAS;IAE3B,YAAY,MAAc,EAAE,UAAkB;QAC5C,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC/B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxB,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAEO,IAAI;QACV,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;0BAYS,IAAI,CAAC,UAAU;;KAEpC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,EAAU,EAAE,SAAmB,EAAE,WAAmC,EAAE;QAC3E,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,uCAAuC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAEnE,IAAI,CAAC,EAAE,CAAC,OAAO,CACb,0GAA0G,CAC3G,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,IAAI,IAAI,EAAE,EAAE,QAAQ,CAAC,UAAU,IAAI,EAAE,EAAE,QAAQ,CAAC,UAAU,IAAI,EAAE,EAAE,QAAQ,CAAC,SAAS,IAAI,EAAE,EAAE,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAEtI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,0DAA0D,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC;IACnH,CAAC;IAED,MAAM,CAAC,KAAe,EAAE,UAA+B,EAAE;QACvD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;QAElC,iFAAiF;QACjF,6EAA6E;QAC7E,oFAAoF;QACpF,MAAM,SAAS,GAAc,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;QAClE,MAAM,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;KAM/B,CAAC,CAAC,GAAG,CAAC,GAAG,SAAS,CAA4C,CAAC;QAEhE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEpC,mEAAmE;QACnE,MAAM,cAAc,GAAa,CAAC,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;QACtF,MAAM,UAAU,GAAc,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAEvD,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YAAC,cAAc,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAAC,CAAC;QACrF,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YAAC,cAAc,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAAC,CAAC;QAEvG,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;cAGvB,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC;KACrC,CAAC,CAAC,GAAG,CAAC,GAAG,UAAU,CAAmH,CAAC;QAExI,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAExD,MAAM,MAAM,GAAG,OAAO;aACnB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACT,MAAM,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAE,CAAC;YAC7B,OAAO;gBACL,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC,QAAQ;gBAC1B,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;aACzH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEL,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEnC,uEAAuE;QACvE,oEAAoE;QACpE,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;QACtC,MAAM,SAAS,GAAG,OAAO,CAAC,iBAAiB,IAAI,GAAG,CAAC;QACnD,MAAM,KAAK,GAAG,QAAQ,GAAG,SAAS,CAAC;QAEnC,OAAO,MAAM;aACV,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,IAAI,KAAK,CAAC;aACpC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACrB,CAAC;IAED,MAAM,CAAC,EAAU;QACf,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,uCAAuC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,KAAK;QACH,OAAQ,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,wCAAwC,CAAC,CAAC,GAAG,EAAoB,CAAC,CAAC,CAAC;IAC9F,CAAC;IAED,KAAK,KAAW,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;CACnC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import type { LlmProvider, LlmResponse, EmbeddingResponse, LlmRequestOptions } from './llm.js';
|
|
2
|
+
interface AnthropicConfig {
|
|
3
|
+
model?: string;
|
|
4
|
+
summary_model?: string;
|
|
5
|
+
embedding_provider?: string;
|
|
6
|
+
}
|
|
7
|
+
export declare class AnthropicBackend implements LlmProvider {
|
|
8
|
+
readonly name = "anthropic";
|
|
9
|
+
private client;
|
|
10
|
+
private model;
|
|
11
|
+
constructor(config?: AnthropicConfig);
|
|
12
|
+
summarize(prompt: string, opts?: LlmRequestOptions): Promise<LlmResponse>;
|
|
13
|
+
embed(_text: string): Promise<EmbeddingResponse>;
|
|
14
|
+
isAvailable(): Promise<boolean>;
|
|
15
|
+
}
|
|
16
|
+
export {};
|
|
17
|
+
//# sourceMappingURL=anthropic.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/intelligence/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAG/F,UAAU,eAAe;IACvB,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,qBAAa,gBAAiB,YAAW,WAAW;IAClD,QAAQ,CAAC,IAAI,eAAe;IAC5B,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,CAAC,EAAE,eAAe;IAK9B,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC;IAiBzE,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAMhD,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;CAOtC"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import Anthropic from '@anthropic-ai/sdk';
|
|
2
|
+
import { LLM_REQUEST_TIMEOUT_MS } from '../constants.js';
|
|
3
|
+
export class AnthropicBackend {
|
|
4
|
+
name = 'anthropic';
|
|
5
|
+
client;
|
|
6
|
+
model;
|
|
7
|
+
constructor(config) {
|
|
8
|
+
this.client = new Anthropic();
|
|
9
|
+
this.model = config?.model ?? config?.summary_model ?? 'claude-haiku-4-5-20251001';
|
|
10
|
+
}
|
|
11
|
+
async summarize(prompt, opts) {
|
|
12
|
+
const maxTokens = opts?.maxTokens ?? 1024;
|
|
13
|
+
const response = await this.client.messages.create({
|
|
14
|
+
model: this.model,
|
|
15
|
+
max_tokens: maxTokens,
|
|
16
|
+
messages: [{ role: 'user', content: prompt }],
|
|
17
|
+
}, { timeout: LLM_REQUEST_TIMEOUT_MS });
|
|
18
|
+
const text = response.content
|
|
19
|
+
.filter((block) => block.type === 'text')
|
|
20
|
+
.map((block) => block.text)
|
|
21
|
+
.join('');
|
|
22
|
+
return { text, model: this.model };
|
|
23
|
+
}
|
|
24
|
+
async embed(_text) {
|
|
25
|
+
throw new Error('AnthropicBackend does not support embeddings. Use a local provider (Ollama or LM Studio) for embeddings.');
|
|
26
|
+
}
|
|
27
|
+
async isAvailable() {
|
|
28
|
+
try {
|
|
29
|
+
return !!process.env.ANTHROPIC_API_KEY;
|
|
30
|
+
}
|
|
31
|
+
catch {
|
|
32
|
+
return false;
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/intelligence/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAE1C,OAAO,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AASzD,MAAM,OAAO,gBAAgB;IAClB,IAAI,GAAG,WAAW,CAAC;IACpB,MAAM,CAAY;IAClB,KAAK,CAAS;IAEtB,YAAY,MAAwB;QAClC,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,IAAI,CAAC,KAAK,GAAG,MAAM,EAAE,KAAK,IAAI,MAAM,EAAE,aAAa,IAAI,2BAA2B,CAAC;IACrF,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,IAAwB;QACtD,MAAM,SAAS,GAAG,IAAI,EAAE,SAAS,IAAI,IAAI,CAAC;QAE1C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACjD,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,UAAU,EAAE,SAAS;YACrB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;SAC9C,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAE,CAAC,CAAC;QAExC,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO;aAC1B,MAAM,CAAC,CAAC,KAAK,EAAgC,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC;aACtE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;aAC1B,IAAI,CAAC,EAAE,CAAC,CAAC;QAEZ,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAa;QACvB,MAAM,IAAI,KAAK,CACb,0GAA0G,CAC3G,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddings.d.ts","sourceRoot":"","sources":["../../../src/intelligence/embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAErE,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,iBAAiB,EAC1B,IAAI,EAAE,MAAM,GACX,OAAO,CAAC,iBAAiB,CAAC,CAO5B"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
export async function generateEmbedding(backend, text) {
|
|
2
|
+
const raw = await backend.embed(text);
|
|
3
|
+
return {
|
|
4
|
+
embedding: normalize(raw.embedding),
|
|
5
|
+
model: raw.model,
|
|
6
|
+
dimensions: raw.dimensions,
|
|
7
|
+
};
|
|
8
|
+
}
|
|
9
|
+
function normalize(vec) {
|
|
10
|
+
const magnitude = Math.sqrt(vec.reduce((sum, v) => sum + v * v, 0));
|
|
11
|
+
if (magnitude === 0)
|
|
12
|
+
return vec;
|
|
13
|
+
return vec.map((v) => v / magnitude);
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=embeddings.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../../../src/intelligence/embeddings.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAA0B,EAC1B,IAAY;IAEZ,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACtC,OAAO;QACL,SAAS,EAAE,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC;QACnC,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,UAAU,EAAE,GAAG,CAAC,UAAU;KAC3B,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,GAAa;IAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACpE,IAAI,SAAS,KAAK,CAAC;QAAE,OAAO,GAAG,CAAC;IAChC,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC;AACvC,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import type { LlmBackend, LlmResponse, EmbeddingResponse, LlmRequestOptions } from './llm.js';
|
|
2
|
+
interface HaikuConfig {
|
|
3
|
+
model?: string;
|
|
4
|
+
summary_model?: string;
|
|
5
|
+
embedding_provider?: string;
|
|
6
|
+
}
|
|
7
|
+
export declare class HaikuBackend implements LlmBackend {
|
|
8
|
+
readonly name = "haiku";
|
|
9
|
+
private client;
|
|
10
|
+
private model;
|
|
11
|
+
constructor(config?: HaikuConfig);
|
|
12
|
+
summarize(prompt: string, opts?: LlmRequestOptions): Promise<LlmResponse>;
|
|
13
|
+
embed(_text: string): Promise<EmbeddingResponse>;
|
|
14
|
+
isAvailable(): Promise<boolean>;
|
|
15
|
+
}
|
|
16
|
+
export {};
|
|
17
|
+
//# sourceMappingURL=haiku.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"haiku.d.ts","sourceRoot":"","sources":["../../../src/intelligence/haiku.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAE9F,UAAU,WAAW;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,qBAAa,YAAa,YAAW,UAAU;IAC7C,QAAQ,CAAC,IAAI,WAAW;IACxB,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,CAAC,EAAE,WAAW;IAK1B,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC;IAiBzE,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAMhD,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;CAOtC"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import Anthropic from '@anthropic-ai/sdk';
|
|
2
|
+
export class HaikuBackend {
|
|
3
|
+
name = 'haiku';
|
|
4
|
+
client;
|
|
5
|
+
model;
|
|
6
|
+
constructor(config) {
|
|
7
|
+
this.client = new Anthropic();
|
|
8
|
+
this.model = config?.model ?? config?.summary_model ?? 'claude-haiku-4-5-20251001';
|
|
9
|
+
}
|
|
10
|
+
async summarize(prompt, opts) {
|
|
11
|
+
const maxTokens = opts?.maxTokens ?? 1024;
|
|
12
|
+
const response = await this.client.messages.create({
|
|
13
|
+
model: this.model,
|
|
14
|
+
max_tokens: maxTokens,
|
|
15
|
+
messages: [{ role: 'user', content: prompt }],
|
|
16
|
+
});
|
|
17
|
+
const text = response.content
|
|
18
|
+
.filter((block) => block.type === 'text')
|
|
19
|
+
.map((block) => block.text)
|
|
20
|
+
.join('');
|
|
21
|
+
return { text, model: this.model };
|
|
22
|
+
}
|
|
23
|
+
async embed(_text) {
|
|
24
|
+
throw new Error('HaikuBackend does not support embeddings. Use a local provider (Ollama or LM Studio) for embeddings.');
|
|
25
|
+
}
|
|
26
|
+
async isAvailable() {
|
|
27
|
+
try {
|
|
28
|
+
return !!process.env.ANTHROPIC_API_KEY;
|
|
29
|
+
}
|
|
30
|
+
catch {
|
|
31
|
+
return false;
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=haiku.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"haiku.js","sourceRoot":"","sources":["../../../src/intelligence/haiku.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAU1C,MAAM,OAAO,YAAY;IACd,IAAI,GAAG,OAAO,CAAC;IAChB,MAAM,CAAY;IAClB,KAAK,CAAS;IAEtB,YAAY,MAAoB;QAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,IAAI,CAAC,KAAK,GAAG,MAAM,EAAE,KAAK,IAAI,MAAM,EAAE,aAAa,IAAI,2BAA2B,CAAC;IACrF,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,IAAwB;QACtD,MAAM,SAAS,GAAG,IAAI,EAAE,SAAS,IAAI,IAAI,CAAC;QAE1C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACjD,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,UAAU,EAAE,SAAS;YACrB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;SAC9C,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO;aAC1B,MAAM,CAAC,CAAC,KAAK,EAAgC,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC;aACtE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;aAC1B,IAAI,CAAC,EAAE,CAAC,CAAC;QAEZ,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAa;QACvB,MAAM,IAAI,KAAK,CACb,sGAAsG,CACvG,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
export interface LlmRequestOptions {
|
|
2
|
+
maxTokens?: number;
|
|
3
|
+
}
|
|
4
|
+
export interface LlmResponse {
|
|
5
|
+
text: string;
|
|
6
|
+
model: string;
|
|
7
|
+
}
|
|
8
|
+
export interface EmbeddingResponse {
|
|
9
|
+
embedding: number[];
|
|
10
|
+
model: string;
|
|
11
|
+
dimensions: number;
|
|
12
|
+
}
|
|
13
|
+
export interface LlmProvider {
|
|
14
|
+
name: string;
|
|
15
|
+
summarize(prompt: string, opts?: LlmRequestOptions): Promise<LlmResponse>;
|
|
16
|
+
isAvailable(): Promise<boolean>;
|
|
17
|
+
}
|
|
18
|
+
export interface EmbeddingProvider {
|
|
19
|
+
name: string;
|
|
20
|
+
embed(text: string): Promise<EmbeddingResponse>;
|
|
21
|
+
isAvailable(): Promise<boolean>;
|
|
22
|
+
}
|
|
23
|
+
interface ProviderConfig {
|
|
24
|
+
provider: string;
|
|
25
|
+
model: string;
|
|
26
|
+
base_url?: string;
|
|
27
|
+
context_window?: number;
|
|
28
|
+
max_tokens?: number;
|
|
29
|
+
}
|
|
30
|
+
export declare function createLlmProvider(config: ProviderConfig): LlmProvider;
|
|
31
|
+
export declare function createEmbeddingProvider(config: ProviderConfig): EmbeddingProvider;
|
|
32
|
+
export {};
|
|
33
|
+
//# sourceMappingURL=llm.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"llm.d.ts","sourceRoot":"","sources":["../../../src/intelligence/llm.ts"],"names":[],"mappings":"AAIA,MAAM,WAAW,iBAAiB;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAC1E,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;CACjC;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAChD,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;CACjC;AAED,UAAU,cAAc;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,cAAc,GAAG,WAAW,CAWrE;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,cAAc,GAAG,iBAAiB,CASjF"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import { OllamaBackend } from './ollama.js';
|
|
2
|
+
import { LmStudioBackend } from './lm-studio.js';
|
|
3
|
+
import { AnthropicBackend } from './anthropic.js';
|
|
4
|
+
export function createLlmProvider(config) {
|
|
5
|
+
switch (config.provider) {
|
|
6
|
+
case 'ollama':
|
|
7
|
+
return new OllamaBackend(config);
|
|
8
|
+
case 'lm-studio':
|
|
9
|
+
return new LmStudioBackend(config);
|
|
10
|
+
case 'anthropic':
|
|
11
|
+
return new AnthropicBackend(config);
|
|
12
|
+
default:
|
|
13
|
+
throw new Error(`Unknown LLM provider: ${config.provider}`);
|
|
14
|
+
}
|
|
15
|
+
}
|
|
16
|
+
export function createEmbeddingProvider(config) {
|
|
17
|
+
switch (config.provider) {
|
|
18
|
+
case 'ollama':
|
|
19
|
+
return new OllamaBackend(config);
|
|
20
|
+
case 'lm-studio':
|
|
21
|
+
return new LmStudioBackend(config);
|
|
22
|
+
default:
|
|
23
|
+
throw new Error(`Provider "${config.provider}" does not support embeddings. Use ollama or lm-studio.`);
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
//# sourceMappingURL=llm.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"llm.js","sourceRoot":"","sources":["../../../src/intelligence/llm.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAqClD,MAAM,UAAU,iBAAiB,CAAC,MAAsB;IACtD,QAAQ,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,QAAQ;YACX,OAAO,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;QACnC,KAAK,WAAW;YACd,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;QACrC,KAAK,WAAW;YACd,OAAO,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACtC;YACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAChE,CAAC;AACH,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,MAAsB;IAC5D,QAAQ,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,QAAQ;YACX,OAAO,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;QACnC,KAAK,WAAW;YACd,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;QACrC;YACE,MAAM,IAAI,KAAK,CAAC,aAAa,MAAM,CAAC,QAAQ,yDAAyD,CAAC,CAAC;IAC3G,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { LlmProvider, EmbeddingProvider, LlmResponse, EmbeddingResponse, LlmRequestOptions } from './llm.js';
|
|
2
|
+
interface LmStudioConfig {
|
|
3
|
+
model?: string;
|
|
4
|
+
base_url?: string;
|
|
5
|
+
max_tokens?: number;
|
|
6
|
+
embedding_model?: string;
|
|
7
|
+
summary_model?: string;
|
|
8
|
+
}
|
|
9
|
+
export declare class LmStudioBackend implements LlmProvider, EmbeddingProvider {
|
|
10
|
+
readonly name = "lm-studio";
|
|
11
|
+
private baseUrl;
|
|
12
|
+
private model;
|
|
13
|
+
private defaultMaxTokens;
|
|
14
|
+
constructor(config?: LmStudioConfig);
|
|
15
|
+
summarize(prompt: string, opts?: LlmRequestOptions): Promise<LlmResponse>;
|
|
16
|
+
embed(text: string): Promise<EmbeddingResponse>;
|
|
17
|
+
isAvailable(): Promise<boolean>;
|
|
18
|
+
}
|
|
19
|
+
export {};
|
|
20
|
+
//# sourceMappingURL=lm-studio.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"lm-studio.d.ts","sourceRoot":"","sources":["../../../src/intelligence/lm-studio.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,iBAAiB,EAAE,WAAW,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAGlH,UAAU,cAAc;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,qBAAa,eAAgB,YAAW,WAAW,EAAE,iBAAiB;IACpE,QAAQ,CAAC,IAAI,eAAe;IAC5B,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,gBAAgB,CAAS;gBAErB,MAAM,CAAC,EAAE,cAAc;IAM7B,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC;IAyBzE,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAuB/C,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;CAUtC"}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
import { LLM_REQUEST_TIMEOUT_MS, EMBEDDING_REQUEST_TIMEOUT_MS, DAEMON_CLIENT_TIMEOUT_MS } from '../constants.js';
|
|
2
|
+
export class LmStudioBackend {
|
|
3
|
+
name = 'lm-studio';
|
|
4
|
+
baseUrl;
|
|
5
|
+
model;
|
|
6
|
+
defaultMaxTokens;
|
|
7
|
+
constructor(config) {
|
|
8
|
+
this.baseUrl = config?.base_url ?? 'http://localhost:1234';
|
|
9
|
+
this.model = config?.model ?? config?.summary_model ?? 'llama3.2';
|
|
10
|
+
this.defaultMaxTokens = config?.max_tokens ?? 1024;
|
|
11
|
+
}
|
|
12
|
+
async summarize(prompt, opts) {
|
|
13
|
+
const maxTokens = opts?.maxTokens ?? this.defaultMaxTokens;
|
|
14
|
+
const response = await fetch(`${this.baseUrl}/v1/chat/completions`, {
|
|
15
|
+
method: 'POST',
|
|
16
|
+
headers: { 'Content-Type': 'application/json' },
|
|
17
|
+
body: JSON.stringify({
|
|
18
|
+
model: this.model,
|
|
19
|
+
messages: [{ role: 'user', content: prompt }],
|
|
20
|
+
max_tokens: maxTokens,
|
|
21
|
+
}),
|
|
22
|
+
signal: AbortSignal.timeout(LLM_REQUEST_TIMEOUT_MS),
|
|
23
|
+
});
|
|
24
|
+
if (!response.ok) {
|
|
25
|
+
throw new Error(`LM Studio summarize failed: ${response.status}`);
|
|
26
|
+
}
|
|
27
|
+
const data = await response.json();
|
|
28
|
+
return { text: data.choices[0].message.content, model: data.model };
|
|
29
|
+
}
|
|
30
|
+
async embed(text) {
|
|
31
|
+
const response = await fetch(`${this.baseUrl}/v1/embeddings`, {
|
|
32
|
+
method: 'POST',
|
|
33
|
+
headers: { 'Content-Type': 'application/json' },
|
|
34
|
+
body: JSON.stringify({
|
|
35
|
+
model: this.model,
|
|
36
|
+
input: text,
|
|
37
|
+
}),
|
|
38
|
+
signal: AbortSignal.timeout(EMBEDDING_REQUEST_TIMEOUT_MS),
|
|
39
|
+
});
|
|
40
|
+
if (!response.ok) {
|
|
41
|
+
throw new Error(`LM Studio embed failed: ${response.status}`);
|
|
42
|
+
}
|
|
43
|
+
const data = await response.json();
|
|
44
|
+
const embedding = data.data[0].embedding;
|
|
45
|
+
return { embedding, model: data.model, dimensions: embedding.length };
|
|
46
|
+
}
|
|
47
|
+
async isAvailable() {
|
|
48
|
+
try {
|
|
49
|
+
const response = await fetch(`${this.baseUrl}/v1/models`, {
|
|
50
|
+
signal: AbortSignal.timeout(DAEMON_CLIENT_TIMEOUT_MS),
|
|
51
|
+
});
|
|
52
|
+
return response.ok;
|
|
53
|
+
}
|
|
54
|
+
catch {
|
|
55
|
+
return false;
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
//# sourceMappingURL=lm-studio.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"lm-studio.js","sourceRoot":"","sources":["../../../src/intelligence/lm-studio.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,EAAE,4BAA4B,EAAE,wBAAwB,EAAE,MAAM,iBAAiB,CAAC;AAWjH,MAAM,OAAO,eAAe;IACjB,IAAI,GAAG,WAAW,CAAC;IACpB,OAAO,CAAS;IAChB,KAAK,CAAS;IACd,gBAAgB,CAAS;IAEjC,YAAY,MAAuB;QACjC,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,QAAQ,IAAI,uBAAuB,CAAC;QAC3D,IAAI,CAAC,KAAK,GAAG,MAAM,EAAE,KAAK,IAAI,MAAM,EAAE,aAAa,IAAI,UAAU,CAAC;QAClE,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,UAAU,IAAI,IAAI,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,IAAwB;QACtD,MAAM,SAAS,GAAG,IAAI,EAAE,SAAS,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAE3D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,sBAAsB,EAAE;YAClE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;gBAC7C,UAAU,EAAE,SAAS;aACtB,CAAC;YACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,sBAAsB,CAAC;SACpD,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+BAA+B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAG/B,CAAC;QACF,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,gBAAgB,EAAE;YAC5D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,IAAI;aACZ,CAAC;YACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,4BAA4B,CAAC;SAC1D,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAG/B,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACzC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,UAAU,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,YAAY,EAAE;gBACxD,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,wBAAwB,CAAC;aACtD,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,EAAE,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { LlmProvider, EmbeddingProvider, LlmResponse, EmbeddingResponse, LlmRequestOptions } from './llm.js';
|
|
2
|
+
interface OllamaConfig {
|
|
3
|
+
model?: string;
|
|
4
|
+
base_url?: string;
|
|
5
|
+
context_window?: number;
|
|
6
|
+
max_tokens?: number;
|
|
7
|
+
embedding_model?: string;
|
|
8
|
+
summary_model?: string;
|
|
9
|
+
}
|
|
10
|
+
export declare class OllamaBackend implements LlmProvider, EmbeddingProvider {
|
|
11
|
+
readonly name = "ollama";
|
|
12
|
+
private baseUrl;
|
|
13
|
+
private model;
|
|
14
|
+
private contextWindow;
|
|
15
|
+
private defaultMaxTokens;
|
|
16
|
+
constructor(config?: OllamaConfig);
|
|
17
|
+
summarize(prompt: string, opts?: LlmRequestOptions): Promise<LlmResponse>;
|
|
18
|
+
embed(text: string): Promise<EmbeddingResponse>;
|
|
19
|
+
isAvailable(): Promise<boolean>;
|
|
20
|
+
}
|
|
21
|
+
export {};
|
|
22
|
+
//# sourceMappingURL=ollama.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../../src/intelligence/ollama.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,iBAAiB,EAAE,WAAW,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAGlH,UAAU,YAAY;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,qBAAa,aAAc,YAAW,WAAW,EAAE,iBAAiB;IAClE,QAAQ,CAAC,IAAI,YAAY;IACzB,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,gBAAgB,CAAS;gBAErB,MAAM,CAAC,EAAE,YAAY;IAO3B,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC;IAyBzE,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAoB/C,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;CAUtC"}
|