utils

type Budget interface {
	Check(ctx context.Context, req BudgetRequest) (BudgetResult, error)
	Consume(ctx context.Context, sessionID string, amount Cost) error
	GetState(sessionID string) BudgetState
	Reset(sessionID string)
	ResetAll()
	RecordToolCall(ctx context.Context, sessionID string)
	RecordTokens(ctx context.Context, sessionID string, tokens int)
	CanExecute(sessionID string) bool
}

type BudgetConfig struct {
	Enabled      bool
	MaxTokens    int
	MaxToolCalls int
	MaxTimeMs    int64
}

type BudgetRequest struct {
	SessionID string
	ToolCalls []string
	Estimated Cost
}

type BudgetResult struct {
	Allowed bool   `json:"allowed"`
	Reason  string `json:"reason"`
	Remain  Cost   `json:"remain"`
}

type BudgetState struct {
	UsedTokens int   `json:"used_tokens"`
	UsedCalls  int   `json:"used_calls"`
	UsedTimeMs int64 `json:"used_time_ms"`
	MaxTokens  int   `json:"max_tokens"`
	MaxCalls   int   `json:"max_calls"`
	MaxTimeMs  int64 `json:"max_time_ms"`
}

type Cost struct {
	Tokens int   `json:"tokens"`
	Calls  int   `json:"calls"`
	TimeMs int64 `json:"time_ms"`
}

type LoopDetectAction struct {
	Type      string    `json:"type"`
	Content   string    `json:"content"`
	Timestamp time.Time `json:"timestamp"`
}

type LoopDetectConfig struct {
	Enabled             bool
	MaxRepeats          int
	WarningRepeats      int
	SimilarityThreshold float64
	TrackResults        bool
	InputMaxRepeats     int
}

type LoopDetectResult struct {
	IsLoop     bool    `json:"is_loop"`
	IsWarning  bool    `json:"is_warning"`
	Count      int     `json:"count"`
	Similarity float64 `json:"similarity"`
	Suggestion string  `json:"suggestion"`
	Level      string  `json:"level"` // "warning" or "critical"
}

type LoopDetectStats struct {
	TotalActions int            `json:"total_actions"`
	ActionCounts map[string]int `json:"action_counts"`
}

type LoopDetector interface {
	Detect(ctx context.Context, sessionID string, action LoopDetectAction) *LoopDetectResult
	Record(sessionID string, action LoopDetectAction)
	RecordWithResult(sessionID string, action LoopDetectAction, resultHash string)
	Reset(sessionID string)
	GetStats(sessionID string) LoopDetectStats
}

type PermAction string

type PermEvaluator struct {
	Ruleset PermRuleset
	// contains filtered or unexported fields
}

type PermRule struct {
	Permission string                 `json:"permission"`
	Action     PermAction             `json:"action"`
	Pattern    string                 `json:"pattern,omitempty"`
	Extra      map[string]interface{} `json:"extra,omitempty"`
}

type PermRuleset []PermRule

type RateLimitStats struct {
	Allowed  int64
	Rejected int64
	WaitTime time.Duration
	HitRate  float64
}

type RateLimiter interface {
	Allow(ctx context.Context) error
	Wait(ctx context.Context) error
	Stats() RateLimitStats
}