agent

package

v0.1.1 Latest Latest Go to latest Published: Mar 7, 2026 License: MIT Imports: 12 Imported by: 0

Details

Package agent provides the agent runtime pool and runner for processing tasks as goroutines with session lifecycle management and cost recording.

This section is empty.

View Source

var ErrPoolShutdown = errors.New("pool is shut down")

ErrPoolShutdown is returned when Submit is called on a shut-down pool.

This section is empty.

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

Pool manages a fixed set of worker goroutines that process agent tasks.

func NewPool(registry *provider.Registry, tracker forge.IssueTracker, st store.Store, costs *cost.Tracker, workers int) *Pool

NewPool creates a pool with the given number of worker goroutines and starts them immediately. If workers is <= 0, it defaults to 3.

func (p *Pool) Done() <-chan struct{}

Done returns a channel that is closed after Shutdown completes.

func (p *Pool) Shutdown()

Shutdown signals all workers to stop and waits for in-flight tasks to complete. After Shutdown returns, Submit will return ErrPoolShutdown.

func (p *Pool) Submit(task Task) error

Submit enqueues a task for processing. Returns ErrPoolShutdown if the pool has been shut down.

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

Runner executes a single agent task: sends the issue to an AI provider, records cost, and posts the response as an issue comment.

func NewRunner(p provider.Provider, model string, tracker forge.IssueTracker, st store.Store, costs *cost.Tracker) *Runner

NewRunner creates a runner bound to a specific provider and model.

func (r *Runner) Run(ctx context.Context, task Task) error

Run processes a single task through the AI provider pipeline.

Steps:

On any error, the session is marked "failed" and an error comment is posted.

type Task struct {
	Issue     *forge.Issue
	AgentType models.AgentType
	SessionID string
}

Task represents a unit of work to be processed by the agent pool.