fix: enum inference type (#1198)

Signed-off-by: axel7083 <[email protected]>
containers · Jun 11, 2024 · b384a3d · b384a3d
1 parent 5ddb5c5
commit b384a3d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/packages/backend/src/utils/inferenceUtils.spec.ts b/packages/backend/src/utils/inferenceUtils.spec.ts
@@ -91,8 +91,8 @@ describe('parseInferenceType', () => {
     expect(parseInferenceType(undefined)).toBe(InferenceType.NONE);
   });
 
-  test('llamacpp should return the proper InferenceType.LLAMA_CPP', () => {
-    expect(parseInferenceType('llamacpp')).toBe(InferenceType.LLAMA_CPP);
+  test('llama-cpp should return the proper InferenceType.LLAMA_CPP', () => {
+    expect(parseInferenceType('llama-cpp')).toBe(InferenceType.LLAMA_CPP);
   });
 });
 
@@ -115,7 +115,7 @@ describe('getInferenceType', () => {
     expect(
       getInferenceType([
         {
-          backend: 'llamacpp',
+          backend: 'llama-cpp',
         } as unknown as ModelInfo,
       ]),
     ).toBe(InferenceType.LLAMA_CPP);
@@ -125,10 +125,10 @@ describe('getInferenceType', () => {
     expect(
       getInferenceType([
         {
-          backend: 'llamacpp',
+          backend: 'llama-cpp',
         },
         {
-          backend: 'llamacpp',
+          backend: 'llama-cpp',
         },
       ] as unknown as ModelInfo[]),
     ).toBe(InferenceType.LLAMA_CPP);
@@ -138,10 +138,10 @@ describe('getInferenceType', () => {
     expect(
       getInferenceType([
         {
-          backend: 'llamacpp',
+          backend: 'llama-cpp',
         },
         {
-          backend: 'whispercpp',
+          backend: 'whisper-cpp',
         },
       ] as unknown as ModelInfo[]),
     ).toBe(InferenceType.NONE);

diff --git a/packages/shared/src/models/IInference.ts b/packages/shared/src/models/IInference.ts
@@ -18,8 +18,8 @@
 import type { ModelInfo } from './IModelInfo';
 
 export enum InferenceType {
-  LLAMA_CPP = 'llamacpp',
-  WHISPER_CPP = 'whispercpp',
+  LLAMA_CPP = 'llama-cpp',
+  WHISPER_CPP = 'whisper-cpp',
   NONE = 'none',
 }