Muthukumarank commited on
Commit
0eb9ef2
·
verified ·
1 Parent(s): 6337da3

Update config.ts: user-configurable base URLs, models, and LLM provider override

Browse files
Files changed (1) hide show
  1. 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: { pro: 'gemini-2.5-pro', flash: 'gemini-2.5-flash', lite: 'gemini-2.5-flash-lite' },
 
 
 
 
33
  },
34
  featherless: {
35
- apiKey: process.env.FEATHERLESS_API_KEY || '',
36
- baseUrl: process.env.FEATHERLESS_BASE_URL || 'https://api.featherless.ai/v1',
37
- model: process.env.FEATHERLESS_MODEL || 'meta-llama/Meta-Llama-3.1-70B-Instruct',
 
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
  };