core

type BaseTool struct {
	ToolUnion        responses.ToolUnion
	RequiresApproval bool
}

type HistorySummarizer interface {
	// Summarize takes a list of messages and returns a summary result.
	// If summarization is not needed, returns a result with KeepFromIndex = -1.
	Summarize(ctx context.Context, msgIdToRunId map[string]string, messages []responses.InputMessageUnion, usage *responses.Usage) (*SummaryResult, error)
}

type RunState struct {
	CurrentStep           Step                            `json:"current_step"`
	LoopIteration         int                             `json:"loop_iteration"`
	Usage                 responses.Usage                 `json:"usage"`
	PendingToolCalls      []responses.FunctionCallMessage `json:"pending_tool_calls,omitempty"`
	ToolsAwaitingApproval []responses.FunctionCallMessage `json:"tools_awaiting_approval,omitempty"`
}

type RunStatus string

type Step string

type StreamBroker interface {
	// Publish sends a response chunk to subscribers of the given channel.
	// The channel is typically the run ID or a unique identifier for the execution.
	Publish(ctx context.Context, channel string, chunk *responses.ResponseChunk) error

	// Subscribe returns a channel that receives response chunks for the given channel.
	// The returned channel will be closed when Close is called or the context is cancelled.
	Subscribe(ctx context.Context, channel string) (<-chan *responses.ResponseChunk, error)

	// Close signals that no more chunks will be published to the channel.
	// This should close all subscriber channels for the given channel.
	Close(ctx context.Context, channel string) error
}

type SummaryResult struct {
	Summary                 *responses.InputMessageUnion // The summary message
	MessagesToKeep          []responses.InputMessageUnion
	LastSummarizedMessageID string // ID of the last message that was summarized
	SummaryID               string // Unique ID for the summary (generated if empty)
}

type SystemPromptProvider interface {
	GetPrompt(ctx context.Context, data map[string]any) (string, error)
}

type Tool interface {
	Execute(ctx context.Context, params *responses.FunctionCallMessage) (*responses.FunctionCallOutputMessage, error)
	Tool(ctx context.Context) *responses.ToolUnion
	NeedApproval() bool
}