diff --git a/tools/ui/src/lib/enums/model.enums.ts b/tools/ui/src/lib/enums/model.enums.ts index b76d4a2494..ea799b6977 100644 --- a/tools/ui/src/lib/enums/model.enums.ts +++ b/tools/ui/src/lib/enums/model.enums.ts @@ -6,7 +6,7 @@ export enum ModelModality { } export enum ModelCapability { - REASONING = 'REASONING', + REASONING = 'reasoning', TOOL_USE = 'tools' } diff --git a/tools/ui/tests/unit/model-id-parser.test.ts b/tools/ui/tests/unit/model-id-parser.test.ts index bf68cdec66..3e2a1c5c6e 100644 --- a/tools/ui/tests/unit/model-id-parser.test.ts +++ b/tools/ui/tests/unit/model-id-parser.test.ts @@ -1,3 +1,4 @@ +import { ModelAuxSidecar, ModelDraftSidecar } from '$lib/enums'; import { ModelsService } from '$lib/services/models.service'; import { describe, expect, it } from 'vitest'; @@ -328,24 +329,24 @@ describe('parseModelId', () => { // sidecar prefix: bare filename or multi-slash path reduces to the filename expect(parseModelId('mtp-Q4_0.gguf')).toMatchObject({ quantization: 'Q4_0', - sidecar: 'mtp' + sidecar: ModelDraftSidecar.MTP }); expect(parseModelId('ggml-org/Model-GGUF/mtp-Q4_0.gguf')).toMatchObject({ quantization: 'Q4_0', - sidecar: 'mtp' + sidecar: ModelDraftSidecar.MTP }); expect(parseModelId('ggml-org/Model-GGUF/mmproj-F16.gguf')).toMatchObject({ quantization: 'F16', - sidecar: 'mmproj' + sidecar: ModelAuxSidecar.MMPROJ }); // embedded-draft suffix: - only strips when preceded by a quant expect(parseModelId('ggml-org/Hy3-IQ1_M-mtp')).toMatchObject({ modelName: 'Hy3', quantization: 'IQ1_M', - sidecar: 'mtp' + sidecar: ModelDraftSidecar.MTP }); // a model literally named MyModel-mtp is not a draft