steerloop

type ActiveLoop struct {
	Loop   *Loop
	Cancel context.CancelFunc
}

type AwaitingInputData struct {
	Question string   `json:"question"`
	Options  []string `json:"options,omitempty"`
	Payload  any      `json:"payload,omitempty"`
}

type Control struct {
	Type ControlType
	Data any
}

type ControlType string

type DoneData struct {
	Reason       DoneReason `json:"reason"`
	Iterations   int        `json:"iterations"`
	FinalContent string     `json:"final_content,omitempty"`
	// Error is the fatal error message when Reason is DoneReasonError.
	Error string `json:"error,omitempty"`
	// Awaiting is set when Reason is DoneReasonAwaitingHuman.
	Awaiting *AwaitingInputData `json:"awaiting,omitempty"`
	// DroppedEvents counts intermediate events dropped because the event
	// channel was full.
	DroppedEvents int             `json:"dropped_events,omitempty"`
	LLMUsage      LLMUsageSummary `json:"llm_usage"`
}

type DoneReason string

type ErrorData struct {
	Error string `json:"error"`
	Fatal bool   `json:"fatal"`
}

type Event struct {
	Type      EventType `json:"type"`
	Data      any       `json:"data"`
	Iteration int       `json:"iteration"`
	Timestamp time.Time `json:"timestamp"`
}

type EventSink func(ctx context.Context, evt Event)

type EventType string

type InjectData struct {
	Message string
}

type LLMTurnData struct {
	TurnIndex              int     `json:"turn_index"`
	InputTokens            int     `json:"input_tokens"`
	OutputTokens           int     `json:"output_tokens"`
	CostAmount             float64 `json:"cost_amount"`
	Provider               string  `json:"provider,omitempty"`
	Model                  string  `json:"model,omitempty"`
	DurationMs             int64   `json:"duration_ms"`
	RequestMessagesJSON    []byte  `json:"request_messages_json,omitempty"`
	ResponseKind           string  `json:"response_kind"`
	AssistantContent       string  `json:"assistant_content,omitempty"`
	AssistantToolCallsJSON []byte  `json:"assistant_tool_calls_json,omitempty"`
	MaxIterations          int     `json:"max_iterations,omitempty"`
	StopCondition          string  `json:"stop_condition,omitempty"`
}

type LLMUsageSummary struct {
	InputTokens  int    `json:"input_tokens"`
	OutputTokens int    `json:"output_tokens"`
	TotalTokens  int    `json:"total_tokens"`
	Model        string `json:"model,omitempty"`
}

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

type LoopConfig struct {
	// MaxIterations caps the number of model calls. The loop never exceeds it.
	MaxIterations int
	// StopCondition decides when the loop ends on its own.
	StopCondition StopCondition
	// Temperature is passed to the model on every call.
	Temperature float64
	// StepTimeout bounds one model call. Zero means no per-call bound.
	StepTimeout time.Duration
	// TotalTimeout bounds the whole run. Zero means no bound.
	TotalTimeout time.Duration

	// EnableReflection adds one self-review call after the final reply in
	// StopOnNoToolCalls mode.
	EnableReflection bool
	// ReflectionPrompt is the instruction for that call. The loop appends the
	// reply under review. Empty means a built-in neutral prompt.
	ReflectionPrompt string

	// StreamEvents turns on intermediate events. When false only error and
	// done events are emitted.
	StreamEvents bool
	// EventBuffer is the capacity of the event channel. Zero means 64.
	// Intermediate events are dropped, and counted, when it is full.
	EventBuffer int
	// EventSink, when set, receives every emitted event.
	EventSink EventSink
	// DoneDeliveryGrace is how long the loop waits for a reader to take the
	// done event when the event channel is full. Zero means 5 seconds.
	DoneDeliveryGrace time.Duration

	// Pauses maps a tool name to a pause function. A call to such a tool is
	// handed to the function instead of the ToolExecutor.
	Pauses map[string]PauseFunc

	// AutoNudgeText is the user message appended when the model replies with
	// text only in StopOnFinishTool mode. Empty means a built-in nudge.
	AutoNudgeText string
	// DisableAutoNudge turns that message off.
	DisableAutoNudge bool

	// RetryMaxAttempts is the number of extra attempts after a transient model
	// failure. Zero means 1; -1 disables retries.
	RetryMaxAttempts int
	// IsTransient classifies a model error as worth retrying. Nil means the
	// built-in classifier (deadline exceeded, EOF, empty reply, idle stream,
	// connection reset).
	IsTransient func(error) bool

	// DisableStreaming makes the loop call Generate instead of Stream.
	// Streaming is what lets the loop detect a hung turn early.
	DisableStreaming bool
	// IdleChunkTimeout is the longest gap allowed between streamed chunks
	// after the first one. Zero means 30 seconds.
	IdleChunkTimeout time.Duration
	// DisableStepTimeoutDecoupling makes each model call inherit the remaining
	// TotalTimeout. By default a call gets the full StepTimeout even when less
	// of TotalTimeout remains; cancellation of the run still reaches it.
	DisableStepTimeoutDecoupling bool

	// HistoryCompactor, when set, is applied to the message history before
	// each model call. Its result is what the model receives; the stored
	// history is not changed.
	HistoryCompactor func(msgs []*schema.Message) []*schema.Message

	// CallOptions builds the per-call model options. Nil means
	// model.WithTemperature(temperature).
	CallOptions func(temperature float32) []model.Option
	// ModelIdentity names the provider and model for llm_turn events. It is
	// called once per turn, so it may change during a run.
	ModelIdentity func() (provider, model string)
	// CostEstimator prices one turn for llm_turn events. Nil means cost 0.
	CostEstimator func(provider, model string, inputTokens, outputTokens int) float64

	// TranscriptLogLevel is the level at which the loop logs the messages
	// added on each iteration. These lines contain prompt and reply text. The
	// zero value is slog.LevelInfo; DefaultConfig sets slog.LevelDebug to keep
	// them out of normal logs.
	TranscriptLogLevel slog.Level
}

type LoopState string

type ModifyConfigData struct {
	MaxIterations    *int
	Temperature      *float64
	EnableReflection *bool
}

type PauseFunc func(ctx context.Context, in PauseIn) (reply string, cont bool)

type PauseIn struct {
	// CallID, Name and Arguments identify the tool call being handled.
	CallID    string
	Name      string
	Arguments string
	// Messages is a copy of the conversation so far.
	Messages []*schema.Message
	// Tools are the tools the loop was created with.
	Tools []*schema.ToolInfo
	// Execute runs a tool through the loop's ToolExecutor.
	Execute func(ctx context.Context, name, args string) (string, error)
	// Call makes a model call with the loop's retry and timeout handling. A
	// non-empty tools list binds the model to those tools for this call.
	Call func(ctx context.Context, tools []*schema.ToolInfo, msgs []*schema.Message) (*schema.Message, error)
	// Emit sends an event on the loop's event channel.
	Emit func(event EventType, data any)
	// Block waits for the next Inject and returns its message. ok is false
	// when the loop was stopped or its context ended while waiting.
	Block func(ctx context.Context) (reply string, ok bool)
}

type ProgressData struct {
	Message string `json:"message"`
	Step    string `json:"step,omitempty"`
}

type ReflectionData struct {
	Content string `json:"content"`
}

type Registry[K comparable] struct {
	// contains filtered or unexported fields
}

type ResponseData struct {
	Content string `json:"content"`
}

type StopCondition int

type ThinkingData struct {
	Message string `json:"message"`
}

type ToolCallData struct {
	ToolCallID string `json:"tool_call_id"`
	Name       string `json:"name"`
	Arguments  string `json:"arguments"`
}

type ToolExecutor interface {
	// ExecuteTool runs the named tool with its JSON arguments and returns the
	// text result.
	ExecuteTool(ctx context.Context, name string, arguments string) (string, error)
}

type ToolExecutorFunc func(ctx context.Context, name string, arguments string) (string, error)

type ToolResultData struct {
	ToolCallID string `json:"tool_call_id"`
	Name       string `json:"name"`
	Result     string `json:"result"`
	Error      string `json:"error,omitempty"`
	DurationMs int64  `json:"duration_ms,omitempty"`
}