Function useChatStreaming

cli/src/hooks/use-chat-streaming.ts:78–235

({
  agentMode,
  inputValue,
  setInputValue,
  terminalWidth,
  separatorWidth,
  isChainInProgressRef,
  activeAgentStreamsRef,
  sendMessageRef,
}: UseChatStreamingOptions)

}

export function useChatStreaming({
  agentMode,
  inputValue,
  setInputValue,
  terminalWidth,
  separatorWidth,
  isChainInProgressRef,
  activeAgentStreamsRef,
  sendMessageRef,
}: UseChatStreamingOptions): UseChatStreamingReturn {
  const queryClient = useQueryClient()
  const [, startUiTransition] = useTransition()

  // Reconnection state
  const [showReconnectionMessage, setShowReconnectionMessage] = useState(false)
  const reconnectionTimeout = useTimeout()

  // Reconnection handler
  const handleReconnection = useCallback(
    (isInitialConnection: boolean) => {
      queryClient.invalidateQueries({ queryKey: authQueryKeys.all })

      startUiTransition(() => {
        if (!isInitialConnection) {
          setShowReconnectionMessage(true)
          reconnectionTimeout.setTimeout(
            'reconnection-message',
            () => {
              startUiTransition(() => {
                setShowReconnectionMessage(false)
              })
            },
            RECONNECTION_MESSAGE_DURATION_MS,
          )
        }
      })
    },
    [queryClient, reconnectionTimeout, startUiTransition],
  )

  // Connection status
  const isConnected = useConnectionStatus(handleReconnection)

  // Timer
  const mainAgentTimer = useElapsedTime()
  const timerStartTime = mainAgentTimer.startTime

  // Pause/resume timer when ask_user tool becomes active/inactive
  const askUserState = useChatStore((state) => state.askUserState)
  useEffect(() => {
    if (askUserState !== null) {
      mainAgentTimer.pause()
    } else if (mainAgentTimer.isPaused) {
      mainAgentTimer.resume()
    }
  }, [askUserState, mainAgentTimer])

  // Message queue