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', triton_trtllm = 'triton_trtllm',
nitro_tensorrt_llm = 'nitro-tensorrt-llm', nitro_tensorrt_llm = 'nitro-tensorrt-llm',
cohere = 'cohere', cohere = 'cohere',
nvdia = 'nvidia', nvidia = 'nvidia',
cortex_llamacpp = 'cortex.llamacpp', cortex_llamacpp = 'cortex.llamacpp',
cortex_onnx = 'cortex.onnx', cortex_onnx = 'cortex.onnx',
cortex_tensorrtllm = 'cortex.tensorrt-llm', 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) => { export const getLogoEngine = (engine: InferenceEngine) => {
switch (engine) { switch (engine) {
case 'anthropic': case InferenceEngine.anthropic:
return 'images/ModelProvider/anthropic.svg' return 'images/ModelProvider/anthropic.svg'
case 'nitro': case InferenceEngine.nitro:
return 'images/ModelProvider/nitro.svg' return 'images/ModelProvider/nitro.svg'
case 'cortex.llamacpp': case InferenceEngine.cortex_llamacpp':
case 'cortex.onnx': case InferenceEngine.cortex_onnx:
case 'cortex.tensorrtllm': case InferenceEngine.cortex_tensorrtllm:
return 'images/ModelProvider/cortex.svg' return 'images/ModelProvider/cortex.svg'
case 'mistral': case InferenceEngine.mistral:
return 'images/ModelProvider/mistral.svg' return 'images/ModelProvider/mistral.svg'
case 'martian': case InferenceEngine.martian:
return 'images/ModelProvider/martian.svg' return 'images/ModelProvider/martian.svg'
case 'openrouter': case InferenceEngine.openrouter:
return 'images/ModelProvider/openrouter.svg' return 'images/ModelProvider/openrouter.svg'
case 'openai': case InferenceEngine.openai:
return 'images/ModelProvider/openai.svg' return 'images/ModelProvider/openai.svg'
case 'groq': case InferenceEngine.groq:
return 'images/ModelProvider/groq.svg' return 'images/ModelProvider/groq.svg'
case 'triton_trtllm': case InferenceEngine.triton_trtllm:
return 'images/ModelProvider/triton_trtllm.svg' return 'images/ModelProvider/triton_trtllm.svg'
case 'cohere': case InferenceEngine.cohere:
return 'images/ModelProvider/cohere.svg' return 'images/ModelProvider/cohere.svg'
case 'nvidia': case InferenceEngine.nvidia:
return 'images/ModelProvider/nvidia.svg' return 'images/ModelProvider/nvidia.svg'
default: default:
return undefined return undefined
@ -41,17 +41,17 @@ export const localEngines = [
export const getTitleByEngine = (engine: InferenceEngine) => { export const getTitleByEngine = (engine: InferenceEngine) => {
switch (engine) { switch (engine) {
case 'nitro': case InferenceEngine.nitro:
return 'Llama.cpp (Nitro)' return 'Llama.cpp (Nitro)'
case 'cortex.llamacpp': case InferenceEngine.cortex_llamacpp:
return 'Llama.cpp (Cortex)' return 'Llama.cpp (Cortex)'
case 'cortex.onnx': case InferenceEngine.cortex_onnx:
return 'Onnx (Cortex)' return 'Onnx (Cortex)'
case 'cortex.tensorrtllm': case InferenceEngine.cortex_tensorrtllm:
return 'TensorRT-LLM (Cortex)' return 'TensorRT-LLM (Cortex)'
case 'openai': case InferenceEngine.openai:
return 'OpenAI' return 'OpenAI'
case 'openrouter': case InferenceEngine.openrouter:
return 'OpenRouter' return 'OpenRouter'
default: default:
return engine.charAt(0).toUpperCase() + engine.slice(1) return engine.charAt(0).toUpperCase() + engine.slice(1)