Support ask mode in chat UI

[zhanba]

Problem Description

Currently, the Pochi chat UI operates in a single mode where users can input messages and receive responses. However, there are scenarios where the assistant needs to ask follow-up questions using the askFollowupQuestion tool, and the current UI doesn't provide an optimized experience for this interactive question-and-answer flow.

Current State

The codebase already has:

• askFollowupQuestion tool implementation
• AskFollowupQuestionTool component that renders questions with clickable follow-up options
• CLI version that handles follow-up questions interactively

Proposed Feature

Add an "ask mode" to the chat UI that provides a specialized interface when the assistant is asking follow-up questions. This mode should:

Core Features

1. Visual Mode Indicator: Clear indication when the UI is in "ask mode" vs normal chat mode
2. Enhanced Question Display: Prominent display of the assistant's question
3. Quick Response Options: Easy-to-click buttons for suggested follow-up responses
4. Streamlined Input: Simplified input area focused on answering the question
5. Context Preservation: Maintain conversation history while highlighting the current question

Technical Implementation Areas

The implementation would likely involve:

• Chat State Management: Extend existing chat state in packages/vscode-webui/src/features/chat/lib/chat-state/
• Form Editor Enhancement: Modify FormEditor to support ask mode
• Tool Integration: Enhance AskFollowupQuestionTool component
• Chat Input Form: Update ChatInputForm to handle mode switching

UI/UX Considerations

• Mode Transitions: Smooth transitions between normal chat and ask mode
• Mobile/Responsive: Ensure ask mode works well in VSCode panels and responsive layouts
• Accessibility: Proper ARIA labels and keyboard navigation for ask mode
• Theme Support: Integration with VSCode theme system

User Benefits

• Improved Workflow: More intuitive interaction when assistant needs clarification
• Faster Responses: Quick selection from suggested options reduces typing
• Better Context: Clear visual separation between questions and regular conversation
• Consistency: Aligns VSCode UI experience with CLI interactive features

Related Code References

• Tool definition: packages/tools/src/ask-followup-question.ts
• Tool component: packages/vscode-webui/src/components/tool-invocation/tools/ask-followup-question.tsx
• Chat state: packages/vscode-webui/src/features/chat/lib/chat-state/
• CLI implementation: packages/cli/src/output-renderer.ts#L223

🤖 Generated with Pochi