agents

type AgentConfig struct {
	Model            models.Model            `json:"-"`
	Tools            []tools.Tool            `json:"-"`
	SystemPrompt     string                  `json:"system_prompt"`
	MaxSteps         int                     `json:"max_steps"`
	Planning         bool                    `json:"planning"`
	PlanningInterval int                     `json:"planning_interval"`
	PromptTemplates  *PromptTemplates        `json:"prompt_templates"`
	Logger           *monitoring.AgentLogger `json:"-"`
	Monitor          *monitoring.Monitor     `json:"-"`
	Additional       map[string]interface{}  `json:"additional"`
}

type BaseMultiStepAgent struct {
	// contains filtered or unexported fields
}

type FinalAnswerPromptTemplate struct {
	PreMessages  string `json:"pre_messages"`
	PostMessages string `json:"post_messages"`
}

type FinalOutput struct {
	Output interface{} `json:"output"`
}

type ManagedAgentPromptTemplate struct {
	Task   string `json:"task"`
	Report string `json:"report"`
}

type MultiStepAgent interface {
	// Core execution methods
	Run(options *RunOptions) (*RunResult, error)
	RunStream(options *RunOptions) (<-chan *StreamStepResult, error)

	// Agent management
	Reset() error
	Interrupt() error

	// Configuration
	GetModel() models.Model
	SetModel(model models.Model)
	GetTools() []tools.Tool
	SetTools(tools []tools.Tool)
	GetMemory() *memory.AgentMemory
	SetMemory(mem *memory.AgentMemory)
	GetSystemPrompt() string
	SetSystemPrompt(prompt string)
	GetMaxSteps() int
	SetMaxSteps(maxSteps int)

	// Serialization
	ToDict() map[string]interface{}
	Save(outputDir string, toolFileName string, makeGradioApp bool) error
	PushToHub(repoID string, commitMessage string, token string, private bool) error

	// Execution state
	IsRunning() bool
	GetStepCount() int
}

type PlanningPromptTemplate struct {
	InitialPlan       string `json:"initial_plan"`
	UpdatePlanPreMsg  string `json:"update_plan_pre_messages"`
	UpdatePlanPostMsg string `json:"update_plan_post_messages"`
}

type PromptTemplates struct {
	SystemPrompt string                     `json:"system_prompt"`
	Planning     PlanningPromptTemplate     `json:"planning"`
	ManagedAgent ManagedAgentPromptTemplate `json:"managed_agent"`
	FinalAnswer  FinalAnswerPromptTemplate  `json:"final_answer"`
}

type ReactCodeAgent struct {
	*BaseMultiStepAgent
	// contains filtered or unexported fields
}

type ReactCodeAgentOptions struct {
	AuthorizedPackages []string
	CodeBlockTags      [2]string
	StreamOutputs      bool
	StructuredOutput   bool
	MaxCodeLength      int
	EnablePlanning     bool
	PlanningInterval   int
	MaxSteps           int
	Verbose            bool
}

type RunOptions struct {
	Task           string                 `json:"task"`
	Stream         bool                   `json:"stream,omitempty"`
	Reset          bool                   `json:"reset,omitempty"`
	Images         []interface{}          `json:"images,omitempty"`
	AdditionalArgs map[string]interface{} `json:"additional_args,omitempty"`
	MaxSteps       *int                   `json:"max_steps,omitempty"`
	Context        context.Context        `json:"-"`
	StepCallbacks  []StepCallback         `json:"-"`
}

type RunResult struct {
	Output     interface{}              `json:"output"`
	State      string                   `json:"state"` // "success", "max_steps_error"
	Messages   []map[string]interface{} `json:"messages"`
	TokenUsage *monitoring.TokenUsage   `json:"token_usage,omitempty"`
	Timing     *monitoring.Timing       `json:"timing"`
	StepCount  int                      `json:"step_count"`
	Metadata   map[string]interface{}   `json:"metadata,omitempty"`
	Error      error                    `json:"error,omitempty"`
}

type StepCallback func(step memory.MemoryStep) error

type StreamStepResult struct {
	StepNumber int                    `json:"step_number"`
	StepType   string                 `json:"step_type"`
	Output     interface{}            `json:"output,omitempty"`
	Error      error                  `json:"error,omitempty"`
	IsComplete bool                   `json:"is_complete"`
	Metadata   map[string]interface{} `json:"metadata,omitempty"`
}