feat(chat): agnet add hooked mark

Author: fumiama

chat/agent.go

@@ -8,7 +8,6 @@ import (
 	"io"
 	"reflect"
 	"strconv"
-	"sync"
 	"time"
 
 	"github.com/RomiChan/syncx"
@@ -34,8 +33,7 @@ var AgentChar []byte
 var IsAgentCharReady = false
 
 var (
-	inActionMu = sync.Mutex{}
-	ags        = syncx.Map[int64, *goba.Agent]{}
+	ags = syncx.Map[int64, *goba.Agent]{}
 )
 
 type charcfg struct {
@@ -92,19 +90,15 @@ var checkgids = map[string]struct{}{
 }
 
 // CallAgent and check group API permission
-func CallAgent(ag *goba.Agent, issudo bool, iter int, api deepinfra.API, p model.Protocol, grp int64, role goba.PermRole) (
-	[]zero.APIRequest, func(),
-) {
-	inActionMu.Lock()
+func CallAgent(ag *goba.Agent, issudo bool, iter int, api deepinfra.API, p model.Protocol, grp int64, role goba.PermRole) []zero.APIRequest {
 	reqs, err := ag.GetAction(api, p, grp, role, iter, false)
 	if err != nil {
-		inActionMu.Unlock()
 		if !errors.Is(err, io.EOF) {
 			logrus.Warnln("[chat] agent err:", err, reqs)
 		} else {
 			logrus.Infoln("[chat] agent end action")
 		}
-		return nil, nil
+		return nil
 	}
 	logrus.Infoln("[chat] agent do:", reqs)
 	checkedreqs := make([]zero.APIRequest, 0, len(reqs))
@@ -153,7 +147,7 @@ func CallAgent(ag *goba.Agent, issudo bool, iter int, api deepinfra.API, p model
 		}
 		checkedreqs = append(checkedreqs, req)
 	}
-	return checkedreqs, inActionMu.Unlock
+	return checkedreqs
 }
 
 func togobaev(ev *zero.Event) *goba.Event {
@@ -220,11 +214,6 @@ func logev(ctx *zero.Ctx) {
 			if countParamsLength(req.Params) > 256 { // skip too long req&resp
 				return
 			}
-			if inActionMu.TryLock() {
-				defer inActionMu.Unlock()
-			} else {
-				return
-			}
 			gid := ctx.Event.GroupID
 			if gid == 0 {
 				gid = -ctx.Event.UserID
@@ -242,11 +231,7 @@ func logev(ctx *zero.Ctx) {
 			})
 		}),
 	)
-	if inActionMu.TryLock() {
-		defer inActionMu.Unlock()
-	} else {
-		return
-	}
+	ctx.State[zero.StateKeyPrefixKeep+"_chat_ag_hooked__"] = struct{}{}
 	gid := ctx.Event.GroupID
 	if gid == 0 {
 		gid = -ctx.Event.UserID

go.mod

@@ -25,7 +25,7 @@ require (
 	github.com/swaggo/gin-swagger v1.6.1
 	github.com/swaggo/swag v1.16.6
 	github.com/tidwall/gjson v1.18.0
-	github.com/wdvxdr1123/ZeroBot v1.8.3-0.20260103120253-8a8f1347f983
+	github.com/wdvxdr1123/ZeroBot v1.8.3-0.20260108153355-cbf2ab0dd527
 	gopkg.in/yaml.v3 v3.0.1
 )
 

go.sum

@@ -189,8 +189,8 @@ github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS
 github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
 github.com/ugorji/go/codec v1.3.1 h1:waO7eEiFDwidsBN6agj1vJQ4AG7lh2yqXyOXqhgQuyY=
 github.com/ugorji/go/codec v1.3.1/go.mod h1:pRBVtBSKl77K30Bv8R2P+cLSGaTtex6fsA2Wjqmfxj4=
-github.com/wdvxdr1123/ZeroBot v1.8.3-0.20260103120253-8a8f1347f983 h1:hb2FUDooAf3u32wCTgJcBBuGPZF1sjgj8NfJaSy529s=
-github.com/wdvxdr1123/ZeroBot v1.8.3-0.20260103120253-8a8f1347f983/go.mod h1:trueIIVRywKJa3ov4QphzVvzYzgCNrlXdf9JvPJOFW8=
+github.com/wdvxdr1123/ZeroBot v1.8.3-0.20260108153355-cbf2ab0dd527 h1:QQAAftj7MvP+Ebxfs1MV5d6fswYi6YJ2YFkoicFpcEY=
+github.com/wdvxdr1123/ZeroBot v1.8.3-0.20260108153355-cbf2ab0dd527/go.mod h1:trueIIVRywKJa3ov4QphzVvzYzgCNrlXdf9JvPJOFW8=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 go.uber.org/mock v0.6.0 h1:hyF9dfmbgIX5EfOdasqLsWD6xqpNZlXblLB/Dbnwv3Y=
 go.uber.org/mock v0.6.0/go.mod h1:KiVJ4BqZJaMj4svdfmHM0AUx4NJYO8ZNpPnZn1Z+BBU=