From 448b21ed967720ef3c8f88159a285ddce9c16adc Mon Sep 17 00:00:00 2001
From: James <james@jan.ai>
Date: Thu, 5 Oct 2023 23:36:25 -0700
Subject: [PATCH] using generic name for search model

Signed-off-by: James <james@jan.ai>
---
 electron/core/plugins/model-management-plugin/index.js  | 6 +++---
 electron/core/plugins/model-management-plugin/module.js | 4 ++--
 web/app/_hooks/useGetDownloadedModels.ts                | 4 ++--
 web/app/_hooks/useGetHuggingFaceModel.ts                | 4 ++--
 web/shared/coreService.ts                               | 2 +-
 5 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/electron/core/plugins/model-management-plugin/index.js b/electron/core/plugins/model-management-plugin/index.js
index 1e5b57eac..702a04374 100644
--- a/electron/core/plugins/model-management-plugin/index.js
+++ b/electron/core/plugins/model-management-plugin/index.js
@@ -38,11 +38,11 @@ const deleteModel = async (path) =>
     }
   });
 
-const searchHfModels = async (params) =>
+const searchModels = async (params) =>
   new Promise(async (resolve) => {
     if (window.electronAPI) {
       window.electronAPI
-        .invokePluginFunc(MODULE_PATH, "searchHfModels", params)
+        .invokePluginFunc(MODULE_PATH, "searchModels", params)
         .then((res) => resolve(res));
     }
   });
@@ -53,5 +53,5 @@ export function init({ register }) {
   register("getAvailableModels", "getAvailableModels", getAvailableModels);
   register("downloadModel", "downloadModel", downloadModel);
   register("deleteModel", "deleteModel", deleteModel);
-  register("searchHfModels", "searchHfModels", searchHfModels);
+  register("searchModels", "searchModels", searchModels);
 }
diff --git a/electron/core/plugins/model-management-plugin/module.js b/electron/core/plugins/model-management-plugin/module.js
index f1378fc8e..ced07973e 100644
--- a/electron/core/plugins/model-management-plugin/module.js
+++ b/electron/core/plugins/model-management-plugin/module.js
@@ -120,7 +120,7 @@ const getNextModels = async (count) => {
   return result;
 };
 
-const searchHfModels = async (params) => {
+const searchModels = async (params) => {
   if (currentSearchOwner === params.search.owner && modelsIterator != null) {
     // paginated search
     console.debug(`Paginated search owner: ${params.search.owner}`);
@@ -173,5 +173,5 @@ function getAvailableModels() {
 module.exports = {
   getDownloadedModels,
   getAvailableModels,
-  searchHfModels,
+  searchModels,
 };
diff --git a/web/app/_hooks/useGetDownloadedModels.ts b/web/app/_hooks/useGetDownloadedModels.ts
index 0060961e4..3784d36d7 100644
--- a/web/app/_hooks/useGetDownloadedModels.ts
+++ b/web/app/_hooks/useGetDownloadedModels.ts
@@ -32,11 +32,11 @@ export async function getModelFiles(): Promise<Product[]> {
   return downloadedModels ?? [];
 }
 
-export async function searchHfModels(
+export async function searchModels(
   params: SearchModelParamHf
 ): Promise<QueryProductResult> {
   const result = await executeSerial(
-    ModelManagementService.SEARCH_HF_MODELS,
+    ModelManagementService.SEARCH_MODELS,
     params
   );
 
diff --git a/web/app/_hooks/useGetHuggingFaceModel.ts b/web/app/_hooks/useGetHuggingFaceModel.ts
index ff29c2063..a1bac1424 100644
--- a/web/app/_hooks/useGetHuggingFaceModel.ts
+++ b/web/app/_hooks/useGetHuggingFaceModel.ts
@@ -1,5 +1,5 @@
 import { useState } from "react";
-import { searchHfModels } from "./useGetDownloadedModels";
+import { searchModels } from "./useGetDownloadedModels";
 import { SearchModelParamHf } from "@/_models/hf/SearchModelParam.hf";
 import { Product } from "@/_models/Product";
 import { useSetAtom } from "jotai";
@@ -22,7 +22,7 @@ export default function useGetHuggingFaceModel() {
       search: { owner },
       limit: 5,
     };
-    const result = await searchHfModels(searchParams);
+    const result = await searchModels(searchParams);
     console.debug("result", JSON.stringify(result));
     if (owner !== currentOwner) {
       setModelList(result.data);
diff --git a/web/shared/coreService.ts b/web/shared/coreService.ts
index ac2f23192..068ca02d2 100644
--- a/web/shared/coreService.ts
+++ b/web/shared/coreService.ts
@@ -38,7 +38,7 @@ export enum ModelManagementService {
   GET_AVAILABLE_MODELS = "getAvailableModels",
   DELETE_MODEL = "deleteModel",
   DOWNLOAD_MODEL = "downloadModel",
-  SEARCH_HF_MODELS = "searchHfModels",
+  SEARCH_MODELS = "searchModels",
 }
 
 export enum PreferenceService {