fix: revert failed deployment (#81)


app/api/vision-agent/route.ts

@@ -11,7 +11,6 @@ import { getPresignedUrl } from '@/lib/aws';
 // export const runtime = 'edge';
 export const dynamic = 'force-dynamic';
 export const maxDuration = 300; // This function can run for a maximum of 5 minutes
-const TIMEOUT_MILI_SECONDS = 5 * 60 * 1000;
 
 const uploadBase64 = async (
   base64: string,
@@ -92,7 +91,7 @@ export const POST = withLogging(
 
     const fetchResponse = await fetch(
       `https://api.dev.landing.ai/v1/agent/chat?agent_class=vision_agent&self_reflection=false`,
-      // `http://localhost:5001/v1/agent/chat?agent_class=vision_agent&self_reflection=false`,
+      // `http://localhost:5001/v1/agent/chat?agent_class=vision_agent&self_reflection=${enableSelfReflection}`,
       {
         method: 'POST',
         headers: {
@@ -149,7 +148,6 @@ export const POST = withLogging(
       const decoder = new TextDecoder('utf-8');
       let maxChunkSize = 0;
       let buffer = '';
-      let time = Date.now();
       const stream = new ReadableStream({
         async start(controller) {
           // const parser = createParser(streamParser);
@@ -160,23 +158,6 @@ export const POST = withLogging(
             const lines = buffer
               .split('\n')
               .filter(line => line.trim().length > 0);
-            if (lines.length === 0) {
-              if (Date.now() - time > TIMEOUT_MILI_SECONDS) {
-                logger.info(
-                  session,
-                  {
-                    message: 'Agent timed out',
-                  },
-                  request,
-                  '__Agent_timeout__',
-                );
-                controller.error(
-                  `Haven't received any response in last ${TIMEOUT_MILI_SECONDS / 60000} minutes, agent timed out.`,
-                );
-              }
-            } else {
-              time = Date.now();
-            }
             buffer = lines.pop() ?? ''; // Save the last incomplete line back to the buffer
             let done = false;
             const parseLine = async (
@@ -185,27 +166,9 @@ export const POST = withLogging(
             ) => {
               try {
                 const msg = JSON.parse(line);
-                if (
-                  msg.type !== 'final_code' &&
-                  (msg.type !== 'code' ||
-                    msg.status === 'started' ||
-                    msg.status === 'running')
-                ) {
+                if (msg.type !== 'final_code') {
                   return line;
                 }
-                if (msg.type === 'code') {
-                  msg.payload.result = JSON.stringify({
-                    ...msg.payload.result,
-                    results: msg.payload.result.results.map((_result: any) => {
-                      return {
-                        ..._result,
-                        png: undefined,
-                        mp4: undefined,
-                      };
-                    }),
-                  });
-                  return JSON.stringify(msg);
-                }
                 const result = JSON.parse(msg.payload.result) as ResultPayload;
                 for (let index = 0; index < result.results.length; index++) {
                   const png = result.results[index].png ?? '';

app/page.tsx

@@ -22,7 +22,7 @@ const EXAMPLES = [
     mediaUrl:
       'https://vision-agent-dev.s3.us-east-2.amazonaws.com/examples/shark3_short.mp4',
     prompt:
-      'Can you detect any surfboards or sharks in the video, draw a green line between the shark and the nearest paddle boarder and add the distance between them in meters assuming 30 pixels is 1 meter. Make the line red if the shark is within 10 meters of a surfboard. Sample the video at 3 frames per second and save the output video as output.mp4.',
+      'Can you detect any surfboards or sharks in the video, draw a green line between the shark and the nearest surfboard and add the distance between them in meters assuming 30 pixels is 1 meter. Make the line red if the shark is within 10 meters of a surfboard. Sample the video at 1 frames per second and save the output video as output.mp4.',
   },
 ];