From ecba7f5aa0aba25bc35997d00be0076c27411e21 Mon Sep 17 00:00:00 2001
From: Harshith Mullapudi <harshithmullapudi@gmail.com>
Date: Thu, 9 Oct 2025 12:59:57 +0530
Subject: [PATCH] fix: add queue for space-assignemt

---
 .../app/trigger/spaces/space-assignment.ts    | 26 +++++++++++++++----
 apps/webapp/app/trigger/utils/queue.ts        |  1 -
 2 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/apps/webapp/app/trigger/spaces/space-assignment.ts b/apps/webapp/app/trigger/spaces/space-assignment.ts
index 2888f52..d43a7cb 100644
--- a/apps/webapp/app/trigger/spaces/space-assignment.ts
+++ b/apps/webapp/app/trigger/spaces/space-assignment.ts
@@ -1,4 +1,4 @@
-import { task } from "@trigger.dev/sdk/v3";
+import { queue, task } from "@trigger.dev/sdk/v3";
 import { logger } from "~/services/logger.service";
 import { SpaceService } from "~/services/space.server";
 import { makeModelCall } from "~/lib/model.server";
@@ -75,8 +75,14 @@ const AssignmentResultSchema = z.array(
   }),
 );
 
+const spaceAssignmentQueue = queue({
+  name: "space-assignment-queue",
+  concurrencyLimit: 1,
+});
+
 export const spaceAssignmentTask = task({
   id: "space-assignment",
+  queue: spaceAssignmentQueue,
   maxDuration: 1800, // 15 minutes timeout
   run: async (payload: SpaceAssignmentPayload) => {
     const {
@@ -704,9 +710,15 @@ async function processBatch(
 
     // Episode-intent matching is MEDIUM complexity (semantic analysis with intent alignment)
     let responseText = "";
-    await makeModelCall(false, prompt, (text: string) => {
-      responseText = text;
-    }, undefined, 'high');
+    await makeModelCall(
+      false,
+      prompt,
+      (text: string) => {
+        responseText = text;
+      },
+      undefined,
+      "high",
+    );
 
     // Response text is now set by the callback
 
@@ -1179,5 +1191,9 @@ function parseLLMResponse(
 
 // Helper function to trigger the task
 export async function triggerSpaceAssignment(payload: SpaceAssignmentPayload) {
-  return await spaceAssignmentTask.trigger(payload);
+  return await spaceAssignmentTask.trigger(payload, {
+    queue: "space-assignment-queue",
+    concurrencyKey: payload.userId,
+    tags: [payload.userId],
+  });
 }
diff --git a/apps/webapp/app/trigger/utils/queue.ts b/apps/webapp/app/trigger/utils/queue.ts
index 5b62236..6ddbbcd 100644
--- a/apps/webapp/app/trigger/utils/queue.ts
+++ b/apps/webapp/app/trigger/utils/queue.ts
@@ -25,7 +25,6 @@ export const addToQueue = async (
 
   const queuePersist = await prisma.ingestionQueue.create({
     data: {
-      spaceId: body.spaceId ? body.spaceId : null,
       data: body,
       type: body.type,
       status: IngestionStatus.PENDING,