const DEFAULT_MAX_TOKENS = 1024;
const DEFAULT_CONTEXT_LENGTH = 4096;
const DEFAULT_TEMPERATURE = 0.5;

const DEFAULT_ARGS = {
    maxTokens: DEFAULT_MAX_TOKENS,
    temperature: DEFAULT_TEMPERATURE,
};

const CONTEXT_LENGTH_FOR_MODEL: { [name: string]: number } = {
    "gpt-3.5-turbo": 4096,
    "gpt-3.5-turbo-0613": 4096,
    "gpt-3.5-turbo-16k": 16_384,
    "gpt-4": 8192,
    "gpt-35-turbo-16k": 16_384,
    "gpt-35-turbo-0613": 4096,
    "gpt-35-turbo": 4096,
    "gpt-4-32k": 32_768,
    "gpt-4-turbo-preview": 128_000,
    "gpt-4-vision": 128_000,
    "gpt-4-0125-preview": 128_000,
    "gpt-4-1106-preview": 128_000,
};

const TOKEN_BUFFER_FOR_SAFETY = 350;
const PROXY_URL = "http://localhost:65433";

const MAX_CHUNK_SIZE = 500; // 512 - buffer for safety (in case of differing tokenizers)

export {
    CONTEXT_LENGTH_FOR_MODEL,
    DEFAULT_ARGS,
    DEFAULT_CONTEXT_LENGTH,
    DEFAULT_MAX_TOKENS,
    MAX_CHUNK_SIZE,
    PROXY_URL,
    TOKEN_BUFFER_FOR_SAFETY,
};