diff --git a/app/utils/constants.ts b/app/utils/constants.ts index 4908735..4132cda 100644 --- a/app/utils/constants.ts +++ b/app/utils/constants.ts @@ -1,4 +1,4 @@ -import type { ModelInfo } from './types'; +import type { ModelInfo, OllamaApiResponse, OllamaModel } from './types'; export const WORK_DIR_NAME = 'project'; export const WORK_DIR = `/home/${WORK_DIR_NAME}`; @@ -48,9 +48,9 @@ export let MODEL_LIST: ModelInfo[] = [...staticModels]; async function getOllamaModels(): Promise { try { const response = await fetch(`http://localhost:11434/api/tags`); - const data = await response.json(); + const data = await response.json() as OllamaApiResponse; - return data.models.map((model: any) => ({ + return data.models.map((model: OllamaModel) => ({ name: model.name, label: `${model.name} (${model.details.parameter_size})`, provider: 'Ollama', diff --git a/app/utils/types.ts b/app/utils/types.ts index 7ace4e6..5fcd2de 100644 --- a/app/utils/types.ts +++ b/app/utils/types.ts @@ -8,7 +8,7 @@ interface OllamaModelDetails { quantization_level: string; } -interface OllamaModel { +export interface OllamaModel { name: string; model: string; modified_at: string;