mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-16 19:59:56 +02:00
Claude
f79b74dac3
KEEP_TURN_IMAGES ne garde plus la conversation au Go. L'état du tour vit dans la façade ; les fichiers et la vision restent au Go. Go (facade_keepimg.go) : - /internal/v1/turn/keepimg : vision réellement active, budget des images gardées ; - /internal/v1/turn/keepimg/store : une image relayée rangée par référence (refRelayImage), relue à l'octet près. Rust : - chat/keepimg.rs : port de keepImages (image à l'essai, mesure par le moteur répartie au prorata des octets, budget, publication, fin de tour). - La boucle du tour : budget demandé pour le fil principal, image rangée par le Go, historique réécrit publié sans les relais non gardés ; les images gardées partent avant tout résumé (filet réactif, en cours de tour, fenêtre pleine). - Début et fin de tour : retirées si le contexte le demande ou si elles ne doivent plus partir (clé retirée, vision absente). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_016tcGfJWefW5kUGoC2wL8cv
88 lines
3.1 KiB
Go
88 lines
3.1 KiB
Go
package loki
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// L'état que la façade demande au début d'un tour : vision réellement active
|
|
// (projecteur déclaré ET vu par le moteur) et budget des images gardées.
|
|
func TestFacadeKeepImgEtat(t *testing.T) {
|
|
testHome(t)
|
|
t.Setenv(internalTokenEnv, jetonTest)
|
|
if err := SetConfigKey("CTX", "2000"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
etat := func() (bool, float64) {
|
|
w := appelOAIInterne(t, "POST", "/internal/v1/turn/keepimg", `{}`)
|
|
var out struct {
|
|
Vision bool `json:"vision"`
|
|
Budget float64 `json:"budget"`
|
|
}
|
|
if err := json.Unmarshal(w.Body.Bytes(), &out); err != nil || w.Code != 200 {
|
|
t.Fatalf("état : %d %s", w.Code, w.Body.String())
|
|
}
|
|
return out.Vision, out.Budget
|
|
}
|
|
if v, b := etat(); v || b != 200 {
|
|
t.Fatalf("sans projecteur : vision=%v budget=%v", v, b)
|
|
}
|
|
visionServer(t, "")
|
|
if v, _ := etat(); !v {
|
|
t.Fatal("projecteur vu par le moteur : vision attendue")
|
|
}
|
|
if w := appelOAIInterne(t, "GET", "/internal/v1/turn/keepimg", ""); w.Code != 405 {
|
|
t.Fatalf("GET : %d", w.Code)
|
|
}
|
|
}
|
|
|
|
// Une image relayée rangée par référence, relue à l'octet près ; sinon elle
|
|
// reste éphémère.
|
|
func TestFacadeKeepImgRangement(t *testing.T) {
|
|
testHome(t)
|
|
t.Setenv(internalTokenEnv, jetonTest)
|
|
store := func(content string) map[string]any {
|
|
w := appelOAIInterne(t, "POST", "/internal/v1/turn/keepimg/store", `{"msg":{"role":"user","content":`+content+`}}`)
|
|
var out map[string]any
|
|
if err := json.Unmarshal(w.Body.Bytes(), &out); err != nil || w.Code != 200 {
|
|
t.Fatalf("rangement : %d %s", w.Code, w.Body.String())
|
|
}
|
|
return out
|
|
}
|
|
out := store(`[{"type":"text","text":"Voici la capture demandée."},{"type":"image_url","image_url":{"url":"data:image/png;base64,iVBORw0KGgo="}}]`)
|
|
if out["ok"] != true || out["weight"] != float64(8) {
|
|
t.Fatalf("rangée : %v", out)
|
|
}
|
|
msg := out["msg"].(map[string]any)
|
|
parts := msg["content"].([]any)
|
|
if parts[0].(map[string]any)["text"] != "Voici la capture demandée." {
|
|
t.Fatalf("légende : %v", parts)
|
|
}
|
|
ref := parts[1].(map[string]any)["image_url"].(map[string]any)["url"].(string)
|
|
if !strings.HasPrefix(ref, imgRefScheme) {
|
|
t.Fatalf("pas par référence : %s", ref)
|
|
}
|
|
if back, ok := chatImageDataURL(ref); !ok || back != "data:image/png;base64,iVBORw0KGgo=" {
|
|
t.Fatalf("relecture : %v %.40s", ok, back)
|
|
}
|
|
if ents, _ := os.ReadDir(chatImgDir()); len(ents) != 1 {
|
|
t.Fatalf("%d fichier(s) rangé(s)", len(ents))
|
|
}
|
|
for _, c := range []string{
|
|
`"texte seul"`,
|
|
`[{"type":"image_url","image_url":{"url":"https://exemple.org/a.png"}}]`,
|
|
`[{"type":"image_url","image_url":{"url":"data:image/x-inconnu;base64,AAAA"}}]`,
|
|
`[{"type":"image_url","image_url":{"url":"data:image/png;base64,pas du base64"}}]`,
|
|
`[{"type":"image_url","image_url":{"url":"data:image/png;base64,iVBORw0KGgo="}},{"type":"image_url","image_url":{"url":"data:image/png;base64,iVBORw0KGgo="}}]`,
|
|
} {
|
|
if out := store(c); out["ok"] != false {
|
|
t.Errorf("%s : %v", c, out)
|
|
}
|
|
}
|
|
if w := appelOAIInterne(t, "POST", "/internal/v1/turn/keepimg/store", `{`); w.Code != 400 {
|
|
t.Fatalf("corps illisible : %d", w.Code)
|
|
}
|
|
}
|