From ba43f3032a1a62678cb20be921f2fae27e1db1a7 Mon Sep 17 00:00:00 2001 From: cbh778899 Date: Tue, 15 Oct 2024 10:41:26 +1100 Subject: [PATCH] change import path Signed-off-by: cbh778899 --- src/utils/start_loader.js | 2 +- src/utils/workers/index.js | 10 +++------- 2 files changed, 4 insertions(+), 8 deletions(-) diff --git a/src/utils/start_loader.js b/src/utils/start_loader.js index 974fb7a..6bc3f1a 100644 --- a/src/utils/start_loader.js +++ b/src/utils/start_loader.js @@ -1,6 +1,6 @@ import { getPlatformSettings } from "./general_settings"; import { instance } from "./idb"; -import { loadModel } from "./workers/worker"; +import { loadModel } from "./workers/wllama-worker"; export default async function loader() { localStorage.setItem('not-first-time', '1'); diff --git a/src/utils/workers/index.js b/src/utils/workers/index.js index 5cdeddb..a994429 100644 --- a/src/utils/workers/index.js +++ b/src/utils/workers/index.js @@ -1,5 +1,5 @@ import { getModelSettings, getPlatformSettings } from "../general_settings"; -import { chatCompletions as WllamaCompletions, abortCompletion as WllamaAbort } from "./worker"; +import { chatCompletions as WllamaCompletions, abortCompletion as WllamaAbort, setClient as WllamaSetClient } from "./wllama-worker"; import { chatCompletions as AwsCompletions, abortCompletion as AwsAbort, setClient as AwsSetClient, formator as AwsFormator } from "./aws-worker" import { chatCompletions as OpenaiCompletions, abortCompletion as OpenaiAbort, setClient as OpenAISetClient } from "./openai-worker"; @@ -45,13 +45,9 @@ export function getCompletionFunctions(platform = null) { return { completions: WllamaCompletions, abort: WllamaAbort, - platform: "Wllama" + platform: "Wllama", + initClient: WllamaSetClient } - // default: - // return { - // completions: WllamaCompletions, abort: WllamaAbort, - // platform: "Wllama" - // } } } \ No newline at end of file