wip: first run of implementation

src/renderer/components/ChatPanel/ChatPanel.tsx

@@ -5,6 +5,7 @@ import { useChatSurfaceState } from '../../navigation/useChatSurfaceState';
 import { getChatSurfaceMode } from '../../navigation/chatSurfaceMode';
 import { dispatchAssistantAction } from '../../navigation/assistantActionDispatcher';
 import { extractAssistantResponseContent, type AssistantPanelElement } from '../../navigation/assistantPanelSpec';
+import { toClarificationElements } from '../../navigation/protocolNeedsInput';
 import { useAppStore } from '../../store';
 import { ChatTranscript } from '../ChatSurface';
 import { AssistantPanelControls } from '../AssistantPanelControls';
@@ -28,6 +29,7 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ conversationId }) => {
   const [apiKeyError, setApiKeyError] = useState('');
   const [isValidating, setIsValidating] = useState(false);
   const [panelElements, setPanelElements] = useState<AssistantPanelElement[]>([]);
+  const [actionPolicies, setActionPolicies] = useState<Record<string, 'silent' | 'confirm' | 'danger'>>({});
   const [actionError, setActionError] = useState<string | null>(null);
   const messagesEndRef = useRef<HTMLDivElement>(null);
   const inputRef = useRef<HTMLTextAreaElement>(null);
@@ -190,21 +192,36 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ conversationId }) => {
       // Fall back to the backend result message if streaming didn't capture the content
       const assistantContent = getStreamingContent() || (result.success ? result.message : '');
 
-      if (assistantContent) {
+      if (result.envelope) {
+        finalizeAssistantTurn(conversationId, result.envelope.assistantText);
+        const uiElements = Array.isArray(result.envelope.ui?.elements)
+          ? (result.envelope.ui?.elements as AssistantPanelElement[])
+          : toClarificationElements(result.envelope.needsInput);
+        setPanelElements(uiElements);
+        setActionPolicies(
+          result.envelope.actions.reduce<Record<string, 'silent' | 'confirm' | 'danger'>>((accumulator, action) => {
+            accumulator[action.action] = action.policy;
+            return accumulator;
+          }, {}),
+        );
+      } else if (assistantContent) {
         const parsedResponse = extractAssistantResponseContent(assistantContent);
         finalizeAssistantTurn(conversationId, parsedResponse.displayText);
         setPanelElements(parsedResponse.panelSpec?.elements ?? []);
+        setActionPolicies({});
       } else if (!result.success) {
         // Backend returned an error (API failure, model unavailable, etc.)
         appendAssistantMessage(conversationId, tr('chat.errorPrefix', { error: result.error || tr('chat.errorNoResponse') }));
         stopStreaming();
         setPanelElements([]);
+        setActionPolicies({});
       } else {
         // No content from streaming AND no error, but also no success message
         // This can happen with some models that don't return content properly
         appendAssistantMessage(conversationId, tr('chat.errorEmptyResponse'));
         stopStreaming();
         setPanelElements([]);
+        setActionPolicies({});
       }
     } catch (error) {
       console.error('Failed to send message:', error);
@@ -226,7 +243,64 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ conversationId }) => {
     }
   };
 
+  const handleNeedsInputSubmit = async (payload?: Record<string, unknown>) => {
+    const values = payload?.values;
+    if (!values || typeof values !== 'object') {
+      setActionError(tr('assistantSidebar.error.actionFailed'));
+      return;
+    }
+
+    const clarificationMessage = `needs_input_response: ${JSON.stringify(values)}`;
+    beginUserTurn(conversationId, clarificationMessage);
+
+    try {
+      const result = await sendChatMessage({
+        conversationId,
+        message: clarificationMessage,
+        metadata: { surface: 'tab' },
+      });
+
+      if (!result.success) {
+        appendAssistantMessage(conversationId, tr('chat.errorPrefix', { error: result.error || tr('chat.errorNoResponse') }));
+        stopStreaming();
+        return;
+      }
+
+      if (result.envelope) {
+        finalizeAssistantTurn(conversationId, result.envelope.assistantText);
+        const uiElements = Array.isArray(result.envelope.ui?.elements)
+          ? (result.envelope.ui?.elements as AssistantPanelElement[])
+          : toClarificationElements(result.envelope.needsInput);
+        setPanelElements(uiElements);
+        setActionPolicies(
+          result.envelope.actions.reduce<Record<string, 'silent' | 'confirm' | 'danger'>>((accumulator, action) => {
+            accumulator[action.action] = action.policy;
+            return accumulator;
+          }, {}),
+        );
+        return;
+      }
+
+      const assistantContent = getStreamingContent() || result.message;
+      if (assistantContent) {
+        const parsedResponse = extractAssistantResponseContent(assistantContent);
+        finalizeAssistantTurn(conversationId, parsedResponse.displayText);
+        setPanelElements(parsedResponse.panelSpec?.elements ?? []);
+        setActionPolicies({});
+      }
+    } catch (error) {
+      console.error('Failed to submit clarification:', error);
+      appendAssistantMessage(conversationId, tr('chat.errorGeneric'));
+      stopStreaming();
+    }
+  };
+
   const handleAssistantAction = (action: string, payload?: Record<string, unknown>) => {
+    if (action === 'submitNeedsInput') {
+      void handleNeedsInputSubmit(payload);
+      return;
+    }
+
     const result = dispatchAssistantAction(
       {
         action,
@@ -377,7 +451,7 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ conversationId }) => {
         />
 
         {panelElements.length > 0 && (
-          <AssistantPanelControls elements={panelElements} onAction={handleAssistantAction} />
+          <AssistantPanelControls elements={panelElements} onAction={handleAssistantAction} actionPolicies={actionPolicies} />
         )}
 
         {actionError && <p className="chat-surface-error">{actionError}</p>}