rag

type BatchEvaluationResult struct {
	// Results 各项结果
	Results []*EvaluationResult `json:"results"`

	// AverageScores 平均得分
	AverageScores *EvaluationResult `json:"average_scores"`

	// TotalSamples 总样本数
	TotalSamples int `json:"total_samples"`

	// SuccessCount 成功数
	SuccessCount int `json:"success_count"`

	// FailureCount 失败数
	FailureCount int `json:"failure_count"`
}

type EvaluationInput struct {
	// Question 用户问题
	Question string `json:"question"`

	// Answer 生成的回答
	Answer string `json:"answer"`

	// Contexts 检索到的上下文
	Contexts []string `json:"contexts"`

	// GroundTruth 真实答案（可选，用于答案正确性评估）
	GroundTruth string `json:"ground_truth,omitempty"`

	// GroundTruthContexts 真实相关上下文（可选，用于召回评估）
	GroundTruthContexts []string `json:"ground_truth_contexts,omitempty"`
}

type EvaluationResult struct {
	// Faithfulness 忠实度得分 (0-1)
	Faithfulness float64 `json:"faithfulness"`

	// Relevancy 相关性得分 (0-1)
	Relevancy float64 `json:"relevancy"`

	// ContextPrecision 上下文精度 (0-1)
	ContextPrecision float64 `json:"context_precision"`

	// ContextRecall 上下文召回 (0-1)
	ContextRecall float64 `json:"context_recall"`

	// AnswerCorrectness 答案正确性 (0-1)
	AnswerCorrectness float64 `json:"answer_correctness,omitempty"`

	// Hallucination 幻觉得分 (0-1，越低越好)
	Hallucination float64 `json:"hallucination"`

	// OverallScore 综合得分 (0-1)
	OverallScore float64 `json:"overall_score"`

	// Details 详细信息
	Details map[string]any `json:"details,omitempty"`
}

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

type EvaluatorConfig struct {
	// EnableDetailedAnalysis 启用详细分析
	EnableDetailedAnalysis bool

	// ParallelEvaluation 并行评估
	ParallelEvaluation bool

	// Timeout 超时时间（秒）
	Timeout int
}

type LLMProvider interface {
	// Complete 执行补全
	Complete(ctx context.Context, prompt string) (string, error)
}

type MetricWeights struct {
	Faithfulness      float64 `json:"faithfulness"`
	Relevancy         float64 `json:"relevancy"`
	ContextPrecision  float64 `json:"context_precision"`
	ContextRecall     float64 `json:"context_recall"`
	AnswerCorrectness float64 `json:"answer_correctness"`
}