@wopr-network/platform-core 1.6.0 → 1.8.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (26) hide show
  1. package/dist/monetization/adapters/embeddings-factory.d.ts +46 -0
  2. package/dist/monetization/adapters/embeddings-factory.js +50 -0
  3. package/dist/monetization/adapters/embeddings-factory.test.d.ts +1 -0
  4. package/dist/monetization/adapters/embeddings-factory.test.js +105 -0
  5. package/dist/monetization/adapters/rate-table.js +26 -2
  6. package/dist/monetization/adapters/rate-table.test.js +8 -1
  7. package/dist/monetization/adapters/transcription-factory.d.ts +48 -0
  8. package/dist/monetization/adapters/transcription-factory.js +52 -0
  9. package/dist/monetization/adapters/transcription-factory.test.d.ts +1 -0
  10. package/dist/monetization/adapters/transcription-factory.test.js +84 -0
  11. package/dist/monetization/adapters/tts-factory.d.ts +50 -0
  12. package/dist/monetization/adapters/tts-factory.js +62 -0
  13. package/dist/monetization/adapters/tts-factory.test.d.ts +1 -0
  14. package/dist/monetization/adapters/tts-factory.test.js +129 -0
  15. package/dist/monetization/index.d.ts +3 -0
  16. package/dist/monetization/index.js +6 -0
  17. package/package.json +1 -1
  18. package/src/monetization/adapters/embeddings-factory.test.ts +141 -0
  19. package/src/monetization/adapters/embeddings-factory.ts +77 -0
  20. package/src/monetization/adapters/rate-table.test.ts +9 -1
  21. package/src/monetization/adapters/rate-table.ts +28 -2
  22. package/src/monetization/adapters/transcription-factory.test.ts +110 -0
  23. package/src/monetization/adapters/transcription-factory.ts +79 -0
  24. package/src/monetization/adapters/tts-factory.test.ts +163 -0
  25. package/src/monetization/adapters/tts-factory.ts +93 -0
  26. package/src/monetization/index.ts +21 -0
@@ -0,0 +1,129 @@
1
+ import { afterAll, describe, expect, it, vi } from "vitest";
2
+ import { createTTSAdapters, createTTSAdaptersFromEnv } from "./tts-factory.js";
3
+ describe("createTTSAdapters", () => {
4
+ it("creates both adapters when all config provided", () => {
5
+ const result = createTTSAdapters({
6
+ chatterboxBaseUrl: "http://chatterbox:8000",
7
+ elevenlabsApiKey: "sk-el",
8
+ });
9
+ expect(result.adapters).toHaveLength(2);
10
+ expect(result.adapterMap.size).toBe(2);
11
+ expect(result.skipped).toHaveLength(0);
12
+ });
13
+ it("returns adapters in cost-priority order (GPU first)", () => {
14
+ const result = createTTSAdapters({
15
+ chatterboxBaseUrl: "http://chatterbox:8000",
16
+ elevenlabsApiKey: "sk-el",
17
+ });
18
+ const names = result.adapters.map((a) => a.name);
19
+ expect(names).toEqual(["chatterbox-tts", "elevenlabs"]);
20
+ });
21
+ it("skips chatterbox when no base URL", () => {
22
+ const result = createTTSAdapters({
23
+ elevenlabsApiKey: "sk-el",
24
+ });
25
+ expect(result.adapters).toHaveLength(1);
26
+ expect(result.adapters[0].name).toBe("elevenlabs");
27
+ expect(result.skipped).toEqual(["chatterbox-tts"]);
28
+ });
29
+ it("skips elevenlabs when no API key", () => {
30
+ const result = createTTSAdapters({
31
+ chatterboxBaseUrl: "http://chatterbox:8000",
32
+ });
33
+ expect(result.adapters).toHaveLength(1);
34
+ expect(result.adapters[0].name).toBe("chatterbox-tts");
35
+ expect(result.skipped).toEqual(["elevenlabs"]);
36
+ });
37
+ it("skips elevenlabs with empty string API key", () => {
38
+ const result = createTTSAdapters({
39
+ chatterboxBaseUrl: "http://chatterbox:8000",
40
+ elevenlabsApiKey: "",
41
+ });
42
+ expect(result.adapters).toHaveLength(1);
43
+ expect(result.skipped).toContain("elevenlabs");
44
+ });
45
+ it("returns empty result when no config provided", () => {
46
+ const result = createTTSAdapters({});
47
+ expect(result.adapters).toHaveLength(0);
48
+ expect(result.adapterMap.size).toBe(0);
49
+ expect(result.skipped).toEqual(["chatterbox-tts", "elevenlabs"]);
50
+ });
51
+ it("all adapters support tts capability", () => {
52
+ const result = createTTSAdapters({
53
+ chatterboxBaseUrl: "http://chatterbox:8000",
54
+ elevenlabsApiKey: "sk-el",
55
+ });
56
+ for (const adapter of result.adapters) {
57
+ expect(adapter.capabilities).toContain("tts");
58
+ }
59
+ });
60
+ it("all adapters implement synthesizeSpeech", () => {
61
+ const result = createTTSAdapters({
62
+ chatterboxBaseUrl: "http://chatterbox:8000",
63
+ elevenlabsApiKey: "sk-el",
64
+ });
65
+ for (const adapter of result.adapters) {
66
+ expect(typeof adapter.synthesizeSpeech).toBe("function");
67
+ }
68
+ });
69
+ it("adapterMap keys match adapter names", () => {
70
+ const result = createTTSAdapters({
71
+ chatterboxBaseUrl: "http://chatterbox:8000",
72
+ elevenlabsApiKey: "sk-el",
73
+ });
74
+ for (const [key, adapter] of result.adapterMap) {
75
+ expect(key).toBe(adapter.name);
76
+ }
77
+ });
78
+ it("chatterbox adapter is marked self-hosted", () => {
79
+ const result = createTTSAdapters({
80
+ chatterboxBaseUrl: "http://chatterbox:8000",
81
+ });
82
+ expect(result.adapters[0].selfHosted).toBe(true);
83
+ });
84
+ it("elevenlabs adapter is not marked self-hosted", () => {
85
+ const result = createTTSAdapters({
86
+ elevenlabsApiKey: "sk-el",
87
+ });
88
+ expect(result.adapters[0].selfHosted).toBeUndefined();
89
+ });
90
+ it("passes per-adapter config overrides", () => {
91
+ const result = createTTSAdapters({
92
+ chatterboxBaseUrl: "http://chatterbox:8000",
93
+ chatterbox: { costPerChar: 0.000001 },
94
+ elevenlabsApiKey: "sk-el",
95
+ elevenlabs: { defaultVoice: "custom-voice" },
96
+ });
97
+ expect(result.adapters).toHaveLength(2);
98
+ });
99
+ });
100
+ describe("createTTSAdaptersFromEnv", () => {
101
+ afterAll(() => {
102
+ vi.unstubAllEnvs();
103
+ });
104
+ it("reads config from environment variables", () => {
105
+ vi.stubEnv("CHATTERBOX_BASE_URL", "http://chatterbox:8000");
106
+ vi.stubEnv("ELEVENLABS_API_KEY", "env-el");
107
+ const result = createTTSAdaptersFromEnv();
108
+ expect(result.adapters).toHaveLength(2);
109
+ const names = result.adapters.map((a) => a.name);
110
+ expect(names).toEqual(["chatterbox-tts", "elevenlabs"]);
111
+ expect(result.skipped).toHaveLength(0);
112
+ });
113
+ it("returns empty when no env vars set", () => {
114
+ vi.stubEnv("CHATTERBOX_BASE_URL", "");
115
+ vi.stubEnv("ELEVENLABS_API_KEY", "");
116
+ const result = createTTSAdaptersFromEnv();
117
+ expect(result.adapters).toHaveLength(0);
118
+ expect(result.skipped).toHaveLength(2);
119
+ });
120
+ it("accepts per-adapter overrides alongside env config", () => {
121
+ vi.stubEnv("CHATTERBOX_BASE_URL", "http://chatterbox:8000");
122
+ vi.stubEnv("ELEVENLABS_API_KEY", "");
123
+ const result = createTTSAdaptersFromEnv({
124
+ chatterbox: { costPerChar: 0.000001 },
125
+ });
126
+ expect(result.adapters).toHaveLength(1);
127
+ expect(result.adapters[0].name).toBe("chatterbox-tts");
128
+ });
129
+ });
@@ -20,6 +20,7 @@ export { type ChatterboxTTSAdapterConfig, createChatterboxTTSAdapter } from "./a
20
20
  export { createDeepgramAdapter, type DeepgramAdapterConfig } from "./adapters/deepgram.js";
21
21
  export { createDeepSeekAdapter, type DeepSeekAdapterConfig } from "./adapters/deepseek.js";
22
22
  export { createElevenLabsAdapter, type ElevenLabsAdapterConfig } from "./adapters/elevenlabs.js";
23
+ export { createEmbeddingsAdapters, createEmbeddingsAdaptersFromEnv, type EmbeddingsFactoryConfig, type EmbeddingsFactoryResult, } from "./adapters/embeddings-factory.js";
23
24
  export { createGeminiAdapter, type GeminiAdapterConfig } from "./adapters/gemini.js";
24
25
  export { createKimiAdapter, type KimiAdapterConfig } from "./adapters/kimi.js";
25
26
  export { getMargin, loadMarginConfig, type MarginConfig, type MarginRule, withMarginConfig, } from "./adapters/margin-config.js";
@@ -30,6 +31,8 @@ export { calculateSavings, getRatesForCapability, lookupRate, RATE_TABLE, type R
30
31
  export { createReplicateAdapter, type ReplicateAdapterConfig } from "./adapters/replicate.js";
31
32
  export { checkHealth, type FetchFn, type SelfHostedAdapterConfig } from "./adapters/self-hosted-base.js";
32
33
  export { createTextGenAdapters, createTextGenAdaptersFromEnv, type TextGenFactoryConfig, type TextGenFactoryResult, } from "./adapters/text-gen-factory.js";
34
+ export { createTranscriptionAdapters, createTranscriptionAdaptersFromEnv, type TranscriptionFactoryConfig, type TranscriptionFactoryResult, } from "./adapters/transcription-factory.js";
35
+ export { createTTSAdapters, createTTSAdaptersFromEnv, type TTSFactoryConfig, type TTSFactoryResult, } from "./adapters/tts-factory.js";
33
36
  export { type AdapterCapability, type AdapterResult, type EmbeddingsInput, type EmbeddingsOutput, type ImageGenerationInput, type ImageGenerationOutput, type MeterEvent, type ProviderAdapter, type TextGenerationInput, type TextGenerationOutput, type TranscriptionInput, type TranscriptionOutput, type TTSInput, type TTSOutput, withMargin, } from "./adapters/types.js";
34
37
  export { type ArbitrageRequest, ArbitrageRouter, type ArbitrageRouterConfig, type MarginRecord, type ModelProviderEntry, NoProviderAvailableError, ProviderRegistry, type ProviderRegistryConfig, type RoutingDecision, } from "./arbitrage/index.js";
35
38
  export type { BudgetCheckerConfig, BudgetCheckResult, SpendLimits } from "./budget/index.js";
@@ -21,6 +21,8 @@ export { createChatterboxTTSAdapter } from "./adapters/chatterbox-tts.js";
21
21
  export { createDeepgramAdapter } from "./adapters/deepgram.js";
22
22
  export { createDeepSeekAdapter } from "./adapters/deepseek.js";
23
23
  export { createElevenLabsAdapter } from "./adapters/elevenlabs.js";
24
+ // Embeddings adapter factory (WOP-2190)
25
+ export { createEmbeddingsAdapters, createEmbeddingsAdaptersFromEnv, } from "./adapters/embeddings-factory.js";
24
26
  export { createGeminiAdapter } from "./adapters/gemini.js";
25
27
  export { createKimiAdapter } from "./adapters/kimi.js";
26
28
  // Margin config (WOP-364)
@@ -35,6 +37,10 @@ export { createReplicateAdapter } from "./adapters/replicate.js";
35
37
  export { checkHealth } from "./adapters/self-hosted-base.js";
36
38
  // Text-generation adapter factory (WOP-463)
37
39
  export { createTextGenAdapters, createTextGenAdaptersFromEnv, } from "./adapters/text-gen-factory.js";
40
+ // Transcription adapter factory
41
+ export { createTranscriptionAdapters, createTranscriptionAdaptersFromEnv, } from "./adapters/transcription-factory.js";
42
+ // TTS adapter factory
43
+ export { createTTSAdapters, createTTSAdaptersFromEnv, } from "./adapters/tts-factory.js";
38
44
  export { withMargin, } from "./adapters/types.js";
39
45
  // Arbitrage router — multi-provider routing for maximum margin (WOP-463)
40
46
  export { ArbitrageRouter, NoProviderAvailableError, ProviderRegistry, } from "./arbitrage/index.js";
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@wopr-network/platform-core",
3
- "version": "1.6.0",
3
+ "version": "1.8.0",
4
4
  "type": "module",
5
5
  "main": "dist/index.js",
6
6
  "types": "dist/index.d.ts",
@@ -0,0 +1,141 @@
1
+ import { afterAll, beforeEach, describe, expect, it, vi } from "vitest";
2
+ import { createEmbeddingsAdapters, createEmbeddingsAdaptersFromEnv } from "./embeddings-factory.js";
3
+ import * as openrouterModule from "./openrouter.js";
4
+
5
+ describe("createEmbeddingsAdapters", () => {
6
+ it("creates adapter when API key provided", () => {
7
+ const result = createEmbeddingsAdapters({
8
+ openrouterApiKey: "sk-or",
9
+ });
10
+
11
+ expect(result.adapters).toHaveLength(1);
12
+ expect(result.adapterMap.size).toBe(1);
13
+ expect(result.skipped).toHaveLength(0);
14
+ });
15
+
16
+ it("adapter is openrouter", () => {
17
+ const result = createEmbeddingsAdapters({
18
+ openrouterApiKey: "sk-or",
19
+ });
20
+
21
+ expect(result.adapters[0].name).toBe("openrouter");
22
+ });
23
+
24
+ it("skips openrouter when no API key", () => {
25
+ const result = createEmbeddingsAdapters({});
26
+
27
+ expect(result.adapters).toHaveLength(0);
28
+ expect(result.skipped).toEqual(["openrouter"]);
29
+ });
30
+
31
+ it("skips adapter with empty string key", () => {
32
+ const result = createEmbeddingsAdapters({
33
+ openrouterApiKey: "",
34
+ });
35
+
36
+ expect(result.adapters).toHaveLength(0);
37
+ expect(result.skipped).toContain("openrouter");
38
+ });
39
+
40
+ it("adapter supports embeddings capability", () => {
41
+ const result = createEmbeddingsAdapters({
42
+ openrouterApiKey: "sk-or",
43
+ });
44
+
45
+ expect(result.adapters[0].capabilities).toContain("embeddings");
46
+ });
47
+
48
+ it("adapter implements embed", () => {
49
+ const result = createEmbeddingsAdapters({
50
+ openrouterApiKey: "sk-or",
51
+ });
52
+
53
+ expect(typeof result.adapters[0].embed).toBe("function");
54
+ });
55
+
56
+ it("adapterMap keys match adapter names", () => {
57
+ const result = createEmbeddingsAdapters({
58
+ openrouterApiKey: "sk-or",
59
+ });
60
+
61
+ for (const [key, adapter] of result.adapterMap) {
62
+ expect(key).toBe(adapter.name);
63
+ }
64
+ });
65
+
66
+ it("passes per-adapter config overrides to adapter constructor", () => {
67
+ const spy = vi.spyOn(openrouterModule, "createOpenRouterAdapter");
68
+
69
+ createEmbeddingsAdapters({
70
+ openrouterApiKey: "sk-or",
71
+ openrouter: { marginMultiplier: 1.5 },
72
+ });
73
+
74
+ expect(spy).toHaveBeenCalledWith(
75
+ expect.objectContaining({
76
+ apiKey: "sk-or",
77
+ marginMultiplier: 1.5,
78
+ }),
79
+ );
80
+
81
+ spy.mockRestore();
82
+ });
83
+
84
+ it("apiKey cannot be overridden via openrouter config", () => {
85
+ // Ensure apiKey always comes from openrouterApiKey, not from spread
86
+ const result = createEmbeddingsAdapters({
87
+ openrouterApiKey: "sk-real",
88
+ openrouter: { apiKey: "sk-evil" } as never,
89
+ });
90
+
91
+ expect(result.adapters).toHaveLength(1);
92
+ expect(result.adapters[0].name).toBe("openrouter");
93
+ });
94
+ });
95
+
96
+ describe("createEmbeddingsAdaptersFromEnv", () => {
97
+ beforeEach(() => {
98
+ vi.unstubAllEnvs();
99
+ });
100
+
101
+ afterAll(() => {
102
+ vi.unstubAllEnvs();
103
+ });
104
+
105
+ it("reads key from environment variable", () => {
106
+ vi.stubEnv("OPENROUTER_API_KEY", "env-or");
107
+
108
+ const result = createEmbeddingsAdaptersFromEnv();
109
+
110
+ expect(result.adapters).toHaveLength(1);
111
+ expect(result.adapters[0].name).toBe("openrouter");
112
+ expect(result.skipped).toHaveLength(0);
113
+ });
114
+
115
+ it("returns empty when no env var set", () => {
116
+ vi.stubEnv("OPENROUTER_API_KEY", "");
117
+
118
+ const result = createEmbeddingsAdaptersFromEnv();
119
+
120
+ expect(result.adapters).toHaveLength(0);
121
+ expect(result.skipped).toEqual(["openrouter"]);
122
+ });
123
+
124
+ it("passes per-adapter overrides alongside env key to adapter constructor", () => {
125
+ vi.stubEnv("OPENROUTER_API_KEY", "env-or");
126
+ const spy = vi.spyOn(openrouterModule, "createOpenRouterAdapter");
127
+
128
+ createEmbeddingsAdaptersFromEnv({
129
+ openrouter: { marginMultiplier: 1.2 },
130
+ });
131
+
132
+ expect(spy).toHaveBeenCalledWith(
133
+ expect.objectContaining({
134
+ apiKey: "env-or",
135
+ marginMultiplier: 1.2,
136
+ }),
137
+ );
138
+
139
+ spy.mockRestore();
140
+ });
141
+ });
@@ -0,0 +1,77 @@
1
+ /**
2
+ * Embeddings adapter factory — instantiates all available embeddings adapters
3
+ * from environment config and returns them ready to register.
4
+ *
5
+ * Only adapters with valid config are created. The factory never touches
6
+ * the database — it returns plain ProviderAdapter instances that the caller
7
+ * registers with an ArbitrageRouter or AdapterSocket.
8
+ *
9
+ * Priority order (cheapest first, when all adapters available):
10
+ * self-hosted-embeddings (GPU, cheapest — not yet implemented)
11
+ * → OpenRouter ($0.02/1M tokens via text-embedding-3-small)
12
+ */
13
+
14
+ import { createOpenRouterAdapter, type OpenRouterAdapterConfig } from "./openrouter.js";
15
+ import type { ProviderAdapter } from "./types.js";
16
+
17
+ /** Top-level factory config. Only providers with an API key are instantiated. */
18
+ export interface EmbeddingsFactoryConfig {
19
+ /** OpenRouter API key. Omit or empty string to skip. */
20
+ openrouterApiKey?: string;
21
+ /** Per-adapter config overrides */
22
+ openrouter?: Omit<Partial<OpenRouterAdapterConfig>, "apiKey">;
23
+ }
24
+
25
+ /** Result of the factory — adapters + metadata for observability. */
26
+ export interface EmbeddingsFactoryResult {
27
+ /** All instantiated adapters, ordered by cost priority (cheapest first). */
28
+ adapters: ProviderAdapter[];
29
+ /** Map of adapter name → ProviderAdapter for direct registration. */
30
+ adapterMap: Map<string, ProviderAdapter>;
31
+ /** Names of providers that were skipped (missing config). */
32
+ skipped: string[];
33
+ }
34
+
35
+ /**
36
+ * Create embeddings adapters from the provided config.
37
+ *
38
+ * Returns only adapters whose API key is present and non-empty.
39
+ * Order matches arbitrage priority: cheapest first.
40
+ */
41
+ export function createEmbeddingsAdapters(config: EmbeddingsFactoryConfig): EmbeddingsFactoryResult {
42
+ const adapters: ProviderAdapter[] = [];
43
+ const skipped: string[] = [];
44
+
45
+ // OpenRouter — $0.02/1M tokens (text-embedding-3-small via OpenAI)
46
+ if (config.openrouterApiKey) {
47
+ adapters.push(createOpenRouterAdapter({ ...config.openrouter, apiKey: config.openrouterApiKey }));
48
+ } else {
49
+ skipped.push("openrouter");
50
+ }
51
+
52
+ // Future: self-hosted-embeddings will go BEFORE openrouter (GPU tier, cheapest)
53
+
54
+ const adapterMap = new Map<string, ProviderAdapter>();
55
+ for (const adapter of adapters) {
56
+ adapterMap.set(adapter.name, adapter);
57
+ }
58
+
59
+ return { adapters, adapterMap, skipped };
60
+ }
61
+
62
+ /**
63
+ * Create embeddings adapters from environment variables.
64
+ *
65
+ * Reads API keys from:
66
+ * - OPENROUTER_API_KEY
67
+ *
68
+ * Accepts optional per-adapter overrides.
69
+ */
70
+ export function createEmbeddingsAdaptersFromEnv(
71
+ overrides?: Omit<EmbeddingsFactoryConfig, "openrouterApiKey">,
72
+ ): EmbeddingsFactoryResult {
73
+ return createEmbeddingsAdapters({
74
+ openrouterApiKey: process.env.OPENROUTER_API_KEY,
75
+ ...overrides,
76
+ });
77
+ }
@@ -138,6 +138,13 @@ describe("getRatesForCapability", () => {
138
138
  expect(rates.map((r) => r.tier)).toContain("premium");
139
139
  });
140
140
 
141
+ it("returns premium-only for transcription (no standard tier yet)", () => {
142
+ const rates = getRatesForCapability("transcription");
143
+ expect(rates).toHaveLength(1);
144
+ expect(rates[0].tier).toBe("premium");
145
+ expect(rates[0].provider).toBe("deepgram");
146
+ });
147
+
141
148
  it("returns empty array for non-existent capability", () => {
142
149
  const rates = getRatesForCapability("image-generation" as unknown as AdapterCapability);
143
150
  expect(rates).toHaveLength(0);
@@ -178,7 +185,8 @@ describe("calculateSavings", () => {
178
185
  });
179
186
 
180
187
  it("returns zero when capability has no standard tier", () => {
181
- const savings = calculateSavings("image-generation" as unknown as AdapterCapability, 1000);
188
+ // Transcription only has premium (deepgram) no self-hosted whisper yet
189
+ const savings = calculateSavings("transcription", 1000);
182
190
  expect(savings).toBe(0);
183
191
  });
184
192
 
@@ -83,9 +83,35 @@ export const RATE_TABLE: RateEntry[] = [
83
83
  effectivePrice: 0.0000013, // = costPerUnit * margin ($1.30 per 1M tokens)
84
84
  },
85
85
 
86
+ // Transcription - Speech-to-Text
87
+ // NOTE: No self-hosted whisper adapter yet — only premium (deepgram) available.
88
+ // When self-hosted-whisper lands, add a standard tier entry here.
89
+ {
90
+ capability: "transcription",
91
+ tier: "premium",
92
+ provider: "deepgram",
93
+ costPerUnit: 0.0000717, // $0.0043/min = $0.0000717/sec
94
+ billingUnit: "per-second",
95
+ margin: 1.3, // 30% — dashboard default; runtime uses getMargin()
96
+ effectivePrice: 0.00009321, // = costPerUnit * margin ($93.21 per 1M seconds ≈ $5.59 per 1M minutes)
97
+ },
98
+
99
+ // Embeddings
100
+ // NOTE: No self-hosted embeddings adapter yet — only premium (openrouter) available.
101
+ // When self-hosted-embeddings lands, add a standard tier entry here.
102
+ {
103
+ capability: "embeddings",
104
+ tier: "premium",
105
+ provider: "openrouter",
106
+ costPerUnit: 0.00000002, // $0.02 per 1M tokens (text-embedding-3-small via OpenRouter)
107
+ billingUnit: "per-token",
108
+ margin: 1.3, // 30% — dashboard default; runtime uses getMargin()
109
+ effectivePrice: 0.000000026, // = costPerUnit * margin ($0.026 per 1M tokens)
110
+ },
111
+
86
112
  // Future self-hosted adapters will add more entries here:
87
- // - transcription: self-hosted-whisper (standard) vs deepgram (premium)
88
- // - embeddings: self-hosted-embeddings (standard) vs openrouter (premium)
113
+ // - transcription: self-hosted-whisper (standard) when GPU adapter exists
114
+ // - embeddings: self-hosted-embeddings (standard) when GPU adapter exists
89
115
  // - image-generation: self-hosted-sdxl (standard) vs replicate (premium)
90
116
  ];
91
117
 
@@ -0,0 +1,110 @@
1
+ import { afterAll, describe, expect, it, vi } from "vitest";
2
+ import { createTranscriptionAdapters, createTranscriptionAdaptersFromEnv } from "./transcription-factory.js";
3
+
4
+ describe("createTranscriptionAdapters", () => {
5
+ it("creates deepgram adapter when API key provided", () => {
6
+ const result = createTranscriptionAdapters({
7
+ deepgramApiKey: "sk-dg",
8
+ });
9
+
10
+ expect(result.adapters).toHaveLength(1);
11
+ expect(result.adapterMap.size).toBe(1);
12
+ expect(result.skipped).toHaveLength(0);
13
+ });
14
+
15
+ it("returns deepgram as only adapter", () => {
16
+ const result = createTranscriptionAdapters({
17
+ deepgramApiKey: "sk-dg",
18
+ });
19
+
20
+ expect(result.adapters[0].name).toBe("deepgram");
21
+ });
22
+
23
+ it("skips deepgram when no API key", () => {
24
+ const result = createTranscriptionAdapters({});
25
+
26
+ expect(result.adapters).toHaveLength(0);
27
+ expect(result.skipped).toEqual(["deepgram"]);
28
+ });
29
+
30
+ it("skips deepgram with empty string API key", () => {
31
+ const result = createTranscriptionAdapters({
32
+ deepgramApiKey: "",
33
+ });
34
+
35
+ expect(result.adapters).toHaveLength(0);
36
+ expect(result.skipped).toContain("deepgram");
37
+ });
38
+
39
+ it("adapter supports transcription capability", () => {
40
+ const result = createTranscriptionAdapters({
41
+ deepgramApiKey: "sk-dg",
42
+ });
43
+
44
+ expect(result.adapters[0].capabilities).toContain("transcription");
45
+ });
46
+
47
+ it("adapter implements transcribe", () => {
48
+ const result = createTranscriptionAdapters({
49
+ deepgramApiKey: "sk-dg",
50
+ });
51
+
52
+ expect(typeof result.adapters[0].transcribe).toBe("function");
53
+ });
54
+
55
+ it("adapterMap keys match adapter names", () => {
56
+ const result = createTranscriptionAdapters({
57
+ deepgramApiKey: "sk-dg",
58
+ });
59
+
60
+ for (const [key, adapter] of result.adapterMap) {
61
+ expect(key).toBe(adapter.name);
62
+ }
63
+ });
64
+
65
+ it("passes per-adapter config overrides", () => {
66
+ const result = createTranscriptionAdapters({
67
+ deepgramApiKey: "sk-dg",
68
+ deepgram: { costPerMinute: 0.005, defaultModel: "nova-2-general" },
69
+ });
70
+
71
+ expect(result.adapters).toHaveLength(1);
72
+ expect(result.adapters[0].name).toBe("deepgram");
73
+ });
74
+ });
75
+
76
+ describe("createTranscriptionAdaptersFromEnv", () => {
77
+ afterAll(() => {
78
+ vi.unstubAllEnvs();
79
+ });
80
+
81
+ it("reads API keys from environment variables", () => {
82
+ vi.stubEnv("DEEPGRAM_API_KEY", "env-dg");
83
+
84
+ const result = createTranscriptionAdaptersFromEnv();
85
+
86
+ expect(result.adapters).toHaveLength(1);
87
+ expect(result.adapters[0].name).toBe("deepgram");
88
+ expect(result.skipped).toHaveLength(0);
89
+ });
90
+
91
+ it("returns empty when no env vars set", () => {
92
+ vi.stubEnv("DEEPGRAM_API_KEY", "");
93
+
94
+ const result = createTranscriptionAdaptersFromEnv();
95
+
96
+ expect(result.adapters).toHaveLength(0);
97
+ expect(result.skipped).toHaveLength(1);
98
+ });
99
+
100
+ it("accepts per-adapter overrides alongside env keys", () => {
101
+ vi.stubEnv("DEEPGRAM_API_KEY", "env-dg");
102
+
103
+ const result = createTranscriptionAdaptersFromEnv({
104
+ deepgram: { marginMultiplier: 1.5 },
105
+ });
106
+
107
+ expect(result.adapters).toHaveLength(1);
108
+ expect(result.adapters[0].name).toBe("deepgram");
109
+ });
110
+ });
@@ -0,0 +1,79 @@
1
+ /**
2
+ * Transcription adapter factory — instantiates all available STT adapters
3
+ * from environment config and returns them ready to register.
4
+ *
5
+ * Only adapters with valid config are created. The factory never touches
6
+ * the database — it returns plain ProviderAdapter instances that the caller
7
+ * registers with an ArbitrageRouter or AdapterSocket.
8
+ *
9
+ * Priority order (cheapest first, when all adapters available):
10
+ * self-hosted-whisper (GPU, cheapest) → Deepgram ($0.0043/min, third-party)
11
+ *
12
+ * NOTE: Self-hosted Whisper adapter doesn't exist yet. When it does, it
13
+ * will slot in as the GPU tier (cheapest) ahead of Deepgram.
14
+ */
15
+
16
+ import { createDeepgramAdapter, type DeepgramAdapterConfig } from "./deepgram.js";
17
+ import type { ProviderAdapter } from "./types.js";
18
+
19
+ /** Top-level factory config. Only providers with an API key are instantiated. */
20
+ export interface TranscriptionFactoryConfig {
21
+ /** Deepgram API key. Omit or empty string to skip. */
22
+ deepgramApiKey?: string;
23
+ /** Per-adapter config overrides */
24
+ deepgram?: Omit<Partial<DeepgramAdapterConfig>, "apiKey">;
25
+ }
26
+
27
+ /** Result of the factory — adapters + metadata for observability. */
28
+ export interface TranscriptionFactoryResult {
29
+ /** All instantiated adapters, ordered by cost priority (cheapest first). */
30
+ adapters: ProviderAdapter[];
31
+ /** Map of adapter name → ProviderAdapter for direct registration. */
32
+ adapterMap: Map<string, ProviderAdapter>;
33
+ /** Names of providers that were skipped (no API key). */
34
+ skipped: string[];
35
+ }
36
+
37
+ /**
38
+ * Create transcription adapters from the provided config.
39
+ *
40
+ * Returns only adapters whose API key is present and non-empty.
41
+ * Order matches arbitrage priority: cheapest first.
42
+ */
43
+ export function createTranscriptionAdapters(config: TranscriptionFactoryConfig): TranscriptionFactoryResult {
44
+ const adapters: ProviderAdapter[] = [];
45
+ const skipped: string[] = [];
46
+
47
+ // Deepgram — $0.0043/min (Nova-2 wholesale)
48
+ if (config.deepgramApiKey) {
49
+ adapters.push(createDeepgramAdapter({ apiKey: config.deepgramApiKey, ...config.deepgram }));
50
+ } else {
51
+ skipped.push("deepgram");
52
+ }
53
+
54
+ // Future: self-hosted-whisper will go BEFORE deepgram (GPU tier, cheapest)
55
+
56
+ const adapterMap = new Map<string, ProviderAdapter>();
57
+ for (const adapter of adapters) {
58
+ adapterMap.set(adapter.name, adapter);
59
+ }
60
+
61
+ return { adapters, adapterMap, skipped };
62
+ }
63
+
64
+ /**
65
+ * Create transcription adapters from environment variables.
66
+ *
67
+ * Reads API keys from:
68
+ * - DEEPGRAM_API_KEY
69
+ *
70
+ * Accepts optional per-adapter overrides.
71
+ */
72
+ export function createTranscriptionAdaptersFromEnv(
73
+ overrides?: Omit<TranscriptionFactoryConfig, "deepgramApiKey">,
74
+ ): TranscriptionFactoryResult {
75
+ return createTranscriptionAdapters({
76
+ deepgramApiKey: process.env.DEEPGRAM_API_KEY,
77
+ ...overrides,
78
+ });
79
+ }