@phmotad/fire-code 0.2.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/LICENSE +21 -0
- package/README.md +379 -0
- package/dist/agents/BaseAgent.d.ts +32 -0
- package/dist/agents/BaseAgent.d.ts.map +1 -0
- package/dist/agents/BaseAgent.js +19 -0
- package/dist/agents/BaseAgent.js.map +1 -0
- package/dist/agents/CodeAgent.d.ts +7 -0
- package/dist/agents/CodeAgent.d.ts.map +1 -0
- package/dist/agents/CodeAgent.js +130 -0
- package/dist/agents/CodeAgent.js.map +1 -0
- package/dist/cli/commands/corpus.d.ts +20 -0
- package/dist/cli/commands/corpus.d.ts.map +1 -0
- package/dist/cli/commands/corpus.js +60 -0
- package/dist/cli/commands/corpus.js.map +1 -0
- package/dist/cli/commands/daemon.d.ts +5 -0
- package/dist/cli/commands/daemon.d.ts.map +1 -0
- package/dist/cli/commands/daemon.js +79 -0
- package/dist/cli/commands/daemon.js.map +1 -0
- package/dist/cli/commands/dev.d.ts +2 -0
- package/dist/cli/commands/dev.d.ts.map +1 -0
- package/dist/cli/commands/dev.js +20 -0
- package/dist/cli/commands/dev.js.map +1 -0
- package/dist/cli/commands/index-cmd.d.ts +6 -0
- package/dist/cli/commands/index-cmd.d.ts.map +1 -0
- package/dist/cli/commands/index-cmd.js +57 -0
- package/dist/cli/commands/index-cmd.js.map +1 -0
- package/dist/cli/commands/init.d.ts +2 -0
- package/dist/cli/commands/init.d.ts.map +1 -0
- package/dist/cli/commands/init.js +173 -0
- package/dist/cli/commands/init.js.map +1 -0
- package/dist/cli/commands/install.d.ts +8 -0
- package/dist/cli/commands/install.d.ts.map +1 -0
- package/dist/cli/commands/install.js +154 -0
- package/dist/cli/commands/install.js.map +1 -0
- package/dist/cli/commands/observe.d.ts +19 -0
- package/dist/cli/commands/observe.d.ts.map +1 -0
- package/dist/cli/commands/observe.js +86 -0
- package/dist/cli/commands/observe.js.map +1 -0
- package/dist/cli/index.d.ts +3 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +162 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/config/defaults.d.ts +3 -0
- package/dist/config/defaults.d.ts.map +1 -0
- package/dist/config/defaults.js +8 -0
- package/dist/config/defaults.js.map +1 -0
- package/dist/config/loader.d.ts +3 -0
- package/dist/config/loader.d.ts.map +1 -0
- package/dist/config/loader.js +96 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/config/types.d.ts +436 -0
- package/dist/config/types.d.ts.map +1 -0
- package/dist/config/types.js +73 -0
- package/dist/config/types.js.map +1 -0
- package/dist/core/ExecutionEngine.d.ts +26 -0
- package/dist/core/ExecutionEngine.d.ts.map +1 -0
- package/dist/core/ExecutionEngine.js +102 -0
- package/dist/core/ExecutionEngine.js.map +1 -0
- package/dist/core/TaskRouter.d.ts +3 -0
- package/dist/core/TaskRouter.d.ts.map +1 -0
- package/dist/core/TaskRouter.js +29 -0
- package/dist/core/TaskRouter.js.map +1 -0
- package/dist/daemon/DaemonClient.d.ts +8 -0
- package/dist/daemon/DaemonClient.d.ts.map +1 -0
- package/dist/daemon/DaemonClient.js +61 -0
- package/dist/daemon/DaemonClient.js.map +1 -0
- package/dist/daemon/DaemonServer.d.ts +17 -0
- package/dist/daemon/DaemonServer.d.ts.map +1 -0
- package/dist/daemon/DaemonServer.js +235 -0
- package/dist/daemon/DaemonServer.js.map +1 -0
- package/dist/db/DatabaseManager.d.ts +79 -0
- package/dist/db/DatabaseManager.d.ts.map +1 -0
- package/dist/db/DatabaseManager.js +212 -0
- package/dist/db/DatabaseManager.js.map +1 -0
- package/dist/db/schema.d.ts +2 -0
- package/dist/db/schema.d.ts.map +1 -0
- package/dist/db/schema.js +127 -0
- package/dist/db/schema.js.map +1 -0
- package/dist/git/CommitFormatter.d.ts +19 -0
- package/dist/git/CommitFormatter.d.ts.map +1 -0
- package/dist/git/CommitFormatter.js +47 -0
- package/dist/git/CommitFormatter.js.map +1 -0
- package/dist/git/GitManager.d.ts +28 -0
- package/dist/git/GitManager.d.ts.map +1 -0
- package/dist/git/GitManager.js +147 -0
- package/dist/git/GitManager.js.map +1 -0
- package/dist/graph/GraphStore.d.ts +54 -0
- package/dist/graph/GraphStore.d.ts.map +1 -0
- package/dist/graph/GraphStore.js +3 -0
- package/dist/graph/GraphStore.js.map +1 -0
- package/dist/graph/InMemoryGraphStore.d.ts +16 -0
- package/dist/graph/InMemoryGraphStore.d.ts.map +1 -0
- package/dist/graph/InMemoryGraphStore.js +75 -0
- package/dist/graph/InMemoryGraphStore.js.map +1 -0
- package/dist/graph/SQLiteGraphStore.d.ts +31 -0
- package/dist/graph/SQLiteGraphStore.d.ts.map +1 -0
- package/dist/graph/SQLiteGraphStore.js +158 -0
- package/dist/graph/SQLiteGraphStore.js.map +1 -0
- package/dist/index.d.ts +16 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +32 -0
- package/dist/index.js.map +1 -0
- package/dist/indexing/ASTParser.d.ts +32 -0
- package/dist/indexing/ASTParser.d.ts.map +1 -0
- package/dist/indexing/ASTParser.js +185 -0
- package/dist/indexing/ASTParser.js.map +1 -0
- package/dist/indexing/EmbeddingGenerator.d.ts +15 -0
- package/dist/indexing/EmbeddingGenerator.d.ts.map +1 -0
- package/dist/indexing/EmbeddingGenerator.js +66 -0
- package/dist/indexing/EmbeddingGenerator.js.map +1 -0
- package/dist/indexing/FileScanner.d.ts +10 -0
- package/dist/indexing/FileScanner.d.ts.map +1 -0
- package/dist/indexing/FileScanner.js +58 -0
- package/dist/indexing/FileScanner.js.map +1 -0
- package/dist/indexing/GraphBuilder.d.ts +4 -0
- package/dist/indexing/GraphBuilder.d.ts.map +1 -0
- package/dist/indexing/GraphBuilder.js +51 -0
- package/dist/indexing/GraphBuilder.js.map +1 -0
- package/dist/indexing/Indexer.d.ts +13 -0
- package/dist/indexing/Indexer.d.ts.map +1 -0
- package/dist/indexing/Indexer.js +68 -0
- package/dist/indexing/Indexer.js.map +1 -0
- package/dist/mcp/server.d.ts +2 -0
- package/dist/mcp/server.d.ts.map +1 -0
- package/dist/mcp/server.js +155 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/mcp/tools/corpus_search.d.ts +14 -0
- package/dist/mcp/tools/corpus_search.d.ts.map +1 -0
- package/dist/mcp/tools/corpus_search.js +33 -0
- package/dist/mcp/tools/corpus_search.js.map +1 -0
- package/dist/mcp/tools/execute.d.ts +19 -0
- package/dist/mcp/tools/execute.d.ts.map +1 -0
- package/dist/mcp/tools/execute.js +38 -0
- package/dist/mcp/tools/execute.js.map +1 -0
- package/dist/mcp/tools/get_context.d.ts +17 -0
- package/dist/mcp/tools/get_context.d.ts.map +1 -0
- package/dist/mcp/tools/get_context.js +50 -0
- package/dist/mcp/tools/get_context.js.map +1 -0
- package/dist/mcp/tools/get_graph.d.ts +26 -0
- package/dist/mcp/tools/get_graph.d.ts.map +1 -0
- package/dist/mcp/tools/get_graph.js +68 -0
- package/dist/mcp/tools/get_graph.js.map +1 -0
- package/dist/mcp/tools/observations.d.ts +23 -0
- package/dist/mcp/tools/observations.d.ts.map +1 -0
- package/dist/mcp/tools/observations.js +72 -0
- package/dist/mcp/tools/observations.js.map +1 -0
- package/dist/mcp/tools/search_code.d.ts +14 -0
- package/dist/mcp/tools/search_code.d.ts.map +1 -0
- package/dist/mcp/tools/search_code.js +38 -0
- package/dist/mcp/tools/search_code.js.map +1 -0
- package/dist/mcp/tools/smart_outline.d.ts +14 -0
- package/dist/mcp/tools/smart_outline.d.ts.map +1 -0
- package/dist/mcp/tools/smart_outline.js +99 -0
- package/dist/mcp/tools/smart_outline.js.map +1 -0
- package/dist/mcp/tools/smart_search.d.ts +20 -0
- package/dist/mcp/tools/smart_search.d.ts.map +1 -0
- package/dist/mcp/tools/smart_search.js +131 -0
- package/dist/mcp/tools/smart_search.js.map +1 -0
- package/dist/memory/FallbackMemory.d.ts +9 -0
- package/dist/memory/FallbackMemory.d.ts.map +1 -0
- package/dist/memory/FallbackMemory.js +70 -0
- package/dist/memory/FallbackMemory.js.map +1 -0
- package/dist/memory/HybridMemory.d.ts +20 -0
- package/dist/memory/HybridMemory.d.ts.map +1 -0
- package/dist/memory/HybridMemory.js +74 -0
- package/dist/memory/HybridMemory.js.map +1 -0
- package/dist/providers/AgentSDKProvider.d.ts +13 -0
- package/dist/providers/AgentSDKProvider.d.ts.map +1 -0
- package/dist/providers/AgentSDKProvider.js +64 -0
- package/dist/providers/AgentSDKProvider.js.map +1 -0
- package/dist/providers/AnthropicProvider.d.ts +11 -0
- package/dist/providers/AnthropicProvider.d.ts.map +1 -0
- package/dist/providers/AnthropicProvider.js +61 -0
- package/dist/providers/AnthropicProvider.js.map +1 -0
- package/dist/providers/LLMProvider.d.ts +12 -0
- package/dist/providers/LLMProvider.d.ts.map +1 -0
- package/dist/providers/LLMProvider.js +3 -0
- package/dist/providers/LLMProvider.js.map +1 -0
- package/dist/providers/OllamaProvider.d.ts +11 -0
- package/dist/providers/OllamaProvider.d.ts.map +1 -0
- package/dist/providers/OllamaProvider.js +71 -0
- package/dist/providers/OllamaProvider.js.map +1 -0
- package/dist/providers/OpenAIProvider.d.ts +11 -0
- package/dist/providers/OpenAIProvider.d.ts.map +1 -0
- package/dist/providers/OpenAIProvider.js +63 -0
- package/dist/providers/OpenAIProvider.js.map +1 -0
- package/dist/providers/OpenRouterProvider.d.ts +12 -0
- package/dist/providers/OpenRouterProvider.d.ts.map +1 -0
- package/dist/providers/OpenRouterProvider.js +89 -0
- package/dist/providers/OpenRouterProvider.js.map +1 -0
- package/dist/providers/ProviderFactory.d.ts +4 -0
- package/dist/providers/ProviderFactory.d.ts.map +1 -0
- package/dist/providers/ProviderFactory.js +27 -0
- package/dist/providers/ProviderFactory.js.map +1 -0
- package/dist/services/CorpusService.d.ts +27 -0
- package/dist/services/CorpusService.d.ts.map +1 -0
- package/dist/services/CorpusService.js +142 -0
- package/dist/services/CorpusService.js.map +1 -0
- package/dist/services/ObservationService.d.ts +17 -0
- package/dist/services/ObservationService.d.ts.map +1 -0
- package/dist/services/ObservationService.js +101 -0
- package/dist/services/ObservationService.js.map +1 -0
- package/dist/services/SessionService.d.ts +10 -0
- package/dist/services/SessionService.d.ts.map +1 -0
- package/dist/services/SessionService.js +38 -0
- package/dist/services/SessionService.js.map +1 -0
- package/dist/services/SummarizationService.d.ts +8 -0
- package/dist/services/SummarizationService.d.ts.map +1 -0
- package/dist/services/SummarizationService.js +65 -0
- package/dist/services/SummarizationService.js.map +1 -0
- package/dist/utils/errors.d.ts +26 -0
- package/dist/utils/errors.d.ts.map +1 -0
- package/dist/utils/errors.js +72 -0
- package/dist/utils/errors.js.map +1 -0
- package/dist/utils/logger.d.ts +4 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +16 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/paths.d.ts +10 -0
- package/dist/utils/paths.d.ts.map +1 -0
- package/dist/utils/paths.js +43 -0
- package/dist/utils/paths.js.map +1 -0
- package/dist/utils/privacy.d.ts +5 -0
- package/dist/utils/privacy.d.ts.map +1 -0
- package/dist/utils/privacy.js +39 -0
- package/dist/utils/privacy.js.map +1 -0
- package/dist/utils/zodToJsonSchema.d.ts +23 -0
- package/dist/utils/zodToJsonSchema.d.ts.map +1 -0
- package/dist/utils/zodToJsonSchema.js +52 -0
- package/dist/utils/zodToJsonSchema.js.map +1 -0
- package/dist/vector/MemoryVectorStore.d.ts +16 -0
- package/dist/vector/MemoryVectorStore.d.ts.map +1 -0
- package/dist/vector/MemoryVectorStore.js +97 -0
- package/dist/vector/MemoryVectorStore.js.map +1 -0
- package/dist/vector/VectorStore.d.ts +19 -0
- package/dist/vector/VectorStore.d.ts.map +1 -0
- package/dist/vector/VectorStore.js +3 -0
- package/dist/vector/VectorStore.js.map +1 -0
- package/dist/vector/VectraVectorStore.d.ts +23 -0
- package/dist/vector/VectraVectorStore.d.ts.map +1 -0
- package/dist/vector/VectraVectorStore.js +93 -0
- package/dist/vector/VectraVectorStore.js.map +1 -0
- package/examples/firecode.config.example.ts +87 -0
- package/package.json +111 -0
- package/plugin/.claude-plugin/plugin.json +17 -0
- package/plugin/hooks/hooks.json +86 -0
- package/plugin/scripts/context-inject.js +208 -0
- package/plugin/scripts/smart-install.js +33 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AnthropicProvider.d.ts","sourceRoot":"","sources":["../../src/providers/AnthropicProvider.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAEvE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAEpD,qBAAa,iBAAkB,YAAW,WAAW;IACnD,QAAQ,CAAC,IAAI,eAAe;IAC5B,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,EAAE,SAAS;IAMvB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAAsB,GAAG,OAAO,CAAC,MAAM,CAAC;IAmBtE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAAsB,GAAG,aAAa,CAAC,MAAM,CAAC;CAsBnF"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.AnthropicProvider = void 0;
|
|
7
|
+
const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
|
|
8
|
+
const errors_js_1 = require("../utils/errors.js");
|
|
9
|
+
class AnthropicProvider {
|
|
10
|
+
name = 'anthropic';
|
|
11
|
+
client;
|
|
12
|
+
model;
|
|
13
|
+
constructor(config) {
|
|
14
|
+
if (!config.apiKey)
|
|
15
|
+
throw new errors_js_1.ProviderError('Anthropic requires an API key (ANTHROPIC_API_KEY)');
|
|
16
|
+
this.client = new sdk_1.default({ apiKey: config.apiKey });
|
|
17
|
+
this.model = config.model ?? 'claude-sonnet-4-6';
|
|
18
|
+
}
|
|
19
|
+
async complete(prompt, opts = {}) {
|
|
20
|
+
try {
|
|
21
|
+
const msg = await this.client.messages.create({
|
|
22
|
+
model: this.model,
|
|
23
|
+
max_tokens: opts.maxTokens ?? 4096,
|
|
24
|
+
system: opts.systemPrompt,
|
|
25
|
+
messages: [{ role: 'user', content: prompt }],
|
|
26
|
+
temperature: opts.temperature ?? 0.2,
|
|
27
|
+
});
|
|
28
|
+
const block = msg.content[0];
|
|
29
|
+
if (block.type !== 'text')
|
|
30
|
+
throw new errors_js_1.ProviderError('Unexpected response type from Anthropic');
|
|
31
|
+
return block.text;
|
|
32
|
+
}
|
|
33
|
+
catch (err) {
|
|
34
|
+
if (err instanceof errors_js_1.ProviderError)
|
|
35
|
+
throw err;
|
|
36
|
+
throw new errors_js_1.ProviderError(`Anthropic request failed: ${String(err)}`);
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
async *stream(prompt, opts = {}) {
|
|
40
|
+
try {
|
|
41
|
+
const stream = this.client.messages.stream({
|
|
42
|
+
model: this.model,
|
|
43
|
+
max_tokens: opts.maxTokens ?? 4096,
|
|
44
|
+
system: opts.systemPrompt,
|
|
45
|
+
messages: [{ role: 'user', content: prompt }],
|
|
46
|
+
temperature: opts.temperature ?? 0.2,
|
|
47
|
+
});
|
|
48
|
+
for await (const event of stream) {
|
|
49
|
+
if (event.type === 'content_block_delta' &&
|
|
50
|
+
event.delta.type === 'text_delta') {
|
|
51
|
+
yield event.delta.text;
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
catch (err) {
|
|
56
|
+
throw new errors_js_1.ProviderError(`Anthropic stream failed: ${String(err)}`);
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
exports.AnthropicProvider = AnthropicProvider;
|
|
61
|
+
//# sourceMappingURL=AnthropicProvider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"AnthropicProvider.js","sourceRoot":"","sources":["../../src/providers/AnthropicProvider.ts"],"names":[],"mappings":";;;;;;AAAA,4DAA0C;AAE1C,kDAAmD;AAGnD,MAAa,iBAAiB;IACnB,IAAI,GAAG,WAAW,CAAC;IACpB,MAAM,CAAY;IAClB,KAAK,CAAS;IAEtB,YAAY,MAAiB;QAC3B,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,MAAM,IAAI,yBAAa,CAAC,mDAAmD,CAAC,CAAC;QACjG,IAAI,CAAC,MAAM,GAAG,IAAI,aAAS,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QACvD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,mBAAmB,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAA0B,EAAE;QACzD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;gBAClC,MAAM,EAAE,IAAI,CAAC,YAAY;gBACzB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;gBAC7C,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,GAAG;aACrC,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;gBAAE,MAAM,IAAI,yBAAa,CAAC,yCAAyC,CAAC,CAAC;YAC9F,OAAO,KAAK,CAAC,IAAI,CAAC;QACpB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,yBAAa;gBAAE,MAAM,GAAG,CAAC;YAC5C,MAAM,IAAI,yBAAa,CAAC,6BAA6B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,MAAc,EAAE,OAA0B,EAAE;QACxD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACzC,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;gBAClC,MAAM,EAAE,IAAI,CAAC,YAAY;gBACzB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;gBAC7C,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,GAAG;aACrC,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,IACE,KAAK,CAAC,IAAI,KAAK,qBAAqB;oBACpC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,EACjC,CAAC;oBACD,MAAM,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,yBAAa,CAAC,4BAA4B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;CACF;AApDD,8CAoDC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
export interface CompletionOptions {
|
|
2
|
+
maxTokens?: number;
|
|
3
|
+
temperature?: number;
|
|
4
|
+
systemPrompt?: string;
|
|
5
|
+
stopSequences?: string[];
|
|
6
|
+
}
|
|
7
|
+
export interface LLMProvider {
|
|
8
|
+
readonly name: string;
|
|
9
|
+
complete(prompt: string, opts?: CompletionOptions): Promise<string>;
|
|
10
|
+
stream(prompt: string, opts?: CompletionOptions): AsyncIterable<string>;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=LLMProvider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"LLMProvider.d.ts","sourceRoot":"","sources":["../../src/providers/LLMProvider.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,iBAAiB;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;CAC1B;AAED,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACpE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,iBAAiB,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;CACzE"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"LLMProvider.js","sourceRoot":"","sources":["../../src/providers/LLMProvider.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { LLMProvider, CompletionOptions } from './LLMProvider.js';
|
|
2
|
+
import type { LLMConfig } from '../config/types.js';
|
|
3
|
+
export declare class OllamaProvider implements LLMProvider {
|
|
4
|
+
readonly name = "ollama";
|
|
5
|
+
private baseUrl;
|
|
6
|
+
private model;
|
|
7
|
+
constructor(config: LLMConfig);
|
|
8
|
+
complete(prompt: string, opts?: CompletionOptions): Promise<string>;
|
|
9
|
+
stream(prompt: string, opts?: CompletionOptions): AsyncIterable<string>;
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=OllamaProvider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OllamaProvider.d.ts","sourceRoot":"","sources":["../../src/providers/OllamaProvider.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAEvE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAEpD,qBAAa,cAAe,YAAW,WAAW;IAChD,QAAQ,CAAC,IAAI,YAAY;IACzB,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,EAAE,SAAS;IAKvB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAAsB,GAAG,OAAO,CAAC,MAAM,CAAC;IAsBtE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAAsB,GAAG,aAAa,CAAC,MAAM,CAAC;CA+BnF"}
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.OllamaProvider = void 0;
|
|
7
|
+
const axios_1 = __importDefault(require("axios"));
|
|
8
|
+
const errors_js_1 = require("../utils/errors.js");
|
|
9
|
+
class OllamaProvider {
|
|
10
|
+
name = 'ollama';
|
|
11
|
+
baseUrl;
|
|
12
|
+
model;
|
|
13
|
+
constructor(config) {
|
|
14
|
+
this.baseUrl = config.baseUrl ?? 'http://localhost:11434';
|
|
15
|
+
this.model = config.model ?? 'llama3';
|
|
16
|
+
}
|
|
17
|
+
async complete(prompt, opts = {}) {
|
|
18
|
+
const messages = [];
|
|
19
|
+
if (opts.systemPrompt)
|
|
20
|
+
messages.push({ role: 'system', content: opts.systemPrompt });
|
|
21
|
+
messages.push({ role: 'user', content: prompt });
|
|
22
|
+
try {
|
|
23
|
+
const res = await axios_1.default.post(`${this.baseUrl}/api/chat`, {
|
|
24
|
+
model: this.model,
|
|
25
|
+
messages,
|
|
26
|
+
stream: false,
|
|
27
|
+
options: {
|
|
28
|
+
num_predict: opts.maxTokens ?? 4096,
|
|
29
|
+
temperature: opts.temperature ?? 0.2,
|
|
30
|
+
stop: opts.stopSequences,
|
|
31
|
+
},
|
|
32
|
+
});
|
|
33
|
+
return res.data.message.content;
|
|
34
|
+
}
|
|
35
|
+
catch (err) {
|
|
36
|
+
throw new errors_js_1.ProviderError(`Ollama request failed (is Ollama running?): ${String(err)}`);
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
async *stream(prompt, opts = {}) {
|
|
40
|
+
const messages = [];
|
|
41
|
+
if (opts.systemPrompt)
|
|
42
|
+
messages.push({ role: 'system', content: opts.systemPrompt });
|
|
43
|
+
messages.push({ role: 'user', content: prompt });
|
|
44
|
+
try {
|
|
45
|
+
const res = await axios_1.default.post(`${this.baseUrl}/api/chat`, {
|
|
46
|
+
model: this.model,
|
|
47
|
+
messages,
|
|
48
|
+
stream: true,
|
|
49
|
+
options: { num_predict: opts.maxTokens ?? 4096, temperature: opts.temperature ?? 0.2 },
|
|
50
|
+
}, { responseType: 'stream' });
|
|
51
|
+
for await (const chunk of res.data) {
|
|
52
|
+
const lines = chunk.toString().split('\n').filter(Boolean);
|
|
53
|
+
for (const line of lines) {
|
|
54
|
+
try {
|
|
55
|
+
const parsed = JSON.parse(line);
|
|
56
|
+
if (parsed.done)
|
|
57
|
+
return;
|
|
58
|
+
if (parsed.message?.content)
|
|
59
|
+
yield parsed.message.content;
|
|
60
|
+
}
|
|
61
|
+
catch { /* skip */ }
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
catch (err) {
|
|
66
|
+
throw new errors_js_1.ProviderError(`Ollama stream failed: ${String(err)}`);
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
exports.OllamaProvider = OllamaProvider;
|
|
71
|
+
//# sourceMappingURL=OllamaProvider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OllamaProvider.js","sourceRoot":"","sources":["../../src/providers/OllamaProvider.ts"],"names":[],"mappings":";;;;;;AAAA,kDAA0B;AAE1B,kDAAmD;AAGnD,MAAa,cAAc;IAChB,IAAI,GAAG,QAAQ,CAAC;IACjB,OAAO,CAAS;IAChB,KAAK,CAAS;IAEtB,YAAY,MAAiB;QAC3B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,wBAAwB,CAAC;QAC1D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,QAAQ,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAA0B,EAAE;QACzD,MAAM,QAAQ,GAAG,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,YAAY;YAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QACrF,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAEjD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,eAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;gBACvD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ;gBACR,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE;oBACP,WAAW,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;oBACnC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,GAAG;oBACpC,IAAI,EAAE,IAAI,CAAC,aAAa;iBACzB;aACF,CAAC,CAAC;YACH,OAAQ,GAAG,CAAC,IAAyC,CAAC,OAAO,CAAC,OAAO,CAAC;QACxE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,yBAAa,CAAC,+CAA+C,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxF,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,MAAc,EAAE,OAA0B,EAAE;QACxD,MAAM,QAAQ,GAAG,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,YAAY;YAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QACrF,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAEjD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,eAAK,CAAC,IAAI,CAC1B,GAAG,IAAI,CAAC,OAAO,WAAW,EAC1B;gBACE,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ;gBACR,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,GAAG,EAAE;aACvF,EACD,EAAE,YAAY,EAAE,QAAQ,EAAE,CAC3B,CAAC;YAEF,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,CAAC,IAA6B,EAAE,CAAC;gBAC5D,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAC3D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAuD,CAAC;wBACtF,IAAI,MAAM,CAAC,IAAI;4BAAE,OAAO;wBACxB,IAAI,MAAM,CAAC,OAAO,EAAE,OAAO;4BAAE,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;oBAC5D,CAAC;oBAAC,MAAM,CAAC,CAAC,UAAU,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,yBAAa,CAAC,yBAAyB,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;CACF;AA/DD,wCA+DC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { LLMProvider, CompletionOptions } from './LLMProvider.js';
|
|
2
|
+
import type { LLMConfig } from '../config/types.js';
|
|
3
|
+
export declare class OpenAIProvider implements LLMProvider {
|
|
4
|
+
readonly name = "openai";
|
|
5
|
+
private client;
|
|
6
|
+
private model;
|
|
7
|
+
constructor(config: LLMConfig);
|
|
8
|
+
complete(prompt: string, opts?: CompletionOptions): Promise<string>;
|
|
9
|
+
stream(prompt: string, opts?: CompletionOptions): AsyncIterable<string>;
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=OpenAIProvider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OpenAIProvider.d.ts","sourceRoot":"","sources":["../../src/providers/OpenAIProvider.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAEvE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAEpD,qBAAa,cAAe,YAAW,WAAW;IAChD,QAAQ,CAAC,IAAI,YAAY;IACzB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,EAAE,SAAS;IAMvB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAAsB,GAAG,OAAO,CAAC,MAAM,CAAC;IAmBtE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAAsB,GAAG,aAAa,CAAC,MAAM,CAAC;CAsBnF"}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.OpenAIProvider = void 0;
|
|
7
|
+
const openai_1 = __importDefault(require("openai"));
|
|
8
|
+
const errors_js_1 = require("../utils/errors.js");
|
|
9
|
+
class OpenAIProvider {
|
|
10
|
+
name = 'openai';
|
|
11
|
+
client;
|
|
12
|
+
model;
|
|
13
|
+
constructor(config) {
|
|
14
|
+
if (!config.apiKey)
|
|
15
|
+
throw new errors_js_1.ProviderError('OpenAI requires an API key (OPENAI_API_KEY)');
|
|
16
|
+
this.client = new openai_1.default({ apiKey: config.apiKey, baseURL: config.baseUrl });
|
|
17
|
+
this.model = config.model ?? 'gpt-4o';
|
|
18
|
+
}
|
|
19
|
+
async complete(prompt, opts = {}) {
|
|
20
|
+
const messages = [];
|
|
21
|
+
if (opts.systemPrompt)
|
|
22
|
+
messages.push({ role: 'system', content: opts.systemPrompt });
|
|
23
|
+
messages.push({ role: 'user', content: prompt });
|
|
24
|
+
try {
|
|
25
|
+
const res = await this.client.chat.completions.create({
|
|
26
|
+
model: this.model,
|
|
27
|
+
messages,
|
|
28
|
+
max_tokens: opts.maxTokens ?? 4096,
|
|
29
|
+
temperature: opts.temperature ?? 0.2,
|
|
30
|
+
stop: opts.stopSequences,
|
|
31
|
+
});
|
|
32
|
+
return res.choices[0].message.content ?? '';
|
|
33
|
+
}
|
|
34
|
+
catch (err) {
|
|
35
|
+
throw new errors_js_1.ProviderError(`OpenAI request failed: ${String(err)}`);
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
async *stream(prompt, opts = {}) {
|
|
39
|
+
const messages = [];
|
|
40
|
+
if (opts.systemPrompt)
|
|
41
|
+
messages.push({ role: 'system', content: opts.systemPrompt });
|
|
42
|
+
messages.push({ role: 'user', content: prompt });
|
|
43
|
+
try {
|
|
44
|
+
const stream = await this.client.chat.completions.create({
|
|
45
|
+
model: this.model,
|
|
46
|
+
messages,
|
|
47
|
+
max_tokens: opts.maxTokens ?? 4096,
|
|
48
|
+
temperature: opts.temperature ?? 0.2,
|
|
49
|
+
stream: true,
|
|
50
|
+
});
|
|
51
|
+
for await (const chunk of stream) {
|
|
52
|
+
const content = chunk.choices[0]?.delta?.content;
|
|
53
|
+
if (content)
|
|
54
|
+
yield content;
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
catch (err) {
|
|
58
|
+
throw new errors_js_1.ProviderError(`OpenAI stream failed: ${String(err)}`);
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
exports.OpenAIProvider = OpenAIProvider;
|
|
63
|
+
//# sourceMappingURL=OpenAIProvider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OpenAIProvider.js","sourceRoot":"","sources":["../../src/providers/OpenAIProvider.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAE5B,kDAAmD;AAGnD,MAAa,cAAc;IAChB,IAAI,GAAG,QAAQ,CAAC;IACjB,MAAM,CAAS;IACf,KAAK,CAAS;IAEtB,YAAY,MAAiB;QAC3B,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,MAAM,IAAI,yBAAa,CAAC,6CAA6C,CAAC,CAAC;QAC3F,IAAI,CAAC,MAAM,GAAG,IAAI,gBAAM,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7E,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,QAAQ,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAA0B,EAAE;QACzD,MAAM,QAAQ,GAA6C,EAAE,CAAC;QAC9D,IAAI,IAAI,CAAC,YAAY;YAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QACrF,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAEjD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACpD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;gBAClC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,GAAG;gBACpC,IAAI,EAAE,IAAI,CAAC,aAAa;aACzB,CAAC,CAAC;YACH,OAAO,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;QAC9C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,yBAAa,CAAC,0BAA0B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,MAAc,EAAE,OAA0B,EAAE;QACxD,MAAM,QAAQ,GAA6C,EAAE,CAAC;QAC9D,IAAI,IAAI,CAAC,YAAY;YAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QACrF,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAEjD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACvD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;gBAClC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,GAAG;gBACpC,MAAM,EAAE,IAAI;aACb,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC;gBACjD,IAAI,OAAO;oBAAE,MAAM,OAAO,CAAC;YAC7B,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,yBAAa,CAAC,yBAAyB,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;CACF;AApDD,wCAoDC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { LLMProvider, CompletionOptions } from './LLMProvider.js';
|
|
2
|
+
import type { LLMConfig } from '../config/types.js';
|
|
3
|
+
export declare class OpenRouterProvider implements LLMProvider {
|
|
4
|
+
readonly name = "openrouter";
|
|
5
|
+
private apiKey;
|
|
6
|
+
private model;
|
|
7
|
+
private baseUrl;
|
|
8
|
+
constructor(config: LLMConfig);
|
|
9
|
+
complete(prompt: string, opts?: CompletionOptions): Promise<string>;
|
|
10
|
+
stream(prompt: string, opts?: CompletionOptions): AsyncIterable<string>;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=OpenRouterProvider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OpenRouterProvider.d.ts","sourceRoot":"","sources":["../../src/providers/OpenRouterProvider.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAEvE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAEpD,qBAAa,kBAAmB,YAAW,WAAW;IACpD,QAAQ,CAAC,IAAI,gBAAgB;IAC7B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,SAAS;IAOvB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAAsB,GAAG,OAAO,CAAC,MAAM,CAAC;IA8BtE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAE,iBAAsB,GAAG,aAAa,CAAC,MAAM,CAAC;CAyCnF"}
|
|
@@ -0,0 +1,89 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.OpenRouterProvider = void 0;
|
|
7
|
+
const axios_1 = __importDefault(require("axios"));
|
|
8
|
+
const errors_js_1 = require("../utils/errors.js");
|
|
9
|
+
class OpenRouterProvider {
|
|
10
|
+
name = 'openrouter';
|
|
11
|
+
apiKey;
|
|
12
|
+
model;
|
|
13
|
+
baseUrl;
|
|
14
|
+
constructor(config) {
|
|
15
|
+
if (!config.apiKey)
|
|
16
|
+
throw new errors_js_1.ProviderError('OpenRouter requires an API key');
|
|
17
|
+
this.apiKey = config.apiKey;
|
|
18
|
+
this.model = config.model ?? 'deepseek/deepseek-coder';
|
|
19
|
+
this.baseUrl = config.baseUrl ?? 'https://openrouter.ai/api/v1';
|
|
20
|
+
}
|
|
21
|
+
async complete(prompt, opts = {}) {
|
|
22
|
+
const messages = [];
|
|
23
|
+
if (opts.systemPrompt)
|
|
24
|
+
messages.push({ role: 'system', content: opts.systemPrompt });
|
|
25
|
+
messages.push({ role: 'user', content: prompt });
|
|
26
|
+
try {
|
|
27
|
+
const res = await axios_1.default.post(`${this.baseUrl}/chat/completions`, {
|
|
28
|
+
model: this.model,
|
|
29
|
+
messages,
|
|
30
|
+
max_tokens: opts.maxTokens ?? 4096,
|
|
31
|
+
temperature: opts.temperature ?? 0.2,
|
|
32
|
+
stop: opts.stopSequences,
|
|
33
|
+
}, {
|
|
34
|
+
headers: {
|
|
35
|
+
Authorization: `Bearer ${this.apiKey}`,
|
|
36
|
+
'HTTP-Referer': 'https://github.com/fire-code/fire-code',
|
|
37
|
+
'X-Title': 'Fire Code',
|
|
38
|
+
'Content-Type': 'application/json',
|
|
39
|
+
},
|
|
40
|
+
});
|
|
41
|
+
return res.data.choices[0].message.content;
|
|
42
|
+
}
|
|
43
|
+
catch (err) {
|
|
44
|
+
throw new errors_js_1.ProviderError(`OpenRouter request failed: ${String(err)}`);
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
async *stream(prompt, opts = {}) {
|
|
48
|
+
const messages = [];
|
|
49
|
+
if (opts.systemPrompt)
|
|
50
|
+
messages.push({ role: 'system', content: opts.systemPrompt });
|
|
51
|
+
messages.push({ role: 'user', content: prompt });
|
|
52
|
+
try {
|
|
53
|
+
const res = await axios_1.default.post(`${this.baseUrl}/chat/completions`, {
|
|
54
|
+
model: this.model,
|
|
55
|
+
messages,
|
|
56
|
+
max_tokens: opts.maxTokens ?? 4096,
|
|
57
|
+
temperature: opts.temperature ?? 0.2,
|
|
58
|
+
stream: true,
|
|
59
|
+
}, {
|
|
60
|
+
headers: {
|
|
61
|
+
Authorization: `Bearer ${this.apiKey}`,
|
|
62
|
+
'HTTP-Referer': 'https://github.com/fire-code/fire-code',
|
|
63
|
+
'X-Title': 'Fire Code',
|
|
64
|
+
},
|
|
65
|
+
responseType: 'stream',
|
|
66
|
+
});
|
|
67
|
+
for await (const chunk of res.data) {
|
|
68
|
+
const lines = chunk.toString().split('\n').filter((l) => l.startsWith('data: '));
|
|
69
|
+
for (const line of lines) {
|
|
70
|
+
const data = line.slice(6);
|
|
71
|
+
if (data === '[DONE]')
|
|
72
|
+
return;
|
|
73
|
+
try {
|
|
74
|
+
const parsed = JSON.parse(data);
|
|
75
|
+
const content = parsed.choices[0]?.delta?.content;
|
|
76
|
+
if (content)
|
|
77
|
+
yield content;
|
|
78
|
+
}
|
|
79
|
+
catch { /* skip malformed */ }
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
catch (err) {
|
|
84
|
+
throw new errors_js_1.ProviderError(`OpenRouter stream failed: ${String(err)}`);
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
exports.OpenRouterProvider = OpenRouterProvider;
|
|
89
|
+
//# sourceMappingURL=OpenRouterProvider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OpenRouterProvider.js","sourceRoot":"","sources":["../../src/providers/OpenRouterProvider.ts"],"names":[],"mappings":";;;;;;AAAA,kDAA0B;AAE1B,kDAAmD;AAGnD,MAAa,kBAAkB;IACpB,IAAI,GAAG,YAAY,CAAC;IACrB,MAAM,CAAS;IACf,KAAK,CAAS;IACd,OAAO,CAAS;IAExB,YAAY,MAAiB;QAC3B,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,MAAM,IAAI,yBAAa,CAAC,gCAAgC,CAAC,CAAC;QAC9E,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,yBAAyB,CAAC;QACvD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,8BAA8B,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAA0B,EAAE;QACzD,MAAM,QAAQ,GAAG,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,YAAY;YAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QACrF,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAEjD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,eAAK,CAAC,IAAI,CAC1B,GAAG,IAAI,CAAC,OAAO,mBAAmB,EAClC;gBACE,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;gBAClC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,GAAG;gBACpC,IAAI,EAAE,IAAI,CAAC,aAAa;aACzB,EACD;gBACE,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACtC,cAAc,EAAE,wCAAwC;oBACxD,SAAS,EAAE,WAAW;oBACtB,cAAc,EAAE,kBAAkB;iBACnC;aACF,CACF,CAAC;YACF,OAAQ,GAAG,CAAC,IAA6D,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QACvG,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,yBAAa,CAAC,8BAA8B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,MAAc,EAAE,OAA0B,EAAE;QACxD,MAAM,QAAQ,GAAG,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,YAAY;YAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QACrF,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAEjD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,eAAK,CAAC,IAAI,CAC1B,GAAG,IAAI,CAAC,OAAO,mBAAmB,EAClC;gBACE,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;gBAClC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,GAAG;gBACpC,MAAM,EAAE,IAAI;aACb,EACD;gBACE,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACtC,cAAc,EAAE,wCAAwC;oBACxD,SAAS,EAAE,WAAW;iBACvB;gBACD,YAAY,EAAE,QAAQ;aACvB,CACF,CAAC;YAEF,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,CAAC,IAA6B,EAAE,CAAC;gBAC5D,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;gBACzF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAC3B,IAAI,IAAI,KAAK,QAAQ;wBAAE,OAAO;oBAC9B,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAwD,CAAC;wBACvF,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC;wBAClD,IAAI,OAAO;4BAAE,MAAM,OAAO,CAAC;oBAC7B,CAAC;oBAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,yBAAa,CAAC,6BAA6B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;CACF;AApFD,gDAoFC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ProviderFactory.d.ts","sourceRoot":"","sources":["../../src/providers/ProviderFactory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAQpD,wBAAgB,cAAc,CAAC,MAAM,EAAE,SAAS,GAAG,WAAW,CAgB7D"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.createProvider = createProvider;
|
|
4
|
+
const OpenRouterProvider_js_1 = require("./OpenRouterProvider.js");
|
|
5
|
+
const AnthropicProvider_js_1 = require("./AnthropicProvider.js");
|
|
6
|
+
const OpenAIProvider_js_1 = require("./OpenAIProvider.js");
|
|
7
|
+
const OllamaProvider_js_1 = require("./OllamaProvider.js");
|
|
8
|
+
const AgentSDKProvider_js_1 = require("./AgentSDKProvider.js");
|
|
9
|
+
const errors_js_1 = require("../utils/errors.js");
|
|
10
|
+
function createProvider(config) {
|
|
11
|
+
switch (config.provider) {
|
|
12
|
+
case 'openrouter':
|
|
13
|
+
return new OpenRouterProvider_js_1.OpenRouterProvider(config);
|
|
14
|
+
case 'anthropic':
|
|
15
|
+
return new AnthropicProvider_js_1.AnthropicProvider(config);
|
|
16
|
+
case 'openai':
|
|
17
|
+
return new OpenAIProvider_js_1.OpenAIProvider(config);
|
|
18
|
+
case 'ollama':
|
|
19
|
+
return new OllamaProvider_js_1.OllamaProvider(config);
|
|
20
|
+
case 'agent-sdk':
|
|
21
|
+
// Uses logged-in Claude CLI session — no API key required
|
|
22
|
+
return new AgentSDKProvider_js_1.AgentSDKProvider({ model: config.model });
|
|
23
|
+
default:
|
|
24
|
+
throw new errors_js_1.ProviderError(`Unknown LLM provider: ${config.provider}`);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=ProviderFactory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ProviderFactory.js","sourceRoot":"","sources":["../../src/providers/ProviderFactory.ts"],"names":[],"mappings":";;AASA,wCAgBC;AAvBD,mEAA6D;AAC7D,iEAA2D;AAC3D,2DAAqD;AACrD,2DAAqD;AACrD,+DAAyD;AACzD,kDAAmD;AAEnD,SAAgB,cAAc,CAAC,MAAiB;IAC9C,QAAQ,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,YAAY;YACf,OAAO,IAAI,0CAAkB,CAAC,MAAM,CAAC,CAAC;QACxC,KAAK,WAAW;YACd,OAAO,IAAI,wCAAiB,CAAC,MAAM,CAAC,CAAC;QACvC,KAAK,QAAQ;YACX,OAAO,IAAI,kCAAc,CAAC,MAAM,CAAC,CAAC;QACpC,KAAK,QAAQ;YACX,OAAO,IAAI,kCAAc,CAAC,MAAM,CAAC,CAAC;QACpC,KAAK,WAAW;YACd,0DAA0D;YAC1D,OAAO,IAAI,sCAAgB,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QACvD;YACE,MAAM,IAAI,yBAAa,CAAC,yBAAyB,MAAM,CAAC,QAAkB,EAAE,CAAC,CAAC;IAClF,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
export declare class CorpusService {
|
|
2
|
+
private cwd;
|
|
3
|
+
private db;
|
|
4
|
+
private project;
|
|
5
|
+
constructor(cwd: string);
|
|
6
|
+
/** Index all documentation files in the project */
|
|
7
|
+
build(opts?: {
|
|
8
|
+
includeCode?: boolean;
|
|
9
|
+
}): Promise<{
|
|
10
|
+
added: number;
|
|
11
|
+
skipped: number;
|
|
12
|
+
}>;
|
|
13
|
+
/** Search the corpus */
|
|
14
|
+
query(query: string, limit?: number): {
|
|
15
|
+
title: string;
|
|
16
|
+
content: string;
|
|
17
|
+
source: string | null;
|
|
18
|
+
}[];
|
|
19
|
+
/** Add a single item manually */
|
|
20
|
+
prime(title: string, content: string, tags?: string[], isPrivate?: boolean): void;
|
|
21
|
+
/** Format corpus results as context string for LLM */
|
|
22
|
+
formatAsContext(results: {
|
|
23
|
+
title: string;
|
|
24
|
+
content: string;
|
|
25
|
+
}[]): string;
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=CorpusService.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"CorpusService.d.ts","sourceRoot":"","sources":["../../src/services/CorpusService.ts"],"names":[],"mappings":"AAiEA,qBAAa,aAAa;IAIZ,OAAO,CAAC,GAAG;IAHvB,OAAO,CAAC,EAAE,CAAkB;IAC5B,OAAO,CAAC,OAAO,CAAS;gBAEJ,GAAG,EAAE,MAAM;IAK/B,mDAAmD;IAC7C,KAAK,CAAC,IAAI,GAAE;QAAE,WAAW,CAAC,EAAE,OAAO,CAAA;KAAO,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;IA0B9F,wBAAwB;IACxB,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,SAAI,GAAG;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAA;KAAE,EAAE;IAI5F,iCAAiC;IACjC,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,SAAS,UAAQ,GAAG,IAAI;IAM/E,sDAAsD;IACtD,eAAe,CAAC,OAAO,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,EAAE,GAAG,MAAM;CAQvE"}
|
|
@@ -0,0 +1,142 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.CorpusService = void 0;
|
|
4
|
+
const fs_1 = require("fs");
|
|
5
|
+
const path_1 = require("path");
|
|
6
|
+
const DatabaseManager_js_1 = require("../db/DatabaseManager.js");
|
|
7
|
+
const paths_js_1 = require("../utils/paths.js");
|
|
8
|
+
const privacy_js_1 = require("../utils/privacy.js");
|
|
9
|
+
const logger_js_1 = require("../utils/logger.js");
|
|
10
|
+
const TEXT_EXTENSIONS = new Set(['.md', '.txt', '.rst', '.mdx', '.mdc']);
|
|
11
|
+
const SKIP_DIRS = new Set(['node_modules', '.git', 'dist', '.firecode', 'coverage', '.next', 'build']);
|
|
12
|
+
const MAX_FILE_BYTES = 100_000;
|
|
13
|
+
function getProjectName(cwd) {
|
|
14
|
+
try {
|
|
15
|
+
const pkg = JSON.parse((0, fs_1.readFileSync)((0, path_1.join)(cwd, 'package.json'), 'utf8'));
|
|
16
|
+
return pkg.name ?? require('path').basename(cwd);
|
|
17
|
+
}
|
|
18
|
+
catch {
|
|
19
|
+
return require('path').basename(cwd);
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
// Walk directory, collecting text-heavy files
|
|
23
|
+
function walkDocs(dir, cwd) {
|
|
24
|
+
const results = [];
|
|
25
|
+
let entries;
|
|
26
|
+
try {
|
|
27
|
+
entries = (0, fs_1.readdirSync)(dir);
|
|
28
|
+
}
|
|
29
|
+
catch {
|
|
30
|
+
return results;
|
|
31
|
+
}
|
|
32
|
+
for (const entry of entries) {
|
|
33
|
+
if (SKIP_DIRS.has(entry))
|
|
34
|
+
continue;
|
|
35
|
+
const fullPath = (0, path_1.join)(dir, entry);
|
|
36
|
+
let stat;
|
|
37
|
+
try {
|
|
38
|
+
stat = (0, fs_1.statSync)(fullPath);
|
|
39
|
+
}
|
|
40
|
+
catch {
|
|
41
|
+
continue;
|
|
42
|
+
}
|
|
43
|
+
if (stat.isDirectory()) {
|
|
44
|
+
results.push(...walkDocs(fullPath, cwd));
|
|
45
|
+
}
|
|
46
|
+
else if (stat.isFile() && TEXT_EXTENSIONS.has((0, path_1.extname)(entry).toLowerCase())) {
|
|
47
|
+
if (stat.size <= MAX_FILE_BYTES)
|
|
48
|
+
results.push(fullPath);
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
return results;
|
|
52
|
+
}
|
|
53
|
+
// Split document into manageable chunks by heading or paragraph
|
|
54
|
+
function chunkDocument(content, maxChunk = 1_500) {
|
|
55
|
+
// Split on markdown headings first
|
|
56
|
+
const sections = content.split(/(?=\n#{1,3} )/);
|
|
57
|
+
const chunks = [];
|
|
58
|
+
for (const section of sections) {
|
|
59
|
+
if (section.length <= maxChunk) {
|
|
60
|
+
chunks.push(section.trim());
|
|
61
|
+
}
|
|
62
|
+
else {
|
|
63
|
+
// Split long sections by paragraph
|
|
64
|
+
const paragraphs = section.split(/\n\n+/);
|
|
65
|
+
let current = '';
|
|
66
|
+
for (const para of paragraphs) {
|
|
67
|
+
if (current.length + para.length > maxChunk && current) {
|
|
68
|
+
chunks.push(current.trim());
|
|
69
|
+
current = para;
|
|
70
|
+
}
|
|
71
|
+
else {
|
|
72
|
+
current += '\n\n' + para;
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
if (current.trim())
|
|
76
|
+
chunks.push(current.trim());
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
return chunks.filter(c => c.length > 50);
|
|
80
|
+
}
|
|
81
|
+
class CorpusService {
|
|
82
|
+
cwd;
|
|
83
|
+
db;
|
|
84
|
+
project;
|
|
85
|
+
constructor(cwd) {
|
|
86
|
+
this.cwd = cwd;
|
|
87
|
+
this.db = DatabaseManager_js_1.DatabaseManager.getInstance((0, paths_js_1.getFireCodeDir)(cwd));
|
|
88
|
+
this.project = getProjectName(cwd);
|
|
89
|
+
}
|
|
90
|
+
/** Index all documentation files in the project */
|
|
91
|
+
async build(opts = {}) {
|
|
92
|
+
const files = walkDocs(this.cwd, this.cwd);
|
|
93
|
+
let added = 0;
|
|
94
|
+
let skipped = 0;
|
|
95
|
+
for (const filePath of files) {
|
|
96
|
+
if ((0, privacy_js_1.isPrivateFile)(filePath)) {
|
|
97
|
+
skipped++;
|
|
98
|
+
continue;
|
|
99
|
+
}
|
|
100
|
+
let raw;
|
|
101
|
+
try {
|
|
102
|
+
raw = (0, fs_1.readFileSync)(filePath, 'utf8');
|
|
103
|
+
}
|
|
104
|
+
catch {
|
|
105
|
+
skipped++;
|
|
106
|
+
continue;
|
|
107
|
+
}
|
|
108
|
+
const content = (0, privacy_js_1.sanitizeForLLM)(raw, filePath);
|
|
109
|
+
const relPath = (0, path_1.relative)(this.cwd, filePath);
|
|
110
|
+
const chunks = chunkDocument(content);
|
|
111
|
+
for (let i = 0; i < chunks.length; i++) {
|
|
112
|
+
const title = chunks.length === 1 ? relPath : `${relPath}#${i + 1}`;
|
|
113
|
+
this.db.upsertCorpus(this.project, title, chunks[i], relPath, [], false);
|
|
114
|
+
added++;
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
logger_js_1.logger.info({ added, skipped }, 'Corpus build complete');
|
|
118
|
+
return { added, skipped };
|
|
119
|
+
}
|
|
120
|
+
/** Search the corpus */
|
|
121
|
+
query(query, limit = 5) {
|
|
122
|
+
return this.db.getCorpus({ project: this.project, query, limit });
|
|
123
|
+
}
|
|
124
|
+
/** Add a single item manually */
|
|
125
|
+
prime(title, content, tags, isPrivate = false) {
|
|
126
|
+
const sanitized = isPrivate ? content : (0, privacy_js_1.sanitizeForLLM)(content);
|
|
127
|
+
this.db.upsertCorpus(this.project, title, sanitized, undefined, tags, isPrivate);
|
|
128
|
+
logger_js_1.logger.debug({ title }, 'Corpus item primed');
|
|
129
|
+
}
|
|
130
|
+
/** Format corpus results as context string for LLM */
|
|
131
|
+
formatAsContext(results) {
|
|
132
|
+
if (results.length === 0)
|
|
133
|
+
return '';
|
|
134
|
+
const lines = ['## Knowledge Corpus\n'];
|
|
135
|
+
for (const r of results) {
|
|
136
|
+
lines.push(`### ${r.title}\n${r.content}\n`);
|
|
137
|
+
}
|
|
138
|
+
return lines.join('\n');
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
exports.CorpusService = CorpusService;
|
|
142
|
+
//# sourceMappingURL=CorpusService.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"CorpusService.js","sourceRoot":"","sources":["../../src/services/CorpusService.ts"],"names":[],"mappings":";;;AAAA,2BAAqE;AACrE,+BAA+C;AAC/C,iEAA2D;AAC3D,gDAAmD;AACnD,oDAAoE;AACpE,kDAA4C;AAE5C,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;AACzE,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;AACvG,MAAM,cAAc,GAAG,OAAO,CAAC;AAE/B,SAAS,cAAc,CAAC,GAAW;IACjC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAA,iBAAY,EAAC,IAAA,WAAI,EAAC,GAAG,EAAE,cAAc,CAAC,EAAE,MAAM,CAAC,CAAsB,CAAC;QAC7F,OAAO,GAAG,CAAC,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QAAC,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAAC,CAAC;AACnD,CAAC;AAED,8CAA8C;AAC9C,SAAS,QAAQ,CAAC,GAAW,EAAE,GAAW;IACxC,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,IAAI,OAAiB,CAAC;IACtB,IAAI,CAAC;QAAC,OAAO,GAAG,IAAA,gBAAW,EAAC,GAAG,CAAC,CAAC;IAAC,CAAC;IAAC,MAAM,CAAC;QAAC,OAAO,OAAO,CAAC;IAAC,CAAC;IAE7D,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,SAAS;QACnC,MAAM,QAAQ,GAAG,IAAA,WAAI,EAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAClC,IAAI,IAAI,CAAC;QACT,IAAI,CAAC;YAAC,IAAI,GAAG,IAAA,aAAQ,EAAC,QAAQ,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC;YAAC,SAAS;QAAC,CAAC;QAEtD,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;YACvB,OAAO,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC;QAC3C,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,eAAe,CAAC,GAAG,CAAC,IAAA,cAAO,EAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;YAC9E,IAAI,IAAI,CAAC,IAAI,IAAI,cAAc;gBAAE,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,gEAAgE;AAChE,SAAS,aAAa,CAAC,OAAe,EAAE,QAAQ,GAAG,KAAK;IACtD,mCAAmC;IACnC,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;IAChD,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC9B,CAAC;aAAM,CAAC;YACN,mCAAmC;YACnC,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC1C,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC9B,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,GAAG,QAAQ,IAAI,OAAO,EAAE,CAAC;oBACvD,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC5B,OAAO,GAAG,IAAI,CAAC;gBACjB,CAAC;qBAAM,CAAC;oBACN,OAAO,IAAI,MAAM,GAAG,IAAI,CAAC;gBAC3B,CAAC;YACH,CAAC;YACD,IAAI,OAAO,CAAC,IAAI,EAAE;gBAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED,MAAa,aAAa;IAIJ;IAHZ,EAAE,CAAkB;IACpB,OAAO,CAAS;IAExB,YAAoB,GAAW;QAAX,QAAG,GAAH,GAAG,CAAQ;QAC7B,IAAI,CAAC,EAAE,GAAG,oCAAe,CAAC,WAAW,CAAC,IAAA,yBAAc,EAAC,GAAG,CAAC,CAAC,CAAC;QAC3D,IAAI,CAAC,OAAO,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;IACrC,CAAC;IAED,mDAAmD;IACnD,KAAK,CAAC,KAAK,CAAC,OAAkC,EAAE;QAC9C,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3C,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC7B,IAAI,IAAA,0BAAa,EAAC,QAAQ,CAAC,EAAE,CAAC;gBAAC,OAAO,EAAE,CAAC;gBAAC,SAAS;YAAC,CAAC;YAErD,IAAI,GAAW,CAAC;YAChB,IAAI,CAAC;gBAAC,GAAG,GAAG,IAAA,iBAAY,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC;gBAAC,OAAO,EAAE,CAAC;gBAAC,SAAS;YAAC,CAAC;YAE5E,MAAM,OAAO,GAAG,IAAA,2BAAc,EAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAC9C,MAAM,OAAO,GAAG,IAAA,eAAQ,EAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAC7C,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;YAEtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpE,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;gBACzE,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QAED,kBAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,uBAAuB,CAAC,CAAC;QACzD,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IAC5B,CAAC;IAED,wBAAwB;IACxB,KAAK,CAAC,KAAa,EAAE,KAAK,GAAG,CAAC;QAC5B,OAAO,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,iCAAiC;IACjC,KAAK,CAAC,KAAa,EAAE,OAAe,EAAE,IAAe,EAAE,SAAS,GAAG,KAAK;QACtE,MAAM,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAA,2BAAc,EAAC,OAAO,CAAC,CAAC;QAChE,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QACjF,kBAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,oBAAoB,CAAC,CAAC;IAChD,CAAC;IAED,sDAAsD;IACtD,eAAe,CAAC,OAA6C;QAC3D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACpC,MAAM,KAAK,GAAG,CAAC,uBAAuB,CAAC,CAAC;QACxC,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AAzDD,sCAyDC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
export interface ToolCallData {
|
|
2
|
+
tool_name: string;
|
|
3
|
+
tool_input?: Record<string, unknown>;
|
|
4
|
+
tool_result?: unknown;
|
|
5
|
+
session_id: string;
|
|
6
|
+
project: string;
|
|
7
|
+
cwd: string;
|
|
8
|
+
}
|
|
9
|
+
export declare class ObservationService {
|
|
10
|
+
private db;
|
|
11
|
+
constructor(cwd: string);
|
|
12
|
+
capture(data: ToolCallData): number;
|
|
13
|
+
search(query: string, project: string, limit?: number): import("../db/DatabaseManager.js").Observation[];
|
|
14
|
+
recent(project: string, limit?: number): import("../db/DatabaseManager.js").Observation[];
|
|
15
|
+
getContext(project: string): string;
|
|
16
|
+
}
|
|
17
|
+
//# sourceMappingURL=ObservationService.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ObservationService.d.ts","sourceRoot":"","sources":["../../src/services/ObservationService.ts"],"names":[],"mappings":"AAKA,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,GAAG,EAAE,MAAM,CAAC;CACb;AA4DD,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,EAAE,CAAkB;gBAEhB,GAAG,EAAE,MAAM;IAIvB,OAAO,CAAC,IAAI,EAAE,YAAY,GAAG,MAAM;IAoBnC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,SAAK;IAIjD,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,SAAK;IAIlC,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;CAGpC"}
|