Merge pull request stackblitz-labs#44 from TarekS93/main

Enhance Ollama Model Integration and Type Definitions
arsenikstiger · Oct 24, 2024 · 65cf12d · 65cf12d
2 parents 5d70255 + f706523
commit 65cf12d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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<ModelInfo[]> {
   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
@@ -8,7 +8,7 @@ interface OllamaModelDetails {
   quantization_level: string;
 }
 
-interface OllamaModel {
+export interface OllamaModel {
   name: string;
   model: string;
   modified_at: string;