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

Fix query command: use filepath instead of file property

Browse Source 
- Fixed querySearch to map r.filepath instead of non-existent r.file
- Fixed vectorSearch to use r.filepath consistently
- Fixed search function to pass filepath to getContextForFile
- All search commands now work correctly with SearchResult type

qmd query now works!
This commit is contained in:
Tobi Lutke 2025-12-13 13:10:19 -05:00
parent d1e9ca150d
commit 4687394106
No known key found for this signature in database
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/qmd.ts
View File

@ -1936,7 +1936,7 @@ function search(query: string, opts: OutputOptions): void {
// Add context to results
const resultsWithContext = results.map(r => ({
...r,
context: getContextForFile(db, r.file),
context: getContextForFile(db, r.filepath),
}));
closeDb();
@ -1986,9 +1986,9 @@ async function vectorSearch(query: string, opts: OutputOptions, model: string =
// searchVec accepts collection name as number parameter for legacy reasons (will be fixed in store.ts)
const vecResults = await searchVec(db, q, model, perQueryLimit, collectionName as any);
for (const r of vecResults) {
const existing = allResults.get(r.file);
const existing = allResults.get(r.filepath);
if (!existing || r.score > existing.score) {
allResults.set(r.file, { file: r.file, displayPath: r.displayPath, title: r.title, body: r.body, score: r.score });
allResults.set(r.filepath, { file: r.filepath, displayPath: r.displayPath, title: r.title, body: r.body || "", score: r.score });
}
}
}
@ -2106,7 +2106,7 @@ async function querySearch(query: string, opts: OutputOptions, embedModel: strin
// searchFTS accepts collection name as number parameter for legacy reasons (will be fixed in store.ts)
const ftsResults = searchFTS(db, q, 20, collectionName as any);
if (ftsResults.length > 0) {
rankedLists.push(ftsResults.map(r => ({ file: r.file, displayPath: r.displayPath, title: r.title, body: r.body, score: r.score })));
rankedLists.push(ftsResults.map(r => ({ file: r.filepath, displayPath: r.displayPath, title: r.title, body: r.body || "", score: r.score })));
}
// Vector search - get ranked results
@ -2114,7 +2114,7 @@ async function querySearch(query: string, opts: OutputOptions, embedModel: strin
// searchVec accepts collection name as number parameter for legacy reasons (will be fixed in store.ts)
const vecResults = await searchVec(db, q, embedModel, 20, collectionName as any);
if (vecResults.length > 0) {
rankedLists.push(vecResults.map(r => ({ file: r.file, displayPath: r.displayPath, title: r.title, body: r.body, score: r.score })));
rankedLists.push(vecResults.map(r => ({ file: r.filepath, displayPath: r.displayPath, title: r.title, body: r.body || "", score: r.score })));
}
}
}