WebSearch: Add KoboldCpp search endpoint.

0441a8725f14892d4e3b5e3d4c67f94767bb6bc6

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

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

 import { decode } from 'html-entities';
 import { readSecret, SECRET_KEYS } from './secrets.js';
 import { jsonParser } from '../express-common.js';
 +import { trimV1 } from '../util.js';
 +import { setAdditionalHeaders } from '../additional-headers.js';
 export const router = express.Router();

+    }
 });
 +router.post('/koboldcpp', jsonParser, async (request, response) => {
 +    try {
 +        const { query, url } = request.body;
++
 +        if (!url) {
 +            console.error('No URL provided for KoboldCpp search');
 +            return response.sendStatus(400);
 +        }
++
 +        console.debug('KoboldCpp search query', query);
++
 +        const baseUrl = trimV1(url);
 +        const args = {
 +            method: 'POST',
 +            headers: {},
 +            body: JSON.stringify({ q: query }),
 +        };
++
 +        setAdditionalHeaders(request, args, baseUrl);
 +        const result = await fetch(`${baseUrl}/api/extra/websearch`, args);
++
 +        if (!result.ok) {
 +            const text = await result.text();
 +            console.error('KoboldCpp request failed', result.statusText, text);
 +            return response.status(500).send(text);
 +        }
++
 +        const data = await result.json();
 +        return response.json(data);
 +    } catch (error) {
 +        console.error(error);
 +        return response.sendStatus(500);
 +    }
 +});
++
 router.post('/visit', jsonParser, async (request, response) => {
     try {
         const url = request.body.url;