timeout

app/api/ask/route.ts

@@ -31,6 +31,10 @@ import { injectDeepSiteBadge, isIndexPage } from "@/lib/inject-badge";
 
 const ipAddresses = new Map();
 
+const STREAMING_TIMEOUT = 180000;
+const REQUEST_TIMEOUT = 240000;
+export const maxDuration = 300;
+
 export async function POST(request: NextRequest) {
   const authHeaders = await headers();
   const userToken = request.cookies.get(MY_TOKEN_KEY())?.value;
@@ -113,6 +117,9 @@ export async function POST(request: NextRequest) {
 
     (async () => {
       // let completeResponse = "";
+      let timeoutId: NodeJS.Timeout | null = null;
+      let isTimedOut = false;
+      
       try {
         const client = new InferenceClient(token);
         
@@ -144,22 +151,51 @@ export async function POST(request: NextRequest) {
           billTo ? { billTo } : {}
         );
 
-        while (true) {
-          const { done, value } = await chatCompletion.next()
-          if (done) {
-            break;
-          }
+        // Set up timeout
+        const timeoutPromise = new Promise((_, reject) => {
+          timeoutId = setTimeout(() => {
+            isTimedOut = true;
+            reject(new Error('Request timeout: The AI model took too long to respond. Please try again with a simpler prompt or try a different model.'));
+          }, STREAMING_TIMEOUT);
+        });
 
-          const chunk = value.choices[0]?.delta?.content;
-          if (chunk) {
-            await writer.write(encoder.encode(chunk));
-          }
-        }
+        // Race between streaming and timeout
+        await Promise.race([
+          (async () => {
+            while (true) {
+              const { done, value } = await chatCompletion.next()
+              if (done) {
+                break;
+              }
+
+              const chunk = value.choices[0]?.delta?.content;
+              if (chunk) {
+                await writer.write(encoder.encode(chunk));
+              }
+            }
+          })(),
+          timeoutPromise
+        ]);
+
+        // Clear timeout if successful
+        if (timeoutId) clearTimeout(timeoutId);
         
         // Explicitly close the writer after successful completion
         await writer.close();
       } catch (error: any) {
-        if (error.message?.includes("exceeded your monthly included credits")) {
+        // Clear timeout on error
+        if (timeoutId) clearTimeout(timeoutId);
+        
+        if (isTimedOut || error.message?.includes('timeout') || error.message?.includes('Request timeout')) {
+          await writer.write(
+            encoder.encode(
+              JSON.stringify({
+                ok: false,
+                message: "Request timeout: The AI model took too long to respond. Please try again with a simpler prompt or try a different model.",
+              })
+            )
+          );
+        } else if (error.message?.includes("exceeded your monthly included credits")) {
           await writer.write(
             encoder.encode(
               JSON.stringify({
@@ -341,17 +377,52 @@ export async function PUT(request: NextRequest) {
       billTo ? { billTo } : {}
     );
 
+    // Set up timeout for AI streaming
     let chunk = "";
-    while (true) {
-      const { done, value } = await chatCompletion.next();
-      if (done) {
-        break;
-      }
+    let timeoutId: NodeJS.Timeout | null = null;
+    let isTimedOut = false;
+
+    const timeoutPromise = new Promise<never>((_, reject) => {
+      timeoutId = setTimeout(() => {
+        isTimedOut = true;
+        reject(new Error('Request timeout: The AI model took too long to respond. Please try again with a simpler prompt or try a different model.'));
+      }, REQUEST_TIMEOUT);
+    });
+
+    try {
+      await Promise.race([
+        (async () => {
+          while (true) {
+            const { done, value } = await chatCompletion.next();
+            if (done) {
+              break;
+            }
 
-      const deltaContent = value.choices[0]?.delta?.content;
-      if (deltaContent) {
-        chunk += deltaContent;
+            const deltaContent = value.choices[0]?.delta?.content;
+            if (deltaContent) {
+              chunk += deltaContent;
+            }
+          }
+        })(),
+        timeoutPromise
+      ]);
+
+      // Clear timeout if successful
+      if (timeoutId) clearTimeout(timeoutId);
+    } catch (timeoutError: any) {
+      // Clear timeout on error
+      if (timeoutId) clearTimeout(timeoutId);
+      
+      if (isTimedOut || timeoutError.message?.includes('timeout') || timeoutError.message?.includes('Request timeout')) {
+        return NextResponse.json(
+          {
+            ok: false,
+            message: "Request timeout: The AI model took too long to respond. Please try again with a simpler prompt or try a different model.",
+          },
+          { status: 504 }
+        );
       }
+      throw timeoutError;
     }
     if (!chunk) {
       return NextResponse.json(
@@ -616,6 +687,15 @@ This project was created with [DeepSite](https://huggingface.co/deepsite).
       );
     }
   } catch (error: any) {
+    if (error.message?.includes('timeout') || error.message?.includes('Request timeout')) {
+      return NextResponse.json(
+        {
+          ok: false,
+          message: "Request timeout: The operation took too long to complete. Please try again with a simpler request or try a different model.",
+        },
+        { status: 504 }
+      );
+    }
     if (error.message?.includes("exceeded your monthly included credits")) {
       return NextResponse.json(
         {

app/api/auth/login-url/route.ts

@@ -15,7 +15,7 @@ export async function GET(req: NextRequest) {
     url +
     "/deepsite/auth/callback";
 
-  const loginRedirectUrl = `https://huggingface.co/oauth/authorize?client_id=${process.env.OAUTH_CLIENT_ID}&redirect_uri=${redirect_uri}&response_type=code&scope=openid%20profile%20contribute-repos%20inference-api&prompt=consent&state=1234567890`;
+  const loginRedirectUrl = `https://huggingface.co/oauth/authorize?client_id=${process.env.OAUTH_CLIENT_ID}&redirect_uri=${redirect_uri}&response_type=code&scope=openid%20profile%20write-repos%20manage-repos%20inference-api&prompt=consent&state=1234567890`;
   
   return NextResponse.json({ loginUrl: loginRedirectUrl });
 }

app/api/mcp/route.ts

@@ -4,6 +4,36 @@ import { COLORS } from "@/lib/utils";
 import { injectDeepSiteBadge, isIndexPage } from "@/lib/inject-badge";
 import { Commit, Page } from "@/types";
 
+// Timeout configuration (in milliseconds)
+const OPERATION_TIMEOUT = 120000; // 2 minutes for HF operations
+
+// Extend the maximum execution time for this route
+export const maxDuration = 180; // 3 minutes
+
+// Utility function to wrap promises with timeout
+async function withTimeout<T>(
+  promise: Promise<T>,
+  timeoutMs: number,
+  errorMessage: string = "Operation timed out"
+): Promise<T> {
+  let timeoutId: NodeJS.Timeout;
+  
+  const timeoutPromise = new Promise<never>((_, reject) => {
+    timeoutId = setTimeout(() => {
+      reject(new Error(errorMessage));
+    }, timeoutMs);
+  });
+
+  try {
+    const result = await Promise.race([promise, timeoutPromise]);
+    clearTimeout(timeoutId!);
+    return result;
+  } catch (error) {
+    clearTimeout(timeoutId!);
+    throw error;
+  }
+}
+
 interface MCPRequest {
   jsonrpc: "2.0";
   id: number | string;
@@ -228,11 +258,15 @@ async function handleCreateProject(params: CreateProjectParams) {
   // Get user info from HF token
   let username: string;
   try {
-    const userResponse = await fetch("https://huggingface.co/api/whoami-v2", {
-      headers: {
-        Authorization: `Bearer ${hf_token}`,
-      },
-    });
+    const userResponse = await withTimeout(
+      fetch("https://huggingface.co/api/whoami-v2", {
+        headers: {
+          Authorization: `Bearer ${hf_token}`,
+        },
+      }),
+      30000, // 30 seconds for authentication
+      "Authentication timeout: Unable to verify Hugging Face token"
+    );
 
     if (!userResponse.ok) {
       throw new Error("Invalid Hugging Face token");
@@ -241,6 +275,9 @@ async function handleCreateProject(params: CreateProjectParams) {
     const userData = await userResponse.json();
     username = userData.name;
   } catch (error: any) {
+    if (error.message?.includes('timeout')) {
+      throw new Error(`Authentication timeout: ${error.message}`);
+    }
     throw new Error(`Authentication failed: ${error.message}`);
   }
 
@@ -300,38 +337,67 @@ This project was created with [DeepSite](https://huggingface.co/deepsite).
   });
 
   try {
-    const { repoUrl } = await createRepo({
-      repo,
-      accessToken: hf_token,
-    });
+    const { repoUrl } = await withTimeout(
+      createRepo({
+        repo,
+        accessToken: hf_token,
+      }),
+      60000, // 1 minute for repo creation
+      "Timeout creating repository. Please try again."
+    );
 
     const commitTitle = !prompt || prompt.trim() === "" ? "Initial project creation via MCP" : prompt;
     
-    await uploadFiles({
-      repo,
-      files,
-      accessToken: hf_token,
-      commitTitle,
-    });
+    await withTimeout(
+      uploadFiles({
+        repo,
+        files,
+        accessToken: hf_token,
+        commitTitle,
+      }),
+      OPERATION_TIMEOUT,
+      "Timeout uploading files. The repository was created but files may not have been uploaded."
+    );
 
     const path = repoUrl.split("/").slice(-2).join("/");
 
     const commits: Commit[] = [];
-    for await (const commit of listCommits({ repo, accessToken: hf_token })) {
-      if (commit.title.includes("initial commit") || commit.title.includes("image(s)") || commit.title.includes("Promote version")) {
-        continue;
-      }
-      commits.push({
-        title: commit.title,
-        oid: commit.oid,
-        date: commit.date,
-      });
+    const commitIterator = listCommits({ repo, accessToken: hf_token });
+    
+    // Wrap the commit listing with a timeout
+    const commitTimeout = new Promise<void>((_, reject) => {
+      setTimeout(() => reject(new Error("Timeout listing commits")), 30000);
+    });
+    
+    try {
+      await Promise.race([
+        (async () => {
+          for await (const commit of commitIterator) {
+            if (commit.title.includes("initial commit") || commit.title.includes("image(s)") || commit.title.includes("Promote version")) {
+              continue;
+            }
+            commits.push({
+              title: commit.title,
+              oid: commit.oid,
+              date: commit.date,
+            });
+          }
+        })(),
+        commitTimeout
+      ]);
+    } catch (error: any) {
+      // If listing commits times out, continue with empty commits array
+      console.error("Failed to list commits:", error.message);
     }
 
-    const space = await spaceInfo({
-      name: repo.name,
-      accessToken: hf_token,
-    });
+    const space = await withTimeout(
+      spaceInfo({
+        name: repo.name,
+        accessToken: hf_token,
+      }),
+      30000, // 30 seconds for space info
+      "Timeout fetching space information"
+    );
 
     const projectUrl = `https://huggingface.co/deepsite/${path}`;
     const spaceUrl = `https://huggingface.co/spaces/${path}`;
@@ -360,6 +426,9 @@ This project was created with [DeepSite](https://huggingface.co/deepsite).
       ],
     };
   } catch (err: any) {
+    if (err.message?.includes('timeout') || err.message?.includes('Timeout')) {
+      throw new Error(err.message || "Operation timed out. Please try again.");
+    }
     throw new Error(err.message || "Failed to create project");
   }
 }

app/api/re-design/route.ts

@@ -1,5 +1,11 @@
 import { NextRequest, NextResponse } from "next/server";
 
+// Timeout configuration (in milliseconds)
+const FETCH_TIMEOUT = 30000; // 30 seconds for external fetch
+
+// Extend the maximum execution time for this route
+export const maxDuration = 60; // 1 minute
+
 export async function PUT(request: NextRequest) {
   const body = await request.json();
   const { url } = body;
@@ -9,28 +15,54 @@ export async function PUT(request: NextRequest) {
   }
 
   try {
-    const response = await fetch(
-      `https://r.jina.ai/${encodeURIComponent(url)}`,
-      {
-        method: "POST",
+    // Create an AbortController for timeout
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), FETCH_TIMEOUT);
+
+    try {
+      const response = await fetch(
+        `https://r.jina.ai/${encodeURIComponent(url)}`,
+        {
+          method: "POST",
+          signal: controller.signal,
+        }
+      );
+      
+      clearTimeout(timeoutId);
+      
+      if (!response.ok) {
+        return NextResponse.json(
+          { error: "Failed to fetch redesign" },
+          { status: 500 }
+        );
       }
-    );
-    if (!response.ok) {
+      const markdown = await response.text();
       return NextResponse.json(
-        { error: "Failed to fetch redesign" },
-        { status: 500 }
+        {
+          ok: true,
+          markdown,
+        },
+        { status: 200 }
       );
+    } catch (fetchError: any) {
+      clearTimeout(timeoutId);
+      
+      if (fetchError.name === 'AbortError') {
+        return NextResponse.json(
+          { error: "Request timeout: The external service took too long to respond. Please try again." },
+          { status: 504 }
+        );
+      }
+      throw fetchError;
     }
-    const markdown = await response.text();
-    return NextResponse.json(
-      {
-        ok: true,
-        markdown,
-      },
-      { status: 200 }
-    );
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
   } catch (error: any) {
+    if (error.name === 'AbortError' || error.message?.includes('timeout')) {
+      return NextResponse.json(
+        { error: "Request timeout: The external service took too long to respond. Please try again." },
+        { status: 504 }
+      );
+    }
     return NextResponse.json(
       { error: error.message || "An error occurred" },
       { status: 500 }