packages/cli/src/ui/contexts/StreamingContext.tsx


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

/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import React, { createContext } from 'react';
import { StreamingState } from '../types.js';

export interface StreamingContextType {
  streamingState: StreamingState;
}

export const StreamingContext = createContext<StreamingContextType | undefined>(
  undefined,
);

export const useStreamingContext = (): StreamingContextType => {
  const context = React.useContext(StreamingContext);
  if (context === undefined) {
    throw new Error(
      'useStreamingContext must be used within a StreamingContextProvider',
    );
  }
  return context;
};