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
JavaScript
;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