Upstream/waveterm
1
0
Fork 0
mirror of https://github.com/wavetermdev/waveterm.git synced 2024-12-21 16:38:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

AI Sliding Window (#1151)

Browse Source 
Only send the 30 most recent ai questions and responses to the model
when making requests. This prevents the amount of data being sent from
getting too big.
This commit is contained in:
Sylvie Crowe 2024-10-27 11:49:34 -07:00 committed by GitHub
parent e235f563f1
commit bbd530c052
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
frontend/app/view/waveai/waveai.tsx
View File

@ -24,6 +24,7 @@ interface ChatMessageType {
}
const outline = "2px solid var(--accent-color)";
const slidingWindowSize = 30;
interface ChatItemProps {
chatItem: ChatMessageType;
@ -233,7 +234,7 @@ export class WaveAiModel implements ViewModel {
return [];
}
const history: Array<OpenAIPromptMessageType> = JSON.parse(new TextDecoder().decode(data));
return history;
return history.slice(Math.max(history.length - slidingWindowSize, 0));
}
giveFocus(): boolean {
@ -294,19 +295,23 @@ export class WaveAiModel implements ViewModel {
fullMsg += msg.text ?? "";
globalStore.set(this.updateLastMessageAtom, msg.text ?? "", true);
if (this.cancel) {
if (fullMsg == "") {
globalStore.set(this.removeLastMessageAtom);
}
break;
}
}
if (fullMsg == "") {
// remove a message if empty
globalStore.set(this.removeLastMessageAtom);
// only save the author's prompt
await BlockService.SaveWaveAiData(blockId, [...history, newPrompt]);
} else {
const responsePrompt: OpenAIPromptMessageType = {
role: "assistant",
content: fullMsg,
};
//mark message as complete
globalStore.set(this.updateLastMessageAtom, "", false);
if (fullMsg != "") {
const responsePrompt: OpenAIPromptMessageType = {
role: "assistant",
content: fullMsg,
};
await BlockService.SaveWaveAiData(blockId, [...history, newPrompt, responsePrompt]);
}
// save a complete message prompt and response
await BlockService.SaveWaveAiData(blockId, [...history, newPrompt, responsePrompt]);
}
} catch (error) {
const updatedHist = [...history, newPrompt];