snxraven/rayai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

scrape each result with 200 chars response

Browse Source
This commit is contained in:
Raven 2024-08-03 16:07:39 -04:00
parent 5e19d00a80
commit e25557190c
1 changed files with 48 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
backend-server/backend-server.js
View File

@ -109,7 +109,7 @@ function trimConversationHistory(messages, maxLength, tolerance) {
}
// Function to scrape web page
async function scrapeWebPage(url) {
async function scrapeWebPage(url, length) {
console.log(`${getTimestamp()} [INFO] Starting to scrape URL: ${url}`);
try {
const res = await fetch(url);
@ -127,7 +127,7 @@ async function scrapeWebPage(url) {
response += `Description: ${pageDescription}\n`;
}
if (pageContent) {
const MAX_CONTENT_LENGTH = process.env.MAX_CONTENT_LENGTH || 2000;
const MAX_CONTENT_LENGTH = length || process.env.MAX_CONTENT_LENGTH || 2000;
let plainTextContent = $('<div>').html(pageContent).text().trim().replace(/[\r\n\t]+/g, ' ');
const codePattern = /\/\/|\/\*|\*\/|\{|\}|\[|\]|\bfunction\b|\bclass\b|\b0x[0-9A-Fa-f]+\b|\b0b[01]+\b/;
const isCode = codePattern.test(plainTextContent);
@ -424,7 +424,7 @@ app.post('/api/v1/chat', async (req, res) => {
}
}
const searchRegex = /\bsearch\s+(.+)\b/i;
const searchRegex = /\b[Ss]earch\s+(.+)\b/;
const searchMatch = userMessage.match(searchRegex);
if (searchMatch) {
@ -432,13 +432,22 @@ app.post('/api/v1/chat', async (req, res) => {
console.log(`${getTimestamp()} [INFO] Detected search query in user message: ${searchQuery}`);
try {
googleIt({ 'query': searchQuery }).then(results => {
googleIt({ 'query': searchQuery }).then(async results => {
let searchResponse = `Search Query: ${searchQuery}\n`;
searchResponse += `Top Google search results:\n`;
results.forEach((result, index) => {
searchResponse += `${index + 1}. ${result.title} - ${result.link}\n`;
});
for (let i = 0; i < results.length; i++) {
const result = results[i];
searchResponse += `${i + 1}. ${result.title} - ${result.link}\n`;
try {
const scrapeResult = await scrapeWebPage(result.link, 200);
searchResponse += `Scraped Data: ${scrapeResult}\n`;
} catch (scrapeErr) {
console.error(`${getTimestamp()} [ERROR] Failed to scrape URL: ${result.link}`, scrapeErr);
searchResponse += `Failed to scrape URL: ${result.link}\n`;
}
}
const lastMessageIndex = conversationHistory[ip].length - 1;
if (lastMessageIndex >= 0) {
@ -463,6 +472,7 @@ app.post('/api/v1/chat', async (req, res) => {
}
}
// End Plugins ---
console.log(`${getTimestamp()} [INFO] Sending request to llama API for response`);