Merge pull request #218 from OpenTreeHole/ai_summary

feat: update trace id

Author: Crnaneo

apis/hole/apis.go

@@ -1049,7 +1049,7 @@ func GenerateSummary(c *fiber.Ctx) error {
 					if cachedData.Data.HoleID != id && cachedData.Data.HoleID != 0 {
 						log.Error().
 							Int("hole_id", id).
-							Str("trace_id", cachedData.TraceID).
+							Str("trace_id", cachedData.Data.TraceID).
 							Int("ai_server_return_id", cachedData.Data.HoleID).
 							Msg("AISummary: hole id error")
 						cachedData.Data.HoleID = id

apis/hole/schemas.go

@@ -133,10 +133,10 @@ func (body ModifyModel) DoNothing() bool {
 
 type Summary struct {
 	Code    int    `json:"code"`
-	TraceID string `json:"trace_id"`
 	Message string `json:"message"`
 	Data    struct {
 		HoleID   int    `json:"hole_id"`
+		TraceID  string `json:"trace_id"`
 		Summary  string `json:"summary"`
 		Branches []struct {
 			ID                   int    `json:"id"`

models/hole.go

@@ -290,7 +290,11 @@ func (holes Holes) Preprocess(c *fiber.Ctx) error {
 		floors = append(floors, hole.Floors...)
 		// set ai_summary_available
 		uid, _ := common.GetUserID(c)
+
+		// for users in whitelist or whitelist is empty, AISummaryAvailable is true,
 		hole.AISummaryAvailable = config.Config.WhiteListUserIds == nil || slices.Contains(config.Config.WhiteListUserIds, uid)
+
+		hole.AISummaryAvailable = !(hole.Locked || hole.Hidden || hole.Frozen)
 		for _, tag := range hole.Tags {
 			if len(tag.Name) > 0 && tag.Name[0] == '*' {
 				hole.AISummaryAvailable = false
@@ -313,7 +317,6 @@ func (holes Holes) Preprocess(c *fiber.Ctx) error {
 					return err
 				}
 
-				// var discard any
 				hole.AISummaryAvailable = hole.Reply > config.Config.SummaryFloorLimit || contentSum >= config.Config.SummaryContentLimit // || utils.GetCache("AISummary"+strconv.Itoa(hole.ID), &discard)
 
 				if hole.AISummaryAvailable {