pruner

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

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

type BM25Scorer struct {
	K1 float64 // term frequency saturation (default 1.2)
	B  float64 // length normalization (default 0.75)
}

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

type Backend interface {
	Prune(ctx context.Context, req PruneRequest) (*PruneResponse, error)
	Health(ctx context.Context) error
	Close() error
}

type Config struct {
	Enabled       bool    `yaml:"enabled"`
	Backend       string  `yaml:"backend"`    // public: "local" only; internal tests may still use bm25/ir
	RemoteURL     string  `yaml:"remote_url"` // deprecated (SWE remote backend removed)
	ModelDir      string  `yaml:"model_dir"`  // deprecated (ONNX cross-encoder removed)
	Threshold     float64 `yaml:"threshold"`
	MinLines      int     `yaml:"min_lines"`
	TimeoutMs     int     `yaml:"timeout_ms"`
	CacheCapacity int     `yaml:"cache_capacity"` // LRU cache slots (default 256)
}

type ContentType int

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

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

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

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

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

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

type ModelDownloadProgress struct {
	File       string  `json:"file"`
	FileIndex  int     `json:"file_index"`
	TotalFiles int     `json:"total_files"`
	Downloaded int64   `json:"downloaded"`
	Total      int64   `json:"total"`
	Percentage float64 `json:"percentage"`
	SpeedHuman string  `json:"speed_human"`
	ETA        string  `json:"eta"`
}

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

type PruneRequest struct {
	Code        string      `json:"code"`    // Backward compat alias
	Content     string      `json:"content"` // Preferred: raw content (code or text)
	Query       string      `json:"query,omitempty"`
	Threshold   float64     `json:"threshold,omitempty"`
	ContentType ContentType `json:"content_type,omitempty"` // Hint (auto-detected if 0)
}

type PruneResponse struct {
	PrunedCode      string      `json:"pruned_code"`
	PrunedContent   string      `json:"pruned_content"`
	ContentType     ContentType `json:"content_type"`
	Score           float64     `json:"score"`
	OriginalLines   int         `json:"original_lines"`
	KeptLines       int         `json:"kept_lines"`
	PrunedLines     int         `json:"pruned_lines"`
	OriginalTokens  int         `json:"original_tokens"`
	PrunedTokens    int         `json:"pruned_tokens"`
	CompressionRate float64     `json:"compression_rate"`
	LatencyMs       float64     `json:"latency_ms"`
}

type PrunerFileStatus struct {
	Filename   string `json:"filename"`
	Downloaded bool   `json:"downloaded"`
	Size       string `json:"size"`
}

type PrunerModelInfo struct {
	Filename string   `json:"filename"`
	URL      string   `json:"url"`
	Mirrors  []string `json:"-"` // fallback URLs (hf-mirror, modelscope, etc.)
	Size     string   `json:"size"`
}

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

type PrunerModelStatus struct {
	Ready       bool                   `json:"ready"`
	Downloading bool                   `json:"downloading"`
	State       string                 `json:"state,omitempty"`
	Error       string                 `json:"error,omitempty"`
	Progress    *ModelDownloadProgress `json:"progress,omitempty"`
	Files       []PrunerFileStatus     `json:"files,omitempty"`
}

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

type RequestPruneStats struct {
	Pruned         bool `json:"pruned"`
	MessagesPruned int  `json:"messages_pruned"`
	TokensBefore   int  `json:"tokens_before"`
	TokensAfter    int  `json:"tokens_after"`
}

type ScoredSegment struct {
	Segment Segment
	Score   float64
}

type Scorer interface {
	Score(query string, segments []Segment) []ScoredSegment
}

type Segment struct {
	StartLine  int
	EndLine    int
	Kind       SegmentKind
	Name       string
	NameTokens []string // pre-tokenized segment name
	Content    string
	Tokens     []string // pre-tokenized content
	TokenCount int      // pre-computed token estimate (0 = not set)
}

type SegmentKind int

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

type StatsSnapshot struct {
	TotalRequests       int64   `json:"total_requests"`
	PrunedRequests      int64   `json:"pruned_requests"`
	PassthroughRequests int64   `json:"passthrough_requests"`
	TotalTokensBefore   int64   `json:"total_tokens_before"`
	TotalTokensAfter    int64   `json:"total_tokens_after"`
	TokensSaved         int64   `json:"tokens_saved"`
	AvgCompressionRate  float64 `json:"avg_compression_rate"`
	AvgLatencyMs        float64 `json:"avg_latency_ms"`
}