Tavily: allow requesting images

a2a6afb73164a0275e9260c8710c3b9c6cbe13aa

Cohee <18619528+Cohee1207@users.noreply.github.com>

src/endpoints/search.js+15 -7 Viewed

             return response.sendStatus(400);
+        }
-        const { query } = request.body;
+        const { query, include_images } = request.body;
         const body = {
             query: query,

             topic: 'general',
             include_answer: true,
             include_raw_content: false,
-            include_images: false,
+            include_images: !!include_images,
             include_image_descriptions: false,
             include_domains: [],
             max_results: 10,

 router.post('/visit', jsonParser, async (request, response) => {
     try {
         const url = request.body.url;
+        const html = Boolean(request.body.html ?? true);
         if (!url) {
             console.log('No url provided for /visit');

+        }
         const contentType = String(result.headers.get('content-type'));
-        if (!contentType.includes('text/html')) {
-            console.log(`Visit failed, content-type is ${contentType}, expected text/html`);
+        if (html) {
-            return response.sendStatus(500);
+            if (!contentType.includes('text/html')) {
+                console.log(`Visit failed, content-type is ${contentType}, expected text/html`);
+                return response.sendStatus(500);
+            }
+            const text = await result.text();
+            return response.send(text);
+        }
-        const text = await result.text();
+        response.setHeader('Content-Type', contentType);
-        return response.send(text);
+        const buffer = await result.arrayBuffer();
+        return response.send(Buffer.from(buffer));
     } catch (error) {
         console.log(error);
         return response.sendStatus(500);