agent

package

v0.1.1 Latest Latest Go to latest Published: Feb 16, 2026 License: MIT Imports: 24 Imported by: 0

Details

Package agent contains yai's core (non-UI) logic.

It resolves the model/provider configuration, prepares the request (including MCP tools), and starts a streaming completion.

This section is empty.

This section is empty.

func ApplyProxyConfig(httpProxy string, providerCfg *fantasybridge.Config) error

ApplyProxyConfig configures the provider HTTP client to use an HTTP proxy.

func NewFantasyClient(cfg fantasybridge.Config) (stream.Client, error)

NewFantasyClient creates the fantasy bridge client.

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

Service is the core orchestration layer for starting LLM streams.

It is intentionally UI-agnostic and can be used by both the TUI and headless commands.

func New(cfg *config.Config, cache *cache.Conversations, mcpSvc *mcp.Service) *Service

New creates an agent service.

func (s *Service) ActionForStreamError(err error, mod config.Model, prompt string) StreamErrorAction

ActionForStreamError decides whether a provider error should be retried, and if so which prompt/model override should be used.

func (s *Service) Stream(ctx context.Context, prompt string) (StreamStart, error)

Stream starts a streaming completion for the given prompt.

type StreamErrorAction struct {
	Retry         bool
	Prompt        string
	ModelOverride string
	Err           errs.Error
}

StreamErrorAction describes how yai should respond to a streaming error.

type StreamStart struct {
	Stream   stream.Stream
	Model    config.Model
	Messages []proto.Message
}

StreamStart contains the stream plus metadata about the resolved request.