@splashcodex/api-key-manager 4.1.0 → 5.0.2

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 (39) hide show
  1. package/README.md +129 -201
  2. package/bin/cli.js +50 -0
  3. package/dist/index.d.ts +7 -2
  4. package/dist/index.js +16 -4
  5. package/dist/index.js.map +1 -1
  6. package/dist/persistence/file.d.ts +43 -0
  7. package/dist/persistence/file.js +82 -0
  8. package/dist/persistence/file.js.map +1 -0
  9. package/dist/persistence/index.d.ts +7 -0
  10. package/dist/persistence/index.js +12 -0
  11. package/dist/persistence/index.js.map +1 -0
  12. package/dist/persistence/memory.d.ts +30 -0
  13. package/dist/persistence/memory.js +43 -0
  14. package/dist/persistence/memory.js.map +1 -0
  15. package/dist/presets/base.d.ts +138 -0
  16. package/dist/presets/base.js +227 -0
  17. package/dist/presets/base.js.map +1 -0
  18. package/dist/presets/gemini.d.ts +60 -0
  19. package/dist/presets/gemini.js +77 -0
  20. package/dist/presets/gemini.js.map +1 -0
  21. package/dist/presets/index.d.ts +10 -0
  22. package/dist/presets/index.js +16 -0
  23. package/dist/presets/index.js.map +1 -0
  24. package/dist/presets/multi.d.ts +120 -0
  25. package/dist/presets/multi.js +210 -0
  26. package/dist/presets/multi.js.map +1 -0
  27. package/dist/presets/openai.d.ts +45 -0
  28. package/dist/presets/openai.js +62 -0
  29. package/dist/presets/openai.js.map +1 -0
  30. package/package.json +88 -5
  31. package/src/index.ts +24 -3
  32. package/src/persistence/file.ts +89 -0
  33. package/src/persistence/index.ts +7 -0
  34. package/src/persistence/memory.ts +43 -0
  35. package/src/presets/base.ts +323 -0
  36. package/src/presets/gemini.ts +84 -0
  37. package/src/presets/index.ts +10 -0
  38. package/src/presets/multi.ts +280 -0
  39. package/src/presets/openai.ts +69 -0
@@ -0,0 +1,210 @@
1
+ "use strict";
2
+ /**
3
+ * MultiManager Preset — Multi-Provider Key Vault
4
+ *
5
+ * Manages API keys across multiple providers from a single entry point.
6
+ * Aggregates keys from multiple environment variables and routes
7
+ * requests to the correct provider pool.
8
+ *
9
+ * @module presets/multi
10
+ *
11
+ * @example
12
+ * ```ts
13
+ * import { MultiManager } from '@splashcodex/api-key-manager/presets/multi';
14
+ *
15
+ * const result = MultiManager.getInstance({
16
+ * providers: {
17
+ * gemini: { envKeys: ['GOOGLE_GEMINI_API_KEY'] },
18
+ * openai: { envKeys: ['OPENAI_API_KEY'] },
19
+ * }
20
+ * });
21
+ *
22
+ * if (result.success) {
23
+ * const vault = result.data;
24
+ *
25
+ * // Route to specific provider
26
+ * const text = await vault.execute(async (key) => {
27
+ * // key is from the gemini pool
28
+ * return callGemini(key, prompt);
29
+ * }, { provider: 'gemini', maxRetries: 3 });
30
+ *
31
+ * // Get stats across all providers
32
+ * const stats = vault.getMultiStats();
33
+ * }
34
+ * ```
35
+ */
36
+ Object.defineProperty(exports, "__esModule", { value: true });
37
+ exports.MultiManager = void 0;
38
+ const index_1 = require("../index");
39
+ const file_1 = require("../persistence/file");
40
+ const path_1 = require("path");
41
+ const os_1 = require("os");
42
+ // ─── MultiManager ───────────────────────────────────────────────────────────
43
+ class MultiManager {
44
+ static instance = null;
45
+ managers = new Map();
46
+ logger;
47
+ constructor(options) {
48
+ this.logger = options.logger || console;
49
+ for (const [providerName, config] of Object.entries(options.providers)) {
50
+ const keys = MultiManager.parseKeysFromEnv(config.envKeys);
51
+ if (keys.length === 0) {
52
+ this.logger.warn(`[MultiManager:${providerName}] No API keys found in: ${config.envKeys.join(', ')}`);
53
+ }
54
+ const storage = new file_1.FileStorage({
55
+ filePath: (0, path_1.join)((0, os_1.tmpdir)(), `codedex_multi_${providerName}_state.json`),
56
+ clearOnInit: true,
57
+ });
58
+ const manager = new index_1.ApiKeyManager(keys, {
59
+ storage,
60
+ strategy: config.strategy || new index_1.LatencyStrategy(),
61
+ concurrency: config.concurrency ?? 20,
62
+ semanticCache: config.semanticCache,
63
+ });
64
+ // Wire events
65
+ manager.on('keyDead', (key) => this.logger.error(`[MultiManager:${providerName}] Key DEAD: ...${key.slice(-4)}`));
66
+ manager.on('circuitOpen', (key) => this.logger.warn(`[MultiManager:${providerName}] Circuit OPEN: ...${key.slice(-4)}`));
67
+ manager.on('keyRecovered', (key) => this.logger.info(`[MultiManager:${providerName}] Key RECOVERED: ...${key.slice(-4)}`));
68
+ manager.on('allKeysExhausted', () => this.logger.error(`[MultiManager:${providerName}] ALL KEYS EXHAUSTED`));
69
+ // Health checks
70
+ const interval = options.healthCheckIntervalMs ?? 300_000;
71
+ if (interval > 0) {
72
+ manager.startHealthChecks(interval);
73
+ }
74
+ this.managers.set(providerName, manager);
75
+ this.logger.info(`[MultiManager:${providerName}] Initialized with ${keys.length} keys`);
76
+ }
77
+ }
78
+ // ─── Factory ────────────────────────────────────────────────────────────
79
+ /**
80
+ * Get or create the singleton MultiManager instance.
81
+ */
82
+ static getInstance(options) {
83
+ if (MultiManager.instance) {
84
+ return { success: true, data: MultiManager.instance };
85
+ }
86
+ try {
87
+ MultiManager.instance = new MultiManager(options);
88
+ return { success: true, data: MultiManager.instance };
89
+ }
90
+ catch (error) {
91
+ return {
92
+ success: false,
93
+ error: error instanceof Error ? error : new Error(String(error)),
94
+ };
95
+ }
96
+ }
97
+ /**
98
+ * Reset the singleton (primarily for testing).
99
+ */
100
+ static reset() {
101
+ if (MultiManager.instance) {
102
+ MultiManager.instance.destroy();
103
+ MultiManager.instance = null;
104
+ }
105
+ }
106
+ // ─── Public API ─────────────────────────────────────────────────────────
107
+ /**
108
+ * Execute a function with a specific provider's key pool.
109
+ *
110
+ * @param fn - The function to execute with a key
111
+ * @param options - Must include `provider` to select the pool
112
+ */
113
+ async execute(fn, options) {
114
+ const manager = this.managers.get(options.provider);
115
+ if (!manager) {
116
+ throw new Error(`[MultiManager] Unknown provider "${options.provider}". ` +
117
+ `Available: ${[...this.managers.keys()].join(', ')}`);
118
+ }
119
+ // Strip `provider` before delegating — each manager only has keys for one provider
120
+ const { provider, ...delegateOptions } = options;
121
+ return manager.execute(fn, delegateOptions);
122
+ }
123
+ /**
124
+ * Execute a streaming function with a specific provider's key pool.
125
+ */
126
+ async *executeStream(fn, options) {
127
+ const manager = this.managers.get(options.provider);
128
+ if (!manager) {
129
+ throw new Error(`[MultiManager] Unknown provider "${options.provider}". ` +
130
+ `Available: ${[...this.managers.keys()].join(', ')}`);
131
+ }
132
+ const { provider, ...delegateOptions } = options;
133
+ yield* manager.executeStream(fn, delegateOptions);
134
+ }
135
+ /**
136
+ * Get a raw key from a specific provider pool.
137
+ */
138
+ getKey(provider) {
139
+ const manager = this.managers.get(provider);
140
+ if (!manager)
141
+ return null;
142
+ return manager.getKey();
143
+ }
144
+ /**
145
+ * Get stats for a specific provider.
146
+ */
147
+ getStats(provider) {
148
+ const manager = this.managers.get(provider);
149
+ if (!manager)
150
+ return null;
151
+ return manager.getStats();
152
+ }
153
+ /**
154
+ * Get aggregate stats across ALL providers.
155
+ */
156
+ getMultiStats() {
157
+ const stats = {};
158
+ for (const [name, manager] of this.managers) {
159
+ stats[name] = manager.getStats();
160
+ }
161
+ return stats;
162
+ }
163
+ /**
164
+ * Get the list of available provider names.
165
+ */
166
+ getProviders() {
167
+ return [...this.managers.keys()];
168
+ }
169
+ /**
170
+ * Get the underlying ApiKeyManager for a specific provider.
171
+ */
172
+ getManager(provider) {
173
+ return this.managers.get(provider);
174
+ }
175
+ /**
176
+ * Stop all health checks and clean up.
177
+ */
178
+ destroy() {
179
+ for (const [, manager] of this.managers) {
180
+ manager.stopHealthChecks();
181
+ }
182
+ this.managers.clear();
183
+ }
184
+ // ─── Helpers ────────────────────────────────────────────────────────────
185
+ static parseKeysFromEnv(envKeys) {
186
+ const keys = [];
187
+ for (const envName of envKeys) {
188
+ const envValue = process.env[envName];
189
+ if (!envValue)
190
+ continue;
191
+ const trimmed = envValue.trim();
192
+ if (!trimmed)
193
+ continue;
194
+ if (trimmed.startsWith('[')) {
195
+ try {
196
+ const parsed = JSON.parse(trimmed);
197
+ if (Array.isArray(parsed)) {
198
+ keys.push(...parsed.filter((k) => typeof k === 'string' && k.trim()));
199
+ continue;
200
+ }
201
+ }
202
+ catch { /* not JSON */ }
203
+ }
204
+ keys.push(...trimmed.split(',').map(k => k.trim()).filter(k => k.length > 0));
205
+ }
206
+ return [...new Set(keys)];
207
+ }
208
+ }
209
+ exports.MultiManager = MultiManager;
210
+ //# sourceMappingURL=multi.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"multi.js","sourceRoot":"","sources":["../../src/presets/multi.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;;;AAEH,oCAOkB;AAClB,8CAAkD;AAClD,+BAA4B;AAC5B,2BAA4B;AAgC5B,+EAA+E;AAE/E,MAAa,YAAY;IACb,MAAM,CAAC,QAAQ,GAAwB,IAAI,CAAC;IAE5C,QAAQ,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC5C,MAAM,CAA6C;IAE3D,YAAoB,OAA4B;QAC5C,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC;QAExC,KAAK,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YACrE,MAAM,IAAI,GAAG,YAAY,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE3D,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,YAAY,2BAA2B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC1G,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,kBAAW,CAAC;gBAC5B,QAAQ,EAAE,IAAA,WAAI,EAAC,IAAA,WAAM,GAAE,EAAE,iBAAiB,YAAY,aAAa,CAAC;gBACpE,WAAW,EAAE,IAAI;aACpB,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,IAAI,qBAAa,CAAC,IAAI,EAAE;gBACpC,OAAO;gBACP,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,IAAI,uBAAe,EAAE;gBAClD,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,EAAE;gBACrC,aAAa,EAAE,MAAM,CAAC,aAAa;aACtC,CAAC,CAAC;YAEH,cAAc;YACd,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,CAC1B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iBAAiB,YAAY,kBAAkB,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACvF,OAAO,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,GAAG,EAAE,EAAE,CAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,YAAY,sBAAsB,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1F,OAAO,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,GAAG,EAAE,EAAE,CAC/B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,YAAY,uBAAuB,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC3F,OAAO,CAAC,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAChC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iBAAiB,YAAY,sBAAsB,CAAC,CAAC,CAAC;YAE5E,gBAAgB;YAChB,MAAM,QAAQ,GAAG,OAAO,CAAC,qBAAqB,IAAI,OAAO,CAAC;YAC1D,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;gBACf,OAAO,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YACxC,CAAC;YAED,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,IAAI,CACZ,iBAAiB,YAAY,sBAAsB,IAAI,CAAC,MAAM,OAAO,CACxE,CAAC;QACN,CAAC;IACL,CAAC;IAED,2EAA2E;IAE3E;;OAEG;IACH,MAAM,CAAC,WAAW,CAAC,OAA4B;QAC3C,IAAI,YAAY,CAAC,QAAQ,EAAE,CAAC;YACxB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,YAAY,CAAC,QAAQ,EAAE,CAAC;QAC1D,CAAC;QAED,IAAI,CAAC;YACD,YAAY,CAAC,QAAQ,GAAG,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC;YAClD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,YAAY,CAAC,QAAQ,EAAE,CAAC;QAC1D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;aACnE,CAAC;QACN,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK;QACR,IAAI,YAAY,CAAC,QAAQ,EAAE,CAAC;YACxB,YAAY,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YAChC,YAAY,CAAC,QAAQ,GAAG,IAAI,CAAC;QACjC,CAAC;IACL,CAAC;IAED,2EAA2E;IAE3E;;;;;OAKG;IACH,KAAK,CAAC,OAAO,CACT,EAAqD,EACrD,OAA+D;QAE/D,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CACX,oCAAoC,OAAO,CAAC,QAAQ,KAAK;gBACzD,cAAc,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACvD,CAAC;QACN,CAAC;QACD,mFAAmF;QACnF,MAAM,EAAE,QAAQ,EAAE,GAAG,eAAe,EAAE,GAAG,OAAO,CAAC;QACjD,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,eAAe,CAAC,CAAC;IAChD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,CAAC,aAAa,CAChB,EAA0E,EAC1E,OAA+D;QAE/D,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CACX,oCAAoC,OAAO,CAAC,QAAQ,KAAK;gBACzD,cAAc,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACvD,CAAC;QACN,CAAC;QACD,MAAM,EAAE,QAAQ,EAAE,GAAG,eAAe,EAAE,GAAG,OAAO,CAAC;QACjD,KAAK,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,EAAE,eAAe,CAAC,CAAC;IACtD,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,QAAgB;QACnB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAC1B,OAAO,OAAO,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,QAAgB;QACrB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAC1B,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,aAAa;QACT,MAAM,KAAK,GAAuC,EAAE,CAAC;QACrD,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC1C,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QACrC,CAAC;QACD,OAAO,KAAK,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,YAAY;QACR,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,QAAgB;QACvB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,OAAO;QACH,KAAK,MAAM,CAAC,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtC,OAAO,CAAC,gBAAgB,EAAE,CAAC;QAC/B,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAED,2EAA2E;IAEnE,MAAM,CAAC,gBAAgB,CAAC,OAAiB;QAC7C,MAAM,IAAI,GAAa,EAAE,CAAC;QAC1B,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACtC,IAAI,CAAC,QAAQ;gBAAE,SAAS;YACxB,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;YAChC,IAAI,CAAC,OAAO;gBAAE,SAAS;YAEvB,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACnC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;wBACxB,IAAI,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;wBAC3E,SAAS;oBACb,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC,CAAC,cAAc,CAAC,CAAC;YAC9B,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QAClF,CAAC;QACD,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAC9B,CAAC;;AAvML,oCAwMC"}
@@ -0,0 +1,45 @@
1
+ /**
2
+ * OpenAI Preset — One-line OpenAI API key management
3
+ *
4
+ * Pre-configured singleton with:
5
+ * - Reads `OPENAI_API_KEY` from env
6
+ * - LatencyStrategy for optimal key selection
7
+ * - File persistence (survives restarts)
8
+ * - Health checks every 5 minutes
9
+ * - Concurrency limit of 20
10
+ *
11
+ * @module presets/openai
12
+ *
13
+ * @example
14
+ * ```ts
15
+ * import { OpenAIManager } from '@splashcodex/api-key-manager/presets/openai';
16
+ *
17
+ * const result = OpenAIManager.getInstance();
18
+ * if (!result.success) throw result.error;
19
+ * const openai = result.data;
20
+ *
21
+ * const response = await openai.execute(async (key) => {
22
+ * const client = new OpenAI({ apiKey: key });
23
+ * const completion = await client.chat.completions.create({
24
+ * model: 'gpt-4o',
25
+ * messages: [{ role: 'user', content: 'Hello!' }],
26
+ * });
27
+ * return completion.choices[0].message.content;
28
+ * }, { maxRetries: 3, timeoutMs: 30000 });
29
+ * ```
30
+ */
31
+ import { BasePreset, PresetOptions, Result } from './base';
32
+ export declare class OpenAIManager extends BasePreset {
33
+ private static readonly PROVIDER;
34
+ private constructor();
35
+ private static getDefaultOptions;
36
+ /**
37
+ * Get or create the singleton OpenAIManager instance.
38
+ *
39
+ * @param overrides - Optional configuration overrides
40
+ * @returns Result<OpenAIManager>
41
+ */
42
+ static getInstance(overrides?: Partial<PresetOptions>): Result<OpenAIManager>;
43
+ static reset(): void;
44
+ }
45
+ export { Result } from './base';
@@ -0,0 +1,62 @@
1
+ "use strict";
2
+ /**
3
+ * OpenAI Preset — One-line OpenAI API key management
4
+ *
5
+ * Pre-configured singleton with:
6
+ * - Reads `OPENAI_API_KEY` from env
7
+ * - LatencyStrategy for optimal key selection
8
+ * - File persistence (survives restarts)
9
+ * - Health checks every 5 minutes
10
+ * - Concurrency limit of 20
11
+ *
12
+ * @module presets/openai
13
+ *
14
+ * @example
15
+ * ```ts
16
+ * import { OpenAIManager } from '@splashcodex/api-key-manager/presets/openai';
17
+ *
18
+ * const result = OpenAIManager.getInstance();
19
+ * if (!result.success) throw result.error;
20
+ * const openai = result.data;
21
+ *
22
+ * const response = await openai.execute(async (key) => {
23
+ * const client = new OpenAI({ apiKey: key });
24
+ * const completion = await client.chat.completions.create({
25
+ * model: 'gpt-4o',
26
+ * messages: [{ role: 'user', content: 'Hello!' }],
27
+ * });
28
+ * return completion.choices[0].message.content;
29
+ * }, { maxRetries: 3, timeoutMs: 30000 });
30
+ * ```
31
+ */
32
+ Object.defineProperty(exports, "__esModule", { value: true });
33
+ exports.OpenAIManager = void 0;
34
+ const base_1 = require("./base");
35
+ class OpenAIManager extends base_1.BasePreset {
36
+ static PROVIDER = 'openai';
37
+ constructor(keys, options) {
38
+ super(keys, options);
39
+ }
40
+ static getDefaultOptions() {
41
+ return {
42
+ envKeys: ['OPENAI_API_KEY'],
43
+ provider: OpenAIManager.PROVIDER,
44
+ concurrency: 20,
45
+ healthCheckIntervalMs: 300_000,
46
+ };
47
+ }
48
+ /**
49
+ * Get or create the singleton OpenAIManager instance.
50
+ *
51
+ * @param overrides - Optional configuration overrides
52
+ * @returns Result<OpenAIManager>
53
+ */
54
+ static getInstance(overrides) {
55
+ return base_1.BasePreset.createInstance(OpenAIManager, OpenAIManager.getDefaultOptions(), overrides);
56
+ }
57
+ static reset() {
58
+ base_1.BasePreset.resetInstance(OpenAIManager.PROVIDER);
59
+ }
60
+ }
61
+ exports.OpenAIManager = OpenAIManager;
62
+ //# sourceMappingURL=openai.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/presets/openai.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;;;AAEH,iCAA2D;AAE3D,MAAa,aAAc,SAAQ,iBAAU;IACjC,MAAM,CAAU,QAAQ,GAAG,QAAQ,CAAC;IAE5C,YAAoB,IAAc,EAAE,OAAsB;QACtD,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACzB,CAAC;IAEO,MAAM,CAAC,iBAAiB;QAC5B,OAAO;YACH,OAAO,EAAE,CAAC,gBAAgB,CAAC;YAC3B,QAAQ,EAAE,aAAa,CAAC,QAAQ;YAChC,WAAW,EAAE,EAAE;YACf,qBAAqB,EAAE,OAAO;SACjC,CAAC;IACN,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,WAAW,CAAC,SAAkC;QACjD,OAAO,iBAAU,CAAC,cAAc,CAC5B,aAAyF,EACzF,aAAa,CAAC,iBAAiB,EAAE,EACjC,SAAS,CACa,CAAC;IAC/B,CAAC;IAED,MAAM,CAAC,KAAK;QACR,iBAAU,CAAC,aAAa,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IACrD,CAAC;;AAhCL,sCAiCC"}
package/package.json CHANGED
@@ -1,23 +1,97 @@
1
1
  {
2
2
  "name": "@splashcodex/api-key-manager",
3
- "version": "4.1.0",
4
- "description": "Universal API Key Rotation System for rate-limited APIs",
3
+ "version": "5.0.2",
4
+ "description": "Universal API Key Management Gateway with provider presets, persistence, and multi-provider vault",
5
+ "bin": {
6
+ "api-key-manager": "./bin/cli.js"
7
+ },
5
8
  "main": "dist/index.js",
6
9
  "types": "dist/index.d.ts",
10
+ "exports": {
11
+ ".": {
12
+ "require": "./dist/index.js",
13
+ "types": "./dist/index.d.ts"
14
+ },
15
+ "./presets": {
16
+ "require": "./dist/presets/index.js",
17
+ "types": "./dist/presets/index.d.ts"
18
+ },
19
+ "./presets/gemini": {
20
+ "require": "./dist/presets/gemini.js",
21
+ "types": "./dist/presets/gemini.d.ts"
22
+ },
23
+ "./presets/openai": {
24
+ "require": "./dist/presets/openai.js",
25
+ "types": "./dist/presets/openai.d.ts"
26
+ },
27
+ "./presets/multi": {
28
+ "require": "./dist/presets/multi.js",
29
+ "types": "./dist/presets/multi.d.ts"
30
+ },
31
+ "./presets/base": {
32
+ "require": "./dist/presets/base.js",
33
+ "types": "./dist/presets/base.d.ts"
34
+ },
35
+ "./persistence": {
36
+ "require": "./dist/persistence/index.js",
37
+ "types": "./dist/persistence/index.d.ts"
38
+ },
39
+ "./persistence/file": {
40
+ "require": "./dist/persistence/file.js",
41
+ "types": "./dist/persistence/file.d.ts"
42
+ },
43
+ "./persistence/memory": {
44
+ "require": "./dist/persistence/memory.js",
45
+ "types": "./dist/persistence/memory.d.ts"
46
+ }
47
+ },
48
+ "typesVersions": {
49
+ "*": {
50
+ "presets": [
51
+ "dist/presets/index.d.ts"
52
+ ],
53
+ "presets/gemini": [
54
+ "dist/presets/gemini.d.ts"
55
+ ],
56
+ "presets/openai": [
57
+ "dist/presets/openai.d.ts"
58
+ ],
59
+ "presets/multi": [
60
+ "dist/presets/multi.d.ts"
61
+ ],
62
+ "presets/base": [
63
+ "dist/presets/base.d.ts"
64
+ ],
65
+ "persistence": [
66
+ "dist/persistence/index.d.ts"
67
+ ],
68
+ "persistence/file": [
69
+ "dist/persistence/file.d.ts"
70
+ ],
71
+ "persistence/memory": [
72
+ "dist/persistence/memory.d.ts"
73
+ ]
74
+ }
75
+ },
7
76
  "files": [
8
77
  "dist",
9
- "src"
78
+ "src",
79
+ "bin"
10
80
  ],
11
81
  "scripts": {
12
82
  "build": "tsc",
13
83
  "test": "jest --verbose",
14
- "prepublishOnly": "npm run build"
84
+ "prepublishOnly": "npm run build",
85
+ "postinstall": "node -e \"console.log('\\n\\x1b[32m🌟 Successfully installed @splashcodex/api-key-manager v5.0!\\x1b[0m\\n\\x1b[36m📚 Documentation: https://www.npmjs.com/package/@splashcodex/ApiKeyManager\\x1b[0m\\n\\x1b[33m🚀 Quick Start:\\x1b[0m\\n import { GeminiManager } from \\'@splashcodex/api-key-manager/presets/gemini\\';\\n const gemini = GeminiManager.getInstance().data;\\n')\"",
86
+ "gateway": "ts-node gateway/server.ts",
87
+ "gateway:dev": "npx nodemon --watch gateway/ --ext ts --exec ts-node gateway/server.ts"
15
88
  },
16
89
  "keywords": [
17
90
  "api",
18
91
  "key",
19
92
  "rotation",
20
93
  "gemini",
94
+ "openai",
21
95
  "rate-limit",
22
96
  "circuit-breaker",
23
97
  "resilience",
@@ -25,7 +99,12 @@
25
99
  "timeout",
26
100
  "retry",
27
101
  "gateway",
28
- "load-balancing"
102
+ "load-balancing",
103
+ "preset",
104
+ "persistence",
105
+ "multi-provider",
106
+ "semantic-cache",
107
+ "streaming"
29
108
  ],
30
109
  "author": "Antigravity",
31
110
  "license": "ISC",
@@ -36,5 +115,9 @@
36
115
  "ts-jest": "^29.4.6",
37
116
  "ts-node": "^10.9.2",
38
117
  "typescript": "^5.3.3"
118
+ },
119
+ "dependencies": {
120
+ "@fastify/cors": "^11.2.0",
121
+ "fastify": "^5.7.4"
39
122
  }
40
123
  }
package/src/index.ts CHANGED
@@ -1,13 +1,31 @@
1
1
  /**
2
- * Universal ApiKeyManager v4.0 — Mastermind Edition
2
+ * Universal ApiKeyManager v5.0 — Ecosystem Edition
3
3
  * Implements: Rotation, Circuit Breaker, Persistence, Exponential Backoff, Strategies,
4
4
  * Event Emitter, Fallback, execute(), Timeout, Auto-Retry, Provider Tags,
5
5
  * Health Checks, Bulkhead/Concurrency
6
+ * NEW in v5.0: Provider Presets (GeminiManager, OpenAIManager, MultiManager),
7
+ * Built-in Persistence (FileStorage, MemoryStorage)
6
8
  * Gemini-Specific: finishReason handling, Safety blocks, RECITATION detection
7
9
  */
8
10
 
9
11
  import { EventEmitter } from 'events';
10
12
 
13
+ // ─── Re-exports: Persistence ─────────────────────────────────────────────────
14
+ // Persistence adapters can be imported from root or via subpath
15
+ export { FileStorage } from './persistence/file';
16
+ export type { FileStorageOptions } from './persistence/file';
17
+ export { MemoryStorage } from './persistence/memory';
18
+
19
+ // ─── Presets ─────────────────────────────────────────────────────────────────
20
+ // Presets are available via subpath imports to avoid circular dependencies:
21
+ // import { GeminiManager } from '@splashcodex/api-key-manager/presets/gemini';
22
+ // import { OpenAIManager } from '@splashcodex/api-key-manager/presets/openai';
23
+ // import { MultiManager } from '@splashcodex/api-key-manager/presets/multi';
24
+ // Or import all at once:
25
+ // import { GeminiManager, OpenAIManager, MultiManager } from '@splashcodex/api-key-manager/presets';
26
+
27
+
28
+
11
29
  // ─── Interfaces & Types ──────────────────────────────────────────────────────
12
30
 
13
31
  export interface KeyState {
@@ -185,12 +203,15 @@ export class WeightedStrategy implements LoadBalancingStrategy {
185
203
  }
186
204
 
187
205
  /**
188
- * Latency Strategy: Pick lowest average latency
206
+ * Latency Strategy: Pick lowest average latency with LRU tie-break
189
207
  */
190
208
  export class LatencyStrategy implements LoadBalancingStrategy {
191
209
  next(candidates: KeyState[]): KeyState | null {
192
210
  if (candidates.length === 0) return null;
193
- candidates.sort((a, b) => a.averageLatency - b.averageLatency);
211
+ candidates.sort((a, b) => {
212
+ if (a.averageLatency !== b.averageLatency) return a.averageLatency - b.averageLatency;
213
+ return a.lastUsed - b.lastUsed; // LRU tie-break
214
+ });
194
215
  return candidates[0];
195
216
  }
196
217
  }
@@ -0,0 +1,89 @@
1
+ /**
2
+ * File-Based Storage Adapter
3
+ *
4
+ * Persists API key state to a JSON file on disk.
5
+ * Survives process restarts so keys don't reset to CLOSED
6
+ * when an app reboots.
7
+ *
8
+ * Extracted from the proven WhatsDeX adapter pattern.
9
+ *
10
+ * @module persistence/file
11
+ */
12
+
13
+ import { writeFileSync, readFileSync, existsSync, mkdirSync, unlinkSync } from 'fs';
14
+ import { dirname, join } from 'path';
15
+ import { tmpdir } from 'os';
16
+
17
+ export interface FileStorageOptions {
18
+ /** Full path to the state file. Defaults to `os.tmpdir()/codedex_api_key_state.json` */
19
+ filePath?: string;
20
+ /** If true, clears any existing state file on construction (fresh start). Default: true */
21
+ clearOnInit?: boolean;
22
+ }
23
+
24
+ /**
25
+ * File-based storage adapter for ApiKeyManager.
26
+ *
27
+ * @example
28
+ * ```ts
29
+ * import { ApiKeyManager } from '@splashcodex/api-key-manager';
30
+ * import { FileStorage } from '@splashcodex/api-key-manager/persistence/file';
31
+ *
32
+ * const manager = new ApiKeyManager(keys, {
33
+ * storage: new FileStorage({ filePath: './state.json' })
34
+ * });
35
+ * ```
36
+ */
37
+ export class FileStorage {
38
+ private filePath: string;
39
+
40
+ constructor(options: FileStorageOptions = {}) {
41
+ this.filePath = options.filePath || join(tmpdir(), 'codedex_api_key_state.json');
42
+
43
+ if (options.clearOnInit !== false) {
44
+ this.clear();
45
+ }
46
+ }
47
+
48
+ getItem(_key: string): string | null {
49
+ try {
50
+ if (existsSync(this.filePath)) {
51
+ return readFileSync(this.filePath, 'utf-8');
52
+ }
53
+ } catch {
54
+ // Silently fail — state will be rebuilt from scratch
55
+ }
56
+ return null;
57
+ }
58
+
59
+ setItem(_key: string, value: string): void {
60
+ try {
61
+ const dir = dirname(this.filePath);
62
+ if (!existsSync(dir)) {
63
+ mkdirSync(dir, { recursive: true });
64
+ }
65
+ writeFileSync(this.filePath, value, 'utf-8');
66
+ } catch {
67
+ // Silently fail — state will be lost on restart
68
+ }
69
+ }
70
+
71
+ /**
72
+ * Delete the persisted state file.
73
+ * Useful for clearing stale dead-key states from a previous session.
74
+ */
75
+ clear(): void {
76
+ try {
77
+ if (existsSync(this.filePath)) {
78
+ unlinkSync(this.filePath);
79
+ }
80
+ } catch {
81
+ // Silently fail
82
+ }
83
+ }
84
+
85
+ /** Get the path to the state file (for debugging/logging) */
86
+ getFilePath(): string {
87
+ return this.filePath;
88
+ }
89
+ }
@@ -0,0 +1,7 @@
1
+ /**
2
+ * Persistence Adapters
3
+ * @module persistence
4
+ */
5
+ export { FileStorage } from './file';
6
+ export type { FileStorageOptions } from './file';
7
+ export { MemoryStorage } from './memory';
@@ -0,0 +1,43 @@
1
+ /**
2
+ * In-Memory Storage Adapter
3
+ *
4
+ * Simple key-value storage that lives only for the process lifetime.
5
+ * Useful for testing, serverless functions, or when persistence isn't needed.
6
+ *
7
+ * @module persistence/memory
8
+ */
9
+
10
+ /**
11
+ * In-memory storage adapter for ApiKeyManager.
12
+ *
13
+ * @example
14
+ * ```ts
15
+ * import { ApiKeyManager } from '@splashcodex/api-key-manager';
16
+ * import { MemoryStorage } from '@splashcodex/api-key-manager/persistence/memory';
17
+ *
18
+ * const manager = new ApiKeyManager(keys, {
19
+ * storage: new MemoryStorage()
20
+ * });
21
+ * ```
22
+ */
23
+ export class MemoryStorage {
24
+ private store: Map<string, string> = new Map();
25
+
26
+ getItem(key: string): string | null {
27
+ return this.store.get(key) ?? null;
28
+ }
29
+
30
+ setItem(key: string, value: string): void {
31
+ this.store.set(key, value);
32
+ }
33
+
34
+ /** Clear all stored state */
35
+ clear(): void {
36
+ this.store.clear();
37
+ }
38
+
39
+ /** Get the number of stored entries */
40
+ get size(): number {
41
+ return this.store.size;
42
+ }
43
+ }