Skip to content

Commit 227255f

Browse files
committed
fix: should use register path to register ai pack
1 parent a9378d2 commit 227255f

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

src/lib/init-tools.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import {
55
ServerTools,
66
} from '@isdk/ai-tool'
77
import { llm, LLMProvider } from '@isdk/ai-tool-llm'
8+
// @ts-ignore
89
import { LocalProviderName, LocalProvider } from '@isdk/ai-tool-llm-local'
910
import { llamaCpp } from '@isdk/ai-tool-llm-llamacpp'
1011
import { openai } from '@isdk/ai-tool-llm-openai'

src/lib/run-script.ts

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,15 +26,16 @@ import {
2626
EventEmitter,
2727
} from '@isdk/ai-tool'
2828
import { llm } from '@isdk/ai-tool-llm'
29+
// @ts-ignore
2930
import { LocalProviderName, LocalProviderProgressEventName } from '@isdk/ai-tool-llm-local'
3031
// @ts-ignore
3132
import { AIScriptServer, LogLevel, LogLevelMap } from '@isdk/ai-tool-agent'
3233
import { detectTextLanguage as detectLang, detectTextLangEx, getLanguageFromIso6391 } from '@isdk/detect-text-language'
3334
import { prompt, setHistoryStore, HistoryStore } from './prompt.js'
3435

35-
import '@isdk/ai-pack-core'
36-
import '@isdk/ai-pack-pro'
37-
import '@isdk/ai-pack-tool'
36+
import '@isdk/ai-pack-core/register'
37+
import '@isdk/ai-pack-pro/register'
38+
import '@isdk/ai-pack-tool/register'
3839
// import { initTools } from './init-tools.js'
3940

4041
const __filename = fileURLToPath(import.meta.url)
@@ -297,7 +298,7 @@ export async function runScript(filename: string, options: IRunScriptOptions) {
297298

298299
let quit = false
299300
const runtime = options.runtime = await script.getRuntime(false) as AIScriptEx
300-
const currentProvider = llm.getCurrentProvider()
301+
const currentProvider = llm.getCurrentProvider()!
301302
if (currentProvider?.name === LocalProviderName && !currentProvider.defaultModelName && !runtime.parameters?.model) {
302303
const default_model_filepath = path.join(options.brainDir, 'DEFAULT_MODEL_NAME.txt');
303304
let defaultModelName: string | undefined

0 commit comments

Comments
 (0)