A couple of fixes based on Nick PR Reviewer

Author: ericallam

apps/webapp/app/routes/_app.orgs.$organizationSlug.projects.$projectParam.env.$envParam.test.tasks.$taskParam/AIPayloadTabContent.tsx

@@ -39,6 +39,7 @@ export function AIPayloadTabContent({
 }) {
   const [prompt, setPrompt] = useState("");
   const [isLoading, setIsLoading] = useState(false);
+  const isLoadingRef = useRef(false);
   const [thinking, setThinking] = useState("");
   const [error, setError] = useState<string | null>(null);
   const [showThinking, setShowThinking] = useState(false);
@@ -54,8 +55,9 @@ export function AIPayloadTabContent({
 
   const submitGeneration = useCallback(
     async (queryPrompt: string) => {
-      if (!queryPrompt.trim() || isLoading) return;
+      if (!queryPrompt.trim() || isLoadingRef.current) return;
 
+      isLoadingRef.current = true;
       setIsLoading(true);
       setThinking("");
       setError(null);
@@ -138,10 +140,11 @@ export function AIPayloadTabContent({
         setError(err instanceof Error ? err.message : "An error occurred");
         setLastResult("error");
       } finally {
+        isLoadingRef.current = false;
         setIsLoading(false);
       }
     },
-    [isLoading, resourcePath, taskIdentifier, payloadSchema, getCurrentPayload]
+    [resourcePath, taskIdentifier, payloadSchema, getCurrentPayload]
   );
 
   const processStreamEvent = useCallback(
@@ -351,11 +354,12 @@ export function AIPayloadTabContent({
             <button
               key={example}
               type="button"
+              disabled={isLoading}
               onClick={() => {
                 setPrompt(example);
                 submitGeneration(example);
               }}
-              className="group flex w-fit items-center gap-2 rounded-full border border-dashed border-charcoal-600 px-4 py-2 transition-colors hover:border-solid hover:border-indigo-500 focus-custom focus-visible:!rounded-full"
+              className="group flex w-fit items-center gap-2 rounded-full border border-dashed border-charcoal-600 px-4 py-2 transition-colors hover:border-solid hover:border-indigo-500 focus-custom focus-visible:!rounded-full disabled:cursor-not-allowed disabled:opacity-50"
             >
               <SparkleListIcon className="size-4 shrink-0 text-text-dimmed transition group-hover:text-indigo-500" />
               <Paragraph

apps/webapp/app/routes/resources.orgs.$organizationSlug.projects.$projectParam.env.$envParam.test.ai-generate-payload.tsx

@@ -7,16 +7,19 @@ import { findProjectBySlug } from "~/models/project.server";
 import { findEnvironmentBySlug } from "~/models/runtimeEnvironment.server";
 import { requireUserId } from "~/services/session.server";
 import { EnvironmentParamSchema } from "~/utils/pathBuilder";
-import { inflateSync } from "node:zlib";
+import { inflate } from "node:zlib";
+import { promisify } from "node:util";
+
+const inflateAsync = promisify(inflate);
 import { $replica } from "~/db.server";
 import { logger } from "~/services/logger.server";
 import { findCurrentWorkerDeployment } from "~/v3/models/workerDeployment.server";
 
 const RequestSchema = z.object({
-  prompt: z.string().min(1, "Prompt is required"),
-  taskIdentifier: z.string(),
-  payloadSchema: z.string().optional(),
-  currentPayload: z.string().optional(),
+  prompt: z.string().min(1, "Prompt is required").max(1000),
+  taskIdentifier: z.string().max(256),
+  payloadSchema: z.string().max(50_000).optional(),
+  currentPayload: z.string().max(50_000).optional(),
 });
 
 export async function action({ request, params }: ActionFunctionArgs) {
@@ -89,6 +92,7 @@ export async function action({ request, params }: ActionFunctionArgs) {
         const result = streamText({
           model: openai(env.AI_RUN_FILTER_MODEL ?? "gpt-5-mini"),
           temperature: 1,
+          abortSignal: request.signal,
           system: systemPrompt,
           prompt,
           tools: {
@@ -201,7 +205,7 @@ async function getTaskSourceCode(
     // File contents are zlib-deflated then base64-encoded by the CLI,
     // stored as Buffer.from(base64String) in Prisma Bytes
     const base64 = Buffer.from(file.contents).toString("utf-8");
-    const decompressed = inflateSync(Buffer.from(base64, "base64")).toString("utf-8");
+    const decompressed = (await inflateAsync(Buffer.from(base64, "base64"))).toString("utf-8");
 
     logger.info("[AI payload] Found task source code", {
       taskIdentifier,