cherry-studio/src/main/reranker/RerankerFactory.ts

import { KnowledgeBaseParams } from '@types'

import BaseReranker from './BaseReranker'
import DashscopeReranker from './DashscopeReranker'
import DefaultReranker from './DefaultReranker'
import JinaReranker from './JinaReranker'
import SiliconFlowReranker from './SiliconFlowReranker'
import VoyageReranker from './VoyageReranker'

export default class RerankerFactory {
  static create(base: KnowledgeBaseParams): BaseReranker {
    if (base.rerankModelProvider === 'silicon') {
      return new SiliconFlowReranker(base)
    } else if (base.rerankModelProvider === 'jina') {
      return new JinaReranker(base)
    } else if (base.rerankModelProvider === 'voyageai') {
      return new VoyageReranker(base)
    } else if (base.rerankModelProvider === 'dashscope') {
      return new DashscopeReranker(base)
    }
    return new DefaultReranker(base)
  }
}