Function TestStopReasonMapping

pkg/runtime/sampling_test.go:194–214

(t *testing.T)

}

func TestStopReasonMapping(t *testing.T) {
	t.Parallel()

	tests := []struct {
		in   chat.FinishReason
		want string
	}{
		{chat.FinishReasonStop, "endTurn"},
		{chat.FinishReasonLength, "maxTokens"},
		{chat.FinishReasonToolCalls, "toolUse"},
		{chat.FinishReasonNull, "endTurn"},
		{"", "endTurn"},
	}

	for _, tt := range tests {
		t.Run(string(tt.in), func(t *testing.T) {
			t.Parallel()
			assert.Equal(t, tt.want, stopReason(tt.in))
		})
	}
}

func TestDataURL(t *testing.T) {
	t.Parallel()