55 ServerTools ,
66} from '@isdk/ai-tool'
77import { llm , LLMProvider } from '@isdk/ai-tool-llm'
8+ import { LocalProviderName , LocalProvider } from '@isdk/ai-tool-llm-local'
89import { LlamaCppProviderName , llamaCpp } from '@isdk/ai-tool-llm-llamacpp'
910import { openai } from '@isdk/ai-tool-llm-openai'
1011import { AIPromptsFunc , AIPromptsName } from '@isdk/ai-tool-prompt'
@@ -19,14 +20,17 @@ export function registerProvider(provider: LLMProvider) {
1920 providers [ provider . name ! ] = provider
2021}
2122
22- function initRegisteredProviders ( ) {
23+ function initRegisteredProviders ( rootDir : string ) {
24+ const local = new LocalProvider ( LocalProviderName , { rootDir} )
25+ registerProvider ( local )
26+ registerProvider ( llamaCpp )
27+ registerProvider ( openai )
28+
2329 for ( const provider of Object . values ( providers ) ) {
2430 provider . register ( )
2531 }
2632}
2733
28- registerProvider ( llamaCpp )
29- registerProvider ( openai )
3034
3135export async function initTools ( this : Hook . Context , userConfig : any , _config : Config ) {
3236 try {
@@ -37,7 +41,7 @@ export async function initTools(this: Hook.Context, userConfig: any, _config: Co
3741 ServerTools . register ( llm )
3842
3943 // llamaCpp.register()
40- initRegisteredProviders ( )
44+ initRegisteredProviders ( userConfig . brainDir )
4145
4246 let currentProviderName = userConfig . provider || LlamaCppProviderName
4347 const providerUriParts = currentProviderName . split ( '://' )
0 commit comments