Upgrade vertex responses to v3 (#8445)

firebase · Aug 20, 2024 · 8a96f4f · 8a96f4f
1 parent 12ba46f
commit 8a96f4f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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"