Replace published on behalf agents

Author: pblazej

Sources/LiveKit/Agent/Participant+Agent.swift

@@ -16,8 +16,6 @@
 
 import Foundation
 
-let publishOnBehalfAttributeKey = "lk.publish_on_behalf"
-
 public extension Participant {
     @objc
     var isAgent: Bool {
@@ -27,13 +25,6 @@ public extension Participant {
         }
     }
 
-    /// Avatar workers publish audio and video on behalf of the agent.
-    /// They should not be used for other purposes or displayed to the user.
-    @objc
-    var isAvatarWorker: Bool {
-        attributes.keys.contains(publishOnBehalfAttributeKey)
-    }
-
     @objc
     var agentState: AgentState {
         guard isAgent else { return .unknown }

Sources/LiveKit/Agent/Room+Agent.swift

@@ -16,24 +16,18 @@
 
 import Foundation
 
+let publishOnBehalfAttributeKey = "lk.publish_on_behalf"
+
 public extension Room {
-    /// All agent participants in the Room, excluding avatar workers
+    /// All agent participants in the Room
     var agentParticipants: [Participant.Identity: RemoteParticipant] {
-        remoteParticipants.filter { $0.value.isAgent && !$0.value.isAvatarWorker }
+        // Filter out agents that are replaced by another agent e.g. avatar worker
+        let onBehalf = Set(remoteParticipants.map(\.value.attributes[publishOnBehalfAttributeKey]).compactMap { $0 })
+        return remoteParticipants.filter { $0.value.isAgent && !onBehalf.contains($0.key.stringValue) }
     }
 
-    /// The first agent participant in the Room, excluding avatar workers
-    var agentParticipant: RemoteParticipant? {
+    /// The first agent participant in the Room
+    var agentParticipant: Participant? {
         agentParticipants.values.first
     }
-
-    /// All avatar workers in the Room
-    var avatarWorkers: [Participant.Identity: RemoteParticipant] {
-        remoteParticipants.filter { $0.value.isAgent && $0.value.isAvatarWorker }
-    }
-
-    /// The first avatar worker in the Room
-    var avatarWorker: RemoteParticipant? {
-        avatarWorkers.values.first
-    }
 }