agenttest

Documentation

Overview

Package agenttest provides test helpers for pi-core-agent-go.

Index

• func NewAgent(t *testing.T, opts Opts) *pi.Agent
• func RunAndCollect(t *testing.T, agent *pi.Agent, prompt string) ([]pi.AgentEvent, pi.PromptResult)
• type Opts

Functions

func NewAgent

func NewAgent(t *testing.T, opts Opts) *pi.Agent

NewAgent creates an Agent with sensible test defaults and registers t.Cleanup.

func RunAndCollect

func RunAndCollect(t *testing.T, agent *pi.Agent, prompt string) ([]pi.AgentEvent, pi.PromptResult)

RunAndCollect prompts the agent, drains both channels, and returns all events + the terminal result.

Types

type Opts

type Opts struct {
	Provider     llm.LLMProvider
	Tools        []pi.RegisteredTool
	Hooks        pi.Hooks
	SystemPrompt string
	Session      pi.SessionRepo
}

Opts carries options for NewAgent.