@composableai/react/src/useInteractionStreaming.ts

React helpers for Composable AI

import { InteractionBase } from "@composableai/client";
import { ExecutionRun, InteractionExecutionPayload } from "@composableai/common";
import { useMemo, useState } from "react";

export function useInteractionStreaming<TProps, TReturn>(interaction: InteractionBase<TProps, TReturn>) {

    const [isRunning, setRunning] = useState(false);
    const [text, setText] = useState('');

    const execute = useMemo(() => (payload?: InteractionExecutionPayload<TProps>): Promise<ExecutionRun<TProps, TReturn>> => {
        if (isRunning) {
            return Promise.reject(new Error('Trying to run the interaction while it is already running.'));
        }
        setRunning(true);
        let chunks: string[] = [];
        return interaction.execute(payload, (chunk: string) => {
            chunks.push(chunk);
            setText(chunks.join(''));
        }).then(run => {
            setText('');
            setRunning(false);
            return run;
        }).finally(() => {
            chunks = []
        });
    }, []);

    return { text, isRunning, execute }
}