Spaces:
Runtime error
Runtime error
File size: 2,362 Bytes
1a12d36 | 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 | import { create } from 'zustand';
import { persist } from 'zustand/middleware';
import type { SessionMeta } from '@/types/agent';
import { useAgentStore } from './agentStore';
interface SessionStore {
sessions: SessionMeta[];
activeSessionId: string | null;
// Actions
createSession: (id: string) => void;
deleteSession: (id: string) => void;
switchSession: (id: string) => void;
setSessionActive: (id: string, isActive: boolean) => void;
updateSessionTitle: (id: string, title: string) => void;
}
export const useSessionStore = create<SessionStore>()(
persist(
(set, get) => ({
sessions: [],
activeSessionId: null,
createSession: (id: string) => {
const newSession: SessionMeta = {
id,
title: `Chat ${get().sessions.length + 1}`,
createdAt: new Date().toISOString(),
isActive: true,
};
set((state) => ({
sessions: [...state.sessions, newSession],
activeSessionId: id,
}));
},
deleteSession: (id: string) => {
// Clean up persisted messages for this session
useAgentStore.getState().deleteSessionMessages(id);
set((state) => {
const newSessions = state.sessions.filter((s) => s.id !== id);
const newActiveId =
state.activeSessionId === id
? newSessions[newSessions.length - 1]?.id || null
: state.activeSessionId;
return {
sessions: newSessions,
activeSessionId: newActiveId,
};
});
},
switchSession: (id: string) => {
set({ activeSessionId: id });
},
setSessionActive: (id: string, isActive: boolean) => {
set((state) => ({
sessions: state.sessions.map((s) =>
s.id === id ? { ...s, isActive } : s
),
}));
},
updateSessionTitle: (id: string, title: string) => {
set((state) => ({
sessions: state.sessions.map((s) =>
s.id === id ? { ...s, title } : s
),
}));
},
}),
{
name: 'hf-agent-sessions',
partialize: (state) => ({
sessions: state.sessions,
activeSessionId: state.activeSessionId,
}),
}
)
);
|