Skip to content

Commit ce1817a

Browse files
committed
fix: omit file path from diff
1 parent 29b60e9 commit ce1817a

File tree

2 files changed

+16
-4
lines changed

2 files changed

+16
-4
lines changed

llm/api_client.go

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ import (
88
"io"
99
"net/http"
1010
"net/url"
11+
"strings"
1112
)
1213

1314
var (
@@ -96,17 +97,28 @@ func (d *DeepCodeLLMBindingImpl) explainRequestBody(options *ExplainOptions) ([]
9697
} else {
9798
requestBody, marshalErr = json.Marshal(explainFixRequest{
9899
RuleId: options.RuleKey,
99-
Diffs: encodeDiffs(options.Diffs),
100+
Diffs: prepareDiffs(options.Diffs),
100101
ExplanationLength: SHORT,
101102
})
102103
logger.Debug().Msg("payload for FixExplanation")
103104
}
104105
return requestBody, marshalErr
105106
}
106107

107-
func encodeDiffs(diffs []string) []string {
108-
var encodedDiffs []string
108+
func prepareDiffs(diffs []string) []string {
109+
cleanedDiffs := make([]string, 0, len(diffs))
109110
for _, diff := range diffs {
111+
diffLines := strings.Split(diff, "\n")
112+
cleanedLines := ""
113+
for i, line := range diffLines {
114+
if !strings.HasPrefix(line, "---") && !strings.HasPrefix(line, "+++") && i > 1 {
115+
cleanedLines += line + "\n"
116+
}
117+
}
118+
cleanedDiffs = append(cleanedDiffs, cleanedLines)
119+
}
120+
var encodedDiffs []string
121+
for _, diff := range cleanedDiffs {
110122
encodedDiffs = append(encodedDiffs, base64.StdEncoding.EncodeToString([]byte(diff)))
111123
}
112124
return encodedDiffs

llm/api_client_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ func TestDeepcodeLLMBinding_explainRequestBody(t *testing.T) {
151151

152152
assert.NotNil(t, request)
153153
assert.Equal(t, "test-rule-key", request.RuleId)
154-
expectedEncodedDiffs := encodeDiffs([]string{"test-Diffs"})
154+
expectedEncodedDiffs := prepareDiffs([]string{"test-Diffs"})
155155
assert.Equal(t, expectedEncodedDiffs, request.Diffs)
156156
assert.Equal(t, SHORT, request.ExplanationLength)
157157
})

0 commit comments

Comments
 (0)