dev_agent
72042516e2
CI / ci (push) Successful in 1m29s
Co-authored-by: dev_agent <dev_agent@cubecraftlabs.com> Co-committed-by: dev_agent <dev_agent@cubecraftlabs.com>
45 lines
1.4 KiB
Go
45 lines
1.4 KiB
Go
package llm
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
)
|
|
|
|
func TestRequestValidate(t *testing.T) {
|
|
neg := -0.1
|
|
valid := func() Request {
|
|
return Request{
|
|
Messages: []Message{{Role: RoleUser, Content: "hi"}, {Role: RoleAssistant, Content: "hello"}},
|
|
Schema: &JSONSchema{Name: "plan", Schema: json.RawMessage(`{"type":"object"}`)},
|
|
}
|
|
}
|
|
if err := valid().Validate(); err != nil {
|
|
t.Fatalf("valid request: %v", err)
|
|
}
|
|
cases := map[string]func(*Request){
|
|
"no messages": func(r *Request) { r.Messages = nil },
|
|
"unknown role": func(r *Request) { r.Messages[0].Role = "system" },
|
|
"negative temp": func(r *Request) { r.Temperature = &neg },
|
|
"negative max": func(r *Request) { r.MaxTokens = -1 },
|
|
"negative time": func(r *Request) { r.Timeout = -1 },
|
|
"unnamed schema": func(r *Request) { r.Schema.Name = "" },
|
|
"schema array": func(r *Request) { r.Schema.Schema = json.RawMessage(`[]`) },
|
|
"schema null": func(r *Request) { r.Schema.Schema = json.RawMessage(`null`) },
|
|
"schema invalid": func(r *Request) { r.Schema.Schema = json.RawMessage(`{`) },
|
|
"schema empty": func(r *Request) { r.Schema.Schema = nil },
|
|
}
|
|
for name, mutate := range cases {
|
|
r := valid()
|
|
mutate(&r)
|
|
if err := r.Validate(); err == nil {
|
|
t.Errorf("%s: Validate() = nil", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestUsageTotal(t *testing.T) {
|
|
if got := (Usage{InputTokens: 12, OutputTokens: 30}).Total(); got != 42 {
|
|
t.Fatalf("Total() = %d, want 42", got)
|
|
}
|
|
}
|