selfrag

type Critic interface {
	// NeedsRetrieval 判断是否需要检索
	// 返回是否需要检索以及置信度
	NeedsRetrieval(ctx context.Context, query string) (bool, float32, error)

	// IsRelevant 评估文档与查询的相关性
	// 返回是否相关以及相关性分数
	IsRelevant(ctx context.Context, query string, doc rag.Document) (bool, float32, error)

	// IsFaithful 验证回答是否忠实于来源文档
	// 返回是否忠实以及忠实度分数
	IsFaithful(ctx context.Context, response string, sources []rag.Document) (bool, float32, error)

	// IsComplete 检查回答是否完整回答了问题
	// 返回是否完整以及完整度分数
	IsComplete(ctx context.Context, query string, response string) (bool, float32, error)
}

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

type Option func(*SelfRAG)

type Response struct {
	// Content 生成的回答
	Content string `json:"content"`

	// Sources 使用的来源文档
	Sources []rag.Document `json:"sources,omitempty"`

	// NeedRetrieval 是否需要检索
	NeedRetrieval bool `json:"need_retrieval"`

	// RelevanceScores 各文档的相关性分数
	RelevanceScores map[string]float32 `json:"relevance_scores,omitempty"`

	// FaithfulnessScore 忠实度分数
	FaithfulnessScore float32 `json:"faithfulness_score"`

	// CompletenessScore 完整度分数
	CompletenessScore float32 `json:"completeness_score"`

	// Retries 重试次数
	Retries int `json:"retries"`
}

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