before_start.go RawHistoryFolder

1package agent
2
3import (
4	"context"
5	"crypto/rand"
6
7	"github.com/patriceckhart/zot/packages/agent/extensions"
8	"github.com/patriceckhart/zot/packages/agent/extproto"
9	"github.com/patriceckhart/zot/packages/core"
10)
11
12func wireBeforeAgentStart(ag *core.Agent, mgr *extensions.Manager, provider string) {
13	runtimeSession := rand.Text()
14	ag.BeforeStart = func(ctx context.Context, system string) string {
15		session := ag.SessionID
16		if session == "" {
17			session, _ = mgr.SessionContext()
18			if session == "" {
19				session = runtimeSession
20			}
21		}
22		return mgr.InterceptBeforeAgentStart(ctx, extproto.EventInterceptFromHost{
23			SystemPrompt: &system, SessionID: session, AgentRunID: rand.Text(),
24			Provider: provider, Model: ag.Model,
25		})
26	}
27}