@theihtisham/budget-llm 1.0.0

package/src/config.ts

@@ -0,0 +1,254 @@
+import { z } from 'zod';
+import dotenv from 'dotenv';
+import path from 'path';
+import type { ProviderConfig, ModelInfo, BudgetConfig } from './types';
+
+dotenv.config({ path: path.resolve(process.cwd(), '.env') });
+
+// ---- Environment Schema ----
+
+const envSchema = z.object({
+  PORT: z.coerce.number().default(3210),
+  NODE_ENV: z.enum(['development', 'production', 'test']).default('development'),
+  ENCRYPTION_KEY: z.string().min(32).default('dev_key_change_in_production_32chars!!'),
+  DEFAULT_DAILY_BUDGET: z.coerce.number().positive().default(10),
+  DEFAULT_MONTHLY_BUDGET: z.coerce.number().positive().default(200),
+  DEFAULT_PER_REQUEST_CAP: z.coerce.number().positive().default(1),
+  CACHE_TTL: z.coerce.number().positive().default(3600),
+  RATE_LIMIT_WINDOW_MS: z.coerce.number().positive().default(60000),
+  RATE_LIMIT_MAX_REQUESTS: z.coerce.number().positive().default(60),
+  LOG_LEVEL: z.enum(['debug', 'info', 'warn', 'error']).default('info'),
+  OPENAI_API_KEY: z.string().optional(),
+  ANTHROPIC_API_KEY: z.string().optional(),
+  GOOGLE_API_KEY: z.string().optional(),
+  DEEPSEEK_API_KEY: z.string().optional(),
+});
+
+export type EnvConfig = z.infer<typeof envSchema>;
+
+function parseEnv(): EnvConfig {
+  const result = envSchema.safeParse(process.env);
+  if (!result.success) {
+    const errors = result.error.issues
+      .map((i) => `${i.path.join('.')}: ${i.message}`)
+      .join('; ');
+    throw new Error(`Configuration error: ${errors}`);
+  }
+  return result.data;
+}
+
+export const env = parseEnv();
+
+// ---- Provider Configurations ----
+
+export function getProviders(): ProviderConfig[] {
+  return [
+    {
+      id: 'openai',
+      name: 'OpenAI',
+      baseUrl: 'https://api.openai.com/v1',
+      apiKey: env.OPENAI_API_KEY ?? '',
+      enabled: !!env.OPENAI_API_KEY,
+      priority: 1,
+      timeoutMs: 30000,
+    },
+    {
+      id: 'anthropic',
+      name: 'Anthropic',
+      baseUrl: 'https://api.anthropic.com/v1',
+      apiKey: env.ANTHROPIC_API_KEY ?? '',
+      enabled: !!env.ANTHROPIC_API_KEY,
+      priority: 2,
+      timeoutMs: 30000,
+    },
+    {
+      id: 'google',
+      name: 'Google',
+      baseUrl: 'https://generativelanguage.googleapis.com/v1beta',
+      apiKey: env.GOOGLE_API_KEY ?? '',
+      enabled: !!env.GOOGLE_API_KEY,
+      priority: 3,
+      timeoutMs: 30000,
+    },
+    {
+      id: 'deepseek',
+      name: 'DeepSeek',
+      baseUrl: 'https://api.deepseek.com/v1',
+      apiKey: env.DEEPSEEK_API_KEY ?? '',
+      enabled: !!env.DEEPSEEK_API_KEY,
+      priority: 4,
+      timeoutMs: 30000,
+    },
+  ];
+}
+
+// ---- Model Catalog ----
+
+export const MODEL_CATALOG: ModelInfo[] = [
+  // OpenAI
+  {
+    id: 'gpt-4o',
+    provider: 'openai',
+    displayName: 'GPT-4o',
+    inputPricePer1M: 2.50,
+    outputPricePer1M: 10.00,
+    contextWindow: 128000,
+    maxOutputTokens: 16384,
+    capabilities: ['code', 'creative', 'reasoning', 'chat', 'analysis', 'math'],
+    qualityScore: 9,
+    speedScore: 7,
+    costScore: 5,
+  },
+  {
+    id: 'gpt-4o-mini',
+    provider: 'openai',
+    displayName: 'GPT-4o Mini',
+    inputPricePer1M: 0.15,
+    outputPricePer1M: 0.60,
+    contextWindow: 128000,
+    maxOutputTokens: 16384,
+    capabilities: ['code', 'chat', 'summarization', 'translation'],
+    qualityScore: 7,
+    speedScore: 9,
+    costScore: 9,
+  },
+  {
+    id: 'gpt-4-turbo',
+    provider: 'openai',
+    displayName: 'GPT-4 Turbo',
+    inputPricePer1M: 10.00,
+    outputPricePer1M: 30.00,
+    contextWindow: 128000,
+    maxOutputTokens: 4096,
+    capabilities: ['code', 'creative', 'reasoning', 'chat', 'analysis', 'math'],
+    qualityScore: 9,
+    speedScore: 6,
+    costScore: 3,
+  },
+  // Anthropic
+  {
+    id: 'claude-sonnet-4-20250514',
+    provider: 'anthropic',
+    displayName: 'Claude Sonnet 4',
+    inputPricePer1M: 3.00,
+    outputPricePer1M: 15.00,
+    contextWindow: 200000,
+    maxOutputTokens: 8192,
+    capabilities: ['code', 'creative', 'reasoning', 'chat', 'analysis'],
+    qualityScore: 9,
+    speedScore: 7,
+    costScore: 5,
+  },
+  {
+    id: 'claude-haiku-3-5-20241022',
+    provider: 'anthropic',
+    displayName: 'Claude 3.5 Haiku',
+    inputPricePer1M: 0.80,
+    outputPricePer1M: 4.00,
+    contextWindow: 200000,
+    maxOutputTokens: 8192,
+    capabilities: ['code', 'chat', 'summarization', 'translation'],
+    qualityScore: 7,
+    speedScore: 9,
+    costScore: 7,
+  },
+  // Google
+  {
+    id: 'gemini-2.0-flash',
+    provider: 'google',
+    displayName: 'Gemini 2.0 Flash',
+    inputPricePer1M: 0.10,
+    outputPricePer1M: 0.40,
+    contextWindow: 1048576,
+    maxOutputTokens: 8192,
+    capabilities: ['code', 'chat', 'summarization', 'translation', 'analysis', 'math'],
+    qualityScore: 7,
+    speedScore: 10,
+    costScore: 10,
+  },
+  {
+    id: 'gemini-2.0-flash-lite',
+    provider: 'google',
+    displayName: 'Gemini 2.0 Flash Lite',
+    inputPricePer1M: 0.075,
+    outputPricePer1M: 0.30,
+    contextWindow: 1048576,
+    maxOutputTokens: 8192,
+    capabilities: ['chat', 'summarization', 'translation'],
+    qualityScore: 6,
+    speedScore: 10,
+    costScore: 10,
+  },
+  // DeepSeek
+  {
+    id: 'deepseek-chat',
+    provider: 'deepseek',
+    displayName: 'DeepSeek V3',
+    inputPricePer1M: 0.27,
+    outputPricePer1M: 1.10,
+    contextWindow: 65536,
+    maxOutputTokens: 8192,
+    capabilities: ['code', 'chat', 'reasoning', 'analysis', 'math'],
+    qualityScore: 8,
+    speedScore: 8,
+    costScore: 8,
+  },
+  {
+    id: 'deepseek-reasoner',
+    provider: 'deepseek',
+    displayName: 'DeepSeek R1',
+    inputPricePer1M: 0.55,
+    outputPricePer1M: 2.19,
+    contextWindow: 65536,
+    maxOutputTokens: 8192,
+    capabilities: ['reasoning', 'math', 'code', 'analysis'],
+    qualityScore: 9,
+    speedScore: 5,
+    costScore: 7,
+  },
+];
+
+// ---- Budget Defaults ----
+
+export function getDefaultBudget(): BudgetConfig {
+  return {
+    dailyBudget: env.DEFAULT_DAILY_BUDGET,
+    monthlyBudget: env.DEFAULT_MONTHLY_BUDGET,
+    perRequestCap: env.DEFAULT_PER_REQUEST_CAP,
+  };
+}
+
+// ---- Database Path ----
+
+export function getDbPath(): string {
+  return path.resolve(process.cwd(), 'data', 'budgetllm.db');
+}
+
+// ---- Logging Helper ----
+
+type LogLevel = 'debug' | 'info' | 'warn' | 'error';
+const LOG_LEVELS: Record<LogLevel, number> = {
+  debug: 0,
+  info: 1,
+  warn: 2,
+  error: 3,
+};
+
+function shouldLog(level: LogLevel): boolean {
+  return LOG_LEVELS[level] >= LOG_LEVELS[env.LOG_LEVEL];
+}
+
+export const log = {
+  debug: (msg: string, data?: unknown) => {
+    if (shouldLog('debug')) console.debug(`[DEBUG] ${msg}`, data ?? '');
+  },
+  info: (msg: string, data?: unknown) => {
+    if (shouldLog('info')) console.info(`[INFO] ${msg}`, data ?? '');
+  },
+  warn: (msg: string, data?: unknown) => {
+    if (shouldLog('warn')) console.warn(`[WARN] ${msg}`, data ?? '');
+  },
+  error: (msg: string, data?: unknown) => {
+    if (shouldLog('error')) console.error(`[ERROR] ${msg}`, data ?? '');
+  },
+};