assistant-stream/dist/utils/json/parse-partial-json-object.js

Streaming utilities for AI assistants

import { fixJson } from "./fix-json.js";
import sjson from "secure-json-parse";
// src/utils/json/parse-partial-json-object.ts
const PARTIAL_JSON_OBJECT_META_SYMBOL = Symbol("aui.parse-partial-json-object.meta");
const getPartialJsonObjectMeta = (obj) => {
	return obj?.[PARTIAL_JSON_OBJECT_META_SYMBOL];
};
/** Mark an already-finalized JSON object as complete without parsing it again. */
const markPartialJsonObjectComplete = (obj) => {
	const copy = Object.fromEntries(Object.entries(obj));
	Object.defineProperty(copy, PARTIAL_JSON_OBJECT_META_SYMBOL, {
		value: {
			state: "complete",
			partialPath: []
		},
		enumerable: true
	});
	return copy;
};
const parsePartialJsonObject = (json) => {
	if (json.length === 0) return { [PARTIAL_JSON_OBJECT_META_SYMBOL]: {
		state: "partial",
		partialPath: []
	} };
	try {
		const res = sjson.parse(json);
		if (typeof res !== "object" || res === null) throw new Error("argsText is expected to be an object");
		res[PARTIAL_JSON_OBJECT_META_SYMBOL] = {
			state: "complete",
			partialPath: []
		};
		return res;
	} catch {
		try {
			const [fixedJson, partialPath] = fixJson(json);
			const res = sjson.parse(fixedJson);
			if (typeof res !== "object" || res === null) throw new Error("argsText is expected to be an object");
			res[PARTIAL_JSON_OBJECT_META_SYMBOL] = {
				state: "partial",
				partialPath
			};
			return res;
		} catch {
			return;
		}
	}
};
const getFieldState = (parent, parentMeta, fieldPath) => {
	if (typeof parent !== "object" || parent === null) return parentMeta.state;
	if (parentMeta.state === "complete") return "complete";
	if (fieldPath.length === 0) return parentMeta.state;
	const [field, ...restPath] = fieldPath;
	if (!Object.hasOwn(parent, field)) return "partial";
	const [partialField, ...restPartialPath] = parentMeta.partialPath;
	if (field !== partialField) return "complete";
	const child = parent[field];
	return getFieldState(child, {
		state: "partial",
		partialPath: restPartialPath
	}, restPath);
};
const getPartialJsonObjectFieldState = (obj, fieldPath) => {
	const meta = getPartialJsonObjectMeta(obj);
	if (!meta) throw new Error("unable to determine object state");
	return getFieldState(obj, meta, fieldPath.map(String));
};
export { getPartialJsonObjectFieldState, getPartialJsonObjectMeta, markPartialJsonObjectComplete, parsePartialJsonObject };

//# sourceMappingURL=parse-partial-json-object.js.map