fix(CE): fixed build error

Multiwoven · Nov 19, 2024 · 83bb7b7 · 83bb7b7
1 parent 95ec55e
commit 83bb7b7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ui/src/services/models.ts b/ui/src/services/models.ts
@@ -58,8 +58,10 @@ export const getModelPreview = async (query: string, connector_id: string): Prom
   return apiRequest(url, { query: query });
 };
 
-export const getAllModels = async ({ type = 'data' }: GetAllModelsProps): Promise<APIData> =>
-  multiwovenFetch<null, APIData>({
+export const getAllModels = async ({
+  type = 'data',
+}: GetAllModelsProps): Promise<ApiResponse<GetAllModelsResponse[]>> =>
+  multiwovenFetch<null, ApiResponse<GetAllModelsResponse[]>>({
     method: 'get',
     url: type ? `/models?query_type=${type}` : '/models',
   });