111 lines
2.9 KiB
JavaScript
111 lines
2.9 KiB
JavaScript
const createOnGoingConvState = () => ({
|
|
msgChunks: {},
|
|
currentRequestKey: null,
|
|
currentAssistantKey: null,
|
|
toolCallBuffers: {}
|
|
})
|
|
|
|
export function useAgentThreadState({
|
|
chatState,
|
|
getCurrentThreadId,
|
|
onStopThread = null,
|
|
onBeforeResetThread = null,
|
|
onBeforeCleanupThread = null
|
|
}) {
|
|
const resetThreadUiState = (threadState) => {
|
|
if (!threadState) return
|
|
threadState.replyLoadingVisible = false
|
|
threadState.pendingRequestId = null
|
|
}
|
|
|
|
const getThreadState = (threadId) => {
|
|
if (!threadId) return null
|
|
if (!chatState.threadStates[threadId]) {
|
|
chatState.threadStates[threadId] = {
|
|
isStreaming: false,
|
|
streamAbortController: null,
|
|
runStreamAbortController: null,
|
|
activeRunId: null,
|
|
runLastSeq: '0',
|
|
lastRetryableJobTry: null,
|
|
replyLoadingVisible: false,
|
|
pendingRequestId: null,
|
|
onGoingConv: createOnGoingConvState(),
|
|
agentState: null
|
|
}
|
|
}
|
|
return chatState.threadStates[threadId]
|
|
}
|
|
|
|
const stopThreadStream = (threadId) => {
|
|
if (!threadId) return
|
|
const threadState = chatState.threadStates[threadId]
|
|
if (typeof onStopThread === 'function') {
|
|
onStopThread(threadId)
|
|
}
|
|
|
|
if (!threadState?.streamAbortController) return
|
|
|
|
threadState.streamAbortController.abort()
|
|
threadState.streamAbortController = null
|
|
threadState.isStreaming = false
|
|
resetThreadUiState(threadState)
|
|
}
|
|
|
|
const cleanupThreadState = (threadId) => {
|
|
if (!threadId) return
|
|
const threadState = chatState.threadStates[threadId]
|
|
if (!threadState) return
|
|
|
|
if (typeof onBeforeCleanupThread === 'function') {
|
|
onBeforeCleanupThread(threadId)
|
|
}
|
|
|
|
if (threadState.streamAbortController) {
|
|
threadState.streamAbortController.abort()
|
|
}
|
|
if (threadState.runStreamAbortController) {
|
|
threadState.runStreamAbortController.abort()
|
|
}
|
|
delete chatState.threadStates[threadId]
|
|
}
|
|
|
|
const resetOnGoingConv = (threadId = null) => {
|
|
const targetThreadId =
|
|
threadId || (typeof getCurrentThreadId === 'function' ? getCurrentThreadId() : null)
|
|
|
|
if (targetThreadId) {
|
|
const threadState = getThreadState(targetThreadId)
|
|
if (!threadState) return
|
|
|
|
if (typeof onBeforeResetThread === 'function') {
|
|
onBeforeResetThread(targetThreadId)
|
|
}
|
|
|
|
if (threadState.streamAbortController) {
|
|
threadState.streamAbortController.abort()
|
|
threadState.streamAbortController = null
|
|
}
|
|
if (threadState.runStreamAbortController) {
|
|
threadState.runStreamAbortController.abort()
|
|
threadState.runStreamAbortController = null
|
|
}
|
|
|
|
threadState.onGoingConv = createOnGoingConvState()
|
|
resetThreadUiState(threadState)
|
|
return
|
|
}
|
|
|
|
Object.keys(chatState.threadStates).forEach((id) => {
|
|
cleanupThreadState(id)
|
|
})
|
|
}
|
|
|
|
return {
|
|
getThreadState,
|
|
cleanupThreadState,
|
|
resetOnGoingConv,
|
|
stopThreadStream
|
|
}
|
|
}
|