TarekS93 commited on
Commit
f706523
·
unverified ·
1 Parent(s): 34ead80

Update constants.ts

Browse files
Files changed (1) hide show
  1. 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: any) => ({
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',