snxraven/llama-cpp-python-djs-bot
1
1
Fork 2
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Removing post proceess of output

Browse Source
This commit is contained in:
Raven Scott 2023-04-16 16:22:57 +02:00
parent 5f8e57d121
commit 7b3e0c1db2
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
llamabot.js
View File

@ -190,12 +190,10 @@ async function generateResponse(conversation) {
console.log(JSON.stringify(responseData)) console.log(JSON.stringify(responseData))
const choice = responseData.choices[0]; const choice = responseData.choices[0];
// Remove "user None:" and any text after it from the response const responseText = choice.message.content
const responseText = choice.message.content.trim();
const startIndex = responseText.indexOf('user None:'); return responseText;
const sanitizedResponse = startIndex === -1 ? responseText : responseText.substring(0, startIndex);
return sanitizedResponse;
} catch (err) { } catch (err) {
throw err; throw err;
} finally { } finally {