From a0c8c47a998cb12fd6bd7705fc5de6f6d48c25f1 Mon Sep 17 00:00:00 2001
From: hhq <hhq1@meitu.com>
Date: Wed, 26 Feb 2025 18:53:28 +0800
Subject: [PATCH] fix: Setting a custom model does not take effect

---
 src/renderer/packages/models/siliconflow.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/renderer/packages/models/siliconflow.ts b/src/renderer/packages/models/siliconflow.ts
index 6422d57b5..54ae88f44 100644
--- a/src/renderer/packages/models/siliconflow.ts
+++ b/src/renderer/packages/models/siliconflow.ts
@@ -10,6 +10,7 @@ interface Options {
     siliconflowCustomModel?: string
     temperature: number
     topP: number
+    openaiCustomModel?: string
 }
 
 export default class SiliconFlow extends Base {
@@ -49,7 +50,7 @@ export default class SiliconFlow extends Base {
 
         const model =
             this.options.siliconCloudModel === 'custom-model'
-                ? this.options.siliconflowCustomModel || ''
+                ? this.options.siliconflowCustomModel || this.options.openaiCustomModel || ''
                 : this.options.siliconCloudModel
         messages = injectModelSystemPrompt(model, messages)