stream

type ContentSafetyMatcher ¶

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

ContentSafetyMatcher scans for blocked patterns in the stream.

func NewContentSafetyMatcher(patterns []ContentSafetyPattern) *ContentSafetyMatcher

func (m *ContentSafetyMatcher) Name() string

func (m *ContentSafetyMatcher) Scan(buf []byte, prevTail string) MatchResult

type ContentSafetyPattern ¶

type ContentSafetyPattern struct {
	Name  string `json:"name"`
	Regex string `json:"regex"`
	// contains filtered or unexported fields
}

ContentSafetyPattern is a named regex pattern for content filtering.

type ContinueAction ¶

type ContinueAction struct{}

type Extractable ¶

type Extractable interface {
	Extract(buf []byte) FilterAction
}

Extractable is implemented by matchers that produce data on FullMatch.

type FilterAction ¶

type FilterAction interface {
	// contains filtered or unexported methods
}

FilterAction is returned by StreamFilter.Write and Flush to signal what happened.

type KillAction ¶

type KillAction struct{ Reason string }

type MatchResult ¶

type MatchResult int

const (
	NoMatch      MatchResult = iota
	PartialMatch             // keep buffering, pattern might be forming
	FullMatch                // pattern confirmed, act on it
)

type Matcher ¶

type Matcher interface {
	Scan(buf []byte, prevTail string) MatchResult
	Name() string
}

Matcher scans a buffer and reports whether a pattern is present. prevTail contains the last N chars of previously emitted text for cross-boundary matching.

type StreamFilter ¶

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

StreamFilter sits between the model output stream and the SSE sender. It maintains a small lookahead buffer, runs matchers against it, and emits tokens from the trailing edge with a fixed delay.

func NewStreamFilter(emitFunc func(string), matchers []Matcher, maxBuffer int) *StreamFilter

func (f *StreamFilter) Flush() FilterAction

func (f *StreamFilter) PrevTail() string

func (f *StreamFilter) Write(token string) FilterAction

type ToolCall ¶

type ToolCall struct {
	Name      string         `json:"name"`
	Arguments map[string]any `json:"arguments"`
}

ToolCall is a provider-agnostic tool call representation.

type ToolCallAction ¶

type ToolCallAction struct{ Calls []ToolCall }

type ToolCallMatcher ¶

type ToolCallMatcher struct{}

ToolCallMatcher detects tool call JSON emitted as text content.

func NewToolCallMatcher() *ToolCallMatcher

func (m *ToolCallMatcher) Extract(buf []byte) FilterAction

func (m *ToolCallMatcher) Name() string

func (m *ToolCallMatcher) Scan(buf []byte, _ string) MatchResult