jan/web/utils/model-engine.ts

import { LocalEngine } from '@janhq/core'

import { ModelHubCategory } from '@/hooks/useModelHub'

export const getTitleByCategory = (category: ModelHubCategory) => {
  if (!category || !category.length) return ''
  switch (category) {
    case 'cortex.llamacpp':
      return 'llama.cpp'
    case 'cortex.onnx':
      return 'Onnx'
    case 'cortex.tensorrt-llm':
      return 'Tensorrt-llm'
    case 'triton_trtllm':
      return 'Triton-trtllm'
    case 'BuiltInModels':
      return 'Built-in Models'
    case 'HuggingFace':
      return 'Hugging Face'
    case 'openai':
      return 'OpenAI'
    case 'openrouter':
      return 'OpenRouter'
    default:
      return category.charAt(0).toUpperCase() + category.slice(1)
  }
}

export const getDescriptionByCategory = (category: ModelHubCategory) => {
  switch (category) {
    case 'BuiltInModels':
      return 'All models used even offline, model performance depends on your device capability.'
    case 'HuggingFace':
      return 'All models used even offline, model performance depends on your device capability.'
    default:
      return ''
  }
}

export const getLogoByCategory = (category: ModelHubCategory) => {
  switch (category) {
    case 'BuiltInModels':
      return 'icons/app_icon.svg'
    case 'HuggingFace':
      return 'icons/ic_hugging_face.svg'
    default:
      return undefined
  }
}

export const getLogoByLocalEngine = (engine: LocalEngine) => {
  switch (engine) {
    case 'cortex.llamacpp':
      return 'icons/llamacpp.svg'

    default:
      return undefined
  }
}