Spaces:
Configuration error
Configuration error
Update config.ts: user-configurable base URLs, models, and LLM provider override
Browse files- multi-agent-system/config.ts +15 -9
multi-agent-system/config.ts
CHANGED
|
@@ -25,24 +25,30 @@ export interface APIConfig {
|
|
| 25 |
export function loadAndValidateConfig(): { config: APIConfig; warnings: string[] } {
|
| 26 |
const warnings: string[] = [];
|
| 27 |
|
|
|
|
| 28 |
const config: APIConfig = {
|
| 29 |
gemini: {
|
| 30 |
apiKey: process.env.GEMINI_API_KEY || '',
|
| 31 |
-
baseUrl: 'https://generativelanguage.googleapis.com/v1beta',
|
| 32 |
-
models: {
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
},
|
| 34 |
featherless: {
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
|
|
|
| 38 |
},
|
| 39 |
huggingface: {
|
| 40 |
token: process.env.HF_TOKEN || '',
|
| 41 |
-
baseUrl: 'https://api-inference.huggingface.co/models',
|
| 42 |
models: {
|
| 43 |
-
ner: 'dbmdz/bert-large-cased-finetuned-conll03-english',
|
| 44 |
-
zeroShot: 'facebook/bart-large-mnli',
|
| 45 |
-
embeddings: 'sentence-transformers/all-MiniLM-L6-v2',
|
| 46 |
},
|
| 47 |
},
|
| 48 |
};
|
|
|
|
| 25 |
export function loadAndValidateConfig(): { config: APIConfig; warnings: string[] } {
|
| 26 |
const warnings: string[] = [];
|
| 27 |
|
| 28 |
+
// ═══ USER-CONFIGURABLE: All base URLs, API keys, and models from .env ═══
|
| 29 |
const config: APIConfig = {
|
| 30 |
gemini: {
|
| 31 |
apiKey: process.env.GEMINI_API_KEY || '',
|
| 32 |
+
baseUrl: process.env.GEMINI_BASE_URL || 'https://generativelanguage.googleapis.com/v1beta',
|
| 33 |
+
models: {
|
| 34 |
+
pro: process.env.GEMINI_MODEL_PRO || 'gemini-2.5-pro',
|
| 35 |
+
flash: process.env.GEMINI_MODEL_FLASH || 'gemini-2.5-flash',
|
| 36 |
+
lite: process.env.GEMINI_MODEL_LITE || 'gemini-2.5-flash-lite',
|
| 37 |
+
},
|
| 38 |
},
|
| 39 |
featherless: {
|
| 40 |
+
// If user provides a custom LLM, use that instead of Featherless
|
| 41 |
+
apiKey: process.env.LLM_API_KEY || process.env.FEATHERLESS_API_KEY || '',
|
| 42 |
+
baseUrl: process.env.LLM_BASE_URL || process.env.FEATHERLESS_BASE_URL || 'https://api.featherless.ai/v1',
|
| 43 |
+
model: process.env.LLM_MODEL || process.env.FEATHERLESS_MODEL || 'meta-llama/Meta-Llama-3.1-70B-Instruct',
|
| 44 |
},
|
| 45 |
huggingface: {
|
| 46 |
token: process.env.HF_TOKEN || '',
|
| 47 |
+
baseUrl: process.env.HF_BASE_URL || 'https://api-inference.huggingface.co/models',
|
| 48 |
models: {
|
| 49 |
+
ner: process.env.HF_MODEL_NER || 'dbmdz/bert-large-cased-finetuned-conll03-english',
|
| 50 |
+
zeroShot: process.env.HF_MODEL_ZERO_SHOT || 'facebook/bart-large-mnli',
|
| 51 |
+
embeddings: process.env.HF_MODEL_EMBEDDINGS || 'sentence-transformers/all-MiniLM-L6-v2',
|
| 52 |
},
|
| 53 |
},
|
| 54 |
};
|