TarekS93
commited on
Update constants.ts
Browse files- app/utils/constants.ts +3 -3
app/utils/constants.ts
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
import type { ModelInfo } from './types';
|
2 |
|
3 |
export const WORK_DIR_NAME = 'project';
|
4 |
export const WORK_DIR = `/home/${WORK_DIR_NAME}`;
|
@@ -37,9 +37,9 @@ export let MODEL_LIST: ModelInfo[] = [...staticModels];
|
|
37 |
async function getOllamaModels(): Promise<ModelInfo[]> {
|
38 |
try {
|
39 |
const response = await fetch(`http://localhost:11434/api/tags`);
|
40 |
-
const data = await response.json();
|
41 |
|
42 |
-
return data.models.map((model:
|
43 |
name: model.name,
|
44 |
label: `${model.name} (${model.details.parameter_size})`,
|
45 |
provider: 'Ollama',
|
|
|
1 |
+
import type { ModelInfo, OllamaApiResponse, OllamaModel } from './types';
|
2 |
|
3 |
export const WORK_DIR_NAME = 'project';
|
4 |
export const WORK_DIR = `/home/${WORK_DIR_NAME}`;
|
|
|
37 |
async function getOllamaModels(): Promise<ModelInfo[]> {
|
38 |
try {
|
39 |
const response = await fetch(`http://localhost:11434/api/tags`);
|
40 |
+
const data = await response.json() as OllamaApiResponse;
|
41 |
|
42 |
+
return data.models.map((model: OllamaModel) => ({
|
43 |
name: model.name,
|
44 |
label: `${model.name} (${model.details.parameter_size})`,
|
45 |
provider: 'Ollama',
|