Add the close method

Author: daymxn

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

@@ -125,6 +125,20 @@ actor LiveSessionService {
     }
   }
 
+  /// Cancel any running tasks and close the websocket.
+  ///
+  /// This method is idempotent; if it's already ran once, it will effectively be a no-op.
+  public func close() {
+    setupTask.cancel()
+    responsesTask?.cancel()
+    messageQueueTask?.cancel()
+    webSocket?.disconnect()
+
+    webSocket = nil
+    responsesTask = nil
+    messageQueueTask = nil
+  }
+
   /// Start a fresh websocket to the backend, and listen for responses.
   ///
   /// Will hold off on sending any messages until the server sends a setupComplete mesage.
@@ -269,20 +283,6 @@ actor LiveSessionService {
     }
   }
 
-  /// Cancel any running tasks and close the websocket.
-  ///
-  /// This method is idempotent; if it's already ran once, it will effectively be a no-op.
-  private func close() {
-    setupTask.cancel()
-    responsesTask?.cancel()
-    messageQueueTask?.cancel()
-    webSocket?.disconnect()
-
-    webSocket = nil
-    responsesTask = nil
-    messageQueueTask = nil
-  }
-
   /// Creates a websocket pointing to the backend.
   ///
   /// Will apply the required app check and auth headers, as the backend expects them.

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

@@ -126,4 +126,15 @@ public final class LiveSession: Sendable {
                           turnComplete: Bool? = nil) async {
     await sendContent([ModelContent(parts: parts)], turnComplete: turnComplete)
   }
+
+  /// Permanently stop the conversation with the model, and close the connection to the server
+  ///
+  /// This method will be called automatically when the ``LiveSession`` is deinitialized, but this method
+  /// can be called manually to explicitly end the session.
+  ///
+  /// Attempting to receive content from a closed session will cause a ``LiveSessionUnexpectedClosureError`` error
+  /// to be thrown.
+  public func close() async {
+    await service.close()
+  }
 }