remediation

type Config struct {
	// CollectionPrefix is the prefix for remediation collections (default: remediations)
	CollectionPrefix string

	// VectorSize is the dimension of embedding vectors (default: 1536)
	VectorSize uint64

	// DefaultConfidence is the initial confidence for new remediations (default: 0.5)
	DefaultConfidence float64

	// FeedbackDelta is how much feedback changes confidence (default: 0.1)
	FeedbackDelta float64

	// MinConfidence is the minimum confidence threshold (default: 0.1)
	MinConfidence float64

	// MaxConfidence is the maximum confidence (default: 1.0)
	MaxConfidence float64
}

type Embedder interface {
	// Embed generates an embedding vector for the given text.
	Embed(ctx context.Context, text string) ([]float32, error)

	// EmbedBatch generates embeddings for multiple texts.
	EmbedBatch(ctx context.Context, texts []string) ([][]float32, error)

	// Dimension returns the dimensionality of the embedding vectors.
	Dimension() int
}

type ErrorCategory string

type FeedbackRating string

type FeedbackRequest struct {
	RemediationID string
	TenantID      string
	Rating        FeedbackRating
	SessionID     string
	Comment       string
}

type QdrantClient interface {
	// Collection operations
	CreateCollection(ctx context.Context, name string, vectorSize uint64) error
	DeleteCollection(ctx context.Context, name string) error
	CollectionExists(ctx context.Context, name string) (bool, error)
	ListCollections(ctx context.Context) ([]string, error)

	// Point operations
	Upsert(ctx context.Context, collection string, points []*QdrantPoint) error
	Search(ctx context.Context, collection string, vector []float32, limit uint64, filter *QdrantFilter) ([]*QdrantScoredPoint, error)
	Get(ctx context.Context, collection string, ids []string) ([]*QdrantPoint, error)
	Delete(ctx context.Context, collection string, ids []string) error

	// Health
	Health(ctx context.Context) error

	// Close closes the client connection
	Close() error
}

type QdrantCondition struct {
	Field string
	Match interface{}
	Range *QdrantRangeCondition
}

type QdrantFilter struct {
	Must    []QdrantCondition
	Should  []QdrantCondition
	MustNot []QdrantCondition
}

type QdrantPoint struct {
	ID      string
	Vector  []float32
	Payload map[string]interface{}
}

type QdrantRangeCondition struct {
	Gte *float64
	Lte *float64
	Gt  *float64
	Lt  *float64
}

type QdrantScoredPoint struct {
	QdrantPoint
	Score float32
}

type RecordRequest struct {
	Title         string
	Problem       string
	Symptoms      []string
	RootCause     string
	Solution      string
	CodeDiff      string
	AffectedFiles []string
	Category      ErrorCategory
	Tags          []string
	Scope         Scope
	TenantID      string
	TeamID        string
	ProjectPath   string
	SessionID     string
	Confidence    float64 // Initial confidence (default: 0.5)
}

type Remediation struct {
	// ID is the unique identifier for this remediation.
	ID string `json:"id"`

	// Title is a short title for the remediation.
	Title string `json:"title"`

	// Problem is a description of the error or issue.
	Problem string `json:"problem"`

	// Symptoms are observable symptoms of this error.
	Symptoms []string `json:"symptoms"`

	// RootCause is the underlying cause of the error.
	RootCause string `json:"root_cause"`

	// Solution is the fix that resolved the error.
	Solution string `json:"solution"`

	// CodeDiff is an optional code diff showing the fix.
	CodeDiff string `json:"code_diff,omitempty"`

	// AffectedFiles are files that were changed.
	AffectedFiles []string `json:"affected_files,omitempty"`

	// Category is the error category.
	Category ErrorCategory `json:"category"`

	// Confidence is the current confidence score (0.0 - 1.0).
	Confidence float64 `json:"confidence"`

	// UsageCount is how many times this remediation has been retrieved.
	UsageCount int64 `json:"usage_count"`

	// Tags are labels for categorization and filtering.
	Tags []string `json:"tags"`

	// Scope determines visibility (project, team, org).
	Scope Scope `json:"scope"`

	// TenantID is the organization this remediation belongs to.
	TenantID string `json:"tenant_id"`

	// TeamID is the team this remediation belongs to (for team scope).
	TeamID string `json:"team_id,omitempty"`

	// ProjectPath is the project this remediation belongs to (for project scope).
	ProjectPath string `json:"project_path,omitempty"`

	// SessionID is the session this remediation was extracted from.
	SessionID string `json:"session_id,omitempty"`

	// CreatedAt is when this remediation was created.
	CreatedAt time.Time `json:"created_at"`

	// UpdatedAt is when this remediation was last updated.
	UpdatedAt time.Time `json:"updated_at"`

	// Vector is the embedding vector for semantic search.
	Vector []float32 `json:"-"`
}

type Scope string

type ScoredRemediation struct {
	Remediation
	Score float64 `json:"score"`
}

type SearchRequest struct {
	// Query is the error message or description to search for.
	Query string

	// Vector is an optional pre-computed embedding vector.
	Vector []float32

	// Limit is the maximum number of results to return.
	Limit int

	// MinConfidence is the minimum confidence threshold.
	MinConfidence float64

	// Category filters by error category (optional).
	Category ErrorCategory

	// Scope filters by scope (optional, searches all if empty).
	Scope Scope

	// TenantID is required for multi-tenant isolation.
	TenantID string

	// TeamID filters to a specific team (optional).
	TeamID string

	// ProjectPath filters to a specific project (optional).
	ProjectPath string

	// Tags filters by tags (optional, any match).
	Tags []string

	// IncludeHierarchy includes parent scopes in search.
	// If searching project scope, also searches team and org.
	IncludeHierarchy bool
}

type Service interface {
	// Search finds remediations by semantic similarity to error message/pattern.
	Search(ctx context.Context, req *SearchRequest) ([]*ScoredRemediation, error)

	// Record creates a new remediation.
	Record(ctx context.Context, req *RecordRequest) (*Remediation, error)

	// Get retrieves a remediation by ID.
	Get(ctx context.Context, tenantID, remediationID string) (*Remediation, error)

	// Feedback records feedback on a remediation, adjusting confidence.
	Feedback(ctx context.Context, req *FeedbackRequest) error

	// Delete removes a remediation.
	Delete(ctx context.Context, tenantID, remediationID string) error

	// Close closes the service.
	Close() error
}