X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/6b21fcf78372995c6d8130c63f2f8aa2609bf7fe..7594b9938b7343717e7a57aa7aede5d40d547483:/model_test.go

diff --git a/model_test.go b/model_test.go
index 84b767e..16946a0 100644
--- a/model_test.go
+++ b/model_test.go
@@ -145,10 +145,12 @@ func TestClientRecordsEachCall(t *testing.T) {
 	client := newTestModelClient(t, server)
 	request := modelTestRequest()
 	request.Node = &NodeKey{2, 5}
+	request.Agent = "GreenOtter"
 	_, err := client.Create(t.Context(), request)
 	require.NoError(t, err)
 	require.NotContains(t, server.Call(0).body, "node")
 	require.NotContains(t, string(server.Call(0).raw), "[2,5]")
+	require.NotContains(t, string(server.Call(0).raw), "GreenOtter")
 
 	_, err = client.Create(t.Context(), modelTestRequest())
 	require.ErrorContains(t, err, "HTTP 429")
@@ -168,12 +170,14 @@ func TestClientRecordsEachCall(t *testing.T) {
 	require.Equal(t, 7, records[0].OutputTokens)
 	require.Equal(t, quota, records[0].Quota)
 	require.Equal(t, &[2]int{2, 5}, records[0].Node)
+	require.Equal(t, "GreenOtter", records[0].Agent)
 	require.Equal(t, len("ok"), records[0].TextBytes)
 	require.Greater(t, records[0].Seconds, 0.0)
 	require.Empty(t, records[0].Error)
 	require.NotEmpty(t, records[0].Date)
 
 	require.Nil(t, records[1].Node)
+	require.Empty(t, records[1].Agent)
 	require.Zero(t, records[1].TextBytes)
 
 	require.Equal(t, http.StatusTooManyRequests, records[1].Status)