Skip to content

Commit b67f394

Browse files
committed
test: update binding mock response from API
1 parent 0c6b240 commit b67f394

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

llm/binding_test.go

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,13 @@ func TestExplainWithOptions(t *testing.T) {
3030

3131
explainResponseJSON := explainResponse{
3232
Status: completeStatus,
33-
Explanation: map[string]string{"explanation1": "This is the first explanation"},
33+
Explanation: map[string]string{
34+
"explanation1": "This is the first explanation",
35+
"explanation2": "this is the second explanation",
36+
"explanation3": "this is the third explanation",
37+
"explanation4": "this is the fourth explanation",
38+
"explanation5": "this is the fifth explanation",
39+
},
3440
}
3541

3642
expectedResponseBody, err := json.Marshal(explainResponseJSON)
@@ -49,7 +55,7 @@ func TestExplainWithOptions(t *testing.T) {
4955
err = json.Unmarshal(expectedResponseBody, &exptectedExplanationsResponse)
5056
assert.NoError(t, err)
5157
expectedResExplanations := exptectedExplanationsResponse.Explanation
52-
assert.Equal(t, expectedResExplanations["explanation1"], explanation[testDiff])
58+
assert.Equal(t, expectedResExplanations["explanation1"], explanation[0])
5359
})
5460

5561
t.Run("runExplain error", func(t *testing.T) {

0 commit comments

Comments
 (0)