Carlos Florêncio
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
// runBudgetCmd dispatches `agentctl kandev budget <subcmd>`. The
|
|
// CEO uses this to check workspace + per-agent spend before
|
|
// expensive operations (hiring, model upgrades, etc.).
|
|
func runBudgetCmd(args []string) int {
|
|
if len(args) == 0 {
|
|
cliError("usage: agentctl kandev budget <get> [flags]")
|
|
return 1
|
|
}
|
|
switch args[0] {
|
|
case subcmdGet:
|
|
return budgetGet(args[1:])
|
|
default:
|
|
cliError("unknown budget subcommand: %s", args[0])
|
|
return 1
|
|
}
|
|
}
|
|
|
|
// budgetGet returns the workspace cost summary. When --agent-id is
|
|
// supplied, the per-agent endpoint is hit instead so the caller
|
|
// sees just that agent's slice.
|
|
func budgetGet(args []string) int {
|
|
fs := flag.NewFlagSet("budget get", flag.ContinueOnError)
|
|
agentID := fs.String("agent-id", "", "Restrict to a single agent's spend")
|
|
if err := fs.Parse(args); err != nil {
|
|
cliError("parse flags: %v", err)
|
|
return 1
|
|
}
|
|
wsID := os.Getenv("KANDEV_WORKSPACE_ID")
|
|
if wsID == "" {
|
|
cliError("KANDEV_WORKSPACE_ID must be set")
|
|
return 1
|
|
}
|
|
if *agentID == "" {
|
|
return getWithParams(
|
|
fmt.Sprintf("/api/v1/office/workspaces/%s/costs/summary", wsID),
|
|
"KANDEV_WORKSPACE_ID", wsID, nil)
|
|
}
|
|
return getWithParams(
|
|
fmt.Sprintf("/api/v1/office/workspaces/%s/costs/by-agent", wsID),
|
|
"KANDEV_WORKSPACE_ID", wsID, map[string]string{"agent_id": *agentID})
|
|
}
|