@lobehub/chat/src/envs/image.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 { createEnv } from '@t3-oss/env-nextjs';
import { z } from 'zod';

import { MAX_DEFAULT_IMAGE_NUM, MIN_DEFAULT_IMAGE_NUM } from '@/const/settings';

export const getImageConfig = () => {
  return createEnv({
    runtimeEnv: {
      AI_IMAGE_DEFAULT_IMAGE_NUM: process.env.AI_IMAGE_DEFAULT_IMAGE_NUM,
    },
    server: {
      AI_IMAGE_DEFAULT_IMAGE_NUM: z.coerce
        .number()
        .min(
          MIN_DEFAULT_IMAGE_NUM,
          `AI_IMAGE_DEFAULT_IMAGE_NUM must be at least ${MIN_DEFAULT_IMAGE_NUM}`,
        )
        .max(
          MAX_DEFAULT_IMAGE_NUM,
          `AI_IMAGE_DEFAULT_IMAGE_NUM must be at most ${MAX_DEFAULT_IMAGE_NUM}`,
        )
        .optional(),
    },
  });
};

export const imageEnv = getImageConfig();