Function TestOpenAIChatCompletions

internal/integrationtest/bridge_test.go:472–634

(t *testing.T)

}

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

	t.Run("single builtin tool", func(t *testing.T) {
		t.Parallel()

		cases := []struct {
			name                                      string
			streaming                                 bool
			expectedInputTokens, expectedOutputTokens int
			expectedToolCallID                        string
		}{
			{
				name:                 "streaming",
				streaming:            true,
				expectedInputTokens:  60,
				expectedOutputTokens: 15,
				expectedToolCallID:   "call_HjeqP7YeRkoNj0de9e3U4X4B",
			},
			{
				name:                 "non-streaming",
				streaming:            false,
				expectedInputTokens:  60,
				expectedOutputTokens: 15,
				expectedToolCallID:   "call_KjzAbhiZC6nk81tQzL7pwlpc",
			},
		}

		for _, tc := range cases {
			t.Run(tc.name, func(t *testing.T) {
				t.Parallel()

				ctx, cancel := context.WithTimeout(t.Context(), testutil.WaitLong)
				t.Cleanup(cancel)

				fix := fixtures.Parse(t, fixtures.OaiChatSingleBuiltinTool)
				upstream := newMockUpstream(ctx, t, newFixtureResponse(fix))

				bridgeServer := newBridgeTestServer(ctx, t, upstream.URL)

				// Make API call to aibridge for OpenAI /v1/chat/completions
				reqBody, err := sjson.SetBytes(fix.Request(), "stream", tc.streaming)
				require.NoError(t, err)
				resp, err := bridgeServer.makeRequest(t, http.MethodPost, pathOpenAIChatCompletions, reqBody)
				require.NoError(t, err)
				defer resp.Body.Close()
				require.Equal(t, http.StatusOK, resp.StatusCode)

				// Response-specific checks.
				if tc.streaming {
					sp := aibridge.NewSSEParser()
					require.NoError(t, sp.Parse(resp.Body))

					// OpenAI sends all events under the same type.
					messageEvents := sp.MessageEvents()
					assert.NotEmpty(t, messageEvents)

					// OpenAI streaming ends with [DONE]