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(u llm.Usage) 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, nil),
)

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(u llm.Usage) llm.Event