File size: 5,481 Bytes
5f40163 | 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 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 | import { create } from "zustand";
import { devtools } from "zustand/middleware";
import { v4 as uuidv4 } from "uuid";
import type { ProfileInfo } from "#/api/profiles-service/profiles-service.api";
export interface ModelListEntry {
id: string;
/**
* Id of the chat event after which this entry should render, or `null` to
* pin it to the top of the chat history (no rendered events at the time
* of /model).
*/
anchorEventId: string | null;
profiles: ProfileInfo[];
switchedTo?: string;
}
/**
* A historical switch to seed via `seedSwitches`. `id` must be stable across
* reloads (derived from the source observation event id) so re-seeding the
* same loaded history is idempotent.
*/
export interface SeededSwitch {
id: string;
anchorEventId: string | null;
profileName: string;
}
interface ModelState {
entriesByConversation: Record<string, ModelListEntry[]>;
/**
* Most-recently-switched profile name per conversation. Updated by
* `recordSwitch` so the UI (button label, popover check mark) reflects the
* new selection instantly, before the conversation refetch from the agent
* server lands.
*/
activeProfileByConversation: Record<string, string>;
}
interface ModelActions {
show: (
conversationId: string,
anchorEventId: string | null,
profiles: ProfileInfo[],
) => void;
recordSwitch: (
conversationId: string,
anchorEventId: string | null,
profileName: string,
) => void;
/**
* Seeds "Switched to" entries reconstructed from loaded history. Skips any
* whose `id` is already present, so it can run on every history (re)load
* without duplicating, and it preserves live-recorded entries.
*/
seedSwitches: (conversationId: string, switches: SeededSwitch[]) => void;
/**
* Sets the optimistic active-profile entry for a conversation without
* appending a chat entry — the reload/history path re-derives the stamp
* from past SwitchLLM observations, which seed their chat entries through
* `seedSwitches` instead of `recordSwitch`.
*/
setActiveProfile: (conversationId: string, profileName: string) => void;
/** Drops only the optimistic active-profile entry for a conversation. */
clearActiveProfile: (conversationId: string) => void;
clear: (conversationId: string) => void;
clearAll: () => void;
}
type ModelStore = ModelState & ModelActions;
const appendEntry = (
state: ModelState,
conversationId: string,
entry: ModelListEntry,
): Pick<ModelState, "entriesByConversation"> => ({
entriesByConversation: {
...state.entriesByConversation,
[conversationId]: [
...(state.entriesByConversation[conversationId] ?? []),
entry,
],
},
});
export const useModelStore = create<ModelStore>()(
devtools(
(set) => ({
entriesByConversation: {},
activeProfileByConversation: {},
show: (conversationId, anchorEventId, profiles) =>
set((s) =>
appendEntry(s, conversationId, {
id: uuidv4(),
anchorEventId,
profiles,
}),
),
recordSwitch: (conversationId, anchorEventId, profileName) =>
set((s) => ({
...appendEntry(s, conversationId, {
id: uuidv4(),
anchorEventId,
profiles: [],
switchedTo: profileName,
}),
activeProfileByConversation: {
...s.activeProfileByConversation,
[conversationId]: profileName,
},
})),
seedSwitches: (conversationId, switches) =>
set((s) => {
const existing = s.entriesByConversation[conversationId] ?? [];
const existingIds = new Set(existing.map((e) => e.id));
const additions = switches
.filter((sw) => !existingIds.has(sw.id))
.map(
(sw): ModelListEntry => ({
id: sw.id,
anchorEventId: sw.anchorEventId,
profiles: [],
switchedTo: sw.profileName,
}),
);
if (additions.length === 0) return s;
return {
entriesByConversation: {
...s.entriesByConversation,
[conversationId]: [...existing, ...additions],
},
};
}),
setActiveProfile: (conversationId, profileName) =>
set((s) => ({
activeProfileByConversation: {
...s.activeProfileByConversation,
[conversationId]: profileName,
},
})),
clearActiveProfile: (conversationId) =>
set((s) => {
if (!(conversationId in s.activeProfileByConversation)) return s;
const activeProfileByConversation = {
...s.activeProfileByConversation,
};
delete activeProfileByConversation[conversationId];
return { activeProfileByConversation };
}),
clear: (conversationId) =>
set((s) => {
const entriesByConversation = { ...s.entriesByConversation };
delete entriesByConversation[conversationId];
const activeProfileByConversation = {
...s.activeProfileByConversation,
};
delete activeProfileByConversation[conversationId];
return { entriesByConversation, activeProfileByConversation };
}),
clearAll: () =>
set({ entriesByConversation: {}, activeProfileByConversation: {} }),
}),
{ name: "ModelStore" },
),
);
|