wehub-resource-sync
a06f331eb8
CI / benchmark (push) Has been skipped
install-script / posix-syntax (push) Successful in 6m1s
CI / build-onnx (push) Failing after 6m43s
init-smoke / dry-run (push) Failing after 15m57s
security / govulncheck (push) Has been cancelled
security / trivy-fs (push) Has been cancelled
CI / test (1.26, ubuntu-latest) (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
CI / test (1.26, macos-latest) (push) Has been cancelled
CI / build-windows (push) Has been cancelled
CI / lint (push) Has been cancelled
install-script / powershell-syntax (push) Has been cancelled
install-script / install (macos-14) (push) Has been cancelled
install-script / install (ubuntu-latest) (push) Has been cancelled
85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
package llm
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestExtractJSON_Variants(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
in string
|
|
want string
|
|
okErr bool // true means "expect not-found"
|
|
}{
|
|
{"raw", `{"a":1}`, `{"a":1}`, false},
|
|
{"with prose", "Here:\n{\"a\":1}\nend", `{"a":1}`, false},
|
|
{"markdown fence", "```json\n{\"a\":1}\n```", `{"a":1}`, false},
|
|
{"plain fence", "```\n{\"a\":1}\n```", `{"a":1}`, false},
|
|
{"array", `[1,2,3]`, `[1,2,3]`, false},
|
|
{"nested braces in strings", `{"k":"a{b}c","v":1}`, `{"k":"a{b}c","v":1}`, false},
|
|
{"no JSON", "I cannot help with that.", "", true},
|
|
{"truncated", `{"a":1,`, "", true},
|
|
}
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
got, ok := ExtractJSON(tc.in)
|
|
if tc.okErr && ok {
|
|
t.Fatalf("ExtractJSON should have failed; got %q", got)
|
|
}
|
|
if !tc.okErr && !ok {
|
|
t.Fatalf("ExtractJSON unexpectedly failed for %q", tc.in)
|
|
}
|
|
if !tc.okErr && got != tc.want {
|
|
t.Errorf("got=%q want=%q", got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestAppendSchemaInstruction_KeepsExistingSystem(t *testing.T) {
|
|
out := AppendSchemaInstruction("be terse", ShapeExpandTerms, nil)
|
|
if !strings.HasPrefix(out, "be terse") {
|
|
t.Errorf("rider must follow original system: %q", out)
|
|
}
|
|
if !strings.Contains(out, "JSON Schema") {
|
|
t.Errorf("rider must reference JSON Schema: %q", out)
|
|
}
|
|
if !strings.Contains(out, "terms") {
|
|
t.Errorf("rider must embed shape property name: %q", out)
|
|
}
|
|
}
|
|
|
|
func TestAppendSchemaInstruction_Freeform(t *testing.T) {
|
|
if out := AppendSchemaInstruction("base", ShapeFreeform, nil); out != "base" {
|
|
t.Errorf("ShapeFreeform must leave the prompt untouched: %q", out)
|
|
}
|
|
}
|
|
|
|
func TestAppendSchemaInstruction_EmptyPrompt(t *testing.T) {
|
|
out := AppendSchemaInstruction("", ShapeRerankOrder, nil)
|
|
if strings.HasPrefix(out, "\n") {
|
|
t.Errorf("empty prompt must not leave a leading separator: %q", out)
|
|
}
|
|
if !strings.Contains(out, "order") {
|
|
t.Errorf("rider must embed shape property name: %q", out)
|
|
}
|
|
}
|
|
|
|
func TestSnippet_TruncatesLong(t *testing.T) {
|
|
long := strings.Repeat("x", 1000)
|
|
out := Snippet([]byte(long))
|
|
if len(out) < 100 {
|
|
t.Errorf("Snippet=%d chars; want a truncated payload", len(out))
|
|
}
|
|
if !strings.HasSuffix(out, "…") {
|
|
t.Error("long Snippet must end with the truncation marker")
|
|
}
|
|
}
|
|
|
|
func TestSnippet_ShortUnchanged(t *testing.T) {
|
|
if out := Snippet([]byte(" short ")); out != "short" {
|
|
t.Errorf("Snippet must trim and pass short input through: %q", out)
|
|
}
|
|
}
|