add chat avatar icon for different roles

server/chat_handler.go

@@ -27,6 +27,7 @@ func (s *Server) ChatHandle(c *gin.Context) {
 	role := c.Query("role")
 	logger.Infof("New websocket connected, IP: %s", c.Request.RemoteAddr)
 	client := NewWsClient(ws)
+	// TODO: 这里需要先判断一下角色是否存在，并且角色是被启用的
 	// 发送打招呼信息
 	replyMessage(types.WsMessage{Type: types.WsStart}, client)
 	replyMessage(types.WsMessage{Type: types.WsMiddle, Content: s.Config.ChatRoles[role].HelloMsg}, client)
@@ -65,7 +66,6 @@ func (s *Server) sendMessage(sessionId string, role string, text string, ws Clie
 	} else {
 		context = s.Config.ChatRoles[role].Context
 	}
-	logger.Info(context)
 	r.Messages = append(context, types.Message{
 		Role:    "user",
 		Content: text,

server/config_handler.go

@@ -151,14 +151,19 @@ func (s *Server) ListApiKeys(c *gin.Context) {
 }
 
 func (s *Server) GetChatRoles(c *gin.Context) {
-	var roles = make(map[string]interface{})
-	for k, v := range s.Config.ChatRoles {
-		roles[k] = struct {
-			Key  string `json:"key"`
-			Name string `json:"name"`
-		}{
-			Key:  v.Key,
-			Name: v.Name,
+	var rolesOrder = []string{"gpt", "programmer", "teacher", "artist", "philosopher", "lu-xun", "english_trainer", "seller"}
+	var roles = make([]interface{}, 0)
+	for _, k := range rolesOrder {
+		if v, ok := s.Config.ChatRoles[k]; ok {
+			roles = append(roles, struct {
+				Key  string `json:"key"`
+				Name string `json:"name"`
+				Icon string `json:"icon"`
+			}{
+				Key:  v.Key,
+				Name: v.Name,
+				Icon: v.Icon,
+			})
 		}
 	}
 	c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: roles})