@@ -13,7 +13,6 @@ import {t} from "../../../lang";
1313import { useUserStore } from "../../../store/modules/user" ;
1414import { watch } from "vue" ;
1515import { AppConfig } from "../../../config" ;
16- import { AbstractModelProvider } from "../provider/driver/base" ;
1716
1817const userStore = useUserStore ( ) ;
1918
@@ -31,7 +30,7 @@ const mapModelError = (e: any, provider: Provider) => {
3130 if ( provider . id === "buildIn" ) {
3231 const msg = e + "" ;
3332 const showCharge = ( ) => {
34- window . $mapi . user
33+ $mapi . user
3534 . open ( {
3635 readyParam : {
3736 page : "ChargeLmApi" ,
@@ -100,7 +99,8 @@ export const modelStore = defineStore("model", {
10099 } ,
101100 } ) ;
102101 }
103- const storageData = await window . $mapi . storage . read ( "models" ) ;
102+ let buildInProviderData : any = null ;
103+ const storageData = await $mapi . storage . read ( "models" ) ;
104104 if ( storageData ) {
105105 if ( storageData . userProviders ) {
106106 storageData . userProviders . forEach ( provider => {
@@ -126,6 +126,7 @@ export const modelStore = defineStore("model", {
126126 } ) ;
127127 }
128128 if ( storageData . providerData ) {
129+ buildInProviderData = storageData . providerData [ "buildIn" ] || null ;
129130 for ( const providerId in storageData . providerData ) {
130131 const provider = results . find ( p => p . id === providerId ) ;
131132 if ( provider ) {
@@ -156,9 +157,9 @@ export const modelStore = defineStore("model", {
156157 }
157158 }
158159 this . providers = results ;
159- await this . refreshBuildIn ( ) ;
160+ await this . refreshBuildIn ( buildInProviderData ) ;
160161 } ,
161- async refreshBuildIn ( ) {
162+ async refreshBuildIn ( buildInProviderData ?: any ) {
162163 if ( userStore . data && userStore . data . lmApi && userStore . data . lmApi . models ) {
163164 const lmApi = userStore . data . lmApi ;
164165 const buildInProvider = this . providers . find ( p => p . id === "buildIn" ) ;
@@ -176,6 +177,11 @@ export const modelStore = defineStore("model", {
176177 } ) ;
177178 }
178179 // console.log("model.init.buildIn", JSON.stringify({lmApi}, null, 2));
180+ let enabled = true ;
181+ if ( buildInProviderData && 'enabled' in buildInProviderData ) {
182+ enabled = buildInProviderData . enabled ;
183+ }
184+ console . log ( 'model.init.buildIn' , { enabled, buildInProviderData} ) ;
179185 this . providers . unshift ( {
180186 id : "buildIn" ,
181187 type : "openai" ,
@@ -192,7 +198,7 @@ export const modelStore = defineStore("model", {
192198 apiKey : lmApi . apiKey ,
193199 apiHost : "" ,
194200 models : models ,
195- enabled : true ,
201+ enabled : enabled ,
196202 } ,
197203 } ) ;
198204 } else {
@@ -406,7 +412,7 @@ export const modelStore = defineStore("model", {
406412 }
407413 } ) ;
408414 const userProviders = model . providers . filter ( provider => ! provider . isSystem ) ;
409- await window . $mapi . storage . write ( "models" , ObjectUtil . clone ( { providerData, userProviders} ) ) ;
415+ await $mapi . storage . write ( "models" , ObjectUtil . clone ( { providerData, userProviders} ) ) ;
410416 } , 200 ) ,
411417 } ,
412418} ) ;
0 commit comments