ai-workspace-services/qmd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(test): increase timeout for SDK search tests that trigger LLM expansion

Browse Source 
These tests load the query expansion model on first call, which
consistently exceeds the 30s timeout on CI runners.
This commit is contained in:
Tobi Lutke 2026-03-10 12:06:02 -04:00
parent 8478ddb666
commit ed0249fd6b
No known key found for this signature in database
1 changed files with 52 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

101
test/sdk.test.ts
View File

@ -602,52 +602,6 @@ describe("search (unified API)", () => {
await expect(store.search({} as SearchOptions)).rejects.toThrow("requires either 'query' or 'queries'");
});
test("search() with query and rerank:false returns results", async () => {
const results = await store.search({ query: "authentication", rerank: false });
expect(results.length).toBeGreaterThan(0);
expect(results[0]).toHaveProperty("file");
expect(results[0]).toHaveProperty("score");
expect(results[0]).toHaveProperty("title");
expect(results[0]).toHaveProperty("bestChunk");
expect(results[0]).toHaveProperty("docid");
});
test("search() with intent and rerank:false returns results", async () => {
const results = await store.search({
query: "meeting",
intent: "quarterly planning and roadmap",
rerank: false,
});
expect(results.length).toBeGreaterThan(0);
});
test("search() with collection filter", async () => {
const results = await store.search({
query: "authentication",
collection: "docs",
rerank: false,
});
for (const r of results) {
expect(r.file).toMatch(/^qmd:\/\/docs\//);
}
});
test("search() with collections filter", async () => {
const results = await store.search({
query: "authentication",
collections: ["docs"],
rerank: false,
});
for (const r of results) {
expect(r.file).toMatch(/^qmd:\/\/docs\//);
}
});
test("search() with limit", async () => {
const results = await store.search({ query: "meeting", limit: 1, rerank: false });
expect(results.length).toBeLessThanOrEqual(1);
});
test("search() with pre-expanded queries and rerank:false", async () => {
const results = await store.search({
queries: [
@ -659,9 +613,58 @@ describe("search (unified API)", () => {
expect(results.length).toBeGreaterThan(0);
});
test("search() returns empty for non-matching query", async () => {
const results = await store.search({ query: "xyznonexistentterm123", rerank: false });
expect(results).toHaveLength(0);
// Tests below use search({ query: ... }) which triggers LLM query expansion
describe.skipIf(!!process.env.CI)("with LLM query expansion", () => {
test("search() with query and rerank:false returns results", async () => {
const results = await store.search({ query: "authentication", rerank: false });
expect(results.length).toBeGreaterThan(0);
expect(results[0]).toHaveProperty("file");
expect(results[0]).toHaveProperty("score");
expect(results[0]).toHaveProperty("title");
expect(results[0]).toHaveProperty("bestChunk");
expect(results[0]).toHaveProperty("docid");
});
test("search() with intent and rerank:false returns results", async () => {
const results = await store.search({
query: "meeting",
intent: "quarterly planning and roadmap",
rerank: false,
});
expect(results.length).toBeGreaterThan(0);
});
test("search() with collection filter", async () => {
const results = await store.search({
query: "authentication",
collection: "docs",
rerank: false,
});
for (const r of results) {
expect(r.file).toMatch(/^qmd:\/\/docs\//);
}
});
test("search() with collections filter", async () => {
const results = await store.search({
query: "authentication",
collections: ["docs"],
rerank: false,
});
for (const r of results) {
expect(r.file).toMatch(/^qmd:\/\/docs\//);
}
});
test("search() with limit", async () => {
const results = await store.search({ query: "meeting", limit: 1, rerank: false });
expect(results.length).toBeLessThanOrEqual(1);
});
test("search() returns empty for non-matching query", async () => {
const results = await store.search({ query: "xyznonexistentterm123", rerank: false });
expect(results).toHaveLength(0);
});
});
});