admin/MonkeyCode
1
0
Fork 0
mirror of https://github.com/chaitin/MonkeyCode.git synced 2026-02-04 15:53:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #112 from yokowu/fix-linecount

Browse Source 
fix: 修复行数计算问题
This commit is contained in:
Yoko
2025-07-18 22:53:58 +08:00
committed by GitHub
parent 160ebfca40 18ac8e674f
commit f95f62f604
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
backend/internal/proxy/repo/proxy.go
View File

@@ -233,25 +233,25 @@ func (r *ProxyRepo) Report(ctx context.Context, req *domain.ReportReq) error {
}
func (r *ProxyRepo) handleFileWritten(ctx context.Context, tx *db.Tx, rc *db.Task, req *domain.ReportReq) error {
lines := 0
lineCount := 0
switch req.Tool {
case "appliedDiff", "editedExistingFile", "insertContent":
if strings.Contains(req.Content, "<<<<<<<") {
lines := diff.ParseConflictsAndCountLines(req.Content)
for _, line := range lines {
rc.CodeLines += int64(line)
lineCount += line
}
} else {
rc.CodeLines = int64(strings.Count(req.Content, "\n"))
lineCount = strings.Count(req.Content, "\n")
}
case "newFileCreated":
rc.CodeLines = int64(strings.Count(req.Content, "\n"))
lineCount = strings.Count(req.Content, "\n")
}
if lines > 0 {
if lineCount > 0 {
if err := tx.Task.
UpdateOneID(rc.ID).
AddCodeLines(int64(lines)).
AddCodeLines(int64(lineCount)).
SetIsAccept(true).
Exec(ctx); err != nil {
return err
@@ -264,7 +264,7 @@ func (r *ProxyRepo) handleFileWritten(ctx context.Context, tx *db.Tx, rc *db.Tas
SetRole(consts.ChatRoleSystem).
SetPrompt("写入文件").
SetCompletion("").
SetCodeLines(int64(lines)).
SetCodeLines(int64(lineCount)).
SetCode(req.Content).
SetOutputTokens(0).
Save(ctx); err != nil {