rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-02-18 21:20:39 +01: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
Show Stats 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);
}
const { query } = request.body;
const { query, include_images } = request.body;
const body = {
query: query,
@ -229,7 +229,7 @@ router.post('/tavily', jsonParser, async (request, response) => {
topic: 'general',
include_answer: true,
include_raw_content: false,
include_images: false,
include_images: !!include_images,
include_image_descriptions: false,
include_domains: [],
max_results: 10,
@ -297,6 +297,7 @@ router.post('/koboldcpp', jsonParser, async (request, response) => {
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');
@ -340,13 +341,20 @@ router.post('/visit', jsonParser, async (request, response) => {
}
const contentType = String(result.headers.get('content-type'));
if (!contentType.includes('text/html')) {
console.log(`Visit failed, content-type is ${contentType}, expected text/html`);
return response.sendStatus(500);
if (html) {
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();
return response.send(text);
response.setHeader('Content-Type', contentType);
const buffer = await result.arrayBuffer();
return response.send(Buffer.from(buffer));
} catch (error) {
console.log(error);
return response.sendStatus(500);