rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Tavily: allow requesting images

Browse Source
This commit is contained in:
Cohee
2025-02-01 02:23:10 +02:00
parent e59114c1c5
commit a2a6afb731
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/endpoints/search.js
View File

@@ -220,7 +220,7 @@ router.post('/tavily', jsonParser, async (request, response) => {
return response.sendStatus(400); return response.sendStatus(400);
} }
const { query } = request.body; const { query, include_images } = request.body;
const body = { const body = {
query: query, query: query,
@@ -229,7 +229,7 @@ router.post('/tavily', jsonParser, async (request, response) => {
topic: 'general', topic: 'general',
include_answer: true, include_answer: true,
include_raw_content: false, include_raw_content: false,
include_images: false, include_images: !!include_images,
include_image_descriptions: false, include_image_descriptions: false,
include_domains: [], include_domains: [],
max_results: 10, max_results: 10,
@@ -297,6 +297,7 @@ router.post('/koboldcpp', jsonParser, async (request, response) => {
router.post('/visit', jsonParser, async (request, response) => { router.post('/visit', jsonParser, async (request, response) => {
try { try {
const url = request.body.url; const url = request.body.url;
const html = Boolean(request.body.html ?? true);
if (!url) { if (!url) {
console.log('No url provided for /visit'); console.log('No url provided for /visit');
@@ -340,13 +341,20 @@ router.post('/visit', jsonParser, async (request, response) => {
} }
const contentType = String(result.headers.get('content-type')); 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) { } catch (error) {
console.log(error); console.log(error);
return response.sendStatus(500); return response.sendStatus(500);