From e146904ad18583462a49c857a7ddb3b921afc02f Mon Sep 17 00:00:00 2001 From: Daniel La Rocque Date: Tue, 20 Aug 2024 14:02:14 -0400 Subject: [PATCH] Upgrade vertex responses to v3 --- packages/vertexai/src/requests/stream-reader.test.ts | 2 +- scripts/update_vertexai_responses.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/packages/vertexai/src/requests/stream-reader.test.ts b/packages/vertexai/src/requests/stream-reader.test.ts index d390666f4b0..eea72f9c7a6 100644 --- a/packages/vertexai/src/requests/stream-reader.test.ts +++ b/packages/vertexai/src/requests/stream-reader.test.ts @@ -210,7 +210,7 @@ describe('processStream', () => { expect(aggregatedResponse.text()).to.include('Quantum mechanics is'); expect( aggregatedResponse.candidates?.[0].citationMetadata?.citations.length - ).to.equal(2); + ).to.equal(3); let foundCitationMetadata = false; for await (const response of result.stream) { expect(response.text()).to.not.be.empty; diff --git a/scripts/update_vertexai_responses.sh b/scripts/update_vertexai_responses.sh index 568612e340d..27c32a09ead 100755 --- a/scripts/update_vertexai_responses.sh +++ b/scripts/update_vertexai_responses.sh @@ -17,7 +17,7 @@ # This script replaces mock response files for Vertex AI unit tests with a fresh # clone of the shared repository of Vertex AI test data. -RESPONSES_VERSION='v2.*' # The major version of mock responses to use +RESPONSES_VERSION='v3.*' # The major version of mock responses to use REPO_NAME="vertexai-sdk-test-data" REPO_LINK="https://github.com/FirebaseExtended/$REPO_NAME.git"