685114d253
### What problem does this PR solve? Feat: Display the pipeline on the agent canvas #9869 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
21 lines
515 B
TypeScript
21 lines
515 B
TypeScript
import { useCancelDataflow } from '@/hooks/use-agent-request';
|
|
import { useCallback } from 'react';
|
|
|
|
export function useCancelCurrentDataflow({
|
|
messageId,
|
|
stopFetchTrace,
|
|
}: {
|
|
messageId: string;
|
|
stopFetchTrace(): void;
|
|
}) {
|
|
const { cancelDataflow } = useCancelDataflow();
|
|
|
|
const handleCancel = useCallback(async () => {
|
|
const code = await cancelDataflow(messageId);
|
|
if (code === 0) {
|
|
stopFetchTrace();
|
|
}
|
|
}, [cancelDataflow, messageId, stopFetchTrace]);
|
|
|
|
return { handleCancel };
|
|
}
|