88 lines
3.5 KiB
Go
88 lines
3.5 KiB
Go
|
package processing
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/descrybe/descrybe-v2/apps/api/internal/aiprompts"
|
||
|
|
)
|
||
|
|
|
||
|
|
type a1SmokeCompleter struct {
|
||
|
|
system, user string
|
||
|
|
reply string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *a1SmokeCompleter) Complete(_ context.Context, system, user string) (Completion, error) {
|
||
|
|
c.system, c.user = system, user
|
||
|
|
return Completion{Text: c.reply, TotalTokens: 12}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *a1SmokeCompleter) Enabled() bool { return true }
|
||
|
|
|
||
|
|
func TestEnhance_A1StyleFormulaLocalSmoke(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
prompt := aiprompts.SectionTitleStart + "\n" +
|
||
|
|
`Napiši novo ime izdelka po formuli: "tip izdelka sentence case", "znamka s pravilno kapitalizacijo", "model", barva v eni besedi. Ne uporabljaj vejic.` + "\n" +
|
||
|
|
aiprompts.SectionTitleEnd + "\n\n" +
|
||
|
|
aiprompts.SectionDescriptionStart + "\n" +
|
||
|
|
`<H2>{Napiši Novo ime izdelka in izpostavi en benefit}</H2>` +
|
||
|
|
`<p>{Napiši odstavek, ki je dolg 100 besed, ki NE vsebuje Novo ime izdelka.}</p>` +
|
||
|
|
`<H2>{Izpostavi en benefit, in NE napiši Novo ime izdelka}</H2>` +
|
||
|
|
`<p>{Napiši odstavek, ki je dolg 100 besed in VKLJUČI tudi tip izdelka lowercase}</p>` +
|
||
|
|
`<b>{Tehnične specifikacije}</b><ul><li>{Napiši 3-10 tehničnih specifikacij v alinejah}</li></ul>` + "\n" +
|
||
|
|
aiprompts.SectionDescriptionEnd + "\n\n" +
|
||
|
|
aiprompts.SectionMetaStart + "\n" +
|
||
|
|
`Najprej napiši Novo ime izdelka in potem nadaljuj dokler nisi zapisal 140 znakov.` + "\n" +
|
||
|
|
aiprompts.SectionMetaEnd
|
||
|
|
|
||
|
|
replyObj := map[string]any{
|
||
|
|
"name": "ovitek PanzerGlass Honor 200 Smart črna",
|
||
|
|
"description": "<h2>Zaščita z elegantnim videzom</h2><p>" + strings.Repeat("Besedilo o ovitku. ", 20) +
|
||
|
|
"</p><h2>Trden oprijem</h2><p>" + strings.Repeat("Še več besedila. ", 20) +
|
||
|
|
"</p><ul><li>Barva: črna</li><li>Model: Honor 200 Smart</li></ul>",
|
||
|
|
"meta_title": "ovitek PanzerGlass Honor 200 Smart črna | zaščita",
|
||
|
|
"meta_description": "ovitek PanzerGlass Honor 200 Smart črna z elegantnim videzom in trdnim oprijemom za vsakodnevno uporabo.",
|
||
|
|
"attrs": map[string]any{"brand": "PanzerGlass"},
|
||
|
|
}
|
||
|
|
raw, _ := json.Marshal(replyObj)
|
||
|
|
cap := &a1SmokeCompleter{reply: string(raw)}
|
||
|
|
eng := &Engine{Completer: cap}
|
||
|
|
|
||
|
|
descTpl := aiprompts.EffectiveDescriptionTemplateAny(nil, prompt)
|
||
|
|
name, desc, _, meta, err := eng.enhance(context.Background(), ProductInput{
|
||
|
|
Name: "ovitek PanzerGlass 200 Smart",
|
||
|
|
Description: "Zaščitni ovitek za telefon.",
|
||
|
|
Language: "sl",
|
||
|
|
CategoryEnhancePrompt: prompt,
|
||
|
|
DescriptionTemplate: descTpl,
|
||
|
|
CategoryUniqueID: "ovitki",
|
||
|
|
}, "Ovitki in mape", map[string]any{"brand": "PanzerGlass", "product_model": "Honor 200 Smart"})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if !strings.Contains(cap.system, "Si tekstopisec") {
|
||
|
|
t.Fatalf("expected A1 system, got: %s", cap.system)
|
||
|
|
}
|
||
|
|
if !strings.Contains(cap.user, "GPT predloga:") || !strings.Contains(cap.user, "<name>{") {
|
||
|
|
t.Fatalf("expected legacy GPT predloga user, got: %s", cap.user)
|
||
|
|
}
|
||
|
|
if !strings.Contains(cap.user, "Staro_ime_izdelka: ovitek PanzerGlass 200 Smart") {
|
||
|
|
t.Fatalf("expected supplier name substituted like legacy STARO IME: %s", cap.user)
|
||
|
|
}
|
||
|
|
if !strings.Contains(cap.user, "<H2>{") && !strings.Contains(cap.user, "<h2>{") {
|
||
|
|
t.Fatalf("expected HTML formula in predloga: %s", cap.user)
|
||
|
|
}
|
||
|
|
if name != "ovitek PanzerGlass Honor 200 Smart črna" {
|
||
|
|
t.Fatalf("name=%q", name)
|
||
|
|
}
|
||
|
|
if !descriptionSatisfiesFormula(desc, descTpl) {
|
||
|
|
t.Fatalf("description must satisfy formula: %s", desc)
|
||
|
|
}
|
||
|
|
st, _ := meta.(map[string]any)
|
||
|
|
if st == nil || st["status"] != "ok" {
|
||
|
|
t.Fatalf("meta=%#v", meta)
|
||
|
|
}
|
||
|
|
}
|