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

fix: rename collectionId to collectionName in searchVec for proper filtering (#61)

Browse Source
This commit is contained in:
jdvmi00 2026-01-28 00:03:02 -06:00 committed by GitHub
parent bfb0eebc3e
commit 64c6e6c2e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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/store.ts
View File

@ -619,7 +619,7 @@ export type Store = {
// Search
searchFTS: (query: string, limit?: number, collectionId?: number) => SearchResult[];
searchVec: (query: string, model: string, limit?: number, collectionId?: number) => Promise<SearchResult[]>;
searchVec: (query: string, model: string, limit?: number, collectionName?: string) => Promise<SearchResult[]>;
// Query expansion & reranking
expandQuery: (query: string, model?: string) => Promise<string[]>;
@ -702,7 +702,7 @@ export function createStore(dbPath?: string): Store {
// Search
searchFTS: (query: string, limit?: number, collectionId?: number) => searchFTS(db, query, limit, collectionId),
searchVec: (query: string, model: string, limit?: number, collectionId?: number) => searchVec(db, query, model, limit, collectionId),
searchVec: (query: string, model: string, limit?: number, collectionName?: string) => searchVec(db, query, model, limit, collectionName),
// Query expansion & reranking
expandQuery: (query: string, model?: string) => expandQuery(query, model, db),
@ -1900,7 +1900,7 @@ export function searchFTS(db: Database, query: string, limit: number = 20, colle
// Vector Search
// =============================================================================
export async function searchVec(db: Database, query: string, model: string, limit: number = 20, collectionId?: number): Promise<SearchResult[]> {
export async function searchVec(db: Database, query: string, model: string, limit: number = 20, collectionName?: string): Promise<SearchResult[]> {
const tableExists = db.prepare(`SELECT name FROM sqlite_master WHERE type='table' AND name='vectors_vec'`).get();
if (!tableExists) return [];
@ -1943,9 +1943,9 @@ export async function searchVec(db: Database, query: string, model: string, limi
`;
const params: string[] = [...hashSeqs];
if (collectionId) {
if (collectionName) {
docSql += ` AND d.collection = ?`;
params.push(String(collectionId));
params.push(collectionName);
}
const docRows = db.prepare(docSql).all(...params) as {