agent

Documentation

Index

• func NewDefaultAgent(t rebuild.Target, deps *AgentDeps) *defaultAgent
• func RunSession(ctx context.Context, req RunSessionReq, deps RunSessionDeps)
• type Agent
• type AgentDeps
• type ProposeOpts
• type RunSessionDeps
• type RunSessionReq

Functions

func NewDefaultAgent

func NewDefaultAgent(t rebuild.Target, deps *AgentDeps) *defaultAgent

func RunSession

func RunSession(ctx context.Context, req RunSessionReq, deps RunSessionDeps)

Types

type Agent

type Agent interface {
	Propose(context.Context, *ProposeOpts) (*schema.StrategyOneOf, error)
	RecordIteration(*schema.AgentIteration)
}

type AgentDeps

type AgentDeps struct {
	Chat *llm.Chat
	// Bucket for logs and rebuild artifact
	MetadataBucket string
	LogsBucket     string
	GCSClient      *gcs.Client
	MaxTurns       int
	GenaiClient    *genai.Client
}

type ProposeOpts

type ProposeOpts struct {
	ChatUploadURL *url.URL // The path to which llm.Chat messages should be stored.
}

type RunSessionDeps

type RunSessionDeps struct {
	Client        *genai.Client
	GCSClient     *gcs.Client
	IterationStub api.StubT[schema.AgentCreateIterationRequest, schema.AgentCreateIterationResponse]
	CompleteStub  api.StubT[schema.AgentCompleteRequest, schema.AgentCompleteResponse]
	// TODO: Should these be asset stores?
	SessionsBucket string
	MetadataBucket string
	LogsBucket     string
}

type RunSessionReq

type RunSessionReq struct {
	SessionID        string
	Target           rebuild.Target
	MaxIterations    int
	InitialIteration *schema.AgentIteration
}