@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.
Files changed (149) hide show
  1. package/dist/server/node_modules/.pnpm/lock.yaml +9 -0
  2. package/dist/server/node_modules/@omega/agent/dist/executor.d.ts +12 -0
  3. package/dist/server/node_modules/@omega/agent/dist/executor.d.ts.map +1 -0
  4. package/dist/server/node_modules/@omega/agent/dist/executor.js +352 -0
  5. package/dist/server/node_modules/@omega/agent/dist/executor.js.map +1 -0
  6. package/dist/server/node_modules/@omega/agent/dist/git.d.ts +15 -0
  7. package/dist/server/node_modules/@omega/agent/dist/git.d.ts.map +1 -0
  8. package/dist/server/node_modules/@omega/agent/dist/git.js +56 -0
  9. package/dist/server/node_modules/@omega/agent/dist/git.js.map +1 -0
  10. package/dist/server/node_modules/@omega/agent/dist/index.d.ts +7 -0
  11. package/dist/server/node_modules/@omega/agent/dist/index.d.ts.map +1 -0
  12. package/dist/server/node_modules/@omega/agent/dist/index.js +7 -0
  13. package/dist/server/node_modules/@omega/agent/dist/index.js.map +1 -0
  14. package/dist/server/node_modules/@omega/agent/dist/planner.d.ts +12 -0
  15. package/dist/server/node_modules/@omega/agent/dist/planner.d.ts.map +1 -0
  16. package/dist/server/node_modules/@omega/agent/dist/planner.js +64 -0
  17. package/dist/server/node_modules/@omega/agent/dist/planner.js.map +1 -0
  18. package/dist/server/node_modules/@omega/agent/dist/prompts.d.ts +7 -0
  19. package/dist/server/node_modules/@omega/agent/dist/prompts.d.ts.map +1 -0
  20. package/dist/server/node_modules/@omega/agent/dist/prompts.js +55 -0
  21. package/dist/server/node_modules/@omega/agent/dist/prompts.js.map +1 -0
  22. package/dist/server/node_modules/@omega/agent/dist/publisher.d.ts +7 -0
  23. package/dist/server/node_modules/@omega/agent/dist/publisher.d.ts.map +1 -0
  24. package/dist/server/node_modules/@omega/agent/dist/publisher.js +57 -0
  25. package/dist/server/node_modules/@omega/agent/dist/publisher.js.map +1 -0
  26. package/dist/server/node_modules/@omega/agent/dist/tools.d.ts +10 -0
  27. package/dist/server/node_modules/@omega/agent/dist/tools.d.ts.map +1 -0
  28. package/dist/server/node_modules/@omega/agent/dist/tools.js +92 -0
  29. package/dist/server/node_modules/@omega/agent/dist/tools.js.map +1 -0
  30. package/dist/server/node_modules/@omega/agent/dist/validator.d.ts +17 -0
  31. package/dist/server/node_modules/@omega/agent/dist/validator.d.ts.map +1 -0
  32. package/dist/server/node_modules/@omega/agent/dist/validator.js +28 -0
  33. package/dist/server/node_modules/@omega/agent/dist/validator.js.map +1 -0
  34. package/dist/server/node_modules/@omega/agent/package.json +19 -0
  35. package/dist/server/node_modules/@omega/core/dist/agent.d.ts +51 -0
  36. package/dist/server/node_modules/@omega/core/dist/agent.d.ts.map +1 -0
  37. package/dist/server/node_modules/@omega/core/dist/agent.js +2 -0
  38. package/dist/server/node_modules/@omega/core/dist/agent.js.map +1 -0
  39. package/dist/server/node_modules/@omega/core/dist/index.d.ts +5 -0
  40. package/dist/server/node_modules/@omega/core/dist/index.d.ts.map +1 -0
  41. package/dist/server/node_modules/@omega/core/dist/index.js +5 -0
  42. package/dist/server/node_modules/@omega/core/dist/index.js.map +1 -0
  43. package/dist/server/node_modules/@omega/core/dist/provider.d.ts +48 -0
  44. package/dist/server/node_modules/@omega/core/dist/provider.d.ts.map +1 -0
  45. package/dist/server/node_modules/@omega/core/dist/provider.js +2 -0
  46. package/dist/server/node_modules/@omega/core/dist/provider.js.map +1 -0
  47. package/dist/server/node_modules/@omega/core/dist/provider.test.d.ts +2 -0
  48. package/dist/server/node_modules/@omega/core/dist/provider.test.d.ts.map +1 -0
  49. package/dist/server/node_modules/@omega/core/dist/provider.test.js +27 -0
  50. package/dist/server/node_modules/@omega/core/dist/provider.test.js.map +1 -0
  51. package/dist/server/node_modules/@omega/core/dist/skill.d.ts +21 -0
  52. package/dist/server/node_modules/@omega/core/dist/skill.d.ts.map +1 -0
  53. package/dist/server/node_modules/@omega/core/dist/skill.js +2 -0
  54. package/dist/server/node_modules/@omega/core/dist/skill.js.map +1 -0
  55. package/dist/server/node_modules/@omega/core/dist/task.d.ts +27 -0
  56. package/dist/server/node_modules/@omega/core/dist/task.d.ts.map +1 -0
  57. package/dist/server/node_modules/@omega/core/dist/task.js +2 -0
  58. package/dist/server/node_modules/@omega/core/dist/task.js.map +1 -0
  59. package/dist/server/node_modules/@omega/core/package.json +13 -0
  60. package/dist/server/node_modules/@omega/db/dist/client.d.ts +8 -0
  61. package/dist/server/node_modules/@omega/db/dist/client.d.ts.map +1 -0
  62. package/dist/server/node_modules/@omega/db/dist/client.js +8 -0
  63. package/dist/server/node_modules/@omega/db/dist/client.js.map +1 -0
  64. package/dist/server/node_modules/@omega/db/dist/index.d.ts +5 -0
  65. package/dist/server/node_modules/@omega/db/dist/index.d.ts.map +1 -0
  66. package/dist/server/node_modules/@omega/db/dist/index.js +5 -0
  67. package/dist/server/node_modules/@omega/db/dist/index.js.map +1 -0
  68. package/dist/server/node_modules/@omega/db/dist/migrate.d.ts +2 -0
  69. package/dist/server/node_modules/@omega/db/dist/migrate.d.ts.map +1 -0
  70. package/dist/server/node_modules/@omega/db/dist/migrate.js +58 -0
  71. package/dist/server/node_modules/@omega/db/dist/migrate.js.map +1 -0
  72. package/dist/server/node_modules/@omega/db/dist/seed.d.ts +2 -0
  73. package/dist/server/node_modules/@omega/db/dist/seed.d.ts.map +1 -0
  74. package/dist/server/node_modules/@omega/db/dist/seed.js +42 -0
  75. package/dist/server/node_modules/@omega/db/dist/seed.js.map +1 -0
  76. package/dist/server/node_modules/@omega/db/generated/client/default.d.ts +1 -0
  77. package/dist/server/node_modules/@omega/db/generated/client/default.js +1 -0
  78. package/dist/server/node_modules/@omega/db/generated/client/edge.d.ts +1 -0
  79. package/dist/server/node_modules/@omega/db/generated/client/edge.js +282 -0
  80. package/dist/server/node_modules/@omega/db/generated/client/index-browser.js +272 -0
  81. package/dist/server/node_modules/@omega/db/generated/client/index.d.ts +12787 -0
  82. package/dist/server/node_modules/@omega/db/generated/client/index.js +303 -0
  83. package/dist/server/node_modules/@omega/db/generated/client/libquery_engine-darwin-arm64.dylib.node +0 -0
  84. package/dist/server/node_modules/@omega/db/generated/client/package.json +124 -0
  85. package/dist/server/node_modules/@omega/db/generated/client/query_engine_bg.js +2 -0
  86. package/dist/server/node_modules/@omega/db/generated/client/query_engine_bg.wasm +0 -0
  87. package/dist/server/node_modules/@omega/db/generated/client/runtime/edge-esm.js +31 -0
  88. package/dist/server/node_modules/@omega/db/generated/client/runtime/edge.js +31 -0
  89. package/dist/server/node_modules/@omega/db/generated/client/runtime/index-browser.d.ts +365 -0
  90. package/dist/server/node_modules/@omega/db/generated/client/runtime/index-browser.js +13 -0
  91. package/dist/server/node_modules/@omega/db/generated/client/runtime/library.d.ts +3403 -0
  92. package/dist/server/node_modules/@omega/db/generated/client/runtime/library.js +143 -0
  93. package/dist/server/node_modules/@omega/db/generated/client/runtime/react-native.js +80 -0
  94. package/dist/server/node_modules/@omega/db/generated/client/runtime/wasm.js +32 -0
  95. package/dist/server/node_modules/@omega/db/generated/client/schema.prisma +119 -0
  96. package/dist/server/node_modules/@omega/db/generated/client/wasm-edge-light-loader.mjs +1 -0
  97. package/dist/server/node_modules/@omega/db/generated/client/wasm-worker-loader.mjs +1 -0
  98. package/dist/server/node_modules/@omega/db/generated/client/wasm.d.ts +1 -0
  99. package/dist/server/node_modules/@omega/db/generated/client/wasm.js +289 -0
  100. package/dist/server/node_modules/@omega/db/package.json +18 -0
  101. package/dist/server/node_modules/@omega/db/prisma/migrations/20260702110000_init/migration.sql +154 -0
  102. package/dist/server/node_modules/@omega/providers/dist/anthropic.d.ts +10 -0
  103. package/dist/server/node_modules/@omega/providers/dist/anthropic.d.ts.map +1 -0
  104. package/dist/server/node_modules/@omega/providers/dist/anthropic.js +50 -0
  105. package/dist/server/node_modules/@omega/providers/dist/anthropic.js.map +1 -0
  106. package/dist/server/node_modules/@omega/providers/dist/gemini.d.ts +9 -0
  107. package/dist/server/node_modules/@omega/providers/dist/gemini.d.ts.map +1 -0
  108. package/dist/server/node_modules/@omega/providers/dist/gemini.js +46 -0
  109. package/dist/server/node_modules/@omega/providers/dist/gemini.js.map +1 -0
  110. package/dist/server/node_modules/@omega/providers/dist/generic.d.ts +4 -0
  111. package/dist/server/node_modules/@omega/providers/dist/generic.d.ts.map +1 -0
  112. package/dist/server/node_modules/@omega/providers/dist/generic.js +4 -0
  113. package/dist/server/node_modules/@omega/providers/dist/generic.js.map +1 -0
  114. package/dist/server/node_modules/@omega/providers/dist/index.d.ts +9 -0
  115. package/dist/server/node_modules/@omega/providers/dist/index.d.ts.map +1 -0
  116. package/dist/server/node_modules/@omega/providers/dist/index.js +31 -0
  117. package/dist/server/node_modules/@omega/providers/dist/index.js.map +1 -0
  118. package/dist/server/node_modules/@omega/providers/dist/kimi.d.ts +8 -0
  119. package/dist/server/node_modules/@omega/providers/dist/kimi.d.ts.map +1 -0
  120. package/dist/server/node_modules/@omega/providers/dist/kimi.js +13 -0
  121. package/dist/server/node_modules/@omega/providers/dist/kimi.js.map +1 -0
  122. package/dist/server/node_modules/@omega/providers/dist/ollama.d.ts +9 -0
  123. package/dist/server/node_modules/@omega/providers/dist/ollama.d.ts.map +1 -0
  124. package/dist/server/node_modules/@omega/providers/dist/ollama.js +38 -0
  125. package/dist/server/node_modules/@omega/providers/dist/ollama.js.map +1 -0
  126. package/dist/server/node_modules/@omega/providers/dist/openai.d.ts +12 -0
  127. package/dist/server/node_modules/@omega/providers/dist/openai.d.ts.map +1 -0
  128. package/dist/server/node_modules/@omega/providers/dist/openai.js +95 -0
  129. package/dist/server/node_modules/@omega/providers/dist/openai.js.map +1 -0
  130. package/dist/server/node_modules/@omega/providers/dist/providers.test.d.ts +2 -0
  131. package/dist/server/node_modules/@omega/providers/dist/providers.test.d.ts.map +1 -0
  132. package/dist/server/node_modules/@omega/providers/dist/providers.test.js +147 -0
  133. package/dist/server/node_modules/@omega/providers/dist/providers.test.js.map +1 -0
  134. package/dist/server/node_modules/@omega/providers/package.json +16 -0
  135. package/dist/server/node_modules/@omega/router/dist/index.d.ts +2 -0
  136. package/dist/server/node_modules/@omega/router/dist/index.d.ts.map +1 -0
  137. package/dist/server/node_modules/@omega/router/dist/index.js +2 -0
  138. package/dist/server/node_modules/@omega/router/dist/index.js.map +1 -0
  139. package/dist/server/node_modules/@omega/router/dist/router.test.d.ts +2 -0
  140. package/dist/server/node_modules/@omega/router/dist/router.test.d.ts.map +1 -0
  141. package/dist/server/node_modules/@omega/router/dist/router.test.js +102 -0
  142. package/dist/server/node_modules/@omega/router/dist/router.test.js.map +1 -0
  143. package/dist/server/node_modules/@omega/router/dist/rules.d.ts +18 -0
  144. package/dist/server/node_modules/@omega/router/dist/rules.d.ts.map +1 -0
  145. package/dist/server/node_modules/@omega/router/dist/rules.js +59 -0
  146. package/dist/server/node_modules/@omega/router/dist/rules.js.map +1 -0
  147. package/dist/server/node_modules/@omega/router/package.json +16 -0
  148. package/dist/server/package.json +3 -0
  149. 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,2 @@
1
+ export {};
2
+ //# sourceMappingURL=providers.test.d.ts.map
@@ -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,2 @@
1
+ export * from './rules.js';
2
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC"}
@@ -0,0 +1,2 @@
1
+ export * from './rules.js';
2
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=router.test.d.ts.map
@@ -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"}