Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: update utils model entity

Browse Source
This commit is contained in:
Faisal Amir 2024-08-16 15:22:36 +07:00
parent 80b23d10b1
commit 098492d1c5
2 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
core/src/types/model/modelEntity.ts
View File

@ -25,7 +25,7 @@ export enum InferenceEngine {
triton_trtllm = 'triton_trtllm',
nitro_tensorrt_llm = 'nitro-tensorrt-llm',
cohere = 'cohere',
nvdia = 'nvidia',
nvidia = 'nvidia',
cortex_llamacpp = 'cortex.llamacpp',
cortex_onnx = 'cortex.onnx',
cortex_tensorrtllm = 'cortex.tensorrt-llm',

38
web/utils/modelEngine.ts
View File

@ -2,29 +2,29 @@ import { InferenceEngine } from '@janhq/core'
export const getLogoEngine = (engine: InferenceEngine) => {
switch (engine) {
case 'anthropic':
case InferenceEngine.anthropic:
return 'images/ModelProvider/anthropic.svg'
case 'nitro':
case InferenceEngine.nitro:
return 'images/ModelProvider/nitro.svg'
case 'cortex.llamacpp':
case 'cortex.onnx':
case 'cortex.tensorrtllm':
case InferenceEngine.cortex_llamacpp':
case InferenceEngine.cortex_onnx:
case InferenceEngine.cortex_tensorrtllm:
return 'images/ModelProvider/cortex.svg'
case 'mistral':
case InferenceEngine.mistral:
return 'images/ModelProvider/mistral.svg'
case 'martian':
case InferenceEngine.martian:
return 'images/ModelProvider/martian.svg'
case 'openrouter':
case InferenceEngine.openrouter:
return 'images/ModelProvider/openrouter.svg'
case 'openai':
case InferenceEngine.openai:
return 'images/ModelProvider/openai.svg'
case 'groq':
case InferenceEngine.groq:
return 'images/ModelProvider/groq.svg'
case 'triton_trtllm':
case InferenceEngine.triton_trtllm:
return 'images/ModelProvider/triton_trtllm.svg'
case 'cohere':
case InferenceEngine.cohere:
return 'images/ModelProvider/cohere.svg'
case 'nvidia':
case InferenceEngine.nvidia:
return 'images/ModelProvider/nvidia.svg'
default:
return undefined
@ -41,17 +41,17 @@ export const localEngines = [
export const getTitleByEngine = (engine: InferenceEngine) => {
switch (engine) {
case 'nitro':
case InferenceEngine.nitro:
return 'Llama.cpp (Nitro)'
case 'cortex.llamacpp':
case InferenceEngine.cortex_llamacpp:
return 'Llama.cpp (Cortex)'
case 'cortex.onnx':
case InferenceEngine.cortex_onnx:
return 'Onnx (Cortex)'
case 'cortex.tensorrtllm':
case InferenceEngine.cortex_tensorrtllm:
return 'TensorRT-LLM (Cortex)'
case 'openai':
case InferenceEngine.openai:
return 'OpenAI'
case 'openrouter':
case InferenceEngine.openrouter:
return 'OpenRouter'
default:
return engine.charAt(0).toUpperCase() + engine.slice(1)