Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

using generic name for search model

Browse Source 
Signed-off-by: James <james@jan.ai>
This commit is contained in:
James 2023-10-05 23:36:25 -07:00 committed by Louis
parent 9a79c3effa
commit 448b21ed96
5 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
electron/core/plugins/model-management-plugin/index.js
View File

@ -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);
}

4
electron/core/plugins/model-management-plugin/module.js
View File

@ -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,
};

4
web/app/_hooks/useGetDownloadedModels.ts
View File

@ -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
);

4
web/app/_hooks/useGetHuggingFaceModel.ts
View File

@ -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);

2
web/shared/coreService.ts
View File

@ -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 {