Skip to content

Commit 27ce3fa

Browse files
committed
Rename live server message type
1 parent adc35d2 commit 27ce3fa

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,7 @@ actor LiveSessionService {
222222
setupComplete.resume()
223223
}
224224
} else if let liveMessage = LiveServerMessage(from: response) {
225-
if case let .goingAwayNotice(message) = liveMessage.messageType {
225+
if case let .goingAwayNotice(message) = liveMessage.payload {
226226
// TODO: (b/444045023) When auto session resumption is enabled, call `connect` again
227227
AILog.debug(
228228
code: .liveSessionGoingAwaySoon,

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

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ public struct LiveServerMessage: Sendable {
1919
let serverMessage: BidiGenerateContentServerMessage
2020

2121
/// The type of message sent from the server.
22-
public enum MessageType: Sendable {
22+
public enum Event: Sendable {
2323
/// Content generated by the model in response to client messages.
2424
case content(LiveServerContent)
2525

@@ -34,8 +34,8 @@ public struct LiveServerMessage: Sendable {
3434
case goingAwayNotice(LiveServerGoingAwayNotice)
3535
}
3636

37-
/// The actual message sent from the server.
38-
public var messageType: MessageType
37+
/// The message sent from the server.
38+
public let payload: Event
3939

4040
/// Metadata on the usage of the cached content.
4141
public var usageMetadata: GenerateContentResponse.UsageMetadata? { serverMessage.usageMetadata }
@@ -47,18 +47,18 @@ public struct LiveServerMessage: Sendable {
4747
@available(watchOS, unavailable)
4848
extension LiveServerMessage {
4949
init?(from serverMessage: BidiGenerateContentServerMessage) {
50-
guard let messageType = LiveServerMessage.MessageType(from: serverMessage.messageType) else {
50+
guard let payload = LiveServerMessage.Event(from: serverMessage.messageType) else {
5151
return nil
5252
}
5353

5454
self.serverMessage = serverMessage
55-
self.messageType = messageType
55+
self.payload = payload
5656
}
5757
}
5858

5959
@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, *)
6060
@available(watchOS, unavailable)
61-
extension LiveServerMessage.MessageType {
61+
extension LiveServerMessage.Event {
6262
init?(from serverMessage: BidiGenerateContentServerMessage.MessageType) {
6363
switch serverMessage {
6464
case .setupComplete:

0 commit comments

Comments
 (0)