From 1ee2832e2bbec3acb0ed611930e82033faf9e99f Mon Sep 17 00:00:00 2001
From: Matteo Ferrando <matteo.ferrando2@gmail.com>
Date: Wed, 25 Oct 2023 12:53:13 -0400
Subject: [PATCH] comments

---
 apps/demo-nextjs-app/pages/index.tsx | 3 ++-
 libs/client/src/function.ts          | 7 ++++++-
 libs/client/src/types.ts             | 7 +------
 3 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/apps/demo-nextjs-app/pages/index.tsx b/apps/demo-nextjs-app/pages/index.tsx
index c56dbfd..55a50f9 100644
--- a/apps/demo-nextjs-app/pages/index.tsx
+++ b/apps/demo-nextjs-app/pages/index.tsx
@@ -71,13 +71,14 @@ export function Index() {
     setLoading(true);
     const start = Date.now();
     try {
-      const result: Result = await fal.queue.subscribe('110602490-lora', {
+      const result: Result = await fal.subscribe('110602490-lora', {
         input: {
           prompt,
           model_name: 'stabilityai/stable-diffusion-xl-base-1.0',
           image_size: 'square_hd',
         },
         pollInterval: 5000, // Default is 1000 (every 1s)
+        logs: true,
         onQueueUpdate(update) {
           setElapsedTime(Date.now() - start);
           if (
diff --git a/libs/client/src/function.ts b/libs/client/src/function.ts
index e74d660..b74440d 100644
--- a/libs/client/src/function.ts
+++ b/libs/client/src/function.ts
@@ -131,7 +131,7 @@ export async function subscribe<Input, Output>(
     const pollInterval = options.pollInterval ?? 1000;
     const poll = async () => {
       try {
-        const requestStatus = await queue.status(id, requestId, true);
+        const requestStatus = await queue.status(id, requestId, options.logs ?? true);
         if (options.onQueueUpdate) {
           options.onQueueUpdate(requestStatus);
         }
@@ -176,6 +176,11 @@ type QueueSubscribeOptions = {
    * @param status - The current status of the queue.
    */
   onQueueUpdate?: (status: QueueStatus) => void;
+
+  /**
+   * If `true`, the response will include the logs for the request.
+   */
+  logs?: boolean;
 };
 
 /**
diff --git a/libs/client/src/types.ts b/libs/client/src/types.ts
index 74ad756..4faf9f6 100644
--- a/libs/client/src/types.ts
+++ b/libs/client/src/types.ts
@@ -6,11 +6,6 @@ export type EnqueueResult = {
   request_id: string;
 };
 
-// export type QueueStatus = {
-//   status: "IN_PROGRESS" | "COMPLETED";
-//   queue: number;
-// };
-
 export type RequestLog = {
   message: string;
   level: 'STDERR' | 'STDOUT' | 'ERROR' | 'INFO' | 'WARN' | 'DEBUG';
@@ -22,7 +17,7 @@ export type QueueStatus =
   | {
       status: 'IN_PROGRESS' | 'COMPLETED';
       response_url: string;
-      logs: RequestLog[];
+      logs: null | RequestLog[];
     }
   | {
       status: 'IN_QUEUE';