show license info in admin active page, optimize markdown generate prompt

api/core/types/config.go

@@ -119,11 +119,11 @@ type RedisConfig struct {
 const LicenseKey = "Geek-AI-License"
 
 type License struct {
-	Key       string // 许可证书密钥
-	MachineId string // 机器码
-	UserNum   int    // 用户数量
-	ExpiredAt int64  // 过期时间
-	IsActive  bool   // 是否激活
+	Key       string `json:"key"`        // 许可证书密钥
+	MachineId string `json:"machine_id"` // 机器码
+	UserNum   int    `json:"user_num"`   // 用户数量
+	ExpiredAt int64  `json:"expired_at"` // 过期时间
+	IsActive  bool   `json:"is_active"`  // 是否激活
 }
 
 func (c RedisConfig) Url() string {

api/handler/admin/config_handler.go

@@ -115,3 +115,9 @@ func (h *ConfigHandler) Active(c *gin.Context) {
 
 	resp.SUCCESS(c, info.HostID)
 }
+
+// GetLicense 获取 License 信息
+func (h *ConfigHandler) GetLicense(c *gin.Context) {
+	license := h.licenseService.GetLicense()
+	resp.SUCCESS(c, license)
+}

api/handler/markmap_handler.go

@@ -103,7 +103,26 @@ func (h *MarkMapHandler) sendMessage(client *types.WsClient, prompt string, mode
 	}
 
 	messages := make([]interface{}, 0)
-	messages = append(messages, types.Message{Role: "system", Content: "你是一位非常优秀的思维导图助手，你会把用户的所有提问都总结成思维导图，然后以 Markdown 格式输出。不要输出任何解释性的语句。"})
+	messages = append(messages, types.Message{Role: "system", Content: `
+你是一位非常优秀的思维导图助手，你会把用户的所有提问都总结成思维导图，然后以 Markdown 格式输出。markdown 只需要输出一级标题，二级标题，三级标题，四级标题，最多输出四级，除此之外不要输出任何其他 markdown 标记。下面是一个合格的例子：
+# Geek-AI 助手
+
+## 完整的开源系统
+### 前端开源
+### 后端开源
+
+## 支持各种大模型
+### OpenAI 
+### Azure 
+### 文心一言
+### 通义千问
+
+## 集成多种收费方式
+### 支付宝
+### 微信
+
+另外，除此之外不要任何解释性语句。
+`})
 	messages = append(messages, types.Message{Role: "user", Content: prompt})
 	var req = types.ApiRequest{
 		Model:    chatModel.Value,

api/main.go

@@ -291,6 +291,7 @@ func main() {
 			group.POST("config/update", h.Update)
 			group.GET("config/get", h.Get)
 			group.POST("active", h.Active)
+			group.GET("config/get/license", h.GetLicense)
 		}),
 		fx.Invoke(func(s *core.AppServer, h *admin.ManagerHandler) {
 			group := s.Engine.Group("/api/admin/")

api/service/license_service.go

@@ -61,16 +61,17 @@ func (s *LicenseService) ActiveLicense(license string, machineId string) error {
 	}
 
 	if res.Code != types.Success {
-		return fmt.Errorf( "激活失败：%v", res.Message)
+		return fmt.Errorf("激活失败：%v", res.Message)
 	}
 
-	err = s.levelDB.Put(types.LicenseKey, types.License{
+	s.license = types.License{
 		Key:       license,
 		MachineId: machineId,
 		UserNum:   res.Data.UserNum,
 		ExpiredAt: res.Data.ExpiredAt,
-		IsActive: true,
-	})
+		IsActive:  true,
+	}
+	err = s.levelDB.Put(types.LicenseKey, s.license)
 	if err != nil {
 		return fmt.Errorf("保存许可证书失败：%v", err)
 	}