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
detections / detections (push) Waiting to run
web / web (push) Waiting to run
docs / links (push) Canceled after 0s
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"log"
|
|
|
|
actool "github.com/Autumn-27/norma/tool"
|
|
|
|
"github.com/Autumn-27/artex/agent"
|
|
"github.com/Autumn-27/artex/db"
|
|
)
|
|
|
|
type toolUsageRecorder interface {
|
|
InsertToolUsage(*db.ToolUsage) error
|
|
}
|
|
|
|
// meteredTool records one row immediately before a catalog tool is invoked. It
|
|
// embeds the resolved tool so schema overrides, defaults and permission behavior
|
|
// remain unchanged.
|
|
type meteredTool struct {
|
|
actool.CoreTool
|
|
recorder toolUsageRecorder
|
|
toolKey string
|
|
agentKey string
|
|
ri agent.RunInfo
|
|
}
|
|
|
|
func meterTool(t actool.CoreTool, recorder toolUsageRecorder, toolKey, agentKey string, ri agent.RunInfo) actool.CoreTool {
|
|
if recorder == nil {
|
|
return t
|
|
}
|
|
return &meteredTool{CoreTool: t, recorder: recorder, toolKey: toolKey, agentKey: agentKey, ri: ri}
|
|
}
|
|
|
|
func (m *meteredTool) Call(ctx context.Context, input json.RawMessage, tc *actool.ToolContext) (actool.Result, error) {
|
|
if err := m.recorder.InsertToolUsage(&db.ToolUsage{
|
|
ToolKey: m.toolKey,
|
|
AgentKey: m.agentKey,
|
|
TaskID: m.ri.TaskID,
|
|
ExplorationID: m.ri.ExplorationID,
|
|
IntentID: m.ri.IntentID,
|
|
SessionID: m.ri.SessionID,
|
|
}); err != nil {
|
|
log.Printf("[toolusage] insert %s: %v", m.toolKey, err)
|
|
}
|
|
return m.CoreTool.Call(ctx, input, tc)
|
|
}
|