Files
automaker/apps/ui/src/components/views/agent-view.tsx
DhanushSantosh 27c6d5a3bb refactor: improve error handling and CLI integration
- Updated CodexProvider to read prompts from stdin to prevent shell escaping issues.
- Enhanced AgentService to handle streamed error messages from providers, ensuring a consistent user experience.
- Modified UI components to display error messages clearly, including visual indicators for errors in chat bubbles.
- Updated CLI status handling to support both Claude and Codex APIs, improving compatibility and user feedback.

These changes enhance the robustness of the application and improve the user experience during error scenarios.
2026-01-07 19:09:25 +05:30

216 lines
6.9 KiB
TypeScript

import { useState, useCallback, useRef, useEffect } from 'react';
import { useAppStore } from '@/store/app-store';
import type { PhaseModelEntry } from '@automaker/types';
import { useElectronAgent } from '@/hooks/use-electron-agent';
import { SessionManager } from '@/components/session-manager';
// Extracted hooks
import {
useAgentScroll,
useFileAttachments,
useAgentShortcuts,
useAgentSession,
} from './agent-view/hooks';
// Extracted components
import { NoProjectState, AgentHeader, ChatArea } from './agent-view/components';
import { AgentInputArea } from './agent-view/input-area';
export function AgentView() {
const { currentProject } = useAppStore();
const [input, setInput] = useState('');
const [currentTool, setCurrentTool] = useState<string | null>(null);
const [showSessionManager, setShowSessionManager] = useState(true);
const [modelSelection, setModelSelection] = useState<PhaseModelEntry>({ model: 'sonnet' });
// Input ref for auto-focus
const inputRef = useRef<HTMLTextAreaElement>(null);
// Ref for quick create session function from SessionManager
const quickCreateSessionRef = useRef<(() => Promise<void>) | null>(null);
// Session management hook
const { currentSessionId, handleSelectSession } = useAgentSession({
projectPath: currentProject?.path,
});
// Use the Electron agent hook (only if we have a session)
const {
messages,
isProcessing,
isConnected,
sendMessage,
clearHistory,
stopExecution,
error: agentError,
serverQueue,
addToServerQueue,
removeFromServerQueue,
clearServerQueue,
} = useElectronAgent({
sessionId: currentSessionId || '',
workingDirectory: currentProject?.path,
model: modelSelection.model,
thinkingLevel: modelSelection.thinkingLevel,
onToolUse: (toolName) => {
setCurrentTool(toolName);
setTimeout(() => setCurrentTool(null), 2000);
},
});
// File attachments hook
const fileAttachments = useFileAttachments({
isProcessing,
isConnected,
});
// Scroll management hook
const { messagesContainerRef, handleScroll } = useAgentScroll({
messagesLength: messages.length,
currentSessionId,
});
// Keyboard shortcuts hook
useAgentShortcuts({
currentProject,
quickCreateSessionRef,
});
// Handle send message
const handleSend = useCallback(async () => {
const {
selectedImages,
selectedTextFiles,
setSelectedImages,
setSelectedTextFiles,
setShowImageDropZone,
} = fileAttachments;
if (!input.trim() && selectedImages.length === 0 && selectedTextFiles.length === 0) return;
const messageContent = input;
const messageImages = selectedImages;
const messageTextFiles = selectedTextFiles;
setInput('');
setSelectedImages([]);
setSelectedTextFiles([]);
setShowImageDropZone(false);
// If already processing, add to server queue instead
if (isProcessing) {
await addToServerQueue(messageContent, messageImages, messageTextFiles);
} else {
await sendMessage(messageContent, messageImages, messageTextFiles);
}
}, [input, fileAttachments, isProcessing, sendMessage, addToServerQueue]);
const handleClearChat = async () => {
if (!confirm('Are you sure you want to clear this conversation?')) return;
await clearHistory();
};
// Auto-focus input when session is selected/changed
useEffect(() => {
if (currentSessionId && inputRef.current) {
setTimeout(() => {
inputRef.current?.focus();
}, 200);
}
}, [currentSessionId]);
// Show welcome message if no messages yet
const displayMessages =
messages.length === 0
? [
{
id: 'welcome',
role: 'assistant' as const,
content:
"Hello! I'm the Automaker Agent. I can help you build software autonomously. I can read and modify files in this project, run commands, and execute tests. What would you like to create today?",
timestamp: new Date().toISOString(),
},
]
: messages;
if (!currentProject) {
return <NoProjectState />;
}
return (
<div className="flex-1 flex overflow-hidden bg-background" data-testid="agent-view">
{/* Session Manager Sidebar */}
{showSessionManager && currentProject && (
<div className="w-80 border-r border-border shrink-0 bg-card/50">
<SessionManager
currentSessionId={currentSessionId}
onSelectSession={handleSelectSession}
projectPath={currentProject.path}
isCurrentSessionThinking={isProcessing}
onQuickCreateRef={quickCreateSessionRef}
/>
</div>
)}
{/* Chat Area */}
<div className="flex-1 flex flex-col overflow-hidden">
{/* Header */}
<AgentHeader
projectName={currentProject.name}
currentSessionId={currentSessionId}
isConnected={isConnected}
isProcessing={isProcessing}
currentTool={currentTool}
messagesCount={messages.length}
showSessionManager={showSessionManager}
onToggleSessionManager={() => setShowSessionManager(!showSessionManager)}
onClearChat={handleClearChat}
/>
{/* Messages */}
<ChatArea
currentSessionId={currentSessionId}
messages={displayMessages}
isProcessing={isProcessing}
showSessionManager={showSessionManager}
messagesContainerRef={messagesContainerRef}
onScroll={handleScroll}
onShowSessionManager={() => setShowSessionManager(true)}
/>
{/* Input Area */}
{currentSessionId && (
<AgentInputArea
input={input}
onInputChange={setInput}
onSend={handleSend}
onStop={stopExecution}
modelSelection={modelSelection}
onModelSelect={setModelSelection}
isProcessing={isProcessing}
isConnected={isConnected}
selectedImages={fileAttachments.selectedImages}
selectedTextFiles={fileAttachments.selectedTextFiles}
showImageDropZone={fileAttachments.showImageDropZone}
isDragOver={fileAttachments.isDragOver}
onImagesSelected={fileAttachments.handleImagesSelected}
onToggleImageDropZone={fileAttachments.toggleImageDropZone}
onRemoveImage={fileAttachments.removeImage}
onRemoveTextFile={fileAttachments.removeTextFile}
onClearAllFiles={fileAttachments.clearAllFiles}
onDragEnter={fileAttachments.handleDragEnter}
onDragLeave={fileAttachments.handleDragLeave}
onDragOver={fileAttachments.handleDragOver}
onDrop={fileAttachments.handleDrop}
onPaste={fileAttachments.handlePaste}
serverQueue={serverQueue}
onRemoveFromQueue={removeFromServerQueue}
onClearQueue={clearServerQueue}
inputRef={inputRef}
/>
)}
</div>
</div>
);
}