From 70d66b6ce7c8cd584956d19991cef9c27f368a74 Mon Sep 17 00:00:00 2001 From: Dhravya Date: Tue, 2 Apr 2024 18:23:50 -0700 Subject: [PATCH] added category support: add and search --- apps/cf-ai-backend/src/routes/query.ts | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/apps/cf-ai-backend/src/routes/query.ts b/apps/cf-ai-backend/src/routes/query.ts index 7439c44e..d5359353 100644 --- a/apps/cf-ai-backend/src/routes/query.ts +++ b/apps/cf-ai-backend/src/routes/query.ts @@ -8,6 +8,7 @@ export async function GET(request: Request, _: CloudflareVectorizeStore, embeddi const query = queryparams.get("q"); const topK = parseInt(queryparams.get("topK") ?? "5"); const user = queryparams.get("user") + const category = queryparams.get("category") const sourcesOnly = (queryparams.get("sourcesOnly") ?? "false") @@ -25,6 +26,12 @@ export async function GET(request: Request, _: CloudflareVectorizeStore, embeddi } } + if (category) { + filter.category = { + $eq: category + } + } + const queryAsVector = await embeddings.embedQuery(query); const resp = await env!.VECTORIZE_INDEX.query(queryAsVector, {