@phmotad/fire-code 0.2.7 → 0.2.9

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.
Files changed (101) hide show
  1. package/dist/cli/commands/index-cmd.d.ts +1 -0
  2. package/dist/cli/commands/index-cmd.d.ts.map +1 -1
  3. package/dist/cli/commands/index-cmd.js +83 -14
  4. package/dist/cli/commands/index-cmd.js.map +1 -1
  5. package/dist/cli/commands/install.d.ts.map +1 -1
  6. package/dist/cli/commands/install.js +290 -171
  7. package/dist/cli/commands/install.js.map +1 -1
  8. package/dist/core/ExecutionEngine.d.ts +3 -1
  9. package/dist/core/ExecutionEngine.d.ts.map +1 -1
  10. package/dist/core/ExecutionEngine.js +36 -17
  11. package/dist/core/ExecutionEngine.js.map +1 -1
  12. package/dist/db/DatabaseManager.d.ts +4 -0
  13. package/dist/db/DatabaseManager.d.ts.map +1 -1
  14. package/dist/db/DatabaseManager.js +13 -0
  15. package/dist/db/DatabaseManager.js.map +1 -1
  16. package/dist/db/schema.d.ts +1 -1
  17. package/dist/db/schema.d.ts.map +1 -1
  18. package/dist/db/schema.js +20 -0
  19. package/dist/db/schema.js.map +1 -1
  20. package/dist/git/GitManager.d.ts +10 -0
  21. package/dist/git/GitManager.d.ts.map +1 -1
  22. package/dist/git/GitManager.js +46 -0
  23. package/dist/git/GitManager.js.map +1 -1
  24. package/dist/graph/GraphStore.d.ts +6 -1
  25. package/dist/graph/GraphStore.d.ts.map +1 -1
  26. package/dist/graph/InMemoryGraphStore.d.ts +1 -0
  27. package/dist/graph/InMemoryGraphStore.d.ts.map +1 -1
  28. package/dist/graph/InMemoryGraphStore.js +13 -2
  29. package/dist/graph/InMemoryGraphStore.js.map +1 -1
  30. package/dist/graph/SQLiteGraphStore.d.ts.map +1 -1
  31. package/dist/graph/SQLiteGraphStore.js +8 -2
  32. package/dist/graph/SQLiteGraphStore.js.map +1 -1
  33. package/dist/indexing/ASTParser.d.ts +1 -1
  34. package/dist/indexing/ASTParser.d.ts.map +1 -1
  35. package/dist/indexing/ASTParser.js +7 -1
  36. package/dist/indexing/ASTParser.js.map +1 -1
  37. package/dist/indexing/EmbeddingGenerator.d.ts.map +1 -1
  38. package/dist/indexing/EmbeddingGenerator.js +2 -25
  39. package/dist/indexing/EmbeddingGenerator.js.map +1 -1
  40. package/dist/indexing/FileScanner.d.ts.map +1 -1
  41. package/dist/indexing/FileScanner.js +2 -1
  42. package/dist/indexing/FileScanner.js.map +1 -1
  43. package/dist/indexing/GitIndexer.d.ts +4 -0
  44. package/dist/indexing/GitIndexer.d.ts.map +1 -0
  45. package/dist/indexing/GitIndexer.js +59 -0
  46. package/dist/indexing/GitIndexer.js.map +1 -0
  47. package/dist/indexing/GraphBuilder.d.ts.map +1 -1
  48. package/dist/indexing/GraphBuilder.js +21 -1
  49. package/dist/indexing/GraphBuilder.js.map +1 -1
  50. package/dist/indexing/Indexer.d.ts +3 -2
  51. package/dist/indexing/Indexer.d.ts.map +1 -1
  52. package/dist/indexing/Indexer.js +25 -6
  53. package/dist/indexing/Indexer.js.map +1 -1
  54. package/dist/mcp/server.d.ts.map +1 -1
  55. package/dist/mcp/server.js +20 -7
  56. package/dist/mcp/server.js.map +1 -1
  57. package/dist/mcp/tools/execute.d.ts +3 -0
  58. package/dist/mcp/tools/execute.d.ts.map +1 -1
  59. package/dist/mcp/tools/execute.js +2 -0
  60. package/dist/mcp/tools/execute.js.map +1 -1
  61. package/dist/mcp/tools/find_similar.d.ts +17 -0
  62. package/dist/mcp/tools/find_similar.d.ts.map +1 -0
  63. package/dist/mcp/tools/find_similar.js +95 -0
  64. package/dist/mcp/tools/find_similar.js.map +1 -0
  65. package/dist/mcp/tools/get_context.d.ts +3 -0
  66. package/dist/mcp/tools/get_context.d.ts.map +1 -1
  67. package/dist/mcp/tools/get_context.js +68 -7
  68. package/dist/mcp/tools/get_context.js.map +1 -1
  69. package/dist/mcp/tools/search_code.d.ts.map +1 -1
  70. package/dist/mcp/tools/search_code.js +22 -4
  71. package/dist/mcp/tools/search_code.js.map +1 -1
  72. package/dist/mcp/tools/smart_search.d.ts.map +1 -1
  73. package/dist/mcp/tools/smart_search.js +122 -29
  74. package/dist/mcp/tools/smart_search.js.map +1 -1
  75. package/dist/memory/HybridMemory.d.ts +4 -0
  76. package/dist/memory/HybridMemory.d.ts.map +1 -1
  77. package/dist/memory/HybridMemory.js +120 -29
  78. package/dist/memory/HybridMemory.js.map +1 -1
  79. package/dist/utils/modelManager.d.ts +37 -0
  80. package/dist/utils/modelManager.d.ts.map +1 -0
  81. package/dist/utils/modelManager.js +90 -0
  82. package/dist/utils/modelManager.js.map +1 -0
  83. package/dist/utils/workerPool.d.ts +8 -0
  84. package/dist/utils/workerPool.d.ts.map +1 -0
  85. package/dist/utils/workerPool.js +75 -0
  86. package/dist/utils/workerPool.js.map +1 -0
  87. package/dist/vector/MemoryVectorStore.d.ts.map +1 -1
  88. package/dist/vector/MemoryVectorStore.js +19 -13
  89. package/dist/vector/MemoryVectorStore.js.map +1 -1
  90. package/dist/vector/SQLiteVectorStore.d.ts +13 -0
  91. package/dist/vector/SQLiteVectorStore.d.ts.map +1 -0
  92. package/dist/vector/SQLiteVectorStore.js +111 -0
  93. package/dist/vector/SQLiteVectorStore.js.map +1 -0
  94. package/dist/vector/VectorStore.d.ts +0 -1
  95. package/dist/vector/VectorStore.d.ts.map +1 -1
  96. package/dist/workers/embeddingWorker.d.ts +2 -0
  97. package/dist/workers/embeddingWorker.d.ts.map +1 -0
  98. package/dist/workers/embeddingWorker.js +42 -0
  99. package/dist/workers/embeddingWorker.js.map +1 -0
  100. package/package.json +3 -1
  101. package/plugin/.claude-plugin/plugin.json +1 -1
@@ -0,0 +1,111 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.SQLiteVectorStore = void 0;
4
+ const modelManager_js_1 = require("../utils/modelManager.js");
5
+ const logger_js_1 = require("../utils/logger.js");
6
+ async function getEmbedding(text) {
7
+ try {
8
+ const embeddings = await (0, modelManager_js_1.embedTexts)([text]);
9
+ return embeddings[0] ?? [];
10
+ }
11
+ catch {
12
+ return [];
13
+ }
14
+ }
15
+ function cosineSimilarity(a, b) {
16
+ if (a.length !== b.length || a.length === 0)
17
+ return 0;
18
+ let dot = 0, normA = 0, normB = 0;
19
+ for (let i = 0; i < a.length; i++) {
20
+ dot += a[i] * b[i];
21
+ normA += a[i] * a[i];
22
+ normB += b[i] * b[i];
23
+ }
24
+ const denom = Math.sqrt(normA) * Math.sqrt(normB);
25
+ return denom === 0 ? 0 : dot / denom;
26
+ }
27
+ function textSimilarity(query, text) {
28
+ const queryTerms = new Set(query.toLowerCase().split(/\W+/).filter(Boolean));
29
+ const textTerms = text.toLowerCase().split(/\W+/).filter(Boolean);
30
+ if (queryTerms.size === 0 || textTerms.length === 0)
31
+ return 0;
32
+ const matches = textTerms.filter(t => queryTerms.has(t)).length;
33
+ return matches / (queryTerms.size + textTerms.length - matches);
34
+ }
35
+ class SQLiteVectorStore {
36
+ db;
37
+ project;
38
+ constructor(db, project) {
39
+ this.db = db;
40
+ this.project = project;
41
+ }
42
+ async add(documents) {
43
+ const insert = this.db.prepare(`
44
+ INSERT OR REPLACE INTO vector_chunks (id, project, text, metadata, embedding)
45
+ VALUES (?, ?, ?, ?, ?)
46
+ `);
47
+ const insertMany = this.db.transaction((docs) => {
48
+ for (const doc of docs) {
49
+ insert.run(doc.id, this.project, doc.text, JSON.stringify(doc.metadata), doc.embedding && doc.embedding.length > 0 ? JSON.stringify(doc.embedding) : null);
50
+ }
51
+ });
52
+ insertMany(documents);
53
+ logger_js_1.logger.debug({ count: documents.length, project: this.project }, 'Vector chunks upserted to SQLite');
54
+ }
55
+ async search(query, k = 5) {
56
+ const rows = this.db.prepare(`SELECT id, text, metadata, embedding FROM vector_chunks WHERE project = ?`).all(this.project);
57
+ if (rows.length === 0)
58
+ return [];
59
+ const withEmbeddings = rows.filter(r => r.embedding !== null);
60
+ let scored;
61
+ if (withEmbeddings.length > 0) {
62
+ const queryEmbedding = await getEmbedding(query);
63
+ if (queryEmbedding.length > 0) {
64
+ scored = withEmbeddings.map(row => ({
65
+ document: {
66
+ id: row.id,
67
+ text: row.text,
68
+ metadata: JSON.parse(row.metadata),
69
+ embedding: JSON.parse(row.embedding),
70
+ },
71
+ score: cosineSimilarity(queryEmbedding, JSON.parse(row.embedding)),
72
+ }));
73
+ }
74
+ else {
75
+ scored = rows.map(row => ({
76
+ document: {
77
+ id: row.id,
78
+ text: row.text,
79
+ metadata: JSON.parse(row.metadata),
80
+ },
81
+ score: textSimilarity(query, row.text),
82
+ }));
83
+ }
84
+ }
85
+ else {
86
+ scored = rows.map(row => ({
87
+ document: {
88
+ id: row.id,
89
+ text: row.text,
90
+ metadata: JSON.parse(row.metadata),
91
+ },
92
+ score: textSimilarity(query, row.text),
93
+ }));
94
+ }
95
+ return scored.sort((a, b) => b.score - a.score).slice(0, k);
96
+ }
97
+ async delete(ids) {
98
+ if (ids.length === 0)
99
+ return;
100
+ const ph = ids.map(() => '?').join(',');
101
+ this.db.prepare(`DELETE FROM vector_chunks WHERE project = ? AND id IN (${ph})`).run(this.project, ...ids);
102
+ }
103
+ async clear() {
104
+ this.db.prepare(`DELETE FROM vector_chunks WHERE project = ?`).run(this.project);
105
+ }
106
+ size() {
107
+ return this.db.prepare(`SELECT COUNT(*) as c FROM vector_chunks WHERE project = ?`).get(this.project).c;
108
+ }
109
+ }
110
+ exports.SQLiteVectorStore = SQLiteVectorStore;
111
+ //# sourceMappingURL=SQLiteVectorStore.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"SQLiteVectorStore.js","sourceRoot":"","sources":["../../src/vector/SQLiteVectorStore.ts"],"names":[],"mappings":";;;AAEA,8DAAsD;AACtD,kDAA4C;AAE5C,KAAK,UAAU,YAAY,CAAC,IAAY;IACtC,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,IAAA,4BAAU,EAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5C,OAAO,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,CAAW,EAAE,CAAW;IAChD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IACtD,IAAI,GAAG,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,KAAK,CAAC;AACvC,CAAC;AAED,SAAS,cAAc,CAAC,KAAa,EAAE,IAAY;IACjD,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IAC7E,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,IAAI,UAAU,CAAC,IAAI,KAAK,CAAC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAC9D,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAChE,OAAO,OAAO,GAAG,CAAC,UAAU,CAAC,IAAI,GAAG,SAAS,CAAC,MAAM,GAAG,OAAO,CAAC,CAAC;AAClE,CAAC;AASD,MAAa,iBAAiB;IACR;IAA+B;IAAnD,YAAoB,EAAqB,EAAU,OAAe;QAA9C,OAAE,GAAF,EAAE,CAAmB;QAAU,YAAO,GAAP,OAAO,CAAQ;IAAG,CAAC;IAEtE,KAAK,CAAC,GAAG,CAAC,SAAqB;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;KAG9B,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAgB,EAAE,EAAE;YAC1D,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,MAAM,CAAC,GAAG,CACR,GAAG,CAAC,EAAE,EACN,IAAI,CAAC,OAAO,EACZ,GAAG,CAAC,IAAI,EACR,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,EAC5B,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CACjF,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,SAAS,CAAC,CAAC;QACtB,kBAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,kCAAkC,CAAC,CAAC;IACvG,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,CAAC,GAAG,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,2EAA2E,CAC5E,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAgB,CAAC;QAEnC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEjC,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC;QAE9D,IAAI,MAAwB,CAAC;QAE7B,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,cAAc,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,CAAC;YACjD,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBAClC,QAAQ,EAAE;wBACR,EAAE,EAAE,GAAG,CAAC,EAAE;wBACV,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAA4B;wBAC7D,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAU,CAAa;qBAClD;oBACD,KAAK,EAAE,gBAAgB,CAAC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAU,CAAa,CAAC;iBAChF,CAAC,CAAC,CAAC;YACN,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACxB,QAAQ,EAAE;wBACR,EAAE,EAAE,GAAG,CAAC,EAAE;wBACV,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAA4B;qBAC9D;oBACD,KAAK,EAAE,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC;iBACvC,CAAC,CAAC,CAAC;YACN,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACxB,QAAQ,EAAE;oBACR,EAAE,EAAE,GAAG,CAAC,EAAE;oBACV,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAA4B;iBAC9D;gBACD,KAAK,EAAE,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC;aACvC,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAa;QACxB,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAC7B,MAAM,EAAE,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,0DAA0D,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,GAAG,CAAC,CAAC;IAC7G,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACnF,CAAC;IAED,IAAI;QACF,OAAQ,IAAI,CAAC,EAAE,CAAC,OAAO,CACrB,2DAA2D,CAC5D,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAmB,CAAC,CAAC,CAAC;IAC1C,CAAC;CACF;AArFD,8CAqFC"}
@@ -14,6 +14,5 @@ export interface VectorStore {
14
14
  delete(ids: string[]): Promise<void>;
15
15
  clear(): Promise<void>;
16
16
  size(): number;
17
- serialize(): string;
18
17
  }
19
18
  //# sourceMappingURL=VectorStore.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"VectorStore.d.ts","sourceRoot":"","sources":["../../src/vector/VectorStore.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAC7D,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACrC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,MAAM,CAAC;IACf,SAAS,IAAI,MAAM,CAAC;CACrB"}
1
+ {"version":3,"file":"VectorStore.d.ts","sourceRoot":"","sources":["../../src/vector/VectorStore.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAC7D,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACrC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,MAAM,CAAC;CAChB"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=embeddingWorker.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"embeddingWorker.d.ts","sourceRoot":"","sources":["../../src/workers/embeddingWorker.ts"],"names":[],"mappings":""}
@@ -0,0 +1,42 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ const worker_threads_1 = require("worker_threads");
4
+ const MODEL_ID = 'Xenova/all-MiniLM-L6-v2';
5
+ let pipe = null;
6
+ async function init() {
7
+ try {
8
+ const { pipeline, env } = await import('@xenova/transformers');
9
+ if (process.env.FIRECODE_MODEL_CACHE) {
10
+ env.cacheDir = process.env.FIRECODE_MODEL_CACHE;
11
+ }
12
+ pipe = (await pipeline('feature-extraction', MODEL_ID));
13
+ worker_threads_1.parentPort.postMessage({ type: 'ready' });
14
+ }
15
+ catch (err) {
16
+ worker_threads_1.parentPort.postMessage({ type: 'error', error: String(err) });
17
+ }
18
+ }
19
+ worker_threads_1.parentPort.on('message', async (req) => {
20
+ if (!pipe) {
21
+ worker_threads_1.parentPort.postMessage({
22
+ id: req.id,
23
+ embeddings: [],
24
+ error: 'Model not ready',
25
+ });
26
+ return;
27
+ }
28
+ try {
29
+ const tensor = await pipe(req.texts, { pooling: 'mean', normalize: true });
30
+ const [batch, dim] = tensor.dims;
31
+ const embeddings = [];
32
+ for (let i = 0; i < batch; i++) {
33
+ embeddings.push(Array.from(tensor.data.slice(i * dim, (i + 1) * dim)));
34
+ }
35
+ worker_threads_1.parentPort.postMessage({ id: req.id, embeddings });
36
+ }
37
+ catch (err) {
38
+ worker_threads_1.parentPort.postMessage({ id: req.id, embeddings: [], error: String(err) });
39
+ }
40
+ });
41
+ init();
42
+ //# sourceMappingURL=embeddingWorker.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"embeddingWorker.js","sourceRoot":"","sources":["../../src/workers/embeddingWorker.ts"],"names":[],"mappings":";;AAAA,mDAA4C;AAgB5C,MAAM,QAAQ,GAAG,yBAAyB,CAAC;AAO3C,IAAI,IAAI,GAAoB,IAAI,CAAC;AAEjC,KAAK,UAAU,IAAI;IACjB,IAAI,CAAC;QACH,MAAM,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;QAC/D,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,CAAC;YACpC,GAA6B,CAAC,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;QAC7E,CAAC;QACD,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,oBAAoB,EAAE,QAAQ,CAAC,CAAwB,CAAC;QAC/E,2BAAW,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC7C,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,2BAAW,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;AACH,CAAC;AAED,2BAAW,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,GAAiB,EAAE,EAAE;IACpD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,2BAAW,CAAC,WAAW,CAAC;YACtB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,UAAU,EAAE,EAAE;YACd,KAAK,EAAE,iBAAiB;SACR,CAAC,CAAC;QACpB,OAAO;IACT,CAAC;IACD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3E,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC;QACjC,MAAM,UAAU,GAAe,EAAE,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QACzE,CAAC;QACD,2BAAW,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,UAAU,EAAmB,CAAC,CAAC;IACvE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,2BAAW,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAmB,CAAC,CAAC;IAC/F,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,EAAE,CAAC"}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@phmotad/fire-code",
3
- "version": "0.2.7",
3
+ "version": "0.2.9",
4
4
  "description": "Intelligent MCP execution engine for AI coding agents — persistent memory, context-aware, Git-traceable",
5
5
  "keywords": [
6
6
  "mcp",
@@ -55,6 +55,8 @@
55
55
  "test": "jest",
56
56
  "test:unit": "jest --testPathPattern=__tests__/unit",
57
57
  "test:integration": "jest --testPathPattern=__tests__/integration",
58
+ "test:bench": "jest --testPathPattern=__tests__/benchmark --testMatch='**/__tests__/**/*.bench.ts' --verbose --testTimeout=30000 --runInBand",
59
+ "bench:emb": "tsx scripts/benchmark-embeddings.ts",
58
60
  "test:coverage": "jest --coverage",
59
61
  "lint": "tsc --noEmit",
60
62
  "prepublishOnly": "npm run build && npm test",
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "fire-code",
3
- "version": "0.2.7",
3
+ "version": "0.2.8",
4
4
  "description": "Intelligent MCP execution engine for AI coding agents — persistent memory, context-aware, Git-traceable",
5
5
  "author": {
6
6
  "name": "FireCode Contributors",