@@ -44,30 +44,31 @@ public struct LiveServerMessage: Sendable {
44
44
45
45
@available ( iOS 15 . 0 , macOS 12 . 0 , macCatalyst 15 . 0 , tvOS 15 . 0 , watchOS 8 . 0 , * )
46
46
extension LiveServerMessage {
47
- static func tryFrom ( _ serverMessage: BidiGenerateContentServerMessage ) -> Self ? {
48
- guard let messageType = LiveServerMessage . MessageType. tryFrom ( serverMessage. messageType) else {
47
+ init ? ( from serverMessage: BidiGenerateContentServerMessage ) {
48
+ guard let messageType = LiveServerMessage . MessageType ( from : serverMessage. messageType) else {
49
49
return nil
50
50
}
51
51
52
- return LiveServerMessage ( serverMessage: serverMessage, messageType: messageType)
52
+ self . serverMessage = serverMessage
53
+ self . messageType = messageType
53
54
}
54
55
}
55
56
56
57
@available ( iOS 15 . 0 , macOS 12 . 0 , macCatalyst 15 . 0 , tvOS 15 . 0 , watchOS 8 . 0 , * )
57
58
extension LiveServerMessage . MessageType {
58
- static func tryFrom ( _ serverMessage: BidiGenerateContentServerMessage . MessageType ) -> Self ? {
59
- return switch serverMessage {
59
+ init ? ( from serverMessage: BidiGenerateContentServerMessage . MessageType ) {
60
+ switch serverMessage {
60
61
case . setupComplete:
61
62
// this is handled internally, and should not be surfaced to users
62
- nil
63
+ return nil
63
64
case let . serverContent( msg) :
64
- . content( LiveServerContent ( msg) )
65
+ self = . content( LiveServerContent ( msg) )
65
66
case let . toolCall( msg) :
66
- . toolCall( LiveServerToolCall ( msg) )
67
+ self = . toolCall( LiveServerToolCall ( msg) )
67
68
case let . toolCallCancellation( msg) :
68
- . toolCallCancellation( LiveServerToolCallCancellation ( msg) )
69
+ self = . toolCallCancellation( LiveServerToolCallCancellation ( msg) )
69
70
case let . goAway( msg) :
70
- . goAway( LiveServerGoAway ( msg) )
71
+ self = . goAway( LiveServerGoAway ( msg) )
71
72
}
72
73
}
73
74
}
0 commit comments