@castlemilk/omega 0.6.0 → 0.6.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/server/node_modules/.pnpm/lock.yaml +9 -0
- package/dist/server/node_modules/@omega/agent/dist/executor.d.ts +12 -0
- package/dist/server/node_modules/@omega/agent/dist/executor.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/executor.js +352 -0
- package/dist/server/node_modules/@omega/agent/dist/executor.js.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/git.d.ts +15 -0
- package/dist/server/node_modules/@omega/agent/dist/git.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/git.js +56 -0
- package/dist/server/node_modules/@omega/agent/dist/git.js.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/index.d.ts +7 -0
- package/dist/server/node_modules/@omega/agent/dist/index.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/index.js +7 -0
- package/dist/server/node_modules/@omega/agent/dist/index.js.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/planner.d.ts +12 -0
- package/dist/server/node_modules/@omega/agent/dist/planner.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/planner.js +64 -0
- package/dist/server/node_modules/@omega/agent/dist/planner.js.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/prompts.d.ts +7 -0
- package/dist/server/node_modules/@omega/agent/dist/prompts.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/prompts.js +55 -0
- package/dist/server/node_modules/@omega/agent/dist/prompts.js.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/publisher.d.ts +7 -0
- package/dist/server/node_modules/@omega/agent/dist/publisher.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/publisher.js +57 -0
- package/dist/server/node_modules/@omega/agent/dist/publisher.js.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/tools.d.ts +10 -0
- package/dist/server/node_modules/@omega/agent/dist/tools.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/tools.js +92 -0
- package/dist/server/node_modules/@omega/agent/dist/tools.js.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/validator.d.ts +17 -0
- package/dist/server/node_modules/@omega/agent/dist/validator.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/agent/dist/validator.js +28 -0
- package/dist/server/node_modules/@omega/agent/dist/validator.js.map +1 -0
- package/dist/server/node_modules/@omega/agent/package.json +19 -0
- package/dist/server/node_modules/@omega/core/dist/agent.d.ts +51 -0
- package/dist/server/node_modules/@omega/core/dist/agent.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/agent.js +2 -0
- package/dist/server/node_modules/@omega/core/dist/agent.js.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/index.d.ts +5 -0
- package/dist/server/node_modules/@omega/core/dist/index.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/index.js +5 -0
- package/dist/server/node_modules/@omega/core/dist/index.js.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/provider.d.ts +48 -0
- package/dist/server/node_modules/@omega/core/dist/provider.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/provider.js +2 -0
- package/dist/server/node_modules/@omega/core/dist/provider.js.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/provider.test.d.ts +2 -0
- package/dist/server/node_modules/@omega/core/dist/provider.test.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/provider.test.js +27 -0
- package/dist/server/node_modules/@omega/core/dist/provider.test.js.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/skill.d.ts +21 -0
- package/dist/server/node_modules/@omega/core/dist/skill.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/skill.js +2 -0
- package/dist/server/node_modules/@omega/core/dist/skill.js.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/task.d.ts +27 -0
- package/dist/server/node_modules/@omega/core/dist/task.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/core/dist/task.js +2 -0
- package/dist/server/node_modules/@omega/core/dist/task.js.map +1 -0
- package/dist/server/node_modules/@omega/core/package.json +13 -0
- package/dist/server/node_modules/@omega/db/dist/client.d.ts +8 -0
- package/dist/server/node_modules/@omega/db/dist/client.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/db/dist/client.js +8 -0
- package/dist/server/node_modules/@omega/db/dist/client.js.map +1 -0
- package/dist/server/node_modules/@omega/db/dist/index.d.ts +5 -0
- package/dist/server/node_modules/@omega/db/dist/index.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/db/dist/index.js +5 -0
- package/dist/server/node_modules/@omega/db/dist/index.js.map +1 -0
- package/dist/server/node_modules/@omega/db/dist/migrate.d.ts +2 -0
- package/dist/server/node_modules/@omega/db/dist/migrate.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/db/dist/migrate.js +58 -0
- package/dist/server/node_modules/@omega/db/dist/migrate.js.map +1 -0
- package/dist/server/node_modules/@omega/db/dist/seed.d.ts +2 -0
- package/dist/server/node_modules/@omega/db/dist/seed.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/db/dist/seed.js +42 -0
- package/dist/server/node_modules/@omega/db/dist/seed.js.map +1 -0
- package/dist/server/node_modules/@omega/db/generated/client/default.d.ts +1 -0
- package/dist/server/node_modules/@omega/db/generated/client/default.js +1 -0
- package/dist/server/node_modules/@omega/db/generated/client/edge.d.ts +1 -0
- package/dist/server/node_modules/@omega/db/generated/client/edge.js +282 -0
- package/dist/server/node_modules/@omega/db/generated/client/index-browser.js +272 -0
- package/dist/server/node_modules/@omega/db/generated/client/index.d.ts +12787 -0
- package/dist/server/node_modules/@omega/db/generated/client/index.js +303 -0
- package/dist/server/node_modules/@omega/db/generated/client/libquery_engine-darwin-arm64.dylib.node +0 -0
- package/dist/server/node_modules/@omega/db/generated/client/package.json +124 -0
- package/dist/server/node_modules/@omega/db/generated/client/query_engine_bg.js +2 -0
- package/dist/server/node_modules/@omega/db/generated/client/query_engine_bg.wasm +0 -0
- package/dist/server/node_modules/@omega/db/generated/client/runtime/edge-esm.js +31 -0
- package/dist/server/node_modules/@omega/db/generated/client/runtime/edge.js +31 -0
- package/dist/server/node_modules/@omega/db/generated/client/runtime/index-browser.d.ts +365 -0
- package/dist/server/node_modules/@omega/db/generated/client/runtime/index-browser.js +13 -0
- package/dist/server/node_modules/@omega/db/generated/client/runtime/library.d.ts +3403 -0
- package/dist/server/node_modules/@omega/db/generated/client/runtime/library.js +143 -0
- package/dist/server/node_modules/@omega/db/generated/client/runtime/react-native.js +80 -0
- package/dist/server/node_modules/@omega/db/generated/client/runtime/wasm.js +32 -0
- package/dist/server/node_modules/@omega/db/generated/client/schema.prisma +119 -0
- package/dist/server/node_modules/@omega/db/generated/client/wasm-edge-light-loader.mjs +1 -0
- package/dist/server/node_modules/@omega/db/generated/client/wasm-worker-loader.mjs +1 -0
- package/dist/server/node_modules/@omega/db/generated/client/wasm.d.ts +1 -0
- package/dist/server/node_modules/@omega/db/generated/client/wasm.js +289 -0
- package/dist/server/node_modules/@omega/db/package.json +18 -0
- package/dist/server/node_modules/@omega/db/prisma/migrations/20260702110000_init/migration.sql +154 -0
- package/dist/server/node_modules/@omega/providers/dist/anthropic.d.ts +10 -0
- package/dist/server/node_modules/@omega/providers/dist/anthropic.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/anthropic.js +50 -0
- package/dist/server/node_modules/@omega/providers/dist/anthropic.js.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/gemini.d.ts +9 -0
- package/dist/server/node_modules/@omega/providers/dist/gemini.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/gemini.js +46 -0
- package/dist/server/node_modules/@omega/providers/dist/gemini.js.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/generic.d.ts +4 -0
- package/dist/server/node_modules/@omega/providers/dist/generic.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/generic.js +4 -0
- package/dist/server/node_modules/@omega/providers/dist/generic.js.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/index.d.ts +9 -0
- package/dist/server/node_modules/@omega/providers/dist/index.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/index.js +31 -0
- package/dist/server/node_modules/@omega/providers/dist/index.js.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/kimi.d.ts +8 -0
- package/dist/server/node_modules/@omega/providers/dist/kimi.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/kimi.js +13 -0
- package/dist/server/node_modules/@omega/providers/dist/kimi.js.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/ollama.d.ts +9 -0
- package/dist/server/node_modules/@omega/providers/dist/ollama.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/ollama.js +38 -0
- package/dist/server/node_modules/@omega/providers/dist/ollama.js.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/openai.d.ts +12 -0
- package/dist/server/node_modules/@omega/providers/dist/openai.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/openai.js +95 -0
- package/dist/server/node_modules/@omega/providers/dist/openai.js.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/providers.test.d.ts +2 -0
- package/dist/server/node_modules/@omega/providers/dist/providers.test.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/providers/dist/providers.test.js +147 -0
- package/dist/server/node_modules/@omega/providers/dist/providers.test.js.map +1 -0
- package/dist/server/node_modules/@omega/providers/package.json +16 -0
- package/dist/server/node_modules/@omega/router/dist/index.d.ts +2 -0
- package/dist/server/node_modules/@omega/router/dist/index.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/router/dist/index.js +2 -0
- package/dist/server/node_modules/@omega/router/dist/index.js.map +1 -0
- package/dist/server/node_modules/@omega/router/dist/router.test.d.ts +2 -0
- package/dist/server/node_modules/@omega/router/dist/router.test.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/router/dist/router.test.js +102 -0
- package/dist/server/node_modules/@omega/router/dist/router.test.js.map +1 -0
- package/dist/server/node_modules/@omega/router/dist/rules.d.ts +18 -0
- package/dist/server/node_modules/@omega/router/dist/rules.d.ts.map +1 -0
- package/dist/server/node_modules/@omega/router/dist/rules.js +59 -0
- package/dist/server/node_modules/@omega/router/dist/rules.js.map +1 -0
- package/dist/server/node_modules/@omega/router/package.json +16 -0
- package/dist/server/package.json +3 -0
- package/package.json +2 -3
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
export { OpenAIProvider } from './openai.js';
|
|
2
|
+
export { AnthropicProvider } from './anthropic.js';
|
|
3
|
+
export { OllamaProvider } from './ollama.js';
|
|
4
|
+
export { GeminiProvider } from './gemini.js';
|
|
5
|
+
export { KimiProvider } from './kimi.js';
|
|
6
|
+
export { GenericProvider } from './generic.js';
|
|
7
|
+
import { OpenAIProvider } from './openai.js';
|
|
8
|
+
import { AnthropicProvider } from './anthropic.js';
|
|
9
|
+
import { OllamaProvider } from './ollama.js';
|
|
10
|
+
import { GeminiProvider } from './gemini.js';
|
|
11
|
+
import { KimiProvider } from './kimi.js';
|
|
12
|
+
import { GenericProvider } from './generic.js';
|
|
13
|
+
export function createProvider(config) {
|
|
14
|
+
switch (config.kind) {
|
|
15
|
+
case 'openai':
|
|
16
|
+
return new OpenAIProvider(config);
|
|
17
|
+
case 'anthropic':
|
|
18
|
+
return new AnthropicProvider(config);
|
|
19
|
+
case 'ollama':
|
|
20
|
+
return new OllamaProvider(config);
|
|
21
|
+
case 'gemini':
|
|
22
|
+
return new GeminiProvider(config);
|
|
23
|
+
case 'kimi':
|
|
24
|
+
return new KimiProvider(config);
|
|
25
|
+
case 'generic':
|
|
26
|
+
return new GenericProvider(config);
|
|
27
|
+
default:
|
|
28
|
+
throw new Error(`Unknown provider kind: ${String(config.kind)}`);
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAG/C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAE/C,MAAM,UAAU,cAAc,CAAC,MAAsB;IACnD,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,KAAK,WAAW;YACd,OAAO,IAAI,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACvC,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,KAAK,MAAM;YACT,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QAClC,KAAK,SAAS;YACZ,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;QACrC;YACE,MAAM,IAAI,KAAK,CAAC,0BAA0B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACrE,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import type { Provider, ProviderConfig } from '@omega/core';
|
|
2
|
+
import { OpenAIProvider } from './openai.js';
|
|
3
|
+
export declare class KimiProvider extends OpenAIProvider implements Provider {
|
|
4
|
+
readonly config: ProviderConfig;
|
|
5
|
+
constructor(config: ProviderConfig);
|
|
6
|
+
protected readonly supportsTemperature: boolean;
|
|
7
|
+
}
|
|
8
|
+
//# sourceMappingURL=kimi.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"kimi.d.ts","sourceRoot":"","sources":["../src/kimi.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,qBAAa,YAAa,SAAQ,cAAe,YAAW,QAAQ;aACtC,MAAM,EAAE,cAAc;gBAAtB,MAAM,EAAE,cAAc;IAOlD,mBAA4B,mBAAmB,EAAE,OAAO,CAAS;CAClE"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import { OpenAIProvider } from './openai.js';
|
|
2
|
+
export class KimiProvider extends OpenAIProvider {
|
|
3
|
+
config;
|
|
4
|
+
constructor(config) {
|
|
5
|
+
super({
|
|
6
|
+
...config,
|
|
7
|
+
baseUrl: config.baseUrl ?? 'https://api.kimi.com/coding/v1',
|
|
8
|
+
});
|
|
9
|
+
this.config = config;
|
|
10
|
+
}
|
|
11
|
+
supportsTemperature = false;
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=kimi.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"kimi.js","sourceRoot":"","sources":["../src/kimi.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,MAAM,OAAO,YAAa,SAAQ,cAAc;IAClB;IAA5B,YAA4B,MAAsB;QAChD,KAAK,CAAC;YACJ,GAAG,MAAM;YACT,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,gCAAgC;SAC5D,CAAC,CAAC;QAJuB,WAAM,GAAN,MAAM,CAAgB;IAKlD,CAAC;IAE2B,mBAAmB,GAAY,KAAK,CAAC;CAClE"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { Provider, ProviderConfig, SendOptions } from '@omega/core';
|
|
2
|
+
export declare class OllamaProvider implements Provider {
|
|
3
|
+
readonly config: ProviderConfig;
|
|
4
|
+
constructor(config: ProviderConfig);
|
|
5
|
+
private get baseUrl();
|
|
6
|
+
listModels(): Promise<string[]>;
|
|
7
|
+
send(prompt: string, opts?: SendOptions): Promise<string>;
|
|
8
|
+
}
|
|
9
|
+
//# sourceMappingURL=ollama.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../src/ollama.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAIzE,qBAAa,cAAe,YAAW,QAAQ;IAC7C,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC;gBAEpB,MAAM,EAAE,cAAc;IAIlC,OAAO,KAAK,OAAO,GAElB;IAEK,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAO/B,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;CAoBhE"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
const DEFAULT_BASE_URL = 'http://localhost:11434';
|
|
2
|
+
export class OllamaProvider {
|
|
3
|
+
config;
|
|
4
|
+
constructor(config) {
|
|
5
|
+
this.config = config;
|
|
6
|
+
}
|
|
7
|
+
get baseUrl() {
|
|
8
|
+
return (this.config.baseUrl ?? DEFAULT_BASE_URL).replace(/\/$/, '');
|
|
9
|
+
}
|
|
10
|
+
async listModels() {
|
|
11
|
+
const res = await fetch(`${this.baseUrl}/api/tags`);
|
|
12
|
+
if (!res.ok)
|
|
13
|
+
return [this.config.defaultModel];
|
|
14
|
+
const data = (await res.json());
|
|
15
|
+
return data.models?.map((m) => m.name) ?? [this.config.defaultModel];
|
|
16
|
+
}
|
|
17
|
+
async send(prompt, opts) {
|
|
18
|
+
const res = await fetch(`${this.baseUrl}/api/chat`, {
|
|
19
|
+
method: 'POST',
|
|
20
|
+
headers: { 'Content-Type': 'application/json' },
|
|
21
|
+
body: JSON.stringify({
|
|
22
|
+
model: opts?.model ?? this.config.defaultModel,
|
|
23
|
+
messages: [
|
|
24
|
+
...(opts?.system ? [{ role: 'system', content: opts.system }] : []),
|
|
25
|
+
{ role: 'user', content: prompt },
|
|
26
|
+
],
|
|
27
|
+
stream: false,
|
|
28
|
+
options: opts?.temperature !== undefined ? { temperature: opts.temperature } : undefined,
|
|
29
|
+
}),
|
|
30
|
+
});
|
|
31
|
+
if (!res.ok) {
|
|
32
|
+
throw new Error(`Ollama request failed: ${res.status.toString()} ${res.statusText}`);
|
|
33
|
+
}
|
|
34
|
+
const data = (await res.json());
|
|
35
|
+
return data.message?.content ?? '';
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
//# sourceMappingURL=ollama.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../src/ollama.ts"],"names":[],"mappings":"AAEA,MAAM,gBAAgB,GAAG,wBAAwB,CAAC;AAElD,MAAM,OAAO,cAAc;IAChB,MAAM,CAAiB;IAEhC,YAAY,MAAsB;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,IAAY,OAAO;QACjB,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,gBAAgB,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,CAAC,CAAC;QACpD,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAoC,CAAC;QACnE,OAAO,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAc,EAAE,IAAkB;QAC3C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;YAClD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,IAAI,EAAE,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY;gBAC9C,QAAQ,EAAE;oBACR,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBACnE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;iBAClC;gBACD,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE,IAAI,EAAE,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS;aACzF,CAAC;SACH,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,0BAA0B,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;QACvF,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAuC,CAAC;QACtE,OAAO,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;IACrC,CAAC;CACF"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { Provider, ProviderConfig, SendOptions, ToolDefinition } from '@omega/core';
|
|
2
|
+
export declare class OpenAIProvider implements Provider {
|
|
3
|
+
readonly config: ProviderConfig;
|
|
4
|
+
constructor(config: ProviderConfig);
|
|
5
|
+
protected get baseUrl(): string;
|
|
6
|
+
protected readonly supportsTemperature: boolean;
|
|
7
|
+
listModels(): Promise<string[]>;
|
|
8
|
+
send(prompt: string, opts?: SendOptions): Promise<string>;
|
|
9
|
+
sendWithTools(prompt: string, tools: ToolDefinition[], opts?: SendOptions): Promise<string>;
|
|
10
|
+
protected authHeaders(): Record<string, string>;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=openai.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../src/openai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAIzF,qBAAa,cAAe,YAAW,QAAQ;IAC7C,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC;gBAEpB,MAAM,EAAE,cAAc;IAIlC,SAAS,KAAK,OAAO,IAAI,MAAM,CAE9B;IAED,SAAS,CAAC,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAQ;IAEjD,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAS/B,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;IA2BzD,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE,IAAI,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;IAyDjG,SAAS,CAAC,WAAW,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;CAGhD"}
|
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
const DEFAULT_BASE_URL = 'https://api.openai.com/v1';
|
|
2
|
+
export class OpenAIProvider {
|
|
3
|
+
config;
|
|
4
|
+
constructor(config) {
|
|
5
|
+
this.config = config;
|
|
6
|
+
}
|
|
7
|
+
get baseUrl() {
|
|
8
|
+
return (this.config.baseUrl ?? DEFAULT_BASE_URL).replace(/\/$/, '');
|
|
9
|
+
}
|
|
10
|
+
supportsTemperature = true;
|
|
11
|
+
async listModels() {
|
|
12
|
+
const res = await fetch(`${this.baseUrl}/models`, {
|
|
13
|
+
headers: this.authHeaders(),
|
|
14
|
+
});
|
|
15
|
+
if (!res.ok)
|
|
16
|
+
return [this.config.defaultModel];
|
|
17
|
+
const data = (await res.json());
|
|
18
|
+
return data.data?.map((m) => m.id) ?? [this.config.defaultModel];
|
|
19
|
+
}
|
|
20
|
+
async send(prompt, opts) {
|
|
21
|
+
const res = await fetch(`${this.baseUrl}/chat/completions`, {
|
|
22
|
+
method: 'POST',
|
|
23
|
+
headers: {
|
|
24
|
+
'Content-Type': 'application/json',
|
|
25
|
+
...this.authHeaders(),
|
|
26
|
+
},
|
|
27
|
+
body: JSON.stringify({
|
|
28
|
+
model: opts?.model ?? this.config.defaultModel,
|
|
29
|
+
messages: [
|
|
30
|
+
...(opts?.system ? [{ role: 'system', content: opts.system }] : []),
|
|
31
|
+
{ role: 'user', content: prompt },
|
|
32
|
+
],
|
|
33
|
+
...(this.supportsTemperature && opts?.temperature !== undefined
|
|
34
|
+
? { temperature: opts.temperature }
|
|
35
|
+
: {}),
|
|
36
|
+
}),
|
|
37
|
+
});
|
|
38
|
+
if (!res.ok) {
|
|
39
|
+
throw new Error(`OpenAI request failed: ${res.status.toString()} ${res.statusText}`);
|
|
40
|
+
}
|
|
41
|
+
const data = (await res.json());
|
|
42
|
+
return data.choices?.[0]?.message?.content ?? '';
|
|
43
|
+
}
|
|
44
|
+
async sendWithTools(prompt, tools, opts) {
|
|
45
|
+
const res = await fetch(`${this.baseUrl}/chat/completions`, {
|
|
46
|
+
method: 'POST',
|
|
47
|
+
headers: {
|
|
48
|
+
'Content-Type': 'application/json',
|
|
49
|
+
...this.authHeaders(),
|
|
50
|
+
},
|
|
51
|
+
body: JSON.stringify({
|
|
52
|
+
model: opts?.model ?? this.config.defaultModel,
|
|
53
|
+
messages: [
|
|
54
|
+
...(opts?.system ? [{ role: 'system', content: opts.system }] : []),
|
|
55
|
+
{ role: 'user', content: prompt },
|
|
56
|
+
],
|
|
57
|
+
tools: tools.map((t) => ({
|
|
58
|
+
type: 'function',
|
|
59
|
+
function: { name: t.name, description: t.description, parameters: t.parameters },
|
|
60
|
+
})),
|
|
61
|
+
...(this.supportsTemperature && opts?.temperature !== undefined
|
|
62
|
+
? { temperature: opts.temperature }
|
|
63
|
+
: {}),
|
|
64
|
+
}),
|
|
65
|
+
});
|
|
66
|
+
if (!res.ok) {
|
|
67
|
+
throw new Error(`OpenAI tools request failed: ${res.status.toString()} ${res.statusText}`);
|
|
68
|
+
}
|
|
69
|
+
const data = (await res.json());
|
|
70
|
+
const message = data.choices?.[0]?.message;
|
|
71
|
+
const toolCalls = message?.tool_calls;
|
|
72
|
+
if (toolCalls && toolCalls.length > 0) {
|
|
73
|
+
const normalized = toolCalls
|
|
74
|
+
.map((tc) => ({
|
|
75
|
+
id: tc.id ?? '',
|
|
76
|
+
name: tc.function?.name ?? '',
|
|
77
|
+
arguments: (() => {
|
|
78
|
+
try {
|
|
79
|
+
return JSON.parse(tc.function?.arguments ?? '{}');
|
|
80
|
+
}
|
|
81
|
+
catch {
|
|
82
|
+
return {};
|
|
83
|
+
}
|
|
84
|
+
})(),
|
|
85
|
+
}))
|
|
86
|
+
.filter((tc) => tc.id && tc.name);
|
|
87
|
+
return JSON.stringify({ tool_calls: normalized });
|
|
88
|
+
}
|
|
89
|
+
return message?.content ?? '';
|
|
90
|
+
}
|
|
91
|
+
authHeaders() {
|
|
92
|
+
return this.config.apiKey ? { Authorization: `Bearer ${this.config.apiKey}` } : {};
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
//# sourceMappingURL=openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../src/openai.ts"],"names":[],"mappings":"AAEA,MAAM,gBAAgB,GAAG,2BAA2B,CAAC;AAErD,MAAM,OAAO,cAAc;IAChB,MAAM,CAAiB;IAEhC,YAAY,MAAsB;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,IAAc,OAAO;QACnB,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,gBAAgB,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACtE,CAAC;IAEkB,mBAAmB,GAAY,IAAI,CAAC;IAEvD,KAAK,CAAC,UAAU;QACd,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,SAAS,EAAE;YAChD,OAAO,EAAE,IAAI,CAAC,WAAW,EAAE;SAC5B,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAgC,CAAC;QAC/D,OAAO,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAc,EAAE,IAAkB;QAC3C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,mBAAmB,EAAE;YAC1D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,IAAI,CAAC,WAAW,EAAE;aACtB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,IAAI,EAAE,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY;gBAC9C,QAAQ,EAAE;oBACR,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBACnE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;iBAClC;gBACD,GAAG,CAAC,IAAI,CAAC,mBAAmB,IAAI,IAAI,EAAE,WAAW,KAAK,SAAS;oBAC7D,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;oBACnC,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;SACH,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,0BAA0B,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;QACvF,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAE7B,CAAC;QACF,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,KAAuB,EAAE,IAAkB;QAC7E,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,mBAAmB,EAAE;YAC1D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,IAAI,CAAC,WAAW,EAAE;aACtB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,IAAI,EAAE,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY;gBAC9C,QAAQ,EAAE;oBACR,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBACnE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;iBAClC;gBACD,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACvB,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,UAAU,EAAE;iBACjF,CAAC,CAAC;gBACH,GAAG,CAAC,IAAI,CAAC,mBAAmB,IAAI,IAAI,EAAE,WAAW,KAAK,SAAS;oBAC7D,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;oBACnC,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;SACH,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,gCAAgC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;QAC7F,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAU7B,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;QAC3C,MAAM,SAAS,GAAG,OAAO,EAAE,UAAU,CAAC;QACtC,IAAI,SAAS,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,UAAU,GAAG,SAAS;iBACzB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACZ,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE;gBACf,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE;gBAC7B,SAAS,EAAE,CAAC,GAAG,EAAE;oBACf,IAAI,CAAC;wBACH,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,SAAS,IAAI,IAAI,CAA4B,CAAC;oBAC/E,CAAC;oBAAC,MAAM,CAAC;wBACP,OAAO,EAAE,CAAC;oBACZ,CAAC;gBACH,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;iBACF,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC;YACpC,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;IAChC,CAAC;IAES,WAAW;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACrF,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providers.test.d.ts","sourceRoot":"","sources":["../src/providers.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,147 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import { createProvider } from './index.js';
|
|
3
|
+
function jsonResponse(body, status = 200) {
|
|
4
|
+
return new Response(JSON.stringify(body), {
|
|
5
|
+
status,
|
|
6
|
+
headers: { 'Content-Type': 'application/json' },
|
|
7
|
+
});
|
|
8
|
+
}
|
|
9
|
+
describe('createProvider', () => {
|
|
10
|
+
let fetchSpy;
|
|
11
|
+
beforeEach(() => {
|
|
12
|
+
fetchSpy = vi.fn();
|
|
13
|
+
vi.stubGlobal('fetch', fetchSpy);
|
|
14
|
+
});
|
|
15
|
+
afterEach(() => {
|
|
16
|
+
vi.unstubAllGlobals();
|
|
17
|
+
});
|
|
18
|
+
const openaiConfig = {
|
|
19
|
+
id: 'openai',
|
|
20
|
+
name: 'OpenAI',
|
|
21
|
+
kind: 'openai',
|
|
22
|
+
apiKey: 'sk-test',
|
|
23
|
+
defaultModel: 'gpt-4o',
|
|
24
|
+
capabilities: [],
|
|
25
|
+
enabled: true,
|
|
26
|
+
};
|
|
27
|
+
it('creates an OpenAI provider and sends a chat request', async () => {
|
|
28
|
+
fetchSpy.mockResolvedValue(jsonResponse({ choices: [{ message: { content: 'Hello' } }] }));
|
|
29
|
+
const provider = createProvider(openaiConfig);
|
|
30
|
+
const result = await provider.send('hi');
|
|
31
|
+
expect(result).toBe('Hello');
|
|
32
|
+
expect(fetchSpy).toHaveBeenCalledWith('https://api.openai.com/v1/chat/completions', expect.objectContaining({
|
|
33
|
+
method: 'POST',
|
|
34
|
+
headers: expect.objectContaining({ Authorization: 'Bearer sk-test' }),
|
|
35
|
+
body: expect.stringContaining('gpt-4o'),
|
|
36
|
+
}));
|
|
37
|
+
});
|
|
38
|
+
it('lists OpenAI models', async () => {
|
|
39
|
+
fetchSpy.mockResolvedValue(jsonResponse({ data: [{ id: 'gpt-4o' }, { id: 'gpt-3.5' }] }));
|
|
40
|
+
const models = await createProvider(openaiConfig).listModels();
|
|
41
|
+
expect(models).toEqual(['gpt-4o', 'gpt-3.5']);
|
|
42
|
+
});
|
|
43
|
+
it('falls back to default model when list endpoint fails', async () => {
|
|
44
|
+
fetchSpy.mockResolvedValue(jsonResponse({}, 500));
|
|
45
|
+
const models = await createProvider(openaiConfig).listModels();
|
|
46
|
+
expect(models).toEqual(['gpt-4o']);
|
|
47
|
+
});
|
|
48
|
+
const anthropicConfig = {
|
|
49
|
+
id: 'anthropic',
|
|
50
|
+
name: 'Anthropic',
|
|
51
|
+
kind: 'anthropic',
|
|
52
|
+
apiKey: 'sk-ant-test',
|
|
53
|
+
defaultModel: 'claude-3',
|
|
54
|
+
capabilities: [],
|
|
55
|
+
enabled: true,
|
|
56
|
+
};
|
|
57
|
+
it('creates an Anthropic provider and sends a message', async () => {
|
|
58
|
+
fetchSpy.mockResolvedValue(jsonResponse({ content: [{ type: 'text', text: 'Hi there' }] }));
|
|
59
|
+
const provider = createProvider(anthropicConfig);
|
|
60
|
+
const result = await provider.send('hello', { system: 'You are helpful', temperature: 0.5 });
|
|
61
|
+
expect(result).toBe('Hi there');
|
|
62
|
+
expect(fetchSpy).toHaveBeenCalledWith('https://api.anthropic.com/v1/messages', expect.objectContaining({
|
|
63
|
+
method: 'POST',
|
|
64
|
+
headers: expect.objectContaining({
|
|
65
|
+
'x-api-key': 'sk-ant-test',
|
|
66
|
+
'anthropic-version': '2023-06-01',
|
|
67
|
+
}),
|
|
68
|
+
}));
|
|
69
|
+
const body = JSON.parse(fetchSpy.mock.calls[0][1].body);
|
|
70
|
+
expect(body.system).toBe('You are helpful');
|
|
71
|
+
expect(body.temperature).toBe(0.5);
|
|
72
|
+
expect(body.max_tokens).toBe(1024);
|
|
73
|
+
});
|
|
74
|
+
const ollamaConfig = {
|
|
75
|
+
id: 'ollama',
|
|
76
|
+
name: 'Ollama',
|
|
77
|
+
kind: 'ollama',
|
|
78
|
+
defaultModel: 'llama3',
|
|
79
|
+
capabilities: [],
|
|
80
|
+
enabled: true,
|
|
81
|
+
};
|
|
82
|
+
it('creates an Ollama provider and sends a chat request', async () => {
|
|
83
|
+
fetchSpy.mockResolvedValue(jsonResponse({ message: { content: 'Howdy' } }));
|
|
84
|
+
const provider = createProvider(ollamaConfig);
|
|
85
|
+
const result = await provider.send('hey');
|
|
86
|
+
expect(result).toBe('Howdy');
|
|
87
|
+
expect(fetchSpy).toHaveBeenCalledWith('http://localhost:11434/api/chat', expect.objectContaining({
|
|
88
|
+
method: 'POST',
|
|
89
|
+
body: expect.stringContaining('"stream":false'),
|
|
90
|
+
}));
|
|
91
|
+
});
|
|
92
|
+
it('lists Ollama models', async () => {
|
|
93
|
+
fetchSpy.mockResolvedValue(jsonResponse({ models: [{ name: 'llama3' }, { name: 'mistral' }] }));
|
|
94
|
+
const models = await createProvider(ollamaConfig).listModels();
|
|
95
|
+
expect(models).toEqual(['llama3', 'mistral']);
|
|
96
|
+
});
|
|
97
|
+
const geminiConfig = {
|
|
98
|
+
id: 'gemini',
|
|
99
|
+
name: 'Gemini',
|
|
100
|
+
kind: 'gemini',
|
|
101
|
+
apiKey: 'gemini-key',
|
|
102
|
+
defaultModel: 'gemini-1.5-flash',
|
|
103
|
+
capabilities: [],
|
|
104
|
+
enabled: true,
|
|
105
|
+
};
|
|
106
|
+
it('creates a Gemini provider and generates content', async () => {
|
|
107
|
+
fetchSpy.mockResolvedValue(jsonResponse({ candidates: [{ content: { parts: [{ text: 'Greetings' }] } }] }));
|
|
108
|
+
const provider = createProvider(geminiConfig);
|
|
109
|
+
const result = await provider.send('hello', { system: 'Be polite', temperature: 0.7 });
|
|
110
|
+
expect(result).toBe('Greetings');
|
|
111
|
+
expect(fetchSpy).toHaveBeenCalledWith(expect.stringContaining('/models/gemini-1.5-flash:generateContent'), expect.objectContaining({ method: 'POST' }));
|
|
112
|
+
const url = fetchSpy.mock.calls[0][0];
|
|
113
|
+
expect(url).toContain('key=gemini-key');
|
|
114
|
+
const body = JSON.parse(fetchSpy.mock.calls[0][1].body);
|
|
115
|
+
expect(body.systemInstruction).toBeDefined();
|
|
116
|
+
expect(body.generationConfig).toEqual({ temperature: 0.7 });
|
|
117
|
+
});
|
|
118
|
+
it('lists Gemini models', async () => {
|
|
119
|
+
fetchSpy.mockResolvedValue(jsonResponse({
|
|
120
|
+
models: [{ name: 'models/gemini-1.5-flash' }, { name: 'models/gemini-pro' }],
|
|
121
|
+
}));
|
|
122
|
+
const models = await createProvider(geminiConfig).listModels();
|
|
123
|
+
expect(models).toEqual(['gemini-1.5-flash', 'gemini-pro']);
|
|
124
|
+
});
|
|
125
|
+
it('extends OpenAI provider for generic kind', async () => {
|
|
126
|
+
fetchSpy.mockResolvedValue(jsonResponse({ choices: [{ message: { content: 'Generic' } }] }));
|
|
127
|
+
const genericConfig = {
|
|
128
|
+
id: 'generic',
|
|
129
|
+
name: 'Generic',
|
|
130
|
+
kind: 'generic',
|
|
131
|
+
baseUrl: 'http://localhost:8080/v1',
|
|
132
|
+
apiKey: 'generic-key',
|
|
133
|
+
defaultModel: 'llama3',
|
|
134
|
+
capabilities: [],
|
|
135
|
+
enabled: true,
|
|
136
|
+
};
|
|
137
|
+
const result = await createProvider(genericConfig).send('hi');
|
|
138
|
+
expect(result).toBe('Generic');
|
|
139
|
+
expect(fetchSpy).toHaveBeenCalledWith('http://localhost:8080/v1/chat/completions', expect.objectContaining({
|
|
140
|
+
headers: expect.objectContaining({ Authorization: 'Bearer generic-key' }),
|
|
141
|
+
}));
|
|
142
|
+
});
|
|
143
|
+
it('throws for unknown provider kind', () => {
|
|
144
|
+
expect(() => createProvider({ ...openaiConfig, kind: 'unknown' })).toThrow('Unknown provider kind');
|
|
145
|
+
});
|
|
146
|
+
});
|
|
147
|
+
//# sourceMappingURL=providers.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providers.test.js","sourceRoot":"","sources":["../src/providers.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAG5C,SAAS,YAAY,CAAC,IAAa,EAAE,MAAM,GAAG,GAAG;IAC/C,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;QACxC,MAAM;QACN,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;KAChD,CAAC,CAAC;AACL,CAAC;AAED,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,QAAkC,CAAC;IAEvC,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACnB,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAmB;QACnC,EAAE,EAAE,QAAQ;QACZ,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,SAAS;QACjB,YAAY,EAAE,QAAQ;QACtB,YAAY,EAAE,EAAE;QAChB,OAAO,EAAE,IAAI;KACd,CAAC;IAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC3F,MAAM,QAAQ,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,4CAA4C,EAC5C,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,aAAa,EAAE,gBAAgB,EAAE,CAAC;YACrE,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC;SACxC,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1F,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,MAAM,eAAe,GAAmB;QACtC,EAAE,EAAE,WAAW;QACf,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,WAAW;QACjB,MAAM,EAAE,aAAa;QACrB,YAAY,EAAE,UAAU;QACxB,YAAY,EAAE,EAAE;QAChB,OAAO,EAAE,IAAI;KACd,CAAC;IAEF,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5F,MAAM,QAAQ,GAAG,cAAc,CAAC,eAAe,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QAC7F,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,uCAAuC,EACvC,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC/B,WAAW,EAAE,aAAa;gBAC1B,mBAAmB,EAAE,YAAY;aAClC,CAAC;SACH,CAAC,CACH,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC5C,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAmB;QACnC,EAAE,EAAE,QAAQ;QACZ,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,QAAQ;QACd,YAAY,EAAE,QAAQ;QACtB,YAAY,EAAE,EAAE;QAChB,OAAO,EAAE,IAAI;KACd,CAAC;IAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAC5E,MAAM,QAAQ,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,iCAAiC,EACjC,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,gBAAgB,CAAC;SAChD,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAChG,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAmB;QACnC,EAAE,EAAE,QAAQ;QACZ,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,QAAQ;QACd,MAAM,EAAE,YAAY;QACpB,YAAY,EAAE,kBAAkB;QAChC,YAAY,EAAE,EAAE;QAChB,OAAO,EAAE,IAAI;KACd,CAAC;IAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,QAAQ,CAAC,iBAAiB,CACxB,YAAY,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAChF,CAAC;QACF,MAAM,QAAQ,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QACvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACjC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,MAAM,CAAC,gBAAgB,CAAC,0CAA0C,CAAC,EACnE,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAC5C,CAAC;QACF,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAW,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7C,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,QAAQ,CAAC,iBAAiB,CACxB,YAAY,CAAC;YACX,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,yBAAyB,EAAE,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;SAC7E,CAAC,CACH,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,kBAAkB,EAAE,YAAY,CAAC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7F,MAAM,aAAa,GAAmB;YACpC,EAAE,EAAE,SAAS;YACb,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,0BAA0B;YACnC,MAAM,EAAE,aAAa;YACrB,YAAY,EAAE,QAAQ;YACtB,YAAY,EAAE,EAAE;YAChB,OAAO,EAAE,IAAI;SACd,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,2CAA2C,EAC3C,MAAM,CAAC,gBAAgB,CAAC;YACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,aAAa,EAAE,oBAAoB,EAAE,CAAC;SAC1E,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,CAAC,GAAG,EAAE,CACV,cAAc,CAAC,EAAE,GAAG,YAAY,EAAE,IAAI,EAAE,SAAmC,EAAE,CAAC,CAC/E,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
{
|
|
2
|
+
"name": "@omega/providers",
|
|
3
|
+
"version": "0.1.0",
|
|
4
|
+
"type": "module",
|
|
5
|
+
"main": "./dist/index.js",
|
|
6
|
+
"types": "./dist/index.d.ts",
|
|
7
|
+
"exports": {
|
|
8
|
+
".": {
|
|
9
|
+
"import": "./dist/index.js",
|
|
10
|
+
"types": "./dist/index.d.ts"
|
|
11
|
+
}
|
|
12
|
+
},
|
|
13
|
+
"dependencies": {
|
|
14
|
+
"@omega/core": "*"
|
|
15
|
+
}
|
|
16
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"router.test.d.ts","sourceRoot":"","sources":["../src/router.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,102 @@
|
|
|
1
|
+
import { describe, expect, it } from 'vitest';
|
|
2
|
+
import { rankCapabilityForTask, selectProvider } from './rules.js';
|
|
3
|
+
function makeTask(overrides = {}) {
|
|
4
|
+
return {
|
|
5
|
+
id: 'task-1',
|
|
6
|
+
projectId: 'project-1',
|
|
7
|
+
title: 'Test task',
|
|
8
|
+
status: 'todo',
|
|
9
|
+
complexity: 'simple',
|
|
10
|
+
tags: [],
|
|
11
|
+
createdAt: new Date(),
|
|
12
|
+
updatedAt: new Date(),
|
|
13
|
+
...overrides,
|
|
14
|
+
};
|
|
15
|
+
}
|
|
16
|
+
function makeProvider(overrides = {}) {
|
|
17
|
+
return {
|
|
18
|
+
id: 'provider-1',
|
|
19
|
+
name: 'Provider 1',
|
|
20
|
+
kind: 'generic',
|
|
21
|
+
defaultModel: 'default-model',
|
|
22
|
+
capabilities: [{ name: 'default', level: 'fast' }],
|
|
23
|
+
enabled: true,
|
|
24
|
+
...overrides,
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
describe('rankCapabilityForTask', () => {
|
|
28
|
+
it('assigns a higher score to a capability that matches task complexity', () => {
|
|
29
|
+
const task = makeTask({ complexity: 'complex' });
|
|
30
|
+
const fastCap = { name: 'fast', level: 'fast' };
|
|
31
|
+
const advancedCap = { name: 'advanced', level: 'advanced' };
|
|
32
|
+
expect(rankCapabilityForTask(advancedCap, task)).toBeGreaterThan(rankCapabilityForTask(fastCap, task));
|
|
33
|
+
});
|
|
34
|
+
it('downgrades vision mismatches', () => {
|
|
35
|
+
const task = makeTask({ tags: ['vision'] });
|
|
36
|
+
const withoutVision = { name: 'capable', level: 'capable', supportsVision: false };
|
|
37
|
+
const withVision = { name: 'capable', level: 'capable', supportsVision: true };
|
|
38
|
+
expect(rankCapabilityForTask(withVision, task)).toBeGreaterThan(rankCapabilityForTask(withoutVision, task));
|
|
39
|
+
});
|
|
40
|
+
it('downgrades tools mismatches', () => {
|
|
41
|
+
const task = makeTask({ tags: ['tools'] });
|
|
42
|
+
const withoutTools = { name: 'capable', level: 'capable', supportsTools: false };
|
|
43
|
+
const withTools = { name: 'capable', level: 'capable', supportsTools: true };
|
|
44
|
+
expect(rankCapabilityForTask(withTools, task)).toBeGreaterThan(rankCapabilityForTask(withoutTools, task));
|
|
45
|
+
});
|
|
46
|
+
});
|
|
47
|
+
describe('selectProvider', () => {
|
|
48
|
+
it('routes complex tasks to an advanced model', () => {
|
|
49
|
+
const configs = [
|
|
50
|
+
makeProvider({ id: 'fast', capabilities: [{ name: 'fast', level: 'fast' }] }),
|
|
51
|
+
makeProvider({ id: 'advanced', capabilities: [{ name: 'advanced', level: 'advanced' }] }),
|
|
52
|
+
];
|
|
53
|
+
const task = makeTask({ complexity: 'complex' });
|
|
54
|
+
const result = selectProvider(configs, [], task);
|
|
55
|
+
expect(result?.provider.id).toBe('advanced');
|
|
56
|
+
});
|
|
57
|
+
it('lets an explicit rule override the capability score', () => {
|
|
58
|
+
const configs = [
|
|
59
|
+
makeProvider({ id: 'fast', capabilities: [{ name: 'fast', level: 'fast' }] }),
|
|
60
|
+
makeProvider({ id: 'advanced', capabilities: [{ name: 'advanced', level: 'advanced' }] }),
|
|
61
|
+
];
|
|
62
|
+
const rules = [
|
|
63
|
+
{
|
|
64
|
+
priority: 10,
|
|
65
|
+
when: { complexity: 'complex' },
|
|
66
|
+
then: { provider: 'fast', model: 'override-model' },
|
|
67
|
+
},
|
|
68
|
+
];
|
|
69
|
+
const task = makeTask({ complexity: 'complex' });
|
|
70
|
+
const result = selectProvider(configs, rules, task);
|
|
71
|
+
expect(result?.provider.id).toBe('fast');
|
|
72
|
+
expect(result?.model).toBe('override-model');
|
|
73
|
+
});
|
|
74
|
+
it('downgrades providers that do not satisfy tag-based requirements', () => {
|
|
75
|
+
const configs = [
|
|
76
|
+
makeProvider({
|
|
77
|
+
id: 'vision',
|
|
78
|
+
capabilities: [{ name: 'vision', level: 'capable', supportsVision: true, supportsTools: false }],
|
|
79
|
+
}),
|
|
80
|
+
makeProvider({
|
|
81
|
+
id: 'tools',
|
|
82
|
+
capabilities: [{ name: 'tools', level: 'capable', supportsVision: false, supportsTools: true }],
|
|
83
|
+
}),
|
|
84
|
+
];
|
|
85
|
+
const task = makeTask({ complexity: 'medium', tags: ['tools'] });
|
|
86
|
+
const result = selectProvider(configs, [], task);
|
|
87
|
+
expect(result?.provider.id).toBe('tools');
|
|
88
|
+
});
|
|
89
|
+
it('does not select disabled providers', () => {
|
|
90
|
+
const configs = [makeProvider({ id: 'disabled', enabled: false })];
|
|
91
|
+
const task = makeTask();
|
|
92
|
+
const result = selectProvider(configs, [], task);
|
|
93
|
+
expect(result).toBeUndefined();
|
|
94
|
+
});
|
|
95
|
+
it('returns undefined when no provider matches', () => {
|
|
96
|
+
const configs = [];
|
|
97
|
+
const task = makeTask();
|
|
98
|
+
const result = selectProvider(configs, [], task);
|
|
99
|
+
expect(result).toBeUndefined();
|
|
100
|
+
});
|
|
101
|
+
});
|
|
102
|
+
//# sourceMappingURL=router.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"router.test.js","sourceRoot":"","sources":["../src/router.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9C,OAAO,EAAE,qBAAqB,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEnE,SAAS,QAAQ,CAAC,YAA2B,EAAE;IAC7C,OAAO;QACL,EAAE,EAAE,QAAQ;QACZ,SAAS,EAAE,WAAW;QACtB,KAAK,EAAE,WAAW;QAClB,MAAM,EAAE,MAAM;QACd,UAAU,EAAE,QAAQ;QACpB,IAAI,EAAE,EAAE;QACR,SAAS,EAAE,IAAI,IAAI,EAAE;QACrB,SAAS,EAAE,IAAI,IAAI,EAAE;QACrB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,YAAqC,EAAE;IAC3D,OAAO;QACL,EAAE,EAAE,YAAY;QAChB,IAAI,EAAE,YAAY;QAClB,IAAI,EAAE,SAAS;QACf,YAAY,EAAE,eAAe;QAC7B,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAClD,OAAO,EAAE,IAAI;QACb,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAe,EAAE,CAAC;QACzD,MAAM,WAAW,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,UAAmB,EAAE,CAAC;QAErE,MAAM,CAAC,qBAAqB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,eAAe,CAC9D,qBAAqB,CAAC,OAAO,EAAE,IAAI,CAAC,CACrC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,aAAa,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,SAAkB,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC;QAC5F,MAAM,UAAU,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,SAAkB,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC;QAExF,MAAM,CAAC,qBAAqB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,eAAe,CAC7D,qBAAqB,CAAC,aAAa,EAAE,IAAI,CAAC,CAC3C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3C,MAAM,YAAY,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,SAAkB,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;QAC1F,MAAM,SAAS,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,SAAkB,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;QAEtF,MAAM,CAAC,qBAAqB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,eAAe,CAC5D,qBAAqB,CAAC,YAAY,EAAE,IAAI,CAAC,CAC1C,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,OAAO,GAAG;YACd,YAAY,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;YAC7E,YAAY,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;SAC1F,CAAC;QACF,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;QAEjD,MAAM,MAAM,GAAG,cAAc,CAAC,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,OAAO,GAAG;YACd,YAAY,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;YAC7E,YAAY,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;SAC1F,CAAC;QACF,MAAM,KAAK,GAAkB;YAC3B;gBACE,QAAQ,EAAE,EAAE;gBACZ,IAAI,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE;gBAC/B,IAAI,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,gBAAgB,EAAE;aACpD;SACF,CAAC;QACF,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;QAEjD,MAAM,MAAM,GAAG,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAEpD,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,MAAM,OAAO,GAAG;YACd,YAAY,CAAC;gBACX,EAAE,EAAE,QAAQ;gBACZ,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;aACjG,CAAC;YACF,YAAY,CAAC;gBACX,EAAE,EAAE,OAAO;gBACX,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,cAAc,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;aAChG,CAAC;SACH,CAAC;QACF,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAEjE,MAAM,MAAM,GAAG,cAAc,CAAC,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,OAAO,GAAG,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QACnE,MAAM,IAAI,GAAG,QAAQ,EAAE,CAAC;QAExB,MAAM,MAAM,GAAG,cAAc,CAAC,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,OAAO,GAAqB,EAAE,CAAC;QACrC,MAAM,IAAI,GAAG,QAAQ,EAAE,CAAC;QAExB,MAAM,MAAM,GAAG,cAAc,CAAC,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;QAEjD,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { Capability, Complexity, ProviderConfig, Task } from '@omega/core';
|
|
2
|
+
export interface RoutingRule {
|
|
3
|
+
priority: number;
|
|
4
|
+
when: {
|
|
5
|
+
complexity?: Complexity;
|
|
6
|
+
tags?: string[];
|
|
7
|
+
};
|
|
8
|
+
then: {
|
|
9
|
+
provider?: string;
|
|
10
|
+
model?: string;
|
|
11
|
+
};
|
|
12
|
+
}
|
|
13
|
+
export declare function rankCapabilityForTask(cap: Capability, task: Task): number;
|
|
14
|
+
export declare function selectProvider(configs: ProviderConfig[], rules: RoutingRule[], task: Task): {
|
|
15
|
+
provider: ProviderConfig;
|
|
16
|
+
model: string;
|
|
17
|
+
} | undefined;
|
|
18
|
+
//# sourceMappingURL=rules.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rules.d.ts","sourceRoot":"","sources":["../src/rules.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAEhF,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE;QACJ,UAAU,CAAC,EAAE,UAAU,CAAC;QACxB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;KACjB,CAAC;IACF,IAAI,EAAE;QACJ,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;CACH;AAcD,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,GAAG,MAAM,CAgBzE;AAED,wBAAgB,cAAc,CAC5B,OAAO,EAAE,cAAc,EAAE,EACzB,KAAK,EAAE,WAAW,EAAE,EACpB,IAAI,EAAE,IAAI,GACT;IAAE,QAAQ,EAAE,cAAc,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAAG,SAAS,CA0CzD"}
|