DBT/clawgo
1
0
Fork 0
mirror of https://github.com/YspCoder/clawgo.git synced 2026-05-18 17:17:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

agent async-loop phase2: add task-id progress notices and completion correlation

Browse Source
This commit is contained in:
DBT
2026-02-28 02:16:04 +00:00
parent 4dcfa92717
commit ca4f78c627
1 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/agent/loop.go
View File

@@ -310,7 +310,7 @@ func (al *AgentLoop) lockSessionRun(sessionKey string) func() {
} }
func (al *AgentLoop) processInbound(ctx context.Context, msg bus.InboundMessage) { func (al *AgentLoop) processInbound(ctx context.Context, msg bus.InboundMessage) {
stopNotice := al.startLongRunNotice(ctx, msg) taskID, noticeCount, stopNotice := al.startLongRunNotice(ctx, msg)
defer stopNotice() defer stopNotice()
response, err := al.processMessage(ctx, msg) response, err := al.processMessage(ctx, msg)
@@ -318,6 +318,10 @@ func (al *AgentLoop) processInbound(ctx context.Context, msg bus.InboundMessage)
response = fmt.Sprintf("Error processing message: %v", err) response = fmt.Sprintf("Error processing message: %v", err)
} }
if noticeCount != nil && *noticeCount > 0 && response != "" {
response = fmt.Sprintf("任务ID %s 已完成。\n\n%s", taskID, response)
}
trigger := al.getTrigger(msg) trigger := al.getTrigger(msg)
suppressed := false suppressed := false
if response != "" { if response != "" {
@@ -333,7 +337,7 @@ func (al *AgentLoop) processInbound(ctx context.Context, msg bus.InboundMessage)
al.audit.Record(trigger, msg.Channel, msg.SessionKey, suppressed, err) al.audit.Record(trigger, msg.Channel, msg.SessionKey, suppressed, err)
} }
func (al *AgentLoop) startLongRunNotice(ctx context.Context, msg bus.InboundMessage) func() { func (al *AgentLoop) startLongRunNotice(ctx context.Context, msg bus.InboundMessage) (string, *int, func()) {
first := 45 * time.Second first := 45 * time.Second
interval := 45 * time.Second interval := 45 * time.Second
if v := os.Getenv("CLAWGO_LONGRUN_NOTICE_SEC"); v != "" { if v := os.Getenv("CLAWGO_LONGRUN_NOTICE_SEC"); v != "" {
@@ -342,11 +346,12 @@ func (al *AgentLoop) startLongRunNotice(ctx context.Context, msg bus.InboundMess
interval = first interval = first
} }
} }
taskID := fmt.Sprintf("%s-%d", shortSessionKey(msg.SessionKey), time.Now().Unix()%100000)
stop := make(chan struct{}) stop := make(chan struct{})
notified := 0
go func() { go func() {
t := time.NewTimer(first) t := time.NewTimer(first)
defer t.Stop() defer t.Stop()
notified := 0
for { for {
select { select {
case <-ctx.Done(): case <-ctx.Done():
@@ -355,7 +360,7 @@ func (al *AgentLoop) startLongRunNotice(ctx context.Context, msg bus.InboundMess
return return
case <-t.C: case <-t.C:
notified++ notified++
text := fmt.Sprintf("任务较复杂，正在持续执行中（第%d次进度通知）…", notified) text := fmt.Sprintf("任务ID %s 正在执行中（第%d次进度通知）…", taskID, notified)
if outbound, ok := al.prepareOutbound(msg, text); ok { if outbound, ok := al.prepareOutbound(msg, text); ok {
al.bus.PublishOutbound(outbound) al.bus.PublishOutbound(outbound)
} }
@@ -363,7 +368,14 @@ func (al *AgentLoop) startLongRunNotice(ctx context.Context, msg bus.InboundMess
} }
} }
}() }()
return func() { close(stop) } return taskID, &notified, func() { close(stop) }
}
func shortSessionKey(s string) string {
if len(s) <= 8 {
return s
}
return s[:8]
} }
func sessionShardCount() int { func sessionShardCount() int {