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

10
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,7 +123,6 @@ 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];
@ -134,9 +133,10 @@ window.chromadb_interceptGeneration = async (chat) => {
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);
}
} }
} }
} }