llmtest

Documentation

Overview

Package llmtest provides helpers for testing code that consumes llm.Stream channels, following the convention of packages like net/http/httptest.

Index

• func CompletedEvent(reason llm.StopReason) llm.Event
• func ErrorEvent(err *llm.ProviderError) llm.Event
• func ReasoningEvent(s string) llm.Event
• func SendEvents(evs ...llm.Event) <-chan llm.Envelope
• func TextEvent(s string) llm.Event
• func ToolEvent(id, name string, args map[string]any) llm.Event
• func UsageEvent(rec usage.Record) llm.Event
• func UsageTokenEvent(provider, model string, inputTokens, outputTokens int) llm.Event

Functions

func CompletedEvent

func CompletedEvent(reason llm.StopReason) llm.Event

func ErrorEvent

func ErrorEvent(err *llm.ProviderError) llm.Event

func ReasoningEvent

func ReasoningEvent(s string) llm.Event

func SendEvents

func SendEvents(evs ...llm.Event) <-chan llm.Envelope

SendEvents builds a buffered, pre-populated Envelope channel and closes it. Use it to construct a fake stream in tests:

ch := llmtest.SendEvents(
    llmtest.TextEvent("hello"),
    llmtest.CompletedEvent(llm.StopReasonEndTurn),
)

func TextEvent

func TextEvent(s string) llm.Event

func ToolEvent

func ToolEvent(id, name string, args map[string]any) llm.Event

func UsageEvent

func UsageEvent(rec usage.Record) llm.Event

UsageEvent builds a UsageUpdatedEvent from a usage.Record. For simple tests that only need token counts, use UsageTokenEvent.

func UsageTokenEvent

func UsageTokenEvent(provider, model string, inputTokens, outputTokens int) llm.Event

UsageTokenEvent builds a minimal UsageUpdatedEvent with the given token counts. provider and model are optional; pass empty strings to omit them.