unclesam
26b0342934
- 在 LLM 客户端中捕获流式响应的 Token 使用量 - 内置多模型官方定价表，支持自定义输入/输出单价覆盖 - 新增 `--price-input` 和 `--price-output` 配置项 - 支持通过 `.airignore` 文件自定义排除评审文件 - 更新中英日文文档，补充费用估算与忽略规则说明
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package reviewer
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/unclesam-ly/git-air/internal/llm"
|
|
)
|
|
|
|
type Reviewer struct {
|
|
client *llm.Client
|
|
repoRoot string
|
|
customPrompt string
|
|
}
|
|
|
|
// NewReviewer 初始化评审引擎
|
|
// customPrompt 可传入用户自定义配置的 Prompt，传空字符串则使用默认 Prompt
|
|
func NewReviewer(client *llm.Client, repoRoot string, customPrompt string) *Reviewer {
|
|
return &Reviewer{
|
|
client: client,
|
|
repoRoot: repoRoot,
|
|
customPrompt: customPrompt,
|
|
}
|
|
}
|
|
|
|
// Execute 执行代码评审，返回 (result, usage, error)
|
|
func (r *Reviewer) Execute(ctx context.Context, diff string, stream bool, onChunk func(chunk string)) (string, *llm.Usage, error) {
|
|
if strings.TrimSpace(diff) == "" {
|
|
return "", nil, fmt.Errorf("diff 内容为空，无需评审")
|
|
}
|
|
|
|
// 动态组装最终 System Prompt
|
|
systemPrompt := BuildSystemPrompt(r.customPrompt, r.repoRoot)
|
|
if stream && onChunk != nil {
|
|
usage, err := r.client.ReviewStream(ctx, systemPrompt, diff, onChunk)
|
|
return "", usage, err
|
|
}
|
|
|
|
result, err := r.client.Review(ctx, systemPrompt, diff)
|
|
return result, nil, err
|
|
}
|