fix: Enforce openai inference to load settings from openai.json onLoad and onModelInit

This commit is contained in:
hiro 2023-12-05 18:10:27 +07:00
parent 5f3cf2bcce
commit e1190ec41c

View File

@ -57,7 +57,7 @@ export default class JanInferenceOpenAIExtension implements InferenceExtension {
*/ */
onLoad(): void { onLoad(): void {
fs.mkdir(JanInferenceOpenAIExtension._homeDir) fs.mkdir(JanInferenceOpenAIExtension._homeDir)
this.writeDefaultEngineSettings() JanInferenceOpenAIExtension.writeDefaultEngineSettings()
// Events subscription // Events subscription
events.on(EventName.OnMessageSent, (data) => events.on(EventName.OnMessageSent, (data) =>
@ -90,7 +90,7 @@ export default class JanInferenceOpenAIExtension implements InferenceExtension {
return return
} }
private async writeDefaultEngineSettings() { static async writeDefaultEngineSettings() {
try { try {
const engine_json = join(JanInferenceOpenAIExtension._homeDir, JanInferenceOpenAIExtension._engineMetadataFileName) const engine_json = join(JanInferenceOpenAIExtension._homeDir, JanInferenceOpenAIExtension._engineMetadataFileName)
if (await fs.checkFileExists(engine_json)) { if (await fs.checkFileExists(engine_json)) {
@ -156,6 +156,7 @@ export default class JanInferenceOpenAIExtension implements InferenceExtension {
if (model.engine !== 'openai') { return } if (model.engine !== 'openai') { return }
else { else {
JanInferenceOpenAIExtension._currentModel = model JanInferenceOpenAIExtension._currentModel = model
JanInferenceOpenAIExtension.writeDefaultEngineSettings()
// Todo: Check model list with API key // Todo: Check model list with API key
events.emit(EventName.OnModelReady, model) events.emit(EventName.OnModelReady, model)
// events.emit(EventName.OnModelFail, model) // events.emit(EventName.OnModelFail, model)