feat(ai): Live API

FirebaseAI/Sources/Types/Internal/Live/AsyncWebSocket.swift

@@ -23,7 +23,7 @@ private import FirebaseCoreInternal
 /// Also surfaces a more fine-grained ``WebSocketClosedError`` for when the web socket is closed.
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, *)
 @available(watchOS, unavailable)
-final class AsyncWebSocket: NSObject, @unchecked Sendable, URLSessionWebSocketDelegate {
+final class AsyncWebSocket: Sendable {
   private let webSocketTask: URLSessionWebSocketTask
   private let stream: AsyncThrowingStream<URLSessionWebSocketTask.Message, Error>
   private let continuation: AsyncThrowingStream<URLSessionWebSocketTask.Message, Error>.Continuation
@@ -107,13 +107,6 @@ final class AsyncWebSocket: NSObject, @unchecked Sendable, URLSessionWebSocketDe
       isFinished = true
     }
   }
-
-  func urlSession(_ session: URLSession,
-                  webSocketTask: URLSessionWebSocketTask,
-                  didCloseWith closeCode: URLSessionWebSocketTask.CloseCode,
-                  reason: Data?) {
-    close(code: closeCode, reason: reason)
-  }
 }
 
 private extension URLSessionWebSocketTask {

FirebaseAI/Sources/Types/Internal/Live/LiveSessionService.swift

@@ -308,11 +308,12 @@ actor LiveSessionService {
   ///
   /// Will apply the required app check and auth headers, as the backend expects them.
   private nonisolated func createWebsocket() async throws -> AsyncWebSocket {
+    let host = apiConfig.service.endpoint.rawValue.withoutPrefix("https://")
     let urlString = switch apiConfig.service {
     case .vertexAI:
-      "wss://firebasevertexai.googleapis.com/ws/google.firebase.vertexai.v1beta.LlmBidiService/BidiGenerateContent/locations/us-central1"
+      "wss://\(host)/ws/google.firebase.vertexai.v1beta.LlmBidiService/BidiGenerateContent/locations/us-central1"
     case .googleAI:
-      "wss://firebasevertexai.googleapis.com/ws/google.firebase.vertexai.v1beta.GenerativeService/BidiGenerateContent"
+      "wss://\(host)/ws/google.firebase.vertexai.v1beta.GenerativeService/BidiGenerateContent"
     }
     guard let url = URL(string: urlString) else {
       throw NSError(
@@ -378,3 +379,16 @@ private extension Data {
     }
   }
 }
+
+private extension String {
+  /// Create a new string with the given prefix removed, if it's present.
+  ///
+  /// If the prefix isn't present, this string will be returned instead.
+  func withoutPrefix(_ prefix: String) -> String {
+    if let index = range(of: prefix, options: .anchored) {
+      return String(self[index.upperBound...])
+    } else {
+      return self
+    }
+  }
+}

FirebaseAI/Sources/Types/Public/Live/LiveServerMessage.swift

@@ -19,7 +19,7 @@ public struct LiveServerMessage: Sendable {
   let serverMessage: BidiGenerateContentServerMessage
 
   /// The type of message sent from the server.
-  public enum Event: Sendable {
+  public enum Payload: Sendable {
     /// Content generated by the model in response to client messages.
     case content(LiveServerContent)
 
@@ -35,7 +35,7 @@ public struct LiveServerMessage: Sendable {
   }
 
   /// The message sent from the server.
-  public let payload: Event
+  public let payload: Payload
 
   /// Metadata on the usage of the cached content.
   public var usageMetadata: GenerateContentResponse.UsageMetadata? { serverMessage.usageMetadata }
@@ -47,7 +47,7 @@ public struct LiveServerMessage: Sendable {
 @available(watchOS, unavailable)
 extension LiveServerMessage {
   init?(from serverMessage: BidiGenerateContentServerMessage) {
-    guard let payload = LiveServerMessage.Event(from: serverMessage.messageType) else {
+    guard let payload = LiveServerMessage.Payload(from: serverMessage.messageType) else {
       return nil
     }
 
@@ -58,7 +58,7 @@ extension LiveServerMessage {
 
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, *)
 @available(watchOS, unavailable)
-extension LiveServerMessage.Event {
+extension LiveServerMessage.Payload {
   init?(from serverMessage: BidiGenerateContentServerMessage.MessageType) {
     switch serverMessage {
     case .setupComplete: