From 098492d1c58ce52242bfa27c03d9ed0c580d034c Mon Sep 17 00:00:00 2001
From: Faisal Amir <urmauur@gmail.com>
Date: Fri, 16 Aug 2024 15:22:36 +0700
Subject: [PATCH] chore: update utils model entity

---
 core/src/types/model/modelEntity.ts |  2 +-
 web/utils/modelEngine.ts            | 38 ++++++++++++++---------------
 2 files changed, 20 insertions(+), 20 deletions(-)

diff --git a/core/src/types/model/modelEntity.ts b/core/src/types/model/modelEntity.ts
index 1e3ecbb9c..ef6d59316 100644
--- a/core/src/types/model/modelEntity.ts
+++ b/core/src/types/model/modelEntity.ts
@@ -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',
diff --git a/web/utils/modelEngine.ts b/web/utils/modelEngine.ts
index 9e7ea0146..610e3c128 100644
--- a/web/utils/modelEngine.ts
+++ b/web/utils/modelEngine.ts
@@ -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)