Spaces:
Paused
Paused
File size: 2,643 Bytes
7d4338a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | import { createStore } from "/js/AlpineStore.js";
import { callJsonApi } from "/js/api.js";
import {
toastFrontendSuccess,
toastFrontendError,
} from "/components/notifications/notification-store.js";
export const store = createStore("compactStore", {
compacting: false,
stats: null,
showModal: false,
presets: [],
selectedPresetName: "",
useChatModel: true,
minTokens: 1000,
get canCompact() {
return this.stats && this.stats.token_count >= this.minTokens;
},
get selectedModelDisplay() {
if (this.selectedPresetName) {
const preset = this.presets.find(
(p) => p.name === this.selectedPresetName
);
if (preset) {
const cfg = this.useChatModel ? preset.chat : preset.utility;
if (cfg?.name) return cfg.name;
}
}
return this.useChatModel
? this.stats?.chat_model_name || "Chat Model"
: this.stats?.utility_model_name || "Utility Model";
},
async fetchStats() {
try {
const ctxid = globalThis.getContext?.();
if (!ctxid) {
toastFrontendError("No active chat", "Compaction");
return;
}
const [statsRes, presetsRes] = await Promise.all([
callJsonApi("/plugins/_chat_compaction/compact_chat", {
context: ctxid,
action: "stats",
}),
callJsonApi("/plugins/_model_config/model_presets", {
action: "get",
}),
]);
if (!statsRes?.ok) {
throw new Error(statsRes?.message || "Failed to fetch stats");
}
this.stats = statsRes.stats;
this.presets = presetsRes?.ok ? presetsRes.presets || [] : [];
this.showModal = true;
} catch (e) {
toastFrontendError(e.message, "Compaction");
}
},
async compact() {
this.compacting = true;
try {
const ctxid = globalThis.getContext?.();
if (!ctxid) {
throw new Error("No active chat");
}
const res = await callJsonApi("/plugins/_chat_compaction/compact_chat", {
context: ctxid,
action: "compact",
use_chat_model: this.useChatModel,
preset_name: this.selectedPresetName || null,
});
if (!res?.ok) {
throw new Error(res?.message || "Compaction failed");
}
toastFrontendSuccess("Compaction started", "Compaction");
this.showModal = false;
} catch (e) {
toastFrontendError(e.message, "Compaction");
} finally {
this.compacting = false;
}
},
closeModal() {
this.showModal = false;
this.stats = null;
this.presets = [];
this.selectedPresetName = "";
this.useChatModel = true;
},
});
|