Carlos Florêncio
112 lines
2.7 KiB
Go
112 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
)
|
|
|
|
func runMemoryCmd(args []string) int {
|
|
if len(args) == 0 {
|
|
cliError("usage: agentctl kandev memory <get|set|summary> [flags]")
|
|
return 1
|
|
}
|
|
switch args[0] {
|
|
case subcmdGet:
|
|
return memoryGet(args[1:])
|
|
case "set":
|
|
return memorySet(args[1:])
|
|
case "summary":
|
|
return memorySummary(args[1:])
|
|
default:
|
|
cliError("unknown memory subcommand: %s", args[0])
|
|
return 1
|
|
}
|
|
}
|
|
|
|
// memoryGet retrieves memory entries for the current agent, with optional
|
|
// layer and key filters.
|
|
func memoryGet(args []string) int {
|
|
fs := flag.NewFlagSet("memory get", flag.ContinueOnError)
|
|
layerFlag := fs.String("layer", "", "Filter by layer")
|
|
keyFlag := fs.String("key", "", "Filter by key")
|
|
if err := fs.Parse(args); err != nil {
|
|
cliError("parse flags: %v", err)
|
|
return 1
|
|
}
|
|
|
|
agentID := os.Getenv("KANDEV_AGENT_ID")
|
|
path := fmt.Sprintf("/api/v1/office/agents/%s/memory", agentID)
|
|
return getWithParams(path, "KANDEV_AGENT_ID", agentID, map[string]string{
|
|
"layer": *layerFlag,
|
|
"key": *keyFlag,
|
|
})
|
|
}
|
|
|
|
// memorySet upserts a memory entry for the current agent.
|
|
func memorySet(args []string) int {
|
|
fs := flag.NewFlagSet("memory set", flag.ContinueOnError)
|
|
layerFlag := fs.String("layer", "", "Memory layer (required)")
|
|
keyFlag := fs.String("key", "", "Memory key (required)")
|
|
contentFlag := fs.String("content", "", "Memory content (required)")
|
|
if err := fs.Parse(args); err != nil {
|
|
cliError("parse flags: %v", err)
|
|
return 1
|
|
}
|
|
|
|
if *layerFlag == "" || *keyFlag == "" || *contentFlag == "" {
|
|
cliError("--layer, --key, and --content are all required")
|
|
return 1
|
|
}
|
|
|
|
client, err := newKandevClient()
|
|
if err != nil {
|
|
cliError("%v", err)
|
|
return 1
|
|
}
|
|
|
|
if client.agentID == "" {
|
|
cliError("KANDEV_AGENT_ID must be set")
|
|
return 1
|
|
}
|
|
|
|
payload := map[string]any{
|
|
"entries": []map[string]string{
|
|
{
|
|
"layer": *layerFlag,
|
|
"key": *keyFlag,
|
|
"content": *contentFlag,
|
|
},
|
|
},
|
|
}
|
|
|
|
path := fmt.Sprintf("/api/v1/office/agents/%s/memory", client.agentID)
|
|
body, status, doErr := client.do(http.MethodPut, path, payload)
|
|
return handleResponse(body, status, doErr)
|
|
}
|
|
|
|
// memorySummary retrieves a summary of the agent's memory entries.
|
|
func memorySummary(args []string) int {
|
|
fs := flag.NewFlagSet("memory summary", flag.ContinueOnError)
|
|
if err := fs.Parse(args); err != nil {
|
|
cliError("parse flags: %v", err)
|
|
return 1
|
|
}
|
|
|
|
client, err := newKandevClient()
|
|
if err != nil {
|
|
cliError("%v", err)
|
|
return 1
|
|
}
|
|
|
|
if client.agentID == "" {
|
|
cliError("KANDEV_AGENT_ID must be set")
|
|
return 1
|
|
}
|
|
|
|
path := fmt.Sprintf("/api/v1/office/agents/%s/memory/summary", client.agentID)
|
|
body, status, doErr := client.do(http.MethodGet, path, nil)
|
|
return handleResponse(body, status, doErr)
|
|
}
|