agent

Documentation

Index

• Constants
• type Completer
• type Loop
  • func NewLoop(client *ai.Client, tools ToolRunner, maxRounds int) *Loop
  • func (l *Loop) Run(ctx context.Context, prompt string, maxTokens int) (string, error)
• type Request
  • func ParseToolRequests(response string) []Request
• type ToolRunner

Constants

const (
	DefaultMaxToolRounds    = 5
	MaxToolRequestsPerRound = 3
	MaxTotalToolExecutions  = 8
	MaxToolResultChars      = 16000
)

Types

type Completer

type Completer interface {
	Complete(ctx context.Context, prompt string, maxTokens int) (string, error)
}

type Loop

type Loop struct {
	Complete  Completer
	Tools     ToolRunner
	MaxRounds int
}

func NewLoop

func NewLoop(client *ai.Client, tools ToolRunner, maxRounds int) *Loop

func (*Loop) Run

func (l *Loop) Run(ctx context.Context, prompt string, maxTokens int) (string, error)

type Request

type Request struct {
	Name string
	Args map[string]string
}

func ParseToolRequests

func ParseToolRequests(response string) []Request

type ToolRunner

type ToolRunner interface {
	Execute(ctx context.Context, tool string, args map[string]string) (string, error)
}