mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-12 01:37:06 +02:00
MichaelandClaude Opus 5.5
a745d2399e
Généralisation du transformSubAgentAliases d'OpenFox. Les petits modèles
ont appris d'autres agents : read_file, str_replace, run_command, ou path /
old_string au lieu de file / old. L'appel tombait sur « outil inconnu »
ou sur un argument manquant, et le tour se perdait en allers-retours.
Le nom et les arguments sont traduits vers l'outil réel quand la cible est
disponible dans ce tour, avant que l'appel soit rangé dans l'historique
(le modèle y relit l'appel tel qu'exécuté). Un sous-agent appelé comme un
outil (« explorer », « code_reviewer ») devient subagent{role, task}. Un
appel déjà correct, ou dont la cible n'est pas offerte, reste intact.
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
57 lines
1.9 KiB
Go
57 lines
1.9 KiB
Go
package loki
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
)
|
|
|
|
func outils(noms ...string) []Tool {
|
|
var ts []Tool
|
|
for _, n := range noms {
|
|
ts = append(ts, Tool{Type: "function", Function: ToolFunction{Name: n}})
|
|
}
|
|
return ts
|
|
}
|
|
|
|
// Noms et arguments appris d'autres agents : traduits vers les outils de Loki.
|
|
func TestReparationDesAppelsHallucines(t *testing.T) {
|
|
tools := outils("read", "edit", "bash", "subagent")
|
|
cas := []struct {
|
|
name, args, wantName string
|
|
wantArgs map[string]any
|
|
}{
|
|
{"read_file", `{"path":"a.go"}`, "read", map[string]any{"file": "a.go"}},
|
|
{"str_replace", `{"file_path":"a.go","old_string":"x","new_string":"y"}`, "edit", map[string]any{"file": "a.go", "old": "x", "new": "y"}},
|
|
{"run_command", `{"cmd":"ls"}`, "bash", map[string]any{"command": "ls"}},
|
|
{"edit", `{"file":"a.go","old_str":"x","new":"y"}`, "edit", map[string]any{"file": "a.go", "old": "x", "new": "y"}},
|
|
{"code_reviewer", `{"prompt":"relis"}`, "subagent", map[string]any{"role": "code-reviewer", "task": "relis"}},
|
|
}
|
|
for _, c := range cas {
|
|
tc := ToolCall{Function: ToolCallFunc{Name: c.name, Arguments: c.args}}
|
|
if !repairToolCall(&tc, tools) {
|
|
t.Errorf("%s : non réparé", c.name)
|
|
continue
|
|
}
|
|
var got map[string]any
|
|
_ = json.Unmarshal([]byte(tc.Function.Arguments), &got)
|
|
if tc.Function.Name != c.wantName || len(got) != len(c.wantArgs) {
|
|
t.Errorf("%s → %s %v", c.name, tc.Function.Name, got)
|
|
continue
|
|
}
|
|
for k, v := range c.wantArgs {
|
|
if got[k] != v {
|
|
t.Errorf("%s : %s=%v, attendu %v", c.name, k, got[k], v)
|
|
}
|
|
}
|
|
}
|
|
// Outil cible absent de ce tour, ou appel déjà correct : intact.
|
|
for _, tc := range []ToolCall{
|
|
{Function: ToolCallFunc{Name: "write_file", Arguments: `{"path":"a"}`}},
|
|
{Function: ToolCallFunc{Name: "read", Arguments: `{"file":"a"}`}},
|
|
} {
|
|
if repairToolCall(&tc, tools) {
|
|
t.Errorf("modifié à tort : %+v", tc)
|
|
}
|
|
}
|
|
}
|