@lobehub/chat/src/server/modules/EdgeConfig/index.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import { EdgeConfigClient, createClient } from '@vercel/edge-config';
import createDebug from 'debug';

import { appEnv } from '@/envs/app';

import { EdgeConfigData, EdgeConfigKeys } from './types';

const debug = createDebug('lobe-server:edge-config');

export class EdgeConfig {
  get client(): EdgeConfigClient {
    if (!appEnv.VERCEL_EDGE_CONFIG) {
      throw new Error('VERCEL_EDGE_CONFIG is not set');
    }
    return createClient(appEnv.VERCEL_EDGE_CONFIG);
  }

  /**
   * Check if Edge Config is enabled
   */
  static isEnabled() {
    const isEnabled = !!appEnv.VERCEL_EDGE_CONFIG;
    debug('VERCEL_EDGE_CONFIG env var: %s', appEnv.VERCEL_EDGE_CONFIG ? 'SET' : 'NOT SET');
    debug('EdgeConfig enabled: %s', isEnabled);
    return isEnabled;
  }

  private async getValue<K extends EdgeConfigKeys>(key: K) {
    return this.client.get<EdgeConfigData[K]>(key);
  }

  private async getValues<const K extends EdgeConfigKeys>(keys: K[]) {
    return this.client.getAll<Pick<EdgeConfigData, K>>(keys);
  }

  getAgentRestrictions = async () => {
    const { assistant_blacklist: blacklist, assistant_whitelist: whitelist } = await this.getValues(
      ['assistant_blacklist', 'assistant_whitelist'],
    );
    return { blacklist, whitelist };
  };

  getFeatureFlags = async () => {
    const featureFlags = await this.getValue('feature_flags');
    debug('Feature flags retrieved: %O', featureFlags);
    return featureFlags;
  };
}