diff --git a/packages/react/src/primitives/contentPart/ContentPartInProgress.tsx b/packages/react/src/primitives/contentPart/ContentPartInProgress.tsx index 63f5c5310..a52a3f235 100644 --- a/packages/react/src/primitives/contentPart/ContentPartInProgress.tsx +++ b/packages/react/src/primitives/contentPart/ContentPartInProgress.tsx @@ -3,11 +3,12 @@ import { useContentPart } from "../../context"; export type ContentPartPrimitiveInProgressProps = PropsWithChildren; -// TODO should this be renamed to IsRunning? export const ContentPartPrimitiveInProgress: FC< ContentPartPrimitiveInProgressProps > = ({ children }) => { - const isInProgress = useContentPart((c) => c.status.type === "running"); + const isInProgress = useContentPart( + (c) => c.status.type === "running" || c.status.type === "requires-action", + ); return isInProgress ? children : null; };