judge

type Decision string

type DownloadProgress struct {
	Event        DownloadProgressEvent
	CurrentBytes int64
	TotalBytes   int64
	Err          error
}

type DownloadProgressEvent string

type DownloadProgressHandler func(DownloadProgress)

type Error struct {
	Kind string
	Err  error
}

type Fixture struct {
	ID                  string                   `json:"id"`
	Description         string                   `json:"description"`
	HookEvent           FixtureHookEvent         `json:"hook_event"`
	NormalizedEvent     FixtureNormalizedEvent   `json:"normalized_event"`
	DeterministicPolicy FixtureDeterministicRule `json:"deterministic_policy"`
	JudgeExpected       FixtureExpected          `json:"judge_expected"`
	Notes               string                   `json:"notes"`
}

type FixtureDeterministicRule struct {
	Decision      string   `json:"decision"`
	MatchedRules  []string `json:"matched_rules,omitempty"`
	PolicyVersion string   `json:"policy_version"`
}

type FixtureExpected struct {
	ShouldCallJudge bool      `json:"should_call_judge"`
	Decision        Decision  `json:"decision"`
	RiskLevel       RiskLevel `json:"risk_level"`
	Categories      []string  `json:"categories"`
	ReasonContains  []string  `json:"reason_contains"`
}

type FixtureHookEvent struct {
	Agent         string         `json:"agent"`
	HookEventName string         `json:"hook_event_name"`
	ToolName      string         `json:"tool_name"`
	ToolInput     map[string]any `json:"tool_input"`
}

type FixtureNormalizedEvent struct {
	Type               string   `json:"type"`
	ProviderCategory   string   `json:"provider_category,omitempty"`
	OperationClass     string   `json:"operation_class,omitempty"`
	ResourceClass      string   `json:"resource_class,omitempty"`
	Environment        string   `json:"environment,omitempty"`
	PathClass          string   `json:"path_class,omitempty"`
	CommandSummary     string   `json:"command_summary,omitempty"`
	RequestSummary     string   `json:"request_summary,omitempty"`
	ExplicitUserIntent bool     `json:"explicit_user_intent"`
	Signals            []string `json:"signals,omitempty"`
}

type HTTPOptions struct {
	BaseURL         string
	Model           string
	Runtime         string
	Timeout         time.Duration
	HTTPClient      *http.Client
	Prompt          string
	DisableThinking bool
}

type Input struct {
	ToolName           string    `json:"tool_name,omitempty"`
	ExplicitUserIntent bool      `json:"explicit_user_intent,omitempty"`
	ToolInput          ToolInput `json:"tool_input"`
}

type Judge interface {
	Decide(context.Context, Input) (Result, error)
}

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

type LlamaServerOptions struct {
	BinaryPath       string
	ModelPath        string
	HFRepo           string
	HFFile           string
	HFRevision       string
	CacheDir         string
	Host             string
	Port             int
	StartupTimeout   time.Duration
	HTTPClient       *http.Client
	Stdout           io.Writer
	Stderr           io.Writer
	DownloadProgress DownloadProgressHandler
}

type Metadata struct {
	Runtime     string
	Model       string
	DurationMs  int64
	FailureKind string
}

type MetadataProvider interface {
	Metadata() Metadata
}

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

type Output struct {
	Decision   Decision  `json:"decision"`
	RiskLevel  RiskLevel `json:"risk_level"`
	Categories []string  `json:"categories"`
	Reason     string    `json:"reason"`
}

type Result struct {
	Output   Output
	Metadata Metadata
}

type RiskLevel string

type ToolInput struct {
	Command string `json:"command,omitempty"`
	Path    string `json:"path,omitempty"`
	Request string `json:"request,omitempty"`
}

type UnavailableJudge struct {
	Runtime string
	Model   string
	Kind    string
	Err     error
}