diff --git a/docs/plans/p5b/_files/internal/digest/retrieval_test.go b/docs/plans/p5b/_files/internal/digest/retrieval_test.go index 017b1c6..c7193c0 100644 --- a/docs/plans/p5b/_files/internal/digest/retrieval_test.go +++ b/docs/plans/p5b/_files/internal/digest/retrieval_test.go @@ -126,7 +126,10 @@ func chatDoc(f *fakeDixie) string { func TestEmbeddingsAreCachedAcrossQuestions(t *testing.T) { f := newFakeDixie(t, false) e := f.engine(t, NewGate(2, 8, 5*time.Second), true) - doc := longDoc(12, 0) // ~12 pages, a few chunks: under EmbedSyncMax + doc := longDoc(30, 0) // ~11.5k tokens: over SmallDoc (so not "whole"), ~10 chunks: under EmbedSyncMax + if EstTokens(doc) <= SmallDoc { + t.Fatalf("test document is %d tokens; it must exceed SmallDoc (%d) to reach retrieval", EstTokens(doc), SmallDoc) + } r := e.Run(context.Background(), doc, "which berth did the tug use?", "https://ex.test/a.pdf", dcache.ClassFleet) if r.Fallback != "" || r.Retrieval != "hybrid" { t.Fatalf("first question: fallback %q retrieval %q", r.Fallback, r.Retrieval)