Spaces:
Sleeping
Sleeping
File size: 1,230 Bytes
47203d3 b49a2dd 47203d3 | 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 | import { create } from 'zustand'
import type { Message } from '../types'
interface InterviewState {
sessionId: string | null
topicId: string | null
messages: Message[]
turnCount: number
maxTurns: number
status: 'idle' | 'loading' | 'waiting' | 'finished'
}
interface InterviewActions {
startSession: (sessionId: string, topicId: string, firstMessage: string) => void
addMessage: (message: Message) => void
setTurnCount: (count: number) => void
setStatus: (status: InterviewState['status']) => void
reset: () => void
}
const initialState: InterviewState = {
sessionId: null,
topicId: null,
messages: [],
turnCount: 0,
maxTurns: 5,
status: 'idle',
}
export const useInterviewStore = create<InterviewState & InterviewActions>((set) => ({
...initialState,
startSession: (sessionId, topicId, firstMessage) =>
set({
sessionId,
topicId,
messages: [{ role: 'ai', content: firstMessage }],
turnCount: 0,
status: 'waiting',
}),
addMessage: (message) =>
set((state) => ({ messages: [...state.messages, message] })),
setTurnCount: (count) => set({ turnCount: count }),
setStatus: (status) => set({ status }),
reset: () => set(initialState),
}))
|