dela
0335d572de
ci / go (push) Waiting to run
ci / go-db (agent) (push) Waiting to run
ci / go-db (config) (push) Waiting to run
ci / go-db (db) (push) Waiting to run
ci / go-db (evidence) (push) Waiting to run
ci / go-db (llmrec) (push) Waiting to run
ci / go-db (server) (push) Waiting to run
web / web (push) Waiting to run
docs / links (push) Canceled after 0s
detections / detections (push) Canceled after 0s
74 lines
2.0 KiB
Go
74 lines
2.0 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"testing"
|
|
|
|
actool "github.com/Autumn-27/norma/tool"
|
|
|
|
"github.com/Autumn-27/artex/agent"
|
|
"github.com/Autumn-27/artex/db"
|
|
)
|
|
|
|
type recordingToolUsage struct {
|
|
rows []*db.ToolUsage
|
|
err error
|
|
}
|
|
|
|
func (r *recordingToolUsage) InsertToolUsage(row *db.ToolUsage) error {
|
|
copy := *row
|
|
r.rows = append(r.rows, ©)
|
|
return r.err
|
|
}
|
|
|
|
func TestMeterToolRecordsAttributionAndDelegates(t *testing.T) {
|
|
calls := 0
|
|
base := actool.Build(actool.Spec{
|
|
Name: "record_fact",
|
|
Run: func(context.Context, json.RawMessage, *actool.ToolContext) (actool.Result, error) {
|
|
calls++
|
|
return actool.Result{}, nil
|
|
},
|
|
})
|
|
recorder := &recordingToolUsage{}
|
|
ri := agent.RunInfo{TaskID: 42, ExplorationID: 8, IntentID: 9, SessionID: "session-1"}
|
|
wrapped := meterTool(base, recorder, "record_fact", "worker", ri)
|
|
|
|
if _, err := wrapped.Call(context.Background(), json.RawMessage(`{"secret":"not persisted"}`), nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if calls != 1 {
|
|
t.Fatalf("delegate calls: want 1, got %d", calls)
|
|
}
|
|
if len(recorder.rows) != 1 {
|
|
t.Fatalf("usage rows: want 1, got %d", len(recorder.rows))
|
|
}
|
|
got := recorder.rows[0]
|
|
if got.ToolKey != "record_fact" || got.AgentKey != "worker" || got.TaskID != 42 ||
|
|
got.ExplorationID != 8 || got.IntentID != 9 || got.SessionID != "session-1" {
|
|
t.Fatalf("unexpected attribution: %+v", got)
|
|
}
|
|
}
|
|
|
|
func TestMeterToolFailureDoesNotBreakInvocation(t *testing.T) {
|
|
calls := 0
|
|
base := actool.Build(actool.Spec{
|
|
Name: "list_facts",
|
|
Run: func(context.Context, json.RawMessage, *actool.ToolContext) (actool.Result, error) {
|
|
calls++
|
|
return actool.Result{}, nil
|
|
},
|
|
})
|
|
recorder := &recordingToolUsage{err: errors.New("ledger unavailable")}
|
|
wrapped := meterTool(base, recorder, "list_facts", "worker", agent.RunInfo{})
|
|
|
|
if _, err := wrapped.Call(context.Background(), nil, nil); err != nil {
|
|
t.Fatalf("metering error leaked into tool call: %v", err)
|
|
}
|
|
if calls != 1 {
|
|
t.Fatalf("delegate calls: want 1, got %d", calls)
|
|
}
|
|
}
|