feat: allow Agents to pass through initial attributes (#852)

* feat: allow Agents to pass through initial attributes * changeset
livekit · Oct 15, 2024 · 689a0fe · 689a0fe
1 parent b754791
commit 689a0fe
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/.changeset/afraid-fans-care.md b/.changeset/afraid-fans-care.md
@@ -0,0 +1,5 @@
+---
+"github.com/livekit/protocol": minor
+---
+
+allow Agents to pass through initial attributes
diff --git a/agent/token.go b/agent/token.go
@@ -7,7 +7,11 @@ import (
 	"github.com/livekit/protocol/livekit"
 )
 
-func BuildAgentToken(apiKey, secret, roomName, participantIdentity, participantName, participantMetadata string, permissions *livekit.ParticipantPermission) (string, error) {
+func BuildAgentToken(
+	apiKey, secret, roomName, participantIdentity, participantName, participantMetadata string,
+	participantAttributes map[string]string,
+	permissions *livekit.ParticipantPermission,
+) (string, error) {
 	grant := &auth.VideoGrant{
 		RoomJoin:             true,
 		Agent:                true,
@@ -26,7 +30,8 @@ func BuildAgentToken(apiKey, secret, roomName, participantIdentity, participantN
 		SetName(participantName).
 		SetKind(livekit.ParticipantInfo_AGENT).
 		SetValidFor(1 * time.Hour).
-		SetMetadata(participantMetadata)
+		SetMetadata(participantMetadata).
+		SetAttributes(participantAttributes)
 
 	return at.ToJWT()
 }