middleware

Documentation

Index

• type ApprovalMiddleware
  • func NewApprovalMiddleware(cfg ApprovalMiddlewareConfig) *ApprovalMiddleware
  • func (m *ApprovalMiddleware) Execute(ctx context.Context, toolCall *sdk.ChatCompletionMessageToolCall, ...) (*domain.ToolExecutionResult, error)
  • func (m *ApprovalMiddleware) ExecuteBatch(ctx context.Context, toolCalls []sdk.ChatCompletionMessageToolCall, ...) ([]*domain.ToolExecutionResult, error)
  • func (m *ApprovalMiddleware) WithRequestID(requestID string) *ApprovalMiddleware
• type ApprovalMiddlewareConfig
• type ToolExecutor

Types

type ApprovalMiddleware

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

ApprovalMiddleware intercepts tool execution and injects approval logic

func NewApprovalMiddleware

func NewApprovalMiddleware(cfg ApprovalMiddlewareConfig) *ApprovalMiddleware

NewApprovalMiddleware creates a new approval middleware

func (*ApprovalMiddleware) Execute

func (m *ApprovalMiddleware) Execute(
	ctx context.Context,
	toolCall *sdk.ChatCompletionMessageToolCall,
	isChatMode bool,
	next ToolExecutor,
) (*domain.ToolExecutionResult, error)

Execute intercepts tool execution and applies approval logic If approval is required, it requests approval from the user via events If approval is granted (or not required), it enriches context and calls the next handler

func (*ApprovalMiddleware) ExecuteBatch

func (m *ApprovalMiddleware) ExecuteBatch(
	ctx context.Context,
	toolCalls []sdk.ChatCompletionMessageToolCall,
	isChatMode bool,
	executor ToolExecutor,
) ([]*domain.ToolExecutionResult, error)

ExecuteBatch executes multiple tool calls, handling approval for each Tools requiring approval are executed sequentially, while approved tools can be executed in parallel

func (*ApprovalMiddleware) WithRequestID

func (m *ApprovalMiddleware) WithRequestID(requestID string) *ApprovalMiddleware

WithRequestID creates a new middleware with a different request ID Useful for different execution contexts

type ApprovalMiddlewareConfig

type ApprovalMiddlewareConfig struct {
	Policy       domain.ApprovalPolicy
	StateManager domain.StateManager
	EventBus     chan<- domain.ChatEvent
	RequestID    string
}

ApprovalMiddlewareConfig contains configuration for the approval middleware

type ToolExecutor

type ToolExecutor func(ctx context.Context, toolCall *sdk.ChatCompletionMessageToolCall) (*domain.ToolExecutionResult, error)

ToolExecutor is a function that executes a tool and returns the result