import { RerankType } from '.';

/**
 * 配置模板
 */
export const ConfigRerank: { [key in RerankType]?: any } = {
  // cohere
  cohere: {
    comm: {
      apiKey: 'API密钥',
    },
    options: [
      {
        field: 'model',
        title: 'Cohere',
        select: [
          'rerank-english-v3.0',
          'rerank-multilingual-v3.0',
          'rerank-english-v2.0',
          'rerank-multilingual-v2.0',
        ],
        default: 'rerank-multilingual-v3.0',
      },
    ],
  },
};
