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

First working version

Browse Source
This commit is contained in:
Mark Ceter
2023-05-21 11:33:10 +00:00
parent f2ecac0d79
commit f62c26a9ba
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
public/scripts/extensions/infinity-context/index.js
View File

@ -57,7 +57,7 @@ function onSplitLengthInput() {
async function addMessages(chat_id, messages) { async function addMessages(chat_id, messages) {
const url = new URL(getApiUrl()); const url = new URL(getApiUrl());
url.pathname = '/api/chroma'; url.pathname = '/api/chromadb';
const messagesDeepCopy = JSON.parse(JSON.stringify(messages)); const messagesDeepCopy = JSON.parse(JSON.stringify(messages));
const splittedMessages = []; const splittedMessages = [];
@ -98,7 +98,7 @@ async function addMessages(chat_id, messages) {
async function queryMessages(chat_id, query) { async function queryMessages(chat_id, query) {
const url = new URL(getApiUrl()); const url = new URL(getApiUrl());
url.pathname = '/api/chroma/query'; url.pathname = '/api/chromadb/query';
const queryMessagesResult = await fetch(url, { const queryMessagesResult = await fetch(url, {
method: 'POST', method: 'POST',
@ -123,20 +123,20 @@ window.chromadb_interceptGeneration = async (chat) => {
if (messagesToStore.length > 0) { if (messagesToStore.length > 0) {
await addMessages(currentChatId, messagesToStore); await addMessages(currentChatId, messagesToStore);
}
const lastMessage = chat[chat.length - 1]; const lastMessage = chat[chat.length - 1];
if (lastMessage) { if (lastMessage) {
const queriedMessages = await queryMessages(currentChatId, lastMessage.mes); const queriedMessages = await queryMessages(currentChatId, lastMessage.mes);
queriedMessages.sort((a, b) => a.date - b.date); queriedMessages.sort((a, b) => a.date - b.date);
const newChat = queriedMessages.map(m => JSON.parse(m.meta)); const newChat = queriedMessages.map(m => JSON.parse(m.meta));
console.log(newChat); chat.splice(0, messagesToStore.length, ...newChat);
chat.splice(0, messagesToStore.length, newChat); console.log('ChromaDB chat after injection', chat);
}
} }
} }
} }
@ -165,4 +165,4 @@ jQuery(async () => {
$('#chromadb_split_length').on('input', onSplitLengthInput); $('#chromadb_split_length').on('input', onSplitLengthInput);
await loadSettings(); await loadSettings();
}); });