mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-16 11:49:43 +02:00
Avec LOKI_CHAT=rust, la façade sert les routes du chat. Deux chemins arrivent pourtant au Go sans passer par elle — le tunnel du relais et sa redistribution /api/e2e/req — et liraient la conversation du Go, périmée, voire l'écriraient en base par-dessus celle de la façade. facade_chat.go : - routes du chat remises à la façade (après requireWebAuth), avec le jeton interne qui y tient lieu d'authentification, flux compris ; - /api/e2e/chat lit le flux de la façade et le rechiffre événement par événement (nombres repris tels quels) ; - les événements d'affichage nés ici (bascule du mode Code, critères) vont au journal de la façade (/internal/facade/v1/delta) ; - l'export lit la discussion enregistrée ; - le déverrouillage de la mémoire fait recharger la façade. L'outil d'exécution ne rend plus les critères : critNotify les pousse lui- même, d'où qu'il soit appelé. Sans LOKI_CHAT=rust, rien ne change. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_016tcGfJWefW5kUGoC2wL8cv
87 lines
3.1 KiB
Go
87 lines
3.1 KiB
Go
package loki
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func appelOutil(t *testing.T, body string) map[string]any {
|
|
t.Helper()
|
|
w := appelOAIInterne(t, "POST", "/internal/v1/tool/exec", body)
|
|
if w.Code != 200 {
|
|
t.Fatalf("code %d : %s", w.Code, w.Body.String())
|
|
}
|
|
var got map[string]any
|
|
if err := json.Unmarshal(w.Body.Bytes(), &got); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return got
|
|
}
|
|
|
|
// Un outil exécuté par l'API interne rend exactement ce que la boucle Go
|
|
// aurait obtenu : même fonction (execTool), mêmes garde-fous.
|
|
func TestOutilInterneMemesResultats(t *testing.T) {
|
|
testHome(t)
|
|
t.Setenv(internalTokenEnv, jetonTest)
|
|
|
|
got := appelOutil(t, `{"name":"outil_qui_n_existe_pas","args":{}}`)
|
|
if got["result"] != "[erreur] outil inconnu: outil_qui_n_existe_pas" {
|
|
t.Fatalf("outil inconnu : %v", got)
|
|
}
|
|
|
|
// Commande catastrophique : refusée AVANT exécution, comme dans la boucle.
|
|
got = appelOutil(t, `{"name":"bash","args":{"command":"rm -rf /"},"caps":{"Agent":true}}`)
|
|
res, _ := got["result"].(string)
|
|
if want := refusedCommandResult(dangerousCommand("rm -rf /")); res != want {
|
|
t.Fatalf("garde-fou bash : %q, attendu %q", res, want)
|
|
}
|
|
if got["label"] != "rm -rf /" {
|
|
t.Fatalf("libellé : %v", got["label"])
|
|
}
|
|
|
|
// Question à l'utilisateur : l'événement remonte à l'appelant.
|
|
got = appelOutil(t, `{"name":"ask","args":{"question":"On continue ?","options":["oui","non"]},"caps":{"Agent":true}}`)
|
|
ask, _ := got["ask"].(map[string]any)
|
|
if r, _ := got["result"].(string); strings.HasPrefix(r, "[erreur]") || ask["question"] != "On continue ?" {
|
|
t.Fatalf("ask : %v", got)
|
|
}
|
|
}
|
|
|
|
func TestOutilInterneAppelIllisible(t *testing.T) {
|
|
testHome(t)
|
|
t.Setenv(internalTokenEnv, jetonTest)
|
|
for _, body := range []string{"", "{", `{"args":{}}`} {
|
|
if w := appelOAIInterne(t, "POST", "/internal/v1/tool/exec", body); w.Code != 400 {
|
|
t.Fatalf("%q : %d", body, w.Code)
|
|
}
|
|
}
|
|
if w := appelOAIInterne(t, "GET", "/internal/v1/tool/exec", ""); w.Code != 405 {
|
|
t.Fatalf("GET : %d", w.Code)
|
|
}
|
|
}
|
|
|
|
// Ce que la boucle tire du résultat après l'exécution : capture de la bulle,
|
|
// image relayée au modèle — rien après une erreur.
|
|
func TestOutilInterneSuitesDuResultat(t *testing.T) {
|
|
testHome(t)
|
|
image, shot, relay := toolStepExtras("web_screenshot", "", toolExecResult{
|
|
Result: "Capture : ",
|
|
})
|
|
if image != "captures/a.png" || shot != "" || relay != nil {
|
|
t.Fatalf("capture : %q %q %v (vision coupée : pas de relais)", image, shot, relay)
|
|
}
|
|
img := map[string]any{"type": "image_url", "image_url": map[string]any{"url": "data:image/png;base64,AA=="}}
|
|
_, _, relay = toolStepExtras("see_image", "photo.png", toolExecResult{Result: "ok", VisionImg: img})
|
|
if relay == nil || relay.Role != "user" {
|
|
t.Fatalf("see_image : %v", relay)
|
|
}
|
|
b, _ := json.Marshal(relay)
|
|
if !strings.Contains(string(b), "Image demandée (photo.png)") || !strings.Contains(string(b), "data:image/png") {
|
|
t.Fatalf("relais : %s", b)
|
|
}
|
|
if _, shot, _ := toolStepExtras("browser_click", "x", toolExecResult{Result: "[erreur] rien"}); shot != "" {
|
|
t.Fatalf("aperçu après une erreur : %q", shot)
|
|
}
|
|
}
|