wehub-resource-sync
61 lines
2.1 KiB
Go
61 lines
2.1 KiB
Go
package memory
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"reasonix/internal/tool"
|
|
)
|
|
|
|
// forgetTool deletes a saved memory the model judges wrong or stale. Like
|
|
// rememberTool it is stateful (bound to one project's Store), so boot constructs
|
|
// it and adds it to the registry.
|
|
type forgetTool struct{ store Store }
|
|
|
|
// NewForgetTool returns the `forget` tool bound to store.
|
|
func NewForgetTool(store Store) tool.Tool { return forgetTool{store: store} }
|
|
|
|
func (forgetTool) Name() string { return "forget" }
|
|
|
|
func (forgetTool) Description() string {
|
|
return "Delete a saved memory by name when it is wrong, stale, or superseded, so it stops loading into future sessions. " +
|
|
"Use the slug from the memory index — the \"<name>\" in \"[label](<name>.md)\". " +
|
|
"Prefer updating a memory with `remember` (reuse its name) over forget-then-recreate; reach for forget only when the fact should no longer exist at all."
|
|
}
|
|
|
|
func (forgetTool) Schema() json.RawMessage {
|
|
return json.RawMessage(`{
|
|
"type": "object",
|
|
"properties": {
|
|
"name": {"type": "string", "description": "Slug of the memory to delete, as shown in the index (the \"<name>\" in \"[label](<name>.md)\")."}
|
|
},
|
|
"required": ["name"]
|
|
}`)
|
|
}
|
|
|
|
func (t forgetTool) Execute(ctx context.Context, args json.RawMessage) (string, error) {
|
|
var in struct {
|
|
Name string `json:"name"`
|
|
}
|
|
if err := json.Unmarshal(args, &in); err != nil {
|
|
return "", fmt.Errorf("invalid arguments: %w", err)
|
|
}
|
|
if in.Name == "" {
|
|
return "", fmt.Errorf("name is required")
|
|
}
|
|
archive, err := t.store.Archive(in.Name)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if q, ok := QueueFromContext(ctx); ok {
|
|
q.QueueMemory("Forgot memory \"" + slug(in.Name) + "\" — disregard its line still shown in the saved-memories index until next session.")
|
|
}
|
|
if archive != "" {
|
|
return fmt.Sprintf("Forgot memory %q (it no longer applies and will not load in future sessions; archived to %s).", in.Name, archive), nil
|
|
}
|
|
return fmt.Sprintf("Forgot memory %q (it no longer applies and will not load in future sessions).", in.Name), nil
|
|
}
|
|
|
|
func (forgetTool) ReadOnly() bool { return false }
|