ai

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

type AttributedSink interface {
	ToolResultAttributed(toolUseID, toolName string, result json.RawMessage, isError bool, attr ToolAttribution)
}

type ChatRequest struct {
	System   string
	Messages []Message
	Tools    []ToolSpec
	Model    string
}

type ContentBlock struct {
	Type string

	// Text is set when Type is ContentBlockText.
	Text string

	// ToolUseID, ToolName, ToolInput are set when Type is
	// ContentBlockToolUse: a tool the model wants to invoke.
	ToolUseID string
	ToolName  string
	ToolInput json.RawMessage

	// ToolResultForID, ToolResultContent, ToolResultIsError are set when
	// Type is ContentBlockToolResult: an already-executed tool's outcome,
	// keyed back to the ContentBlockToolUse.ToolUseID it answers.
	ToolResultForID   string
	ToolResultContent string
	ToolResultIsError bool
}

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

type Message struct {
	Role    string
	Content []ContentBlock
}

type Provider interface {
	Complete(ctx context.Context, req ChatRequest, onText TextDeltaFunc) (TurnResult, error)
}

type ProviderFactory func(provider, apiKey string) (Provider, error)

type SecretsResolver interface {
	Resolve(ctx context.Context, serviceName, envKey string) (string, error)
	Exists(ctx context.Context, serviceName, envKey string) (bool, error)
}

type Sink interface {
	TextDelta(text string)
	ToolCallProposed(confirmationID, toolUseID, toolName string, arguments json.RawMessage)
	ToolResult(toolUseID, toolName string, result json.RawMessage, isError bool)
}

type Store interface {
	ListAIChatMessages(ctx context.Context, sessionID string) ([]store.AIChatMessage, error)
	SaveAIChatMessage(ctx context.Context, m store.AIChatMessage) error
	UpdateAIChatMessageToolCalls(ctx context.Context, messageID string, toolCalls []store.AIChatToolCall) error
	GetAIChatMessage(ctx context.Context, id string) (store.AIChatMessage, error)
	SaveAIChatConfirmation(ctx context.Context, c store.AIChatConfirmation) error
	GetAIChatConfirmation(ctx context.Context, id string) (store.AIChatConfirmation, error)
	ResolveAIChatConfirmation(ctx context.Context, id, status string, result json.RawMessage, resolvedAt time.Time) error
	ListPendingAIChatConfirmations(ctx context.Context, messageID string) ([]store.AIChatConfirmation, error)
	GetAIAssistantSettings(ctx context.Context) (store.AIAssistantSettings, error)
}

type TextDeltaFunc func(delta string)

type ToolAttribution struct {
	ConfirmationID string
	ActingAs       string
	Abilities      []string
}

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

type ToolExecutor interface {
	ListTools(ctx context.Context) ([]ToolSpec, error)
	Call(ctx context.Context, name string, args json.RawMessage) (result string, isError bool, err error)
}

type ToolFactory func(ctx context.Context, token string) (ToolExecutor, error)

type ToolSpec struct {
	Name        string
	Description string
	InputSchema json.RawMessage
	Traits      ToolTraits
}

type ToolTraits struct {
	Known           bool
	ReadOnly        bool
	Destructive     bool
	OpenWorld       bool
	Sensitive       bool
	UntrustedOutput bool
}

type ToolUseCall struct {
	ID    string
	Name  string
	Input json.RawMessage
}

type TurnCredential struct {
	// Token is a short-lived bearer minted for this turn only, capped to the sender.
	Token string
	// ActingAs names the human principal the token stands in for.
	ActingAs string
	// AbilitiesUsed drains the abilities the API checked since its last call; nil is allowed.
	AbilitiesUsed func() []string
}

type TurnResult struct {
	Text       string
	ToolCalls  []ToolUseCall
	StopReason string
}