All files / src/components/Chatbot InputContainer.tsx

100% Statements 22/22
100% Branches 4/4
100% Functions 1/1
100% Lines 22/22

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42    1x     1x           1x 1x 1x   1x 2x 2x   2x 2x 2x 2x 2x       2x 2x   2x 2x 2x     2x   2x 2x   2x  
// Libs
import { InputProps } from '@copilotkit/react-ui';
import { useCopilotMessagesContext } from '@copilotkit/react-core';
 
// Utils
import {
  isAssistantStillGenerating,
  isLastAssistantMessageComplete,
} from '@/utils/chat';
 
// Components
import StopGenerate from './StopGenerate';
import { CustomInput } from './CustomInput';
import { CopilotSuggestions } from './CopilotSuggestions';
 
export const InputContainer = (props: InputProps) => {
  const { inProgress } = props;
  const { messages } = useCopilotMessagesContext();
 
  const lastMessage = messages[messages.length - 1];
  const shouldShowStopButton = isAssistantStillGenerating(
    inProgress,
    lastMessage,
  );
 
  // Check message state to show suggestions immediately, without waiting
  // for inProgress to flip after the last message is complete.
  const shouldShowSuggestions =
    !inProgress || isLastAssistantMessageComplete(lastMessage);
 
  return (
    <div className="flex flex-col gap-2 dark:bg-gray-800">
      {shouldShowSuggestions && <CopilotSuggestions />}
 
      {/* Stop generating button */}
      {shouldShowStopButton && <StopGenerate />}
 
      <CustomInput {...props} />
    </div>
  );
};