add AgentSimulationClient, export CreateSourceTarball and MultipartUpload

Adds new client for the agent simulation API. Exports CreateSourceTarball
and MultipartUpload for use by external callers (livekit-cli).

Author: theomonnom

agent_simulation_client.go

@@ -0,0 +1,77 @@
+// Copyright 2025 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 lksdk
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/twitchtv/twirp"
+
+	"github.com/livekit/protocol/livekit"
+)
+
+type AgentSimulationClient struct {
+	simulationClient livekit.AgentSimulation
+	authBase
+}
+
+func NewAgentSimulationClient(url string, apiKey string, apiSecret string, opts ...twirp.ClientOption) *AgentSimulationClient {
+	client := livekit.NewAgentSimulationProtobufClient(url, &http.Client{}, opts...)
+	return &AgentSimulationClient{
+		simulationClient: client,
+		authBase:         authBase{apiKey, apiSecret},
+	}
+}
+
+func (c *AgentSimulationClient) CreateSimulationRun(ctx context.Context, req *livekit.SimulationRun_Create_Request) (*livekit.SimulationRun_Create_Response, error) {
+	ctx, err := c.withAuth(ctx, withVideoGrant{})
+	if err != nil {
+		return nil, err
+	}
+	return c.simulationClient.CreateSimulationRun(ctx, req)
+}
+
+func (c *AgentSimulationClient) ConfirmSimulationSourceUpload(ctx context.Context, req *livekit.SimulationRun_ConfirmSourceUpload_Request) (*livekit.SimulationRun_ConfirmSourceUpload_Response, error) {
+	ctx, err := c.withAuth(ctx, withVideoGrant{})
+	if err != nil {
+		return nil, err
+	}
+	return c.simulationClient.ConfirmSimulationSourceUpload(ctx, req)
+}
+
+func (c *AgentSimulationClient) GetSimulationRun(ctx context.Context, req *livekit.SimulationRun_Get_Request) (*livekit.SimulationRun_Get_Response, error) {
+	ctx, err := c.withAuth(ctx, withVideoGrant{})
+	if err != nil {
+		return nil, err
+	}
+	return c.simulationClient.GetSimulationRun(ctx, req)
+}
+
+func (c *AgentSimulationClient) ListSimulationRuns(ctx context.Context, req *livekit.SimulationRun_List_Request) (*livekit.SimulationRun_List_Response, error) {
+	ctx, err := c.withAuth(ctx, withVideoGrant{})
+	if err != nil {
+		return nil, err
+	}
+	return c.simulationClient.ListSimulationRuns(ctx, req)
+}
+
+func (c *AgentSimulationClient) CancelSimulationRun(ctx context.Context, req *livekit.SimulationRun_Cancel_Request) (*livekit.SimulationRun_Cancel_Response, error) {
+	ctx, err := c.withAuth(ctx, withVideoGrant{})
+	if err != nil {
+		return nil, err
+	}
+	return c.simulationClient.CancelSimulationRun(ctx, req)
+}

pkg/cloudagents/tar.go

@@ -28,7 +28,7 @@ import (
 	"github.com/moby/patternmatcher"
 )
 
-func createSourceTarball(
+func CreateSourceTarball(
 	directory fs.FS,
 	excludeFiles []string,
 	w io.Writer,

pkg/cloudagents/upload.go

@@ -55,11 +55,11 @@ func uploadSource(
 	excludeFiles []string,
 ) error {
 	var buf bytes.Buffer
-	if err := createSourceTarball(directory, excludeFiles, &buf); err != nil {
+	if err := CreateSourceTarball(directory, excludeFiles, &buf); err != nil {
 		return fmt.Errorf("failed to sanitize source: %w", err)
 	}
 	if presignedPostRequest != nil {
-		if err := multipartUpload(presignedPostRequest.Url, presignedPostRequest.Values, &buf); err != nil {
+		if err := MultipartUpload(presignedPostRequest.Url, presignedPostRequest.Values, &buf); err != nil {
 			return fmt.Errorf("multipart upload failed: %w", err)
 		}
 	} else {
@@ -89,7 +89,7 @@ func upload(presignedUrl string, buf *bytes.Buffer) error {
 	return nil
 }
 
-func multipartUpload(presignedURL string, fields map[string]string, buf *bytes.Buffer) error {
+func MultipartUpload(presignedURL string, fields map[string]string, buf *bytes.Buffer) error {
 	var b bytes.Buffer
 	w := multipart.NewWriter(&b)
 	fileName, ok := fields["key"]