change import path

Signed-off-by: cbh778899 <[email protected]>
SkywardAI · Oct 14, 2024 · ba43f30 · ba43f30
1 parent 5e3ffc3
commit ba43f30
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git 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
@@ -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"
-        //     }
     }
 
 }