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, include_images } = request.body;
         const body = {
             query: query,

             topic: 'general',
             include_answer: true,
             include_raw_content: false,
             include_images: false!!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);