WebSearch: Add endpoint for Tavily

0f320dd362940454de63238ce50861f4be93df89

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

Parents: b5cdb29bf3776fbc4c9c956b3e96076dbd4afae0

public/scripts/secrets.js+1 -0 Viewed

     STABILITY: 'api_key_stability',
     BLOCKENTROPY: 'api_key_blockentropy',
     CUSTOM_OPENAI_TTS: 'api_key_custom_openai_tts',
 +    TAVILY: 'api_key_tavily',
 };
 const INPUT_MAP = {

src/endpoints/search.js+48 -0 Viewed

+    }
 });
 +router.post('/tavily', jsonParser, async (request, response) => {
 +    try {
 +        const apiKey = readSecret(request.user.directories, SECRET_KEYS.TAVILY);
++
 +        if (!apiKey) {
 +            console.log('No Tavily key found');
 +            return response.sendStatus(400);
 +        }
++
 +        const { query } = request.body;
++
 +        const body = {
 +            query: query,
 +            api_key: apiKey,
 +            search_depth: 'basic',
 +            topic: 'general',
 +            include_answer: true,
 +            include_raw_content: false,
 +            include_images: false,
 +            include_image_descriptions: false,
 +            include_domains: [],
 +            max_results: 10,
 +        };
++
 +        const result = await fetch('https://api.tavily.com/search', {
 +            method: 'POST',
 +            headers: {
 +                'Content-Type': 'application/json',
 +            },
 +            body: JSON.stringify(body),
 +        });
++
 +        console.log('Tavily query', query);
++
 +        if (!result.ok) {
 +            const text = await result.text();
 +            console.log('Tavily request failed', result.statusText, text);
 +            return response.status(500).send(text);
 +        }
++
 +        const data = await result.json();
 +        return response.json(data);
 +    } catch (error) {
 +        console.log(error);
 +        return response.sendStatus(500);
 +    }
 +});
++
 router.post('/visit', jsonParser, async (request, response) => {
     try {
         const url = request.body.url;

src/endpoints/secrets.js+1 -0 Viewed

     STABILITY: 'api_key_stability',
     BLOCKENTROPY: 'api_key_blockentropy',
     CUSTOM_OPENAI_TTS: 'api_key_custom_openai_tts',
 +    TAVILY: 'api_key_tavily',
 };
 // These are the keys that are safe to expose, even if allowKeysExposure is false