From b794df9b6352d8bf433792960c81f12173f20554 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AF=BB=E6=AC=A2?= <lxh@cxh.cn>
Date: Tue, 16 Jul 2024 15:39:45 +0800
Subject: [PATCH] =?UTF-8?q?:art:=20=E7=AE=A1=E7=90=86=E6=8C=87=E5=AE=9A?=
 =?UTF-8?q?=E7=BE=A4=E7=BB=84=E6=96=B0=E5=A2=9E=E6=A8=A1=E5=9E=8B=E5=8F=AF?=
 =?UTF-8?q?=E9=80=89=E8=8C=83=E5=9B=B4=E9=99=90=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config.yaml            | 3 +++
 config/ai.go           | 5 +++--
 views/manager-one.html | 2 +-
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/config.yaml b/config.yaml
index 41e0fab..2c02051 100644
--- a/config.yaml
+++ b/config.yaml
@@ -102,6 +102,7 @@ ai:
   models:
     - name: ChatGPT-4
       model: gpt-4
+      canManager: false
     - name: 讯飞星火v3.1
       model: SparkDesk-v3.1
     - name: 讯飞星火v3.5
@@ -116,8 +117,10 @@ ai:
       model: StepChat
     - name: 豆包Lite-4k
       model: Doubao-lite-4k
+      canManager: true
     - name: 豆包Pro-4k
       model: Doubao-pro-4k
+      canManager: true
 
 # 资源配置
 # map[k]v结构，k 会变成全小写，所以这儿不能用大写字母
diff --git a/config/ai.go b/config/ai.go
index cdc44a3..7d987a2 100644
--- a/config/ai.go
+++ b/config/ai.go
@@ -15,6 +15,7 @@ type ai struct {
 // aiModel
 // @description: AI模型
 type aiModel struct {
-	Name  string `json:"name" yaml:"name"` // 模型名称
-	Model string `json:"dto" yaml:"dto"`   // 模型代码
+	Name       string `json:"name" yaml:"name"`             // 模型名称
+	Model      string `json:"dto" yaml:"dto"`               // 模型代码
+	CanManager bool   `json:"canManager" yaml:"canManager"` // 是否可以在manager页面中选择
 }
diff --git a/views/manager-one.html b/views/manager-one.html
index c44b588..1e841bf 100644
--- a/views/manager-one.html
+++ b/views/manager-one.html
@@ -53,7 +53,7 @@
                   <select class="block w-full rounded-md border-0 py-1.5 pl-3 pr-10 text-gray-900 ring-1 ring-inset ring-gray-300 focus:ring-2 focus:ring-green-600 sm:text-sm sm:leading-6" onchange="aiModelChange(event, {{.info.Wxid}})">
                     {{$useModel := .info.AiModel}}
                     {{ range $.aiModels }}
-                    <option value="{{.Model}}" {{ if eq $useModel .Model}}selected{{ end }}>
+                    <option value="{{.Model}}" {{ if eq $useModel .Model}}selected{{ end }} {{ if not .CanManager}}disabled{{ end }}>
                       {{.Name}}
                     </option>
                     {{ end }}