DBT/clawgo
1
0
Fork 0
mirror of https://github.com/YspCoder/clawgo.git synced 2026-04-13 01:57:28 +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) {
stopNotice := al.startLongRunNotice(ctx, msg)
taskID, noticeCount, stopNotice := al.startLongRunNotice(ctx, msg)
defer stopNotice()
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)
}
if noticeCount != nil && *noticeCount > 0 && response != "" {
response = fmt.Sprintf("任务ID %s 已完成。\n\n%s", taskID, response)
}
trigger := al.getTrigger(msg)
suppressed := false
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)
}
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
interval := 45 * time.Second
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
}
}
taskID := fmt.Sprintf("%s-%d", shortSessionKey(msg.SessionKey), time.Now().Unix()%100000)
stop := make(chan struct{})
notified := 0
go func() {
t := time.NewTimer(first)
defer t.Stop()
notified := 0
for {
select {
case <-ctx.Done():
@@ -355,7 +360,7 @@ func (al *AgentLoop) startLongRunNotice(ctx context.Context, msg bus.InboundMess
return
case <-t.C:
notified++
text := fmt.Sprintf("任务较复杂，正在持续执行中（第%d次进度通知）…", notified)
text := fmt.Sprintf("任务ID %s 正在执行中（第%d次进度通知）…", taskID, notified)
if outbound, ok := al.prepareOutbound(msg, text); ok {
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 {