diff --git a/packages/react/src/use-completion.ts b/packages/react/src/use-completion.ts index fedd8c314154..6054155a9a9a 100644 --- a/packages/react/src/use-completion.ts +++ b/packages/react/src/use-completion.ts @@ -198,6 +198,7 @@ export function useCompletion({ const handleSubmit = useCallback( (event?: { preventDefault?: () => void }) => { event?.preventDefault?.(); + setInput('') return input ? complete(input) : undefined; }, [input, complete], diff --git a/packages/react/src/use-completion.ui.test.tsx b/packages/react/src/use-completion.ui.test.tsx index e8df76392e8f..4c1741f5867b 100644 --- a/packages/react/src/use-completion.ui.test.tsx +++ b/packages/react/src/use-completion.ui.test.tsx @@ -163,4 +163,12 @@ describe('text stream', () => { 'Hello, world.', ); }); + + it('should reset text input after the submission', async () => { + await userEvent.type( + screen.getByTestId('input-text-stream'), + 'test input{enter};', + ); + expect(screen.getByTestId('input-text-stream')).toHaveTextContent(''); + }); });