xinmi/internal/workbench/structured_output_test.go

package workbench

import "testing"

func TestStructuredGenerationRequestsProviderJSONMode(t *testing.T) {
	s, ctx, _, model := generationHarness(t)
	model.output = `{"days":[]}`
	if _, err := s.completeJSON(ctx, "agent", "Return JSON", "course plan"); err != nil {
		t.Fatal(err)
	}
	if string(model.options.Format) != `{"type":"object"}` {
		t.Fatal("structured operation omitted provider JSON mode")
	}
	model.output = "Accepted manuscript"
	if _, err := s.complete(ctx, "agent", "Return prose", "content"); err != nil {
		t.Fatal(err)
	}
	if len(model.options.Format) != 0 {
		t.Fatal("manuscript generation must retain its prose output format")
	}
}