27 lines
636 B
TypeScript
27 lines
636 B
TypeScript
|
/**
|
||
|
|
* @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;
|
||
|
|
};
|