snapshot of https://github.com/livekit/livekit.git @ 08c5433449c106943231037d0b892895f6e92d3b (2026-09-15)

pkg/service/agent_dispatch_service.go

@@ -0,0 +1,131 @@
+// Copyright 2023 LiveKit, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package service
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/livekit/livekit-server/pkg/config"
+	"github.com/livekit/livekit-server/pkg/routing"
+	"github.com/livekit/protocol/agent"
+	"github.com/livekit/protocol/livekit"
+	"github.com/livekit/protocol/logger"
+	"github.com/livekit/protocol/rpc"
+	"github.com/livekit/protocol/utils"
+	"github.com/livekit/protocol/utils/guid"
+	"github.com/livekit/psrpc"
+)
+
+type AgentDispatchService struct {
+	limitConf           config.LimitConfig
+	agentDispatchClient rpc.TypedAgentDispatchInternalClient
+	topicFormatter      rpc.TopicFormatter
+	roomAllocator       RoomAllocator
+	router              routing.MessageRouter
+}
+
+func NewAgentDispatchService(
+	limitConf config.LimitConfig,
+	agentDispatchClient rpc.TypedAgentDispatchInternalClient,
+	topicFormatter rpc.TopicFormatter,
+	roomAllocator RoomAllocator,
+	router routing.MessageRouter,
+) *AgentDispatchService {
+	return &AgentDispatchService{
+		limitConf:           limitConf,
+		agentDispatchClient: agentDispatchClient,
+		topicFormatter:      topicFormatter,
+		roomAllocator:       roomAllocator,
+		router:              router,
+	}
+}
+
+func (ag *AgentDispatchService) CreateDispatch(ctx context.Context, req *livekit.CreateAgentDispatchRequest) (*livekit.AgentDispatch, error) {
+	AppendLogFields(ctx, "room", req.Room, "request", logger.Proto(redactCreateAgentDispatchRequest(req)))
+	err := EnsureAdminPermission(ctx, livekit.RoomName(req.Room))
+	if err != nil {
+		return nil, twirpAuthError(err)
+	}
+
+	if err := agent.ValidateDeployment(req.GetDeployment()); err != nil {
+		return nil, psrpc.NewError(psrpc.InvalidArgument, err)
+	}
+
+	if !ag.limitConf.CheckMetadataSize(req.Metadata) {
+		return nil, ErrMetadataExceedsLimits
+	}
+	if !ag.limitConf.CheckAttributesSize(req.Attributes) {
+		return nil, ErrAttributeExceedsLimits
+	}
+
+	if ag.roomAllocator.AutoCreateEnabled(ctx) {
+		err := ag.roomAllocator.SelectRoomNode(ctx, livekit.RoomName(req.Room), "")
+		if err != nil {
+			return nil, err
+		}
+
+		_, err = ag.router.CreateRoom(ctx, &livekit.CreateRoomRequest{Name: req.Room})
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	dispatch := &livekit.AgentDispatch{
+		Id:            DeterministicID(guid.AgentDispatchPrefix, RequestID(ctx)),
+		AgentName:     req.AgentName,
+		Room:          req.Room,
+		Metadata:      req.Metadata,
+		RestartPolicy: req.RestartPolicy,
+		Deployment:    req.Deployment,
+		Attributes:    req.Attributes,
+	}
+	return ag.agentDispatchClient.CreateDispatch(ctx, ag.topicFormatter.RoomTopic(ctx, livekit.RoomName(req.Room)), dispatch)
+}
+
+func (ag *AgentDispatchService) DeleteDispatch(ctx context.Context, req *livekit.DeleteAgentDispatchRequest) (*livekit.AgentDispatch, error) {
+	AppendLogFields(ctx, "room", req.Room, "request", logger.Proto(req))
+	err := EnsureAdminPermission(ctx, livekit.RoomName(req.Room))
+	if err != nil {
+		return nil, twirpAuthError(err)
+	}
+
+	return ag.agentDispatchClient.DeleteDispatch(ctx, ag.topicFormatter.RoomTopic(ctx, livekit.RoomName(req.Room)), req)
+}
+
+func (ag *AgentDispatchService) ListDispatch(ctx context.Context, req *livekit.ListAgentDispatchRequest) (*livekit.ListAgentDispatchResponse, error) {
+	AppendLogFields(ctx, "room", req.Room, "request", logger.Proto(req))
+	err := EnsureAdminPermission(ctx, livekit.RoomName(req.Room))
+	if err != nil {
+		return nil, twirpAuthError(err)
+	}
+
+	return ag.agentDispatchClient.ListDispatch(ctx, ag.topicFormatter.RoomTopic(ctx, livekit.RoomName(req.Room)), req)
+}
+
+func redactCreateAgentDispatchRequest(req *livekit.CreateAgentDispatchRequest) *livekit.CreateAgentDispatchRequest {
+	if req.Metadata == "" {
+		return req
+	}
+
+	clone := utils.CloneProto(req)
+
+	// replace with size of metadata to provide visibility on request size
+	if clone.Metadata != "" {
+		clone.Metadata = fmt.Sprintf("__size: %d", len(clone.Metadata))
+	}
+
+	return clone
+}