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) => { new Promise(async (resolve) => {
if (window.electronAPI) { if (window.electronAPI) {
window.electronAPI window.electronAPI
.invokePluginFunc(MODULE_PATH, "searchHfModels", params) .invokePluginFunc(MODULE_PATH, "searchModels", params)
.then((res) => resolve(res)); .then((res) => resolve(res));
} }
}); });
@ -53,5 +53,5 @@ export function init({ register }) {
register("getAvailableModels", "getAvailableModels", getAvailableModels); register("getAvailableModels", "getAvailableModels", getAvailableModels);
register("downloadModel", "downloadModel", downloadModel); register("downloadModel", "downloadModel", downloadModel);
register("deleteModel", "deleteModel", deleteModel); 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; return result;
}; };
const searchHfModels = async (params) => { const searchModels = async (params) => {
if (currentSearchOwner === params.search.owner && modelsIterator != null) { if (currentSearchOwner === params.search.owner && modelsIterator != null) {
// paginated search // paginated search
console.debug(`Paginated search owner: ${params.search.owner}`); console.debug(`Paginated search owner: ${params.search.owner}`);
@ -173,5 +173,5 @@ function getAvailableModels() {
module.exports = { module.exports = {
getDownloadedModels, getDownloadedModels,
getAvailableModels, getAvailableModels,
searchHfModels, searchModels,
}; };

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

@ -32,11 +32,11 @@ export async function getModelFiles(): Promise<Product[]> {
return downloadedModels ?? []; return downloadedModels ?? [];
} }
export async function searchHfModels( export async function searchModels(
params: SearchModelParamHf params: SearchModelParamHf
): Promise<QueryProductResult> { ): Promise<QueryProductResult> {
const result = await executeSerial( const result = await executeSerial(
ModelManagementService.SEARCH_HF_MODELS, ModelManagementService.SEARCH_MODELS,
params params
); );

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

@ -1,5 +1,5 @@
import { useState } from "react"; import { useState } from "react";
import { searchHfModels } from "./useGetDownloadedModels"; import { searchModels } from "./useGetDownloadedModels";
import { SearchModelParamHf } from "@/_models/hf/SearchModelParam.hf"; import { SearchModelParamHf } from "@/_models/hf/SearchModelParam.hf";
import { Product } from "@/_models/Product"; import { Product } from "@/_models/Product";
import { useSetAtom } from "jotai"; import { useSetAtom } from "jotai";
@ -22,7 +22,7 @@ export default function useGetHuggingFaceModel() {
search: { owner }, search: { owner },
limit: 5, limit: 5,
}; };
const result = await searchHfModels(searchParams); const result = await searchModels(searchParams);
console.debug("result", JSON.stringify(result)); console.debug("result", JSON.stringify(result));
if (owner !== currentOwner) { if (owner !== currentOwner) {
setModelList(result.data); setModelList(result.data);

2
web/shared/coreService.ts
View File

@ -38,7 +38,7 @@ export enum ModelManagementService {
GET_AVAILABLE_MODELS = "getAvailableModels", GET_AVAILABLE_MODELS = "getAvailableModels",
DELETE_MODEL = "deleteModel", DELETE_MODEL = "deleteModel",
DOWNLOAD_MODEL = "downloadModel", DOWNLOAD_MODEL = "downloadModel",
SEARCH_HF_MODELS = "searchHfModels", SEARCH_MODELS = "searchModels",
} }
export enum PreferenceService { export enum PreferenceService {