Apikeys agent (#1716)

* Adding commands for Agent API Keys

* Adding changes for API Keys and list version

* Incorporating review comments

* Updating modules modifying vendor

* Fixing unit tests

Author: kishlay-singh-DO

args.go

@@ -794,4 +794,13 @@ const (
 
 	// ArgFunctionRouteOutputSchema is the output schema of the function route
 	ArgFunctionRouteOutputSchema = "output-schema"
+
+	// ArgAgentAPIKeyName is the name of API Key of the agent.
+	ArgAgentAPIKeyName = "name"
+
+	// ArgAgentUUID is the uuid of the apikey.
+	ArgAPIkeyUUID = "api-key-uuid"
+
+	// ArgAPIKeyForce forces API Key deletion without confirmation.
+	ArgAPIKeyForce = "force"
 )

commands/displayers/genai.go

@@ -9,6 +9,13 @@ import (
 type Agent struct {
 	Agents do.Agents
 }
+type ApiKeyInfo struct {
+	ApiKeyInfo do.ApiKeys
+}
+
+type AgentVersion struct {
+	AgentVersions do.AgentVersions
+}
 
 var _ Displayable = &Agent{}
 
@@ -293,3 +300,106 @@ func (a *AgentRoute) KV() []map[string]any {
 
 	return out
 }
+
+var _ Displayable = &ApiKeyInfo{}
+
+func (v *ApiKeyInfo) JSON(out io.Writer) error {
+	return writeJSON(v.ApiKeyInfo, out)
+}
+
+func (a *ApiKeyInfo) Cols() []string {
+	return []string{
+		"Id",
+		"Name",
+		"CreatedBy",
+		"SecretKey",
+		"DeletedAt",
+		"CreatedAt",
+	}
+}
+
+func (a *ApiKeyInfo) ColMap() map[string]string {
+	return map[string]string{
+		"Id":        "ID",
+		"Name":      "Name",
+		"SecretKey": "Secret Key",
+		"CreatedBy": "Created By",
+		"DeletedAt": "Deleted At",
+		"CreatedAt": "Created At",
+	}
+}
+
+func (a *ApiKeyInfo) KV() []map[string]any {
+	if a == nil || a.ApiKeyInfo == nil {
+		return []map[string]any{}
+	}
+	out := make([]map[string]any, 0, len(a.ApiKeyInfo))
+	for _, apikey := range a.ApiKeyInfo {
+
+		out = append(out, map[string]any{
+			"Id":        apikey.Uuid,
+			"Name":      apikey.Name,
+			"SecretKey": apikey.SecretKey,
+			"CreatedBy": apikey.CreatedBy,
+			"CreatedAt": apikey.CreatedAt,
+			"DeletedAt": apikey.DeletedAt,
+		})
+	}
+	return out
+}
+
+var _ Displayable = &AgentVersion{}
+
+func (v *AgentVersion) JSON(out io.Writer) error {
+	return writeJSON(v.AgentVersions, out)
+}
+
+func (a *AgentVersion) Cols() []string {
+	return []string{
+		"AgentUuid",
+		"CanRollback",
+		"CreatedAt",
+		"CreatedByEmail",
+		"CurrentlyApplied",
+		"ID",
+		"ModelName",
+		"Name",
+		"VersionHash",
+	}
+}
+
+func (a *AgentVersion) ColMap() map[string]string {
+	return map[string]string{
+		"AgentUuid":        "Agent UUID",
+		"CanRollback":      "Can Rollback",
+		"CreatedAt":        "Created At",
+		"CreatedByEmail":   "Created By Email",
+		"CurrentlyApplied": "Currently Applied",
+		"ID":               "ID",
+		"ModelName":        "Model Name",
+		"Name":             "Name",
+		"VersionHash":      "Version Hash",
+	}
+}
+
+func (a *AgentVersion) KV() []map[string]any {
+	if a == nil || a.AgentVersions == nil {
+		return []map[string]any{}
+	}
+	out := make([]map[string]any, 0, len(a.AgentVersions))
+	for _, v := range a.AgentVersions {
+
+		out = append(out, map[string]any{
+			"AgentUuid":        v.AgentUuid,
+			"CanRollback":      v.CanRollback,
+			"CreatedAt":        v.CreatedAt,
+			"CreatedByEmail":   v.CreatedByEmail,
+			"CurrentlyApplied": v.CurrentlyApplied,
+			"ID":               v.ID,
+			"ModelName":        v.ModelName,
+			"Name":             v.Name,
+			"VersionHash":      v.VersionHash,
+		})
+	}
+	return out
+}

commands/genai_agent.go

@@ -20,6 +20,7 @@ func AgentCmd() *Command {
 			Long:    "The subcommands of `doctl genai agent` manage your GenAI agents.",
 		},
 	}
+	cmd.AddCommand(AgentAPIKeyCmd())
 
 	// Add the agent route command as a subcommand to genai agent
 	cmd.AddCommand(AgentRouteCmd())
@@ -137,6 +138,30 @@ func AgentCmd() *Command {
 	AddStringFlag(cmdAgentUpdateVisibility, "visibility", "", "", "Agent deployment visibility. Possible Options: `VISIBILITY_PLAYGROUND`, `VISIBILITY_PUBLIC`, `VISIBILITY_PRIVATE`. Default: `VISIBILITY_UNKNOWN`", requiredOpt())
 	cmdAgentUpdateVisibility.Example = `The following example updates the visibility of an agent with the ID ` + "`" + `12345678-1234-1234-1234-123456789012` + "`" + ` to ` + "`" + `VISIBILITY_PUBLIC` + "`" + `: doctl genai agent update-visibility 12345678-1234-1234-1234-123456789012 --visibility 'VISIBILITY_PUBLIC'`
 
+	AgentVersionDetails := `
+	- The Agent ID
+	- Can it be rollbacked
+	- The Version creation date, in ISO8601 combined date and time format
+	- The Email of the user who created the agent
+	- Whether the version is currently active,
+	- The Agent version ID
+	- The Agent version model name
+	- The Agent version name
+	- The Agent version hash`
+
+	cmdAgentListVersions := CmdBuilder(
+		cmd,
+		RunAgentListVersions,
+		"list-versions",
+		"List versions for an agent",
+		"Retrieves a list of all the versions for an agent on your account including the following information for each version:\n"+
+			AgentVersionDetails,
+		Writer,
+		aliasOpt("lv", "list-versions"),
+		displayerType(&displayers.AgentVersion{}),
+	)
+	cmdAgentListVersions.Example = `The following example retrieves a list of all versions for an Agent with ID ` + "`" + `12345678-1234-1234-1234-123456789012` + "`" + ` : doctl genai agent list-versions 12345678-1234-1234-1234-123456789012`
+
 	return cmd
 }
 
@@ -314,3 +339,21 @@ func RunAgentUpdateVisibility(c *CmdConfig) error {
 	}
 	return c.Display(&displayers.Agent{Agents: do.Agents{*agent}})
 }
+
+// RunAgentListVersions lists all versions for an agent.
+func RunAgentListVersions(c *CmdConfig) error {
+	if len(c.Args) < 1 {
+		return doctl.NewMissingArgsErr(c.NS)
+	}
+	agentID := c.Args[0]
+
+	agentVersions, err := c.GenAI().ListAgentVersions(agentID)
+	if err != nil {
+		return err
+	}
+
+	filtered := make(do.AgentVersions, 0, len(agentVersions))
+	filtered = append(filtered, agentVersions...)
+
+	return c.Display(&displayers.AgentVersion{AgentVersions: filtered})
+}