Operation   : Linux premium131.web-hosting.com 4.18.0-553.44.1.lve.el8.x86_64 #1 SMP Thu Mar 13 14:29:12 UTC 2025 x86_64
Software     : Apache
Server IP    : 162.0.232.56 | Your IP: 216.73.216.111
Domains      : 1034 Domain(s)
Permission   : [ 0755 ]

import { useEffect, useState } from '@wordpress/element';
import { completion } from '@draft/api/Data';

export const useCompletion = (
	prompt,
	promptType,
	systemMessageKey,
	details,
) => {
	const [result, setResult] = useState('');
	const [error, setError] = useState(null);
	const [loading, setLoading] = useState(false);

	useEffect(() => {
		let cancelled = false;
		let reader;
		const decoder = new TextDecoder();

		if (!prompt) {
			setLoading(false);
			setResult('');
			cancelled = true;
			return;
		}

		const fetchData = async () => {
			setResult('');
			setError(false);
			setLoading(true);

			const response = await completion(
				prompt,
				promptType,
				systemMessageKey,
				details,
			);
			reader = response.body.getReader();

			let done = false;
			while (!done) {
				const { value, done: readerDone } = await reader.read();

				done = readerDone;

				if (value && !cancelled) {
					const decodedValue = decoder.decode(value);
					setResult((prevResult) => prevResult + decodedValue);
				}
			}
		};

		fetchData()
			.finally(() => {
				if (!cancelled) {
					setLoading(false);
				}
			})
			.catch((error) => {
				if (!cancelled) {
					setError(error);
				}
			});

		return () => {
			cancelled = true;
			if (reader) {
				reader.cancel();
			}
		};
	}, [prompt, systemMessageKey, promptType, details]);

	return { completion: result, error, loading };
};