first-cut adding agentName to lk commands

Author: shawnyang-coder

pkg/cloudagents/client.go

@@ -75,15 +75,17 @@ func New(opts ...ClientOption) (*Client, error) {
 // CreateAgent creates a new agent by building from source.
 func (c *Client) CreateAgent(
 	ctx context.Context,
+	agentName string,
 	source fs.FS,
 	secrets []*lkproto.AgentSecret,
 	regions []string,
 	excludeFiles []string,
 	buildLogStreamWriter io.Writer,
 ) (*lkproto.CreateAgentResponse, error) {
 	resp, err := c.AgentClient.CreateAgent(ctx, &lkproto.CreateAgentRequest{
-		Secrets: secrets,
-		Regions: regions,
+		AgentName: agentName,
+		Secrets:   secrets,
+		Regions:   regions,
 	})
 	if err != nil {
 		return nil, err
@@ -121,14 +123,16 @@ func (c *Client) CreateAgentV2(
 func (c *Client) DeployAgent(
 	ctx context.Context,
 	agentID string,
+	agentName string,
 	source fs.FS,
 	secrets []*lkproto.AgentSecret,
 	excludeFiles []string,
 	buildLogStreamWriter io.Writer,
 ) error {
 	resp, err := c.AgentClient.DeployAgent(ctx, &lkproto.DeployAgentRequest{
-		AgentId: agentID,
-		Secrets: secrets,
+		AgentId:   agentID,
+		AgentName: agentName,
+		Secrets:   secrets,
 	})
 	if err != nil {
 		return err
@@ -164,10 +168,11 @@ func (c *Client) DeployAgentV2(
 
 // RegisterAgent creates an agent record without uploading source or triggering a build.
 // Use this when you intend to push a prebuilt image immediately after via GetPushTarget.
-func (c *Client) RegisterAgent(ctx context.Context, secrets []*lkproto.AgentSecret, regions []string) (string, error) {
+func (c *Client) RegisterAgent(ctx context.Context, agentName string, secrets []*lkproto.AgentSecret, regions []string) (string, error) {
 	resp, err := c.AgentClient.CreateAgent(ctx, &lkproto.CreateAgentRequest{
-		Secrets: secrets,
-		Regions: regions,
+		AgentName: agentName,
+		Secrets:   secrets,
+		Regions:   regions,
 	})
 	if err != nil {
 		return "", err