From 85f543fa656093d749c66588d29bbf56cd45a2f1 Mon Sep 17 00:00:00 2001 From: ripark Date: Sat, 6 Dec 2025 00:31:27 +0000 Subject: [PATCH] Token counts aren't contractual for our client, no need to test for it. --- sdk/ai/azopenai/client_chat_completions_test.go | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/sdk/ai/azopenai/client_chat_completions_test.go b/sdk/ai/azopenai/client_chat_completions_test.go index 51375c747382..d3c9585adfae 100644 --- a/sdk/ai/azopenai/client_chat_completions_test.go +++ b/sdk/ai/azopenai/client_chat_completions_test.go @@ -57,17 +57,10 @@ func TestClient_GetChatCompletions(t *testing.T) { require.NotEmpty(t, choice.Message.Content) require.Equal(t, "stop", choice.FinishReason) - require.Equal(t, openai.CompletionUsage{ - // these change depending on which model you use. These #'s work for gpt-4, which is - // what I'm using for these tests. - CompletionTokens: 29, - PromptTokens: 42, - TotalTokens: 71, - }, openai.CompletionUsage{ - CompletionTokens: resp.Usage.CompletionTokens, - PromptTokens: resp.Usage.PromptTokens, - TotalTokens: resp.Usage.TotalTokens, - }) + // let's just make sure that the #'s are filled out. + require.Greater(t, resp.Usage.CompletionTokens, int64(0)) + require.Greater(t, resp.Usage.PromptTokens, int64(0)) + require.Greater(t, resp.Usage.TotalTokens, int64(0)) } t.Run("AzureOpenAI", func(t *testing.T) {