1package agent
2
3import (
4 "context"
5 "fmt"
6 "strings"
7 "sync"
8
9 "github.com/bnema/zut/packages/agent/extensions"
10 "github.com/bnema/zut/packages/agent/scheduler"
11 "github.com/bnema/zut/packages/core"
12 "github.com/bnema/zut/packages/provider"
13)
14
15// runScheduledSession reconstructs an inactive session in the current zut
16// process and appends the resulting turn to its existing transcript. The
17// caller serializes this with session transitions before invoking it.
18func runScheduledSession(ctx context.Context, task scheduler.Task, args Args, base Resolved, extMgr *extensions.Manager, prepareRegistry func(core.Registry) core.Registry) error {
19 path, err := core.FindManagedSessionByID(ctx, ZutHome(), task.SessionID)
20 if err != nil {
21 return fmt.Errorf("find scheduled session %q: %w", task.SessionID, err)
22 }
23 if path == "" {
24 return fmt.Errorf("scheduled session %q no longer exists", task.SessionID)
25 }
26 sess, messages, err := core.OpenSession(path)
27 if err != nil {
28 return fmt.Errorf("open scheduled session: %w", err)
29 }
30 if sess.ID != task.SessionID {
31 closeErr := sess.Close()
32 if closeErr != nil {
33 return fmt.Errorf("scheduled session %q changed during resume: %w", task.SessionID, closeErr)
34 }
35 return fmt.Errorf("scheduled session %q changed during resume", task.SessionID)
36 }
37 defer sess.Close()
38
39 providerName := strings.TrimSpace(sess.Meta.Provider)
40 if providerName == "" {
41 providerName = base.Provider
42 }
43 model := strings.TrimSpace(sess.Meta.Model)
44 if model == "" {
45 model = base.Model
46 }
47 ag, _, _, err := buildNonInteractiveSessionAgentWithRegistry(ctx, args, base, extMgr, providerName, model, prepareRegistry)
48 if err != nil {
49 return fmt.Errorf("build scheduled session agent: %w", err)
50 }
51 ag.SetMessages(trimMessagesForResume(messages, 100))
52 if cumulative, lastTurn, usageErr := core.SessionUsageDetail(path); usageErr == nil {
53 ag.SeedCost(cumulative)
54 ag.SeedLastTurnUsage(lastTurn)
55 }
56 ag.SetSessionTimeContext(sess.Meta.Started, sess.Meta.Timezone, sess.Meta.TimezoneOffset)
57 var appendMu sync.Mutex
58 var appendErr error
59 recordAppendError := func(err error) {
60 if err == nil {
61 return
62 }
63 appendMu.Lock()
64 defer appendMu.Unlock()
65 if appendErr == nil {
66 appendErr = err
67 }
68 }
69 ag.OnMessageAppended = func(message provider.Message) {
70 recordAppendError(sess.AppendMessage(message))
71 }
72 ag.OnUsage = func(cumulative provider.Usage) {
73 recordAppendError(sess.AppendUsage(cumulative, cumulative))
74 }
75 ag.OnTranscriptCompacted = func(messages []provider.Message) {
76 recordAppendError(sess.AppendCompaction(messages))
77 }
78 if err := ag.Prompt(ctx, task.Message, nil, func(core.AgentEvent) {}); err != nil {
79 return err
80 }
81 appendMu.Lock()
82 err = appendErr
83 appendMu.Unlock()
84 if err != nil {
85 return fmt.Errorf("append scheduled transcript: %w", err)
86 }
87 if err := sess.Flush(); err != nil {
88 return fmt.Errorf("flush scheduled session: %w", err)
89 }
90 return nil
91}