UNPKG

stream-hooks

Version:

<div align="center"> <a aria-label="NPM version" href="https://twitter.com/dimitrikennedy"> <img alt="stream-hooks" src="https://img.shields.io/twitter/follow/dimitrikennedy?style=social&labelColor=000000"> </a> <a aria-label="GH Issues" href="h

70 lines (64 loc) 2.71 kB
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }"use client" var _chunkVBU466HQjs = require('./chunk-VBU466HQ.js'); // src/hooks/use-json-stream.tsx var _react = require('react'); var _zodstream = require('zod-stream'); var _zodstream2 = _interopRequireDefault(_zodstream); function useJsonStream({ onReceive, onEnd, schema, onBeforeStart, onStop, defaultData = {} }) { const streamClient = _react.useRef.call(void 0, new (0, _zodstream2.default)({})); const stubbedValue = streamClient.current.getSchemaStub({ schema, defaultData }); const [json, setJson] = _react.useState.call(void 0, stubbedValue); const [loading, setLoading] = _react.useState.call(void 0, false); const { startStream: startStreamBase, stopStream: stopStreamBase } = _chunkVBU466HQjs.useStream.call(void 0, { onBeforeStart, onStop }); const stopStream = _react.useCallback.call(void 0, () => { setLoading(false); stopStreamBase(); }, [stopStreamBase]); const startStream = _react.useCallback.call(void 0, async (streamProps) => { setLoading(true); try { const extractionStream = await streamClient.current.create({ completionPromise: async () => await startStreamBase(streamProps), response_model: { schema } }); let final = {}; for await (const data of extractionStream) { final = data; onReceive && onReceive(data); setJson(data); } setLoading(false); onEnd && onEnd(final); } catch (err) { setLoading(false); if (_optionalChain([err, 'optionalAccess', _ => _.name]) === "AbortError") { console.warn("useJsonStream: aborted", err); return null; } stopStream(); throw err; } finally { setLoading(false); } }, [onEnd, onReceive, schema, startStreamBase, stopStream] ); return { startStream, stopStream, loading, data: json }; } exports.useJsonStream = useJsonStream; //# sourceMappingURL=chunk-XPHI5HCQ.js.map