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;