📦 NEW: onCancel function

Author: msaaddev

packages/langbase/src/react/use-pipe.ts

@@ -1,5 +1,5 @@
-import { Message, Role, RunResponse } from '@/langbase/langbase';
-import { getRunner, Runner } from '@/lib/helpers';
+import {Message, Role, RunResponse} from '@/langbase/langbase';
+import {getRunner, Runner} from '@/lib/helpers';
 import React, {useCallback, useMemo, useRef, useState} from 'react';
 import {z} from 'zod';
 
@@ -16,6 +16,7 @@ interface UsePipeOptions {
 	onFinish?: (messages: Message[], threadId: string) => void;
 	onConnect?: () => void;
 	onError?: (error: Error) => void;
+	onCancel?: (messages: Message[]) => void;
 	threadId?: string;
 	initialMessages?: Message[];
 	stream?: boolean;
@@ -33,6 +34,7 @@ export function usePipe({
 	threadId: initialThreadId,
 	initialMessages = [],
 	stream = true,
+	onCancel,
 }: UsePipeOptions = {}) {
 	const [messages, setMessages] = useState<Message[]>(initialMessages);
 	const [input, setInput] = useState('');
@@ -256,6 +258,7 @@ export function usePipe({
 
 	const stop = useCallback(() => {
 		abortControllerRef.current?.abort();
+		onCancel?.(messagesRef.current);
 		setIsLoading(false);
 	}, []);