From 636b37f5415f938a4e212504c89bc532500f0aac Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Tue, 2 Apr 2024 14:48:35 +0000
Subject: [PATCH] Update to prod endpoint in Vertex AI (#12657)

---
 FirebaseVertexAI/Sources/GenerativeAISwift.swift                | 2 +-
 .../Unit/GenerateContentResponses/unary-failure-api-key.json    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/FirebaseVertexAI/Sources/GenerativeAISwift.swift b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
index 0f0af2bc3e2..933a19c4759 100644
--- a/FirebaseVertexAI/Sources/GenerativeAISwift.swift
+++ b/FirebaseVertexAI/Sources/GenerativeAISwift.swift
@@ -21,5 +21,5 @@ import Foundation
 /// Constants associated with the GenerativeAISwift SDK
 @available(iOS 15.0, macOS 11.0, macCatalyst 15.0, *)
 public enum GenerativeAISwift {
-  static let baseURL = "https://staging-firebaseml.sandbox.googleapis.com"
+  static let baseURL = "https://firebaseml.googleapis.com"
 }
diff --git a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
index d3b5677366e..e9b472fd031 100644
--- a/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
+++ b/FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-failure-api-key.json
@@ -9,7 +9,7 @@
         "reason": "API_KEY_INVALID",
         "domain": "googleapis.com",
         "metadata": {
-          "service": "staging-firebaseml.sandbox.googleapis.com"
+          "service": "firebaseml.googleapis.com"
         }
       },
       {