orka-cli 1.0.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/CHANGELOG.md +25 -0
- package/CONTRIBUTING.md +59 -0
- package/README.md +208 -0
- package/dist/agents/base-agent.d.ts +48 -0
- package/dist/agents/base-agent.d.ts.map +1 -0
- package/dist/agents/base-agent.js +59 -0
- package/dist/agents/base-agent.js.map +1 -0
- package/dist/agents/builder.d.ts +53 -0
- package/dist/agents/builder.d.ts.map +1 -0
- package/dist/agents/builder.js +147 -0
- package/dist/agents/builder.js.map +1 -0
- package/dist/agents/coordinator.d.ts +45 -0
- package/dist/agents/coordinator.d.ts.map +1 -0
- package/dist/agents/coordinator.js +201 -0
- package/dist/agents/coordinator.js.map +1 -0
- package/dist/agents/index.d.ts +8 -0
- package/dist/agents/index.d.ts.map +1 -0
- package/dist/agents/index.js +7 -0
- package/dist/agents/index.js.map +1 -0
- package/dist/agents/reviewer.d.ts +39 -0
- package/dist/agents/reviewer.d.ts.map +1 -0
- package/dist/agents/reviewer.js +119 -0
- package/dist/agents/reviewer.js.map +1 -0
- package/dist/agents/scout.d.ts +35 -0
- package/dist/agents/scout.d.ts.map +1 -0
- package/dist/agents/scout.js +76 -0
- package/dist/agents/scout.js.map +1 -0
- package/dist/agents/tester.d.ts +25 -0
- package/dist/agents/tester.d.ts.map +1 -0
- package/dist/agents/tester.js +76 -0
- package/dist/agents/tester.js.map +1 -0
- package/dist/cli/board.d.ts +28 -0
- package/dist/cli/board.d.ts.map +1 -0
- package/dist/cli/board.js +178 -0
- package/dist/cli/board.js.map +1 -0
- package/dist/cli/cost.d.ts +31 -0
- package/dist/cli/cost.d.ts.map +1 -0
- package/dist/cli/cost.js +85 -0
- package/dist/cli/cost.js.map +1 -0
- package/dist/cli/credentials.d.ts +23 -0
- package/dist/cli/credentials.d.ts.map +1 -0
- package/dist/cli/credentials.js +77 -0
- package/dist/cli/credentials.js.map +1 -0
- package/dist/cli/dry-run.d.ts +3 -0
- package/dist/cli/dry-run.d.ts.map +1 -0
- package/dist/cli/dry-run.js +21 -0
- package/dist/cli/dry-run.js.map +1 -0
- package/dist/cli/extras.d.ts +21 -0
- package/dist/cli/extras.d.ts.map +1 -0
- package/dist/cli/extras.js +39 -0
- package/dist/cli/extras.js.map +1 -0
- package/dist/cli/fix.d.ts +11 -0
- package/dist/cli/fix.d.ts.map +1 -0
- package/dist/cli/fix.js +72 -0
- package/dist/cli/fix.js.map +1 -0
- package/dist/cli/git-session.d.ts +21 -0
- package/dist/cli/git-session.d.ts.map +1 -0
- package/dist/cli/git-session.js +62 -0
- package/dist/cli/git-session.js.map +1 -0
- package/dist/cli/index.d.ts +4 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +546 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/init.d.ts +16 -0
- package/dist/cli/init.d.ts.map +1 -0
- package/dist/cli/init.js +75 -0
- package/dist/cli/init.js.map +1 -0
- package/dist/cli/login.d.ts +9 -0
- package/dist/cli/login.d.ts.map +1 -0
- package/dist/cli/login.js +114 -0
- package/dist/cli/login.js.map +1 -0
- package/dist/cli/logs.d.ts +16 -0
- package/dist/cli/logs.d.ts.map +1 -0
- package/dist/cli/logs.js +37 -0
- package/dist/cli/logs.js.map +1 -0
- package/dist/cli/memory-learner.d.ts +7 -0
- package/dist/cli/memory-learner.d.ts.map +1 -0
- package/dist/cli/memory-learner.js +36 -0
- package/dist/cli/memory-learner.js.map +1 -0
- package/dist/cli/memory.d.ts +13 -0
- package/dist/cli/memory.d.ts.map +1 -0
- package/dist/cli/memory.js +52 -0
- package/dist/cli/memory.js.map +1 -0
- package/dist/cli/plugin.d.ts +5 -0
- package/dist/cli/plugin.d.ts.map +1 -0
- package/dist/cli/plugin.js +42 -0
- package/dist/cli/plugin.js.map +1 -0
- package/dist/cli/resume.d.ts +18 -0
- package/dist/cli/resume.d.ts.map +1 -0
- package/dist/cli/resume.js +84 -0
- package/dist/cli/resume.js.map +1 -0
- package/dist/cli/rollback.d.ts +11 -0
- package/dist/cli/rollback.d.ts.map +1 -0
- package/dist/cli/rollback.js +12 -0
- package/dist/cli/rollback.js.map +1 -0
- package/dist/cli/rules.d.ts +7 -0
- package/dist/cli/rules.d.ts.map +1 -0
- package/dist/cli/rules.js +20 -0
- package/dist/cli/rules.js.map +1 -0
- package/dist/cli/run.d.ts +35 -0
- package/dist/cli/run.d.ts.map +1 -0
- package/dist/cli/run.js +294 -0
- package/dist/cli/run.js.map +1 -0
- package/dist/cli/session.d.ts +5 -0
- package/dist/cli/session.d.ts.map +1 -0
- package/dist/cli/session.js +46 -0
- package/dist/cli/session.js.map +1 -0
- package/dist/cli/ui/Board.d.ts +22 -0
- package/dist/cli/ui/Board.d.ts.map +1 -0
- package/dist/cli/ui/Board.js +40 -0
- package/dist/cli/ui/Board.js.map +1 -0
- package/dist/cli/ui/StatusBar.d.ts +10 -0
- package/dist/cli/ui/StatusBar.d.ts.map +1 -0
- package/dist/cli/ui/StatusBar.js +6 -0
- package/dist/cli/ui/StatusBar.js.map +1 -0
- package/dist/config.d.ts +4 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +79 -0
- package/dist/config.js.map +1 -0
- package/dist/core/bus.d.ts +19 -0
- package/dist/core/bus.d.ts.map +1 -0
- package/dist/core/bus.js +25 -0
- package/dist/core/bus.js.map +1 -0
- package/dist/core/context.d.ts +10 -0
- package/dist/core/context.d.ts.map +1 -0
- package/dist/core/context.js +42 -0
- package/dist/core/context.js.map +1 -0
- package/dist/core/dispatcher.d.ts +25 -0
- package/dist/core/dispatcher.d.ts.map +1 -0
- package/dist/core/dispatcher.js +76 -0
- package/dist/core/dispatcher.js.map +1 -0
- package/dist/core/errors.d.ts +8 -0
- package/dist/core/errors.d.ts.map +1 -0
- package/dist/core/errors.js +13 -0
- package/dist/core/errors.js.map +1 -0
- package/dist/core/file-ownership.d.ts +25 -0
- package/dist/core/file-ownership.d.ts.map +1 -0
- package/dist/core/file-ownership.js +81 -0
- package/dist/core/file-ownership.js.map +1 -0
- package/dist/core/index.d.ts +9 -0
- package/dist/core/index.d.ts.map +1 -0
- package/dist/core/index.js +6 -0
- package/dist/core/index.js.map +1 -0
- package/dist/core/mcp-client-pool.d.ts +9 -0
- package/dist/core/mcp-client-pool.d.ts.map +1 -0
- package/dist/core/mcp-client-pool.js +26 -0
- package/dist/core/mcp-client-pool.js.map +1 -0
- package/dist/core/mcp-client.d.ts +27 -0
- package/dist/core/mcp-client.d.ts.map +1 -0
- package/dist/core/mcp-client.js +63 -0
- package/dist/core/mcp-client.js.map +1 -0
- package/dist/core/plugin-loader.d.ts +41 -0
- package/dist/core/plugin-loader.d.ts.map +1 -0
- package/dist/core/plugin-loader.js +62 -0
- package/dist/core/plugin-loader.js.map +1 -0
- package/dist/core/plugin-sdk.d.ts +5 -0
- package/dist/core/plugin-sdk.d.ts.map +1 -0
- package/dist/core/plugin-sdk.js +2 -0
- package/dist/core/plugin-sdk.js.map +1 -0
- package/dist/core/state.d.ts +77 -0
- package/dist/core/state.d.ts.map +1 -0
- package/dist/core/state.js +357 -0
- package/dist/core/state.js.map +1 -0
- package/dist/core/task.d.ts +13 -0
- package/dist/core/task.d.ts.map +1 -0
- package/dist/core/task.js +87 -0
- package/dist/core/task.js.map +1 -0
- package/dist/git/__tests__/helpers.d.ts +3 -0
- package/dist/git/__tests__/helpers.d.ts.map +1 -0
- package/dist/git/__tests__/helpers.js +18 -0
- package/dist/git/__tests__/helpers.js.map +1 -0
- package/dist/git/branch.d.ts +7 -0
- package/dist/git/branch.d.ts.map +1 -0
- package/dist/git/branch.js +62 -0
- package/dist/git/branch.js.map +1 -0
- package/dist/git/commit.d.ts +3 -0
- package/dist/git/commit.d.ts.map +1 -0
- package/dist/git/commit.js +21 -0
- package/dist/git/commit.js.map +1 -0
- package/dist/git/conflict.d.ts +19 -0
- package/dist/git/conflict.d.ts.map +1 -0
- package/dist/git/conflict.js +86 -0
- package/dist/git/conflict.js.map +1 -0
- package/dist/git/index.d.ts +7 -0
- package/dist/git/index.d.ts.map +1 -0
- package/dist/git/index.js +7 -0
- package/dist/git/index.js.map +1 -0
- package/dist/git/merge.d.ts +16 -0
- package/dist/git/merge.d.ts.map +1 -0
- package/dist/git/merge.js +101 -0
- package/dist/git/merge.js.map +1 -0
- package/dist/git/safety.d.ts +16 -0
- package/dist/git/safety.d.ts.map +1 -0
- package/dist/git/safety.js +64 -0
- package/dist/git/safety.js.map +1 -0
- package/dist/git/utils.d.ts +8 -0
- package/dist/git/utils.d.ts.map +1 -0
- package/dist/git/utils.js +39 -0
- package/dist/git/utils.js.map +1 -0
- package/dist/mcp-server/index.d.ts +2 -0
- package/dist/mcp-server/index.d.ts.map +1 -0
- package/dist/mcp-server/index.js +39 -0
- package/dist/mcp-server/index.js.map +1 -0
- package/dist/mcp-server/tools.d.ts +70 -0
- package/dist/mcp-server/tools.d.ts.map +1 -0
- package/dist/mcp-server/tools.js +112 -0
- package/dist/mcp-server/tools.js.map +1 -0
- package/dist/models/item.d.ts +6 -0
- package/dist/models/item.d.ts.map +1 -0
- package/dist/models/item.js +2 -0
- package/dist/models/item.js.map +1 -0
- package/dist/orchestration/glue.d.ts +31 -0
- package/dist/orchestration/glue.d.ts.map +1 -0
- package/dist/orchestration/glue.js +204 -0
- package/dist/orchestration/glue.js.map +1 -0
- package/dist/orchestration/index.d.ts +3 -0
- package/dist/orchestration/index.d.ts.map +1 -0
- package/dist/orchestration/index.js +3 -0
- package/dist/orchestration/index.js.map +1 -0
- package/dist/orchestration/runtime.d.ts +35 -0
- package/dist/orchestration/runtime.d.ts.map +1 -0
- package/dist/orchestration/runtime.js +165 -0
- package/dist/orchestration/runtime.js.map +1 -0
- package/dist/providers/base-provider.d.ts +28 -0
- package/dist/providers/base-provider.d.ts.map +1 -0
- package/dist/providers/base-provider.js +2 -0
- package/dist/providers/base-provider.js.map +1 -0
- package/dist/providers/claude.d.ts +10 -0
- package/dist/providers/claude.d.ts.map +1 -0
- package/dist/providers/claude.js +93 -0
- package/dist/providers/claude.js.map +1 -0
- package/dist/providers/gemini.d.ts +16 -0
- package/dist/providers/gemini.d.ts.map +1 -0
- package/dist/providers/gemini.js +101 -0
- package/dist/providers/gemini.js.map +1 -0
- package/dist/providers/index.d.ts +21 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +68 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/mcp-provider.d.ts +11 -0
- package/dist/providers/mcp-provider.d.ts.map +1 -0
- package/dist/providers/mcp-provider.js +42 -0
- package/dist/providers/mcp-provider.js.map +1 -0
- package/dist/providers/mock.d.ts +10 -0
- package/dist/providers/mock.d.ts.map +1 -0
- package/dist/providers/mock.js +66 -0
- package/dist/providers/mock.js.map +1 -0
- package/dist/providers/model-registry.d.ts +9 -0
- package/dist/providers/model-registry.d.ts.map +1 -0
- package/dist/providers/model-registry.js +153 -0
- package/dist/providers/model-registry.js.map +1 -0
- package/dist/providers/ollama.d.ts +11 -0
- package/dist/providers/ollama.d.ts.map +1 -0
- package/dist/providers/ollama.js +141 -0
- package/dist/providers/ollama.js.map +1 -0
- package/dist/providers/openai.d.ts +17 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +109 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/zai.d.ts +6 -0
- package/dist/providers/zai.d.ts.map +1 -0
- package/dist/providers/zai.js +17 -0
- package/dist/providers/zai.js.map +1 -0
- package/dist/server/api.d.ts +5 -0
- package/dist/server/api.d.ts.map +1 -0
- package/dist/server/api.js +153 -0
- package/dist/server/api.js.map +1 -0
- package/dist/server/index.d.ts +14 -0
- package/dist/server/index.d.ts.map +1 -0
- package/dist/server/index.js +98 -0
- package/dist/server/index.js.map +1 -0
- package/dist/server/middleware.d.ts +6 -0
- package/dist/server/middleware.d.ts.map +1 -0
- package/dist/server/middleware.js +35 -0
- package/dist/server/middleware.js.map +1 -0
- package/dist/server/tunnel.d.ts +7 -0
- package/dist/server/tunnel.d.ts.map +1 -0
- package/dist/server/tunnel.js +44 -0
- package/dist/server/tunnel.js.map +1 -0
- package/dist/server/websocket.d.ts +11 -0
- package/dist/server/websocket.d.ts.map +1 -0
- package/dist/server/websocket.js +48 -0
- package/dist/server/websocket.js.map +1 -0
- package/dist/types.d.ts +90 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/package.json +62 -0
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { AgentRole } from '../types.js';
|
|
2
|
+
import type { ChatChunk, ChatMessage, ChatOptions, ChatResponse, LLMProvider } from './base-provider.js';
|
|
3
|
+
export declare class MockProvider implements LLMProvider {
|
|
4
|
+
private readonly role;
|
|
5
|
+
readonly name = "mock";
|
|
6
|
+
constructor(role: AgentRole);
|
|
7
|
+
chat(_messages: ChatMessage[], _opts?: ChatOptions): Promise<ChatResponse>;
|
|
8
|
+
stream(_messages: ChatMessage[], _opts?: ChatOptions): AsyncIterable<ChatChunk>;
|
|
9
|
+
}
|
|
10
|
+
//# sourceMappingURL=mock.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mock.d.ts","sourceRoot":"","sources":["../../src/providers/mock.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAA;AAC5C,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAwCxG,qBAAa,YAAa,YAAW,WAAW;IAGlC,OAAO,CAAC,QAAQ,CAAC,IAAI;IAFjC,QAAQ,CAAC,IAAI,UAAS;gBAEO,IAAI,EAAE,SAAS;IAEtC,IAAI,CAAC,SAAS,EAAE,WAAW,EAAE,EAAE,KAAK,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAYzE,MAAM,CAAC,SAAS,EAAE,WAAW,EAAE,EAAE,KAAK,CAAC,EAAE,WAAW,GAAG,aAAa,CAAC,SAAS,CAAC;CAiBvF"}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
const RESPONSES = {
|
|
2
|
+
scout: 'Project: TypeScript monorepo. Key files: src/index.ts, package.json. Stack: Node.js, TypeScript. Dependencies: commander, better-sqlite3, ink.',
|
|
3
|
+
coordinator: JSON.stringify({
|
|
4
|
+
tasks: [
|
|
5
|
+
{ id: 'task-1', goal: 'Buat model dan schema', files: ['src/models/item.ts'], dependencies: [] },
|
|
6
|
+
{ id: 'task-2', goal: 'Buat service layer', files: ['src/services/item.ts'], dependencies: ['task-1'] },
|
|
7
|
+
{ id: 'task-3', goal: 'Buat API routes', files: ['src/routes/item.ts'], dependencies: ['task-2'] },
|
|
8
|
+
],
|
|
9
|
+
}),
|
|
10
|
+
builder: JSON.stringify({ files: [{ path: 'src/models/item.ts', content: 'export interface Item {\n id: string\n name: string\n createdAt: number\n}\n' }] }),
|
|
11
|
+
reviewer: JSON.stringify({ approved: true, feedback: 'Implementasi sudah sesuai dengan requirement dan best practices.' }),
|
|
12
|
+
tester: '',
|
|
13
|
+
};
|
|
14
|
+
const USAGE = {
|
|
15
|
+
scout: { inputTokens: 150, outputTokens: 80 },
|
|
16
|
+
coordinator: { inputTokens: 300, outputTokens: 120 },
|
|
17
|
+
builder: { inputTokens: 400, outputTokens: 250 },
|
|
18
|
+
reviewer: { inputTokens: 200, outputTokens: 60 },
|
|
19
|
+
tester: { inputTokens: 0, outputTokens: 0 },
|
|
20
|
+
};
|
|
21
|
+
const DELAYS_MS = {
|
|
22
|
+
scout: [600, 900],
|
|
23
|
+
coordinator: [800, 1200],
|
|
24
|
+
builder: [1500, 2500],
|
|
25
|
+
reviewer: [700, 1000],
|
|
26
|
+
tester: [0, 0],
|
|
27
|
+
};
|
|
28
|
+
function delay(ms) {
|
|
29
|
+
return new Promise(resolve => setTimeout(resolve, ms));
|
|
30
|
+
}
|
|
31
|
+
function rand(min, max) {
|
|
32
|
+
return min + Math.floor(Math.random() * (max - min));
|
|
33
|
+
}
|
|
34
|
+
export class MockProvider {
|
|
35
|
+
role;
|
|
36
|
+
name = 'mock';
|
|
37
|
+
constructor(role) {
|
|
38
|
+
this.role = role;
|
|
39
|
+
}
|
|
40
|
+
async chat(_messages, _opts) {
|
|
41
|
+
const [min, max] = DELAYS_MS[this.role];
|
|
42
|
+
await delay(rand(min, max));
|
|
43
|
+
let content = RESPONSES[this.role];
|
|
44
|
+
if (this.role === 'reviewer' && Math.random() < 0.1) {
|
|
45
|
+
content = JSON.stringify({ approved: false, feedback: 'Tambahkan error handling untuk edge case null input.' });
|
|
46
|
+
}
|
|
47
|
+
return { content, usage: USAGE[this.role] };
|
|
48
|
+
}
|
|
49
|
+
async *stream(_messages, _opts) {
|
|
50
|
+
const [min, max] = DELAYS_MS[this.role];
|
|
51
|
+
await delay(rand(min, max) / 3);
|
|
52
|
+
const content = RESPONSES[this.role];
|
|
53
|
+
const chunkSize = Math.ceil(content.length / 3);
|
|
54
|
+
for (let i = 0; i < content.length; i += chunkSize) {
|
|
55
|
+
const isLast = i + chunkSize >= content.length;
|
|
56
|
+
yield {
|
|
57
|
+
content: content.slice(i, i + chunkSize),
|
|
58
|
+
done: isLast,
|
|
59
|
+
usage: isLast ? USAGE[this.role] : undefined,
|
|
60
|
+
};
|
|
61
|
+
if (!isLast)
|
|
62
|
+
await delay(150);
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
//# sourceMappingURL=mock.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mock.js","sourceRoot":"","sources":["../../src/providers/mock.ts"],"names":[],"mappings":"AAGA,MAAM,SAAS,GAA8B;IAC3C,KAAK,EAAE,gJAAgJ;IACvJ,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC;QAC1B,KAAK,EAAE;YACL,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,uBAAuB,EAAE,KAAK,EAAE,CAAC,oBAAoB,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE;YAChG,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAAC,sBAAsB,CAAC,EAAE,YAAY,EAAE,CAAC,QAAQ,CAAC,EAAE;YACvG,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,oBAAoB,CAAC,EAAE,YAAY,EAAE,CAAC,QAAQ,CAAC,EAAE;SACnG;KACF,CAAC;IACF,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,OAAO,EAAE,iFAAiF,EAAE,CAAC,EAAE,CAAC;IAChK,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,kEAAkE,EAAE,CAAC;IAC1H,MAAM,EAAE,EAAE;CACX,CAAA;AAED,MAAM,KAAK,GAAqE;IAC9E,KAAK,EAAQ,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE;IACnD,WAAW,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,EAAE;IACpD,OAAO,EAAM,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,EAAE;IACpD,QAAQ,EAAK,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE;IACnD,MAAM,EAAO,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;CACjD,CAAA;AAED,MAAM,SAAS,GAAwC;IACrD,KAAK,EAAQ,CAAC,GAAG,EAAE,GAAG,CAAC;IACvB,WAAW,EAAE,CAAC,GAAG,EAAE,IAAI,CAAC;IACxB,OAAO,EAAM,CAAC,IAAI,EAAE,IAAI,CAAC;IACzB,QAAQ,EAAK,CAAC,GAAG,EAAE,IAAI,CAAC;IACxB,MAAM,EAAO,CAAC,CAAC,EAAE,CAAC,CAAC;CACpB,CAAA;AAED,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAA;AACxD,CAAC;AAED,SAAS,IAAI,CAAC,GAAW,EAAE,GAAW;IACpC,OAAO,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,CAAA;AACtD,CAAC;AAED,MAAM,OAAO,YAAY;IAGM;IAFpB,IAAI,GAAG,MAAM,CAAA;IAEtB,YAA6B,IAAe;QAAf,SAAI,GAAJ,IAAI,CAAW;IAAG,CAAC;IAEhD,KAAK,CAAC,IAAI,CAAC,SAAwB,EAAE,KAAmB;QACtD,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACvC,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;QAE3B,IAAI,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAClC,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,EAAE,CAAC;YACpD,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,sDAAsD,EAAE,CAAC,CAAA;QACjH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAA;IAC7C,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,SAAwB,EAAE,KAAmB;QACzD,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACvC,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAA;QAE/B,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;QAE/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACnD,MAAM,MAAM,GAAG,CAAC,GAAG,SAAS,IAAI,OAAO,CAAC,MAAM,CAAA;YAC9C,MAAM;gBACJ,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC;gBACxC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;aAC7C,CAAA;YACD,IAAI,CAAC,MAAM;gBAAE,MAAM,KAAK,CAAC,GAAG,CAAC,CAAA;QAC/B,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Fetch available models from provider APIs dynamically.
|
|
3
|
+
* Cached in-memory per process with 1-hour TTL — no stale hardcoded lists.
|
|
4
|
+
*/
|
|
5
|
+
export declare const PROVIDER_DEFAULT_MODEL: Record<string, string>;
|
|
6
|
+
export declare function fetchModels(provider: string, apiKey: string, baseUrl?: string): Promise<string[]>;
|
|
7
|
+
export declare function clearModelCache(): void;
|
|
8
|
+
export declare function getCachedModels(provider: string, apiKey: string): string[] | null;
|
|
9
|
+
//# sourceMappingURL=model-registry.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-registry.d.ts","sourceRoot":"","sources":["../../src/providers/model-registry.ts"],"names":[],"mappings":"AAAA;;;GAGG;AA4CH,eAAO,MAAM,sBAAsB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAQzD,CAAA;AA+DD,wBAAsB,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAyCvG;AAED,wBAAgB,eAAe,IAAI,IAAI,CAEtC;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,IAAI,CAKjF"}
|
|
@@ -0,0 +1,153 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Fetch available models from provider APIs dynamically.
|
|
3
|
+
* Cached in-memory per process with 1-hour TTL — no stale hardcoded lists.
|
|
4
|
+
*/
|
|
5
|
+
const CACHE_TTL = 60 * 60 * 1000; // 1 hour
|
|
6
|
+
const cache = new Map();
|
|
7
|
+
// Fallback lists — used only when API is unreachable
|
|
8
|
+
const FALLBACK_MODELS = {
|
|
9
|
+
claude: [
|
|
10
|
+
'claude-opus-4-6',
|
|
11
|
+
'claude-sonnet-4-6',
|
|
12
|
+
'claude-haiku-4-5',
|
|
13
|
+
],
|
|
14
|
+
openai: [
|
|
15
|
+
'gpt-4o',
|
|
16
|
+
'gpt-4o-mini',
|
|
17
|
+
'o1',
|
|
18
|
+
'o1-mini',
|
|
19
|
+
],
|
|
20
|
+
openrouter: [],
|
|
21
|
+
gemini: [
|
|
22
|
+
'gemini-2.0-flash',
|
|
23
|
+
'gemini-1.5-pro',
|
|
24
|
+
'gemini-1.5-flash',
|
|
25
|
+
],
|
|
26
|
+
zai: [
|
|
27
|
+
'glm-4-flash',
|
|
28
|
+
'glm-4-plus',
|
|
29
|
+
],
|
|
30
|
+
opencode: [
|
|
31
|
+
'claude-sonnet-4-6',
|
|
32
|
+
'claude-haiku-4-5',
|
|
33
|
+
'minimax-m2.5',
|
|
34
|
+
'kimi-k2.5',
|
|
35
|
+
'glm-5',
|
|
36
|
+
],
|
|
37
|
+
ollama: [],
|
|
38
|
+
};
|
|
39
|
+
// Default model to use when --provider flag is passed without explicit model
|
|
40
|
+
export const PROVIDER_DEFAULT_MODEL = {
|
|
41
|
+
claude: 'claude-sonnet-4-6',
|
|
42
|
+
openai: 'gpt-4o',
|
|
43
|
+
openrouter: 'anthropic/claude-sonnet-4-5',
|
|
44
|
+
gemini: 'gemini-2.0-flash',
|
|
45
|
+
ollama: 'deepseek-coder-v2',
|
|
46
|
+
zai: 'glm-4-flash',
|
|
47
|
+
opencode: 'claude-sonnet-4-6',
|
|
48
|
+
};
|
|
49
|
+
const PROVIDER_ENDPOINTS = {
|
|
50
|
+
claude: ({ apiKey }) => ({
|
|
51
|
+
url: 'https://api.anthropic.com/v1/models',
|
|
52
|
+
headers: { 'x-api-key': apiKey, 'anthropic-version': '2023-06-01' },
|
|
53
|
+
}),
|
|
54
|
+
openai: ({ apiKey }) => ({
|
|
55
|
+
url: 'https://api.openai.com/v1/models',
|
|
56
|
+
headers: { Authorization: `Bearer ${apiKey}` },
|
|
57
|
+
}),
|
|
58
|
+
openrouter: ({ apiKey }) => ({
|
|
59
|
+
url: 'https://openrouter.ai/api/v1/models',
|
|
60
|
+
headers: { Authorization: `Bearer ${apiKey}` },
|
|
61
|
+
}),
|
|
62
|
+
gemini: ({ apiKey }) => ({
|
|
63
|
+
url: `https://generativelanguage.googleapis.com/v1beta/models?key=${apiKey}`,
|
|
64
|
+
headers: {},
|
|
65
|
+
}),
|
|
66
|
+
zai: ({ apiKey }) => ({
|
|
67
|
+
url: 'https://api.z.ai/v1/models',
|
|
68
|
+
headers: { Authorization: `Bearer ${apiKey}` },
|
|
69
|
+
}),
|
|
70
|
+
opencode: ({ apiKey }) => ({
|
|
71
|
+
url: 'https://opencode.ai/zen/v1/models',
|
|
72
|
+
headers: { Authorization: `Bearer ${apiKey}` },
|
|
73
|
+
}),
|
|
74
|
+
};
|
|
75
|
+
function parseModelIds(provider, data) {
|
|
76
|
+
if (!data || typeof data !== 'object')
|
|
77
|
+
return [];
|
|
78
|
+
// OpenAI-compatible: { data: [{ id }] }
|
|
79
|
+
if ('data' in data && Array.isArray(data.data)) {
|
|
80
|
+
return data.data
|
|
81
|
+
.map(m => m.id)
|
|
82
|
+
.filter((id) => typeof id === 'string');
|
|
83
|
+
}
|
|
84
|
+
// Anthropic: { data: [{ id }] } — same shape, covered above
|
|
85
|
+
// Gemini: { models: [{ name: "models/gemini-..." }] }
|
|
86
|
+
if ('models' in data && Array.isArray(data.models)) {
|
|
87
|
+
return data.models
|
|
88
|
+
.map(m => m.name?.replace('models/', '') ?? '')
|
|
89
|
+
.filter(Boolean);
|
|
90
|
+
}
|
|
91
|
+
// Ollama: { models: [{ name }] }
|
|
92
|
+
if (provider === 'ollama' && 'models' in data) {
|
|
93
|
+
return data.models
|
|
94
|
+
.map(m => m.name ?? '')
|
|
95
|
+
.filter(Boolean);
|
|
96
|
+
}
|
|
97
|
+
return [];
|
|
98
|
+
}
|
|
99
|
+
export async function fetchModels(provider, apiKey, baseUrl) {
|
|
100
|
+
const cacheKey = `${provider}:${apiKey.slice(-8)}`; // key by provider + last 8 chars of key
|
|
101
|
+
const cached = cache.get(cacheKey);
|
|
102
|
+
if (cached && Date.now() - cached.fetchedAt < CACHE_TTL) {
|
|
103
|
+
return cached.models;
|
|
104
|
+
}
|
|
105
|
+
// Ollama: hit local tags endpoint
|
|
106
|
+
if (provider === 'ollama') {
|
|
107
|
+
const base = baseUrl ?? 'http://localhost:11434';
|
|
108
|
+
try {
|
|
109
|
+
const res = await fetch(`${base}/api/tags`, { signal: AbortSignal.timeout(5000) });
|
|
110
|
+
if (!res.ok)
|
|
111
|
+
return FALLBACK_MODELS.ollama ?? [];
|
|
112
|
+
const data = await res.json();
|
|
113
|
+
const models = parseModelIds('ollama', data);
|
|
114
|
+
cache.set(cacheKey, { models, fetchedAt: Date.now() });
|
|
115
|
+
return models;
|
|
116
|
+
}
|
|
117
|
+
catch {
|
|
118
|
+
return FALLBACK_MODELS.ollama ?? [];
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
const endpointFn = PROVIDER_ENDPOINTS[provider];
|
|
122
|
+
if (!endpointFn)
|
|
123
|
+
return FALLBACK_MODELS[provider] ?? [];
|
|
124
|
+
const { url, headers } = endpointFn({ apiKey, baseUrl: baseUrl ?? '' });
|
|
125
|
+
try {
|
|
126
|
+
const res = await fetch(url, {
|
|
127
|
+
headers,
|
|
128
|
+
signal: AbortSignal.timeout(8000),
|
|
129
|
+
});
|
|
130
|
+
if (!res.ok)
|
|
131
|
+
return FALLBACK_MODELS[provider] ?? [];
|
|
132
|
+
const data = await res.json();
|
|
133
|
+
const models = parseModelIds(provider, data);
|
|
134
|
+
if (models.length === 0)
|
|
135
|
+
return FALLBACK_MODELS[provider] ?? [];
|
|
136
|
+
cache.set(cacheKey, { models, fetchedAt: Date.now() });
|
|
137
|
+
return models;
|
|
138
|
+
}
|
|
139
|
+
catch {
|
|
140
|
+
return FALLBACK_MODELS[provider] ?? [];
|
|
141
|
+
}
|
|
142
|
+
}
|
|
143
|
+
export function clearModelCache() {
|
|
144
|
+
cache.clear();
|
|
145
|
+
}
|
|
146
|
+
export function getCachedModels(provider, apiKey) {
|
|
147
|
+
const cacheKey = `${provider}:${apiKey.slice(-8)}`;
|
|
148
|
+
const cached = cache.get(cacheKey);
|
|
149
|
+
if (cached && Date.now() - cached.fetchedAt < CACHE_TTL)
|
|
150
|
+
return cached.models;
|
|
151
|
+
return null;
|
|
152
|
+
}
|
|
153
|
+
//# sourceMappingURL=model-registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-registry.js","sourceRoot":"","sources":["../../src/providers/model-registry.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAOH,MAAM,SAAS,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAA,CAAC,SAAS;AAC1C,MAAM,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAA;AAE3C,qDAAqD;AACrD,MAAM,eAAe,GAA6B;IAChD,MAAM,EAAE;QACN,iBAAiB;QACjB,mBAAmB;QACnB,kBAAkB;KACnB;IACD,MAAM,EAAE;QACN,QAAQ;QACR,aAAa;QACb,IAAI;QACJ,SAAS;KACV;IACD,UAAU,EAAE,EAAE;IACd,MAAM,EAAE;QACN,kBAAkB;QAClB,gBAAgB;QAChB,kBAAkB;KACnB;IACD,GAAG,EAAE;QACH,aAAa;QACb,YAAY;KACb;IACD,QAAQ,EAAE;QACR,mBAAmB;QACnB,kBAAkB;QAClB,cAAc;QACd,WAAW;QACX,OAAO;KACR;IACD,MAAM,EAAE,EAAE;CACX,CAAA;AAED,6EAA6E;AAC7E,MAAM,CAAC,MAAM,sBAAsB,GAA2B;IAC5D,MAAM,EAAE,mBAAmB;IAC3B,MAAM,EAAE,QAAQ;IAChB,UAAU,EAAE,6BAA6B;IACzC,MAAM,EAAE,kBAAkB;IAC1B,MAAM,EAAE,mBAAmB;IAC3B,GAAG,EAAE,aAAa;IAClB,QAAQ,EAAE,mBAAmB;CAC9B,CAAA;AAOD,MAAM,kBAAkB,GAA2F;IACjH,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACvB,GAAG,EAAE,qCAAqC;QAC1C,OAAO,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,mBAAmB,EAAE,YAAY,EAAE;KACpE,CAAC;IACF,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACvB,GAAG,EAAE,kCAAkC;QACvC,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;KAC/C,CAAC;IACF,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAC3B,GAAG,EAAE,qCAAqC;QAC1C,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;KAC/C,CAAC;IACF,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACvB,GAAG,EAAE,+DAA+D,MAAM,EAAE;QAC5E,OAAO,EAAE,EAAE;KACZ,CAAC;IACF,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACpB,GAAG,EAAE,4BAA4B;QACjC,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;KAC/C,CAAC;IACF,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACzB,GAAG,EAAE,mCAAmC;QACxC,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;KAC/C,CAAC;CACH,CAAA;AAED,SAAS,aAAa,CAAC,QAAgB,EAAE,IAAa;IACpD,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO,EAAE,CAAA;IAEhD,wCAAwC;IACxC,IAAI,MAAM,IAAI,IAAI,IAAI,KAAK,CAAC,OAAO,CAAE,IAA0B,CAAC,IAAI,CAAC,EAAE,CAAC;QACtE,OAAQ,IAAoC,CAAC,IAAI;aAC9C,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACd,MAAM,CAAC,CAAC,EAAE,EAAgB,EAAE,CAAC,OAAO,EAAE,KAAK,QAAQ,CAAC,CAAA;IACzD,CAAC;IAED,4DAA4D;IAE5D,sDAAsD;IACtD,IAAI,QAAQ,IAAI,IAAI,IAAI,KAAK,CAAC,OAAO,CAAE,IAA4B,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5E,OAAQ,IAAwC,CAAC,MAAM;aACpD,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;aAC9C,MAAM,CAAC,OAAO,CAAC,CAAA;IACpB,CAAC;IAED,iCAAiC;IACjC,IAAI,QAAQ,KAAK,QAAQ,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;QAC9C,OAAQ,IAAwC,CAAC,MAAM;aACpD,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;aACtB,MAAM,CAAC,OAAO,CAAC,CAAA;IACpB,CAAC;IAED,OAAO,EAAE,CAAA;AACX,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,QAAgB,EAAE,MAAc,EAAE,OAAgB;IAClF,MAAM,QAAQ,GAAG,GAAG,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA,CAAC,wCAAwC;IAC3F,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;IAClC,IAAI,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,SAAS,GAAG,SAAS,EAAE,CAAC;QACxD,OAAO,MAAM,CAAC,MAAM,CAAA;IACtB,CAAC;IAED,kCAAkC;IAClC,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,MAAM,IAAI,GAAG,OAAO,IAAI,wBAAwB,CAAA;QAChD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,WAAW,EAAE,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;YAClF,IAAI,CAAC,GAAG,CAAC,EAAE;gBAAE,OAAO,eAAe,CAAC,MAAM,IAAI,EAAE,CAAA;YAChD,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAa,CAAA;YACxC,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;YAC5C,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;YACtD,OAAO,MAAM,CAAA;QACf,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,eAAe,CAAC,MAAM,IAAI,EAAE,CAAA;QACrC,CAAC;IACH,CAAC;IAED,MAAM,UAAU,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAA;IAC/C,IAAI,CAAC,UAAU;QAAE,OAAO,eAAe,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAA;IAEvD,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,EAAE,CAAC,CAAA;IAEvE,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAC3B,OAAO;YACP,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;SAClC,CAAC,CAAA;QACF,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,OAAO,eAAe,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAA;QACnD,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAa,CAAA;QACxC,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;QAC5C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,eAAe,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAA;QAC/D,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;QACtD,OAAO,MAAM,CAAA;IACf,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,eAAe,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAA;IACxC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,eAAe;IAC7B,KAAK,CAAC,KAAK,EAAE,CAAA;AACf,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,QAAgB,EAAE,MAAc;IAC9D,MAAM,QAAQ,GAAG,GAAG,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;IAClD,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;IAClC,IAAI,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,SAAS,GAAG,SAAS;QAAE,OAAO,MAAM,CAAC,MAAM,CAAA;IAC7E,OAAO,IAAI,CAAA;AACb,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { ChatChunk, ChatMessage, ChatOptions, ChatResponse, LLMProvider } from './base-provider.js';
|
|
2
|
+
export declare class OllamaProvider implements LLMProvider {
|
|
3
|
+
private readonly baseUrl;
|
|
4
|
+
private readonly model;
|
|
5
|
+
readonly name = "ollama";
|
|
6
|
+
constructor(baseUrl: string, model: string);
|
|
7
|
+
chat(messages: ChatMessage[], options?: ChatOptions): Promise<ChatResponse>;
|
|
8
|
+
stream(messages: ChatMessage[], options?: ChatOptions): AsyncGenerator<ChatChunk>;
|
|
9
|
+
private buildOptions;
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=ollama.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../src/providers/ollama.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAcxG,qBAAa,cAAe,YAAW,WAAW;IAI9C,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,KAAK;IAJxB,QAAQ,CAAC,IAAI,YAAW;gBAGL,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM;IAG1B,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAsC1E,MAAM,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,cAAc,CAAC,SAAS,CAAC;IA4FxF,OAAO,CAAC,YAAY;CAqBrB"}
|
|
@@ -0,0 +1,141 @@
|
|
|
1
|
+
import { OrkaError } from '../core/errors.js';
|
|
2
|
+
function mapOllamaError(err, model) {
|
|
3
|
+
const e = err;
|
|
4
|
+
if (e.code === 'ECONNREFUSED' || e.message?.includes('ECONNREFUSED') || e.message?.includes('fetch failed')) {
|
|
5
|
+
return new OrkaError('ollama_unreachable', 'Ollama tidak berjalan. Jalankan `ollama serve` terlebih dahulu.', err);
|
|
6
|
+
}
|
|
7
|
+
if (e.status === 404) {
|
|
8
|
+
return new OrkaError('ollama_model_not_found', `Model ${model} tidak tersedia. Jalankan \`ollama pull ${model}\`.`, err);
|
|
9
|
+
}
|
|
10
|
+
return new OrkaError('provider_unreachable', `Ollama error: ${e.message ?? String(err)}`, err);
|
|
11
|
+
}
|
|
12
|
+
export class OllamaProvider {
|
|
13
|
+
baseUrl;
|
|
14
|
+
model;
|
|
15
|
+
name = 'ollama';
|
|
16
|
+
constructor(baseUrl, model) {
|
|
17
|
+
this.baseUrl = baseUrl;
|
|
18
|
+
this.model = model;
|
|
19
|
+
}
|
|
20
|
+
async chat(messages, options) {
|
|
21
|
+
try {
|
|
22
|
+
const response = await fetch(`${this.baseUrl}/api/chat`, {
|
|
23
|
+
method: 'POST',
|
|
24
|
+
headers: { 'Content-Type': 'application/json' },
|
|
25
|
+
body: JSON.stringify({
|
|
26
|
+
model: this.model,
|
|
27
|
+
messages,
|
|
28
|
+
stream: false,
|
|
29
|
+
options: this.buildOptions(options),
|
|
30
|
+
}),
|
|
31
|
+
});
|
|
32
|
+
if (!response.ok) {
|
|
33
|
+
const body = await response.text();
|
|
34
|
+
const err = new Error(`Ollama error ${response.status}: ${body}`);
|
|
35
|
+
err.status = response.status;
|
|
36
|
+
throw err;
|
|
37
|
+
}
|
|
38
|
+
const data = (await response.json());
|
|
39
|
+
return {
|
|
40
|
+
content: data.message?.content ?? '',
|
|
41
|
+
usage: {
|
|
42
|
+
inputTokens: data.prompt_eval_count ?? 0,
|
|
43
|
+
outputTokens: data.eval_count ?? 0,
|
|
44
|
+
},
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
catch (err) {
|
|
48
|
+
throw mapOllamaError(err, this.model);
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
async *stream(messages, options) {
|
|
52
|
+
try {
|
|
53
|
+
const response = await fetch(`${this.baseUrl}/api/chat`, {
|
|
54
|
+
method: 'POST',
|
|
55
|
+
headers: { 'Content-Type': 'application/json' },
|
|
56
|
+
body: JSON.stringify({
|
|
57
|
+
model: this.model,
|
|
58
|
+
messages,
|
|
59
|
+
stream: true,
|
|
60
|
+
options: this.buildOptions(options),
|
|
61
|
+
}),
|
|
62
|
+
});
|
|
63
|
+
if (!response.ok) {
|
|
64
|
+
const body = await response.text();
|
|
65
|
+
const err = new Error(`Ollama error ${response.status}: ${body}`);
|
|
66
|
+
err.status = response.status;
|
|
67
|
+
throw err;
|
|
68
|
+
}
|
|
69
|
+
const reader = response.body?.getReader();
|
|
70
|
+
if (!reader) {
|
|
71
|
+
return;
|
|
72
|
+
}
|
|
73
|
+
const decoder = new TextDecoder();
|
|
74
|
+
let buffer = '';
|
|
75
|
+
while (true) {
|
|
76
|
+
const { done, value } = await reader.read();
|
|
77
|
+
if (done) {
|
|
78
|
+
break;
|
|
79
|
+
}
|
|
80
|
+
buffer += decoder.decode(value, { stream: true });
|
|
81
|
+
const lines = buffer.split('\n');
|
|
82
|
+
buffer = lines.pop() ?? '';
|
|
83
|
+
for (const line of lines) {
|
|
84
|
+
if (!line.trim()) {
|
|
85
|
+
continue;
|
|
86
|
+
}
|
|
87
|
+
const data = JSON.parse(line);
|
|
88
|
+
if (data.done) {
|
|
89
|
+
yield {
|
|
90
|
+
content: data.message?.content ?? '',
|
|
91
|
+
done: true,
|
|
92
|
+
usage: {
|
|
93
|
+
inputTokens: data.prompt_eval_count ?? 0,
|
|
94
|
+
outputTokens: data.eval_count ?? 0,
|
|
95
|
+
},
|
|
96
|
+
};
|
|
97
|
+
}
|
|
98
|
+
else {
|
|
99
|
+
yield {
|
|
100
|
+
content: data.message?.content ?? '',
|
|
101
|
+
done: false,
|
|
102
|
+
};
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
if (buffer.trim()) {
|
|
107
|
+
const data = JSON.parse(buffer);
|
|
108
|
+
yield {
|
|
109
|
+
content: data.message?.content ?? '',
|
|
110
|
+
done: data.done ?? true,
|
|
111
|
+
usage: data.done
|
|
112
|
+
? {
|
|
113
|
+
inputTokens: data.prompt_eval_count ?? 0,
|
|
114
|
+
outputTokens: data.eval_count ?? 0,
|
|
115
|
+
}
|
|
116
|
+
: undefined,
|
|
117
|
+
};
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
catch (err) {
|
|
121
|
+
throw mapOllamaError(err, this.model);
|
|
122
|
+
}
|
|
123
|
+
}
|
|
124
|
+
buildOptions(options) {
|
|
125
|
+
if (!options) {
|
|
126
|
+
return undefined;
|
|
127
|
+
}
|
|
128
|
+
const built = {};
|
|
129
|
+
if (options.temperature !== undefined) {
|
|
130
|
+
built.temperature = options.temperature;
|
|
131
|
+
}
|
|
132
|
+
if (options.maxTokens !== undefined) {
|
|
133
|
+
built.num_predict = options.maxTokens;
|
|
134
|
+
}
|
|
135
|
+
if (options.stopSequences?.length) {
|
|
136
|
+
built.stop = options.stopSequences;
|
|
137
|
+
}
|
|
138
|
+
return Object.keys(built).length > 0 ? built : undefined;
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
//# sourceMappingURL=ollama.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/providers/ollama.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAA;AAE7C,SAAS,cAAc,CAAC,GAAY,EAAE,KAAa;IACjD,MAAM,CAAC,GAAG,GAA2D,CAAA;IACrE,IAAI,CAAC,CAAC,IAAI,KAAK,cAAc,IAAI,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;QAC5G,OAAO,IAAI,SAAS,CAAC,oBAAoB,EAAE,iEAAiE,EAAE,GAAY,CAAC,CAAA;IAC7H,CAAC;IACD,IAAI,CAAC,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACrB,OAAO,IAAI,SAAS,CAAC,wBAAwB,EAAE,SAAS,KAAK,2CAA2C,KAAK,KAAK,EAAE,GAAY,CAAC,CAAA;IACnI,CAAC;IACD,OAAO,IAAI,SAAS,CAAC,sBAAsB,EAAE,iBAAiB,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,GAAY,CAAC,CAAA;AACzG,CAAC;AAED,MAAM,OAAO,cAAc;IAIN;IACA;IAJV,IAAI,GAAG,QAAQ,CAAA;IAExB,YACmB,OAAe,EACf,KAAa;QADb,YAAO,GAAP,OAAO,CAAQ;QACf,UAAK,GAAL,KAAK,CAAQ;IAC7B,CAAC;IAEJ,KAAK,CAAC,IAAI,CAAC,QAAuB,EAAE,OAAqB;QACvD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;gBACvD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,QAAQ;oBACR,MAAM,EAAE,KAAK;oBACb,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;iBACpC,CAAC;aACH,CAAC,CAAA;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAA;gBAClC,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,gBAAgB,QAAQ,CAAC,MAAM,KAAK,IAAI,EAAE,CAA0C,CAAA;gBAC1G,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAA;gBAC5B,MAAM,GAAG,CAAA;YACX,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAIlC,CAAA;YAED,OAAO;gBACL,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE;gBACpC,KAAK,EAAE;oBACL,WAAW,EAAE,IAAI,CAAC,iBAAiB,IAAI,CAAC;oBACxC,YAAY,EAAE,IAAI,CAAC,UAAU,IAAI,CAAC;iBACnC;aACF,CAAA;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,CAAA;QACvC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,QAAuB,EAAE,OAAqB;QAC1D,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;gBACvD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,QAAQ;oBACR,MAAM,EAAE,IAAI;oBACZ,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;iBACpC,CAAC;aACH,CAAC,CAAA;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAA;gBAClC,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,gBAAgB,QAAQ,CAAC,MAAM,KAAK,IAAI,EAAE,CAA0C,CAAA;gBAC1G,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAA;gBAC5B,MAAM,GAAG,CAAA;YACX,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAA;YACzC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAM;YACR,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;YACjC,IAAI,MAAM,GAAG,EAAE,CAAA;YAEf,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAA;gBAC3C,IAAI,IAAI,EAAE,CAAC;oBACT,MAAK;gBACP,CAAC;gBAED,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAA;gBACjD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;gBAChC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAA;gBAE1B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;wBACjB,SAAQ;oBACV,CAAC;oBAED,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAK3B,CAAA;oBAED,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBACd,MAAM;4BACJ,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE;4BACpC,IAAI,EAAE,IAAI;4BACV,KAAK,EAAE;gCACL,WAAW,EAAE,IAAI,CAAC,iBAAiB,IAAI,CAAC;gCACxC,YAAY,EAAE,IAAI,CAAC,UAAU,IAAI,CAAC;6BACnC;yBACF,CAAA;oBACH,CAAC;yBAAM,CAAC;wBACN,MAAM;4BACJ,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE;4BACpC,IAAI,EAAE,KAAK;yBACZ,CAAA;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAK7B,CAAA;gBAED,MAAM;oBACJ,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE;oBACpC,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI;oBACvB,KAAK,EAAE,IAAI,CAAC,IAAI;wBACd,CAAC,CAAC;4BACE,WAAW,EAAE,IAAI,CAAC,iBAAiB,IAAI,CAAC;4BACxC,YAAY,EAAE,IAAI,CAAC,UAAU,IAAI,CAAC;yBACnC;wBACH,CAAC,CAAC,SAAS;iBACd,CAAA;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,CAAA;QACvC,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,OAAqB;QACxC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,SAAS,CAAA;QAClB,CAAC;QAED,MAAM,KAAK,GAA4B,EAAE,CAAA;QAEzC,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YACtC,KAAK,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAA;QACzC,CAAC;QAED,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACpC,KAAK,CAAC,WAAW,GAAG,OAAO,CAAC,SAAS,CAAA;QACvC,CAAC;QAED,IAAI,OAAO,CAAC,aAAa,EAAE,MAAM,EAAE,CAAC;YAClC,KAAK,CAAC,IAAI,GAAG,OAAO,CAAC,aAAa,CAAA;QACpC,CAAC;QAED,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAA;IAC1D,CAAC;CACF"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import type { ChatChunk, ChatMessage, ChatOptions, ChatResponse, LLMProvider } from './base-provider.js';
|
|
2
|
+
export interface OpenAIProviderOptions {
|
|
3
|
+
apiKey: string;
|
|
4
|
+
model?: string;
|
|
5
|
+
baseUrl?: string;
|
|
6
|
+
}
|
|
7
|
+
export declare const RETRY_DELAYS: number[];
|
|
8
|
+
export declare class OpenAIProvider implements LLMProvider {
|
|
9
|
+
readonly name: string;
|
|
10
|
+
private readonly client;
|
|
11
|
+
private readonly model;
|
|
12
|
+
private readonly providerLabel;
|
|
13
|
+
constructor(options: OpenAIProviderOptions);
|
|
14
|
+
chat(messages: ChatMessage[], opts?: ChatOptions): Promise<ChatResponse>;
|
|
15
|
+
stream(messages: ChatMessage[], opts?: ChatOptions): AsyncIterable<ChatChunk>;
|
|
16
|
+
}
|
|
17
|
+
//# sourceMappingURL=openai.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAExG,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,OAAO,CAAC,EAAE,MAAM,CAAA;CACjB;AAED,eAAO,MAAM,YAAY,UAAuB,CAAA;AA6ChD,qBAAa,cAAe,YAAW,WAAW;IAChD,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;IACrB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAQ;IAC/B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAQ;IAC9B,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAQ;gBAE1B,OAAO,EAAE,qBAAqB;IAkBpC,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,IAAI,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAkBvE,MAAM,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,IAAI,CAAC,EAAE,WAAW,GAAG,aAAa,CAAC,SAAS,CAAC;CAyBrF"}
|
|
@@ -0,0 +1,109 @@
|
|
|
1
|
+
import OpenAI from 'openai';
|
|
2
|
+
import { OrkaError } from '../core/errors.js';
|
|
3
|
+
export const RETRY_DELAYS = [5000, 15000, 30000];
|
|
4
|
+
function mapError(err, providerLabel = 'OpenAI') {
|
|
5
|
+
const e = err;
|
|
6
|
+
// OpenCode-specific: CreditsError comes as 401 with error.type in body
|
|
7
|
+
if (e.status === 401 && e.error?.type === 'CreditsError') {
|
|
8
|
+
return new OrkaError('rate_limit', `Saldo ${providerLabel} habis. Top up di dashboard provider kamu.`, err);
|
|
9
|
+
}
|
|
10
|
+
if (e.status === 401 || e.status === 403) {
|
|
11
|
+
return new OrkaError('api_key_invalid', `API key ${providerLabel} tidak valid.`, err);
|
|
12
|
+
}
|
|
13
|
+
if (e.code === 'context_length_exceeded' ||
|
|
14
|
+
(e.status === 400 && e.message?.includes('maximum context length'))) {
|
|
15
|
+
return new OrkaError('context_limit', `Konteks melebihi batas model ${providerLabel}. Coba goal lebih spesifik.`, err);
|
|
16
|
+
}
|
|
17
|
+
if (e.status === 429) {
|
|
18
|
+
return new OrkaError('rate_limit', `Rate limit ${providerLabel} tercapai. Coba lagi dalam beberapa saat.`, err);
|
|
19
|
+
}
|
|
20
|
+
if (e.code === 'ECONNREFUSED' || e.message?.includes('fetch failed')) {
|
|
21
|
+
return new OrkaError('provider_unreachable', `Tidak bisa terhubung ke ${providerLabel} API. Cek koneksi internet.`, err);
|
|
22
|
+
}
|
|
23
|
+
return new OrkaError('provider_unreachable', `${providerLabel} error: ${e.message ?? String(err)}`, err);
|
|
24
|
+
}
|
|
25
|
+
async function withRateLimitRetry(fn, providerLabel) {
|
|
26
|
+
for (let attempt = 0; attempt <= RETRY_DELAYS.length; attempt++) {
|
|
27
|
+
try {
|
|
28
|
+
return await fn();
|
|
29
|
+
}
|
|
30
|
+
catch (err) {
|
|
31
|
+
const orkaErr = err instanceof OrkaError ? err : mapError(err, providerLabel);
|
|
32
|
+
if (orkaErr.code !== 'rate_limit' || attempt === RETRY_DELAYS.length)
|
|
33
|
+
throw orkaErr;
|
|
34
|
+
await new Promise(r => setTimeout(r, RETRY_DELAYS[attempt]));
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
throw new OrkaError('rate_limit', `Rate limit ${providerLabel} — semua retry habis.`);
|
|
38
|
+
}
|
|
39
|
+
export class OpenAIProvider {
|
|
40
|
+
name;
|
|
41
|
+
client;
|
|
42
|
+
model;
|
|
43
|
+
providerLabel;
|
|
44
|
+
constructor(options) {
|
|
45
|
+
this.model = options.model ?? 'gpt-4o';
|
|
46
|
+
this.client = new OpenAI({ apiKey: options.apiKey, baseURL: options.baseUrl });
|
|
47
|
+
if (options.baseUrl?.includes('openrouter')) {
|
|
48
|
+
this.providerLabel = 'OpenRouter';
|
|
49
|
+
this.name = 'openrouter';
|
|
50
|
+
}
|
|
51
|
+
else if (options.baseUrl?.includes('opencode.ai')) {
|
|
52
|
+
this.providerLabel = 'OpenCode';
|
|
53
|
+
this.name = 'opencode';
|
|
54
|
+
}
|
|
55
|
+
else if (options.baseUrl?.includes('z.ai')) {
|
|
56
|
+
this.providerLabel = 'Z.ai';
|
|
57
|
+
this.name = 'zai';
|
|
58
|
+
}
|
|
59
|
+
else {
|
|
60
|
+
this.providerLabel = 'OpenAI';
|
|
61
|
+
this.name = 'openai';
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
async chat(messages, opts) {
|
|
65
|
+
return withRateLimitRetry(async () => {
|
|
66
|
+
const res = await this.client.chat.completions.create({
|
|
67
|
+
model: this.model,
|
|
68
|
+
messages: messages.map(m => ({ role: m.role, content: m.content })),
|
|
69
|
+
max_tokens: opts?.maxTokens,
|
|
70
|
+
stream: false,
|
|
71
|
+
});
|
|
72
|
+
return {
|
|
73
|
+
content: res.choices[0]?.message.content ?? '',
|
|
74
|
+
usage: {
|
|
75
|
+
inputTokens: res.usage?.prompt_tokens ?? 0,
|
|
76
|
+
outputTokens: res.usage?.completion_tokens ?? 0,
|
|
77
|
+
},
|
|
78
|
+
};
|
|
79
|
+
}, this.providerLabel);
|
|
80
|
+
}
|
|
81
|
+
async *stream(messages, opts) {
|
|
82
|
+
try {
|
|
83
|
+
const s = await this.client.chat.completions.create({
|
|
84
|
+
model: this.model,
|
|
85
|
+
messages: messages.map(m => ({ role: m.role, content: m.content })),
|
|
86
|
+
max_tokens: opts?.maxTokens,
|
|
87
|
+
stream: true,
|
|
88
|
+
stream_options: { include_usage: true },
|
|
89
|
+
});
|
|
90
|
+
let lastUsage;
|
|
91
|
+
for await (const chunk of s) {
|
|
92
|
+
if (chunk.usage) {
|
|
93
|
+
lastUsage = {
|
|
94
|
+
inputTokens: chunk.usage.prompt_tokens,
|
|
95
|
+
outputTokens: chunk.usage.completion_tokens,
|
|
96
|
+
};
|
|
97
|
+
}
|
|
98
|
+
const content = chunk.choices[0]?.delta?.content ?? '';
|
|
99
|
+
if (content)
|
|
100
|
+
yield { content, done: false };
|
|
101
|
+
}
|
|
102
|
+
yield { content: '', done: true, usage: lastUsage };
|
|
103
|
+
}
|
|
104
|
+
catch (err) {
|
|
105
|
+
throw mapError(err, this.providerLabel);
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
//# sourceMappingURL=openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAA;AAC3B,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAA;AAS7C,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAA;AAEhD,SAAS,QAAQ,CAAC,GAAY,EAAE,aAAa,GAAG,QAAQ;IACtD,MAAM,CAAC,GAAG,GAAwG,CAAA;IAElH,uEAAuE;IACvE,IAAI,CAAC,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,KAAK,EAAE,IAAI,KAAK,cAAc,EAAE,CAAC;QACzD,OAAO,IAAI,SAAS,CAClB,YAAY,EACZ,SAAS,aAAa,4CAA4C,EAClE,GAAY,CACb,CAAA;IACH,CAAC;IAED,IAAI,CAAC,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACzC,OAAO,IAAI,SAAS,CAAC,iBAAiB,EAAE,WAAW,aAAa,eAAe,EAAE,GAAY,CAAC,CAAA;IAChG,CAAC;IACD,IACG,CAAuB,CAAC,IAAI,KAAK,yBAAyB;QAC3D,CAAC,CAAC,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,wBAAwB,CAAC,CAAC,EACnE,CAAC;QACD,OAAO,IAAI,SAAS,CAAC,eAAe,EAAE,gCAAgC,aAAa,6BAA6B,EAAE,GAAY,CAAC,CAAA;IACjI,CAAC;IACD,IAAI,CAAC,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACrB,OAAO,IAAI,SAAS,CAAC,YAAY,EAAE,cAAc,aAAa,2CAA2C,EAAE,GAAY,CAAC,CAAA;IAC1H,CAAC;IACD,IAAI,CAAC,CAAC,IAAI,KAAK,cAAc,IAAI,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;QACrE,OAAO,IAAI,SAAS,CAAC,sBAAsB,EAAE,2BAA2B,aAAa,6BAA6B,EAAE,GAAY,CAAC,CAAA;IACnI,CAAC;IACD,OAAO,IAAI,SAAS,CAAC,sBAAsB,EAAE,GAAG,aAAa,WAAW,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,GAAY,CAAC,CAAA;AACnH,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAI,EAAoB,EAAE,aAAqB;IAC9E,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC;QAChE,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAA;QACnB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,aAAa,CAAC,CAAA;YAC7E,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,IAAI,OAAO,KAAK,YAAY,CAAC,MAAM;gBAAE,MAAM,OAAO,CAAA;YACnF,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;QAC9D,CAAC;IACH,CAAC;IACD,MAAM,IAAI,SAAS,CAAC,YAAY,EAAE,cAAc,aAAa,uBAAuB,CAAC,CAAA;AACvF,CAAC;AAED,MAAM,OAAO,cAAc;IAChB,IAAI,CAAQ;IACJ,MAAM,CAAQ;IACd,KAAK,CAAQ;IACb,aAAa,CAAQ;IAEtC,YAAY,OAA8B;QACxC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,QAAQ,CAAA;QACtC,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAA;QAC9E,IAAI,OAAO,CAAC,OAAO,EAAE,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YAC5C,IAAI,CAAC,aAAa,GAAG,YAAY,CAAA;YACjC,IAAI,CAAC,IAAI,GAAG,YAAY,CAAA;QAC1B,CAAC;aAAM,IAAI,OAAO,CAAC,OAAO,EAAE,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACpD,IAAI,CAAC,aAAa,GAAG,UAAU,CAAA;YAC/B,IAAI,CAAC,IAAI,GAAG,UAAU,CAAA;QACxB,CAAC;aAAM,IAAI,OAAO,CAAC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7C,IAAI,CAAC,aAAa,GAAG,MAAM,CAAA;YAC3B,IAAI,CAAC,IAAI,GAAG,KAAK,CAAA;QACnB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,aAAa,GAAG,QAAQ,CAAA;YAC7B,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAA;QACtB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAuB,EAAE,IAAkB;QACpD,OAAO,kBAAkB,CAAC,KAAK,IAAI,EAAE;YACnC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACpD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;gBACnE,UAAU,EAAE,IAAI,EAAE,SAAS;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAA;YACF,OAAO;gBACL,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;gBAC9C,KAAK,EAAE;oBACL,WAAW,EAAE,GAAG,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;oBAC1C,YAAY,EAAE,GAAG,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;iBAChD;aACF,CAAA;QACH,CAAC,EAAE,IAAI,CAAC,aAAa,CAAC,CAAA;IACxB,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,QAAuB,EAAE,IAAkB;QACvD,IAAI,CAAC;YACH,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBAClD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;gBACnE,UAAU,EAAE,IAAI,EAAE,SAAS;gBAC3B,MAAM,EAAE,IAAI;gBACZ,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;aACxC,CAAC,CAAA;YACF,IAAI,SAAoE,CAAA;YACxE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBAChB,SAAS,GAAG;wBACV,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,aAAa;wBACtC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,iBAAiB;qBAC5C,CAAA;gBACH,CAAC;gBACD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,IAAI,EAAE,CAAA;gBACtD,IAAI,OAAO;oBAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAA;YAC7C,CAAC;YACD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAA;QACrD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAA;QACzC,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import type { LLMProvider } from './base-provider.js';
|
|
2
|
+
export declare const ZAI_BASE_URL = "https://api.z.ai/v1";
|
|
3
|
+
export declare const ZAI_DEFAULT_MODEL = "glm-4-flash";
|
|
4
|
+
export declare const ZAI_MODELS: readonly ["glm-4.7", "glm-4.7-flash", "glm-4-flash", "glm-4"];
|
|
5
|
+
export declare function createZaiProvider(apiKey: string, model?: string): LLMProvider;
|
|
6
|
+
//# sourceMappingURL=zai.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"zai.d.ts","sourceRoot":"","sources":["../../src/providers/zai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAErD,eAAO,MAAM,YAAY,wBAAwB,CAAA;AACjD,eAAO,MAAM,iBAAiB,gBAAgB,CAAA;AAE9C,eAAO,MAAM,UAAU,+DAKb,CAAA;AAEV,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,WAAW,CAM7E"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { OpenAIProvider } from './openai.js';
|
|
2
|
+
export const ZAI_BASE_URL = 'https://api.z.ai/v1';
|
|
3
|
+
export const ZAI_DEFAULT_MODEL = 'glm-4-flash';
|
|
4
|
+
export const ZAI_MODELS = [
|
|
5
|
+
'glm-4.7',
|
|
6
|
+
'glm-4.7-flash',
|
|
7
|
+
'glm-4-flash',
|
|
8
|
+
'glm-4',
|
|
9
|
+
];
|
|
10
|
+
export function createZaiProvider(apiKey, model) {
|
|
11
|
+
return new OpenAIProvider({
|
|
12
|
+
apiKey,
|
|
13
|
+
model: model ?? ZAI_DEFAULT_MODEL,
|
|
14
|
+
baseUrl: ZAI_BASE_URL,
|
|
15
|
+
});
|
|
16
|
+
}
|
|
17
|
+
//# sourceMappingURL=zai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"zai.js","sourceRoot":"","sources":["../../src/providers/zai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AAG5C,MAAM,CAAC,MAAM,YAAY,GAAG,qBAAqB,CAAA;AACjD,MAAM,CAAC,MAAM,iBAAiB,GAAG,aAAa,CAAA;AAE9C,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,SAAS;IACT,eAAe;IACf,aAAa;IACb,OAAO;CACC,CAAA;AAEV,MAAM,UAAU,iBAAiB,CAAC,MAAc,EAAE,KAAc;IAC9D,OAAO,IAAI,cAAc,CAAC;QACxB,MAAM;QACN,KAAK,EAAE,KAAK,IAAI,iBAAiB;QACjC,OAAO,EAAE,YAAY;KACtB,CAAC,CAAA;AACJ,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import type { IncomingMessage, ServerResponse } from 'node:http';
|
|
2
|
+
import type { StateStore } from '../core/state.js';
|
|
3
|
+
import type { ServeOptions } from './index.js';
|
|
4
|
+
export declare function createApiRouter(store: StateStore, options: ServeOptions): (req: IncomingMessage, res: ServerResponse, url: URL) => void;
|
|
5
|
+
//# sourceMappingURL=api.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"api.d.ts","sourceRoot":"","sources":["../../src/server/api.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAIlD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAA;AA2B9C,wBAAgB,eAAe,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,IAiI/C,KAAK,eAAe,EAAE,KAAK,cAAc,EAAE,KAAK,GAAG,KAAG,IAAI,CAelF"}
|