add optional id param in customContent

alfianlosari · Jun 8, 2024 · c32325f · c32325f
1 parent 7196e50
commit c32325f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Sources/ChatGPTUI/Models/MessageRow.swift b/Sources/ChatGPTUI/Models/MessageRow.swift
@@ -33,7 +33,7 @@ public struct MessageRow<CustomContent: View>: Identifiable {
 public enum MessageRowType<CustomContent: View> {
     case attributed(AttributedOutput)
     case rawText(String)
-    case customContent(() -> CustomContent)
+    case customContent((_ messageRowId: UUID?) -> CustomContent)
 
     public var text: String {
         switch self {

diff --git a/Sources/ChatGPTUI/Views/MessageRowView.swift b/Sources/ChatGPTUI/Views/MessageRowView.swift
@@ -68,7 +68,7 @@ public struct MessageRowView<CustomContent: View>: View {
                 }
 
             case .customContent(let customViewProvider):
-                customViewProvider()
+                customViewProvider(message.id)
             }
 
             if let error = responseError {

diff --git a/Sources/ChatGPTUI/Views/VoiceChatView.swift b/Sources/ChatGPTUI/Views/VoiceChatView.swift
@@ -32,7 +32,7 @@ public struct VoiceChatView<CustomContent: View>: View {
                         }
 
                     case .customContent(let customViewProvider):
-                        customViewProvider()
+                        customViewProvider(nil)
                     }
                 }
             }