Tool Picker

This commit is contained in:
Connor Yoh 2025-08-19 15:28:27 +01:00
parent 58e2a50b74
commit b9b52f2269
7 changed files with 425 additions and 85 deletions

View File

@ -5,8 +5,8 @@ import { ToolRegistryEntry } from "../../data/toolsTaxonomy";
import ToolButton from "./toolPicker/ToolButton";
import "./toolPicker/ToolPicker.css";
import { useToolSections } from "../../hooks/useToolSections";
import SubcategoryHeader from "./shared/SubcategoryHeader";
import NoToolsFound from "./shared/NoToolsFound";
import { renderToolButtons } from "./shared/renderToolButtons";
interface ToolPickerProps {
selectedToolKey: string | null;
@ -15,31 +15,6 @@ interface ToolPickerProps {
isSearching?: boolean;
}
// Helper function to render tool buttons for a subcategory
const renderToolButtons = (
subcategory: any,
selectedToolKey: string | null,
onSelect: (id: string) => void,
showSubcategoryHeader: boolean = true
) => (
<Box key={subcategory.subcategory} w="100%">
{showSubcategoryHeader && (
<SubcategoryHeader label={subcategory.subcategory} />
)}
<Stack gap="xs">
{subcategory.tools.map(({ id, tool }: { id: string; tool: any }) => (
<ToolButton
key={id}
id={id}
tool={tool}
isSelected={selectedToolKey === id}
onSelect={onSelect}
/>
))}
</Stack>
</Box>
);
const ToolPicker = ({ selectedToolKey, onSelect, filteredTools, isSearching = false }: ToolPickerProps) => {
const { t } = useTranslation();
const [quickHeaderHeight, setQuickHeaderHeight] = useState(0);

View File

@ -1,31 +1,29 @@
import React, { useState, useEffect } from 'react';
import { useTranslation } from 'react-i18next';
import {
Button,
Text,
Title,
Stack,
Group,
Select,
TextInput,
import {
Button,
Text,
Title,
Stack,
Group,
TextInput,
ActionIcon,
Divider
} from '@mantine/core';
import AddIcon from '@mui/icons-material/Add';
import DeleteIcon from '@mui/icons-material/Delete';
import SettingsIcon from '@mui/icons-material/Settings';
import ArrowBackIcon from '@mui/icons-material/ArrowBack';
import CheckIcon from '@mui/icons-material/Check';
import CloseIcon from '@mui/icons-material/Close';
import { useToolWorkflow } from '../../../contexts/ToolWorkflowContext';
import { useFlatToolRegistry } from '../../../data/useTranslatedToolRegistry';
import ToolConfigurationModal from './ToolConfigurationModal';
import AutomationEntry from './AutomationEntry';
import ToolSelector from './ToolSelector';
interface AutomationCreationProps {
mode: 'custom' | 'suggested' | 'create';
existingAutomation?: any;
onBack: () => void;
onComplete: () => void;
onComplete: (automation: any) => void;
}
interface AutomationTool {
@ -38,8 +36,8 @@ interface AutomationTool {
export default function AutomationCreation({ mode, existingAutomation, onBack, onComplete }: AutomationCreationProps) {
const { t } = useTranslation();
const { toolRegistry } = useToolWorkflow();
const toolRegistry = useFlatToolRegistry();
const [automationName, setAutomationName] = useState('');
const [selectedTools, setSelectedTools] = useState<AutomationTool[]>([]);
const [configModalOpen, setConfigModalOpen] = useState(false);
@ -49,7 +47,7 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
useEffect(() => {
if (mode === 'suggested' && existingAutomation) {
setAutomationName(existingAutomation.name);
const tools = existingAutomation.operations.map((op: string) => ({
id: `${op}-${Date.now()}`,
operation: op,
@ -57,7 +55,7 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
configured: false,
parameters: {}
}));
setSelectedTools(tools);
}
}, [mode, existingAutomation]);
@ -67,20 +65,8 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
return tool?.name || t(`tools.${operation}.name`, operation);
};
const getAvailableTools = () => {
if (!toolRegistry) return [];
return Object.entries(toolRegistry)
.filter(([key]) => key !== 'automate')
.map(([key, tool]) => ({
value: key,
label: (tool as any).name
}));
};
const addTool = (operation: string) => {
const addTool = (operation: string | null) => {
if (!operation) return;
const newTool: AutomationTool = {
id: `${operation}-${Date.now()}`,
operation,
@ -88,7 +74,7 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
configured: false,
parameters: {}
};
setSelectedTools([...selectedTools, newTool]);
};
@ -143,7 +129,7 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
try {
const { automationStorage } = await import('../../../services/automationStorage');
await automationStorage.saveAutomation(automation);
onComplete();
onComplete(automation);
} catch (error) {
console.error('Error saving automation:', error);
}
@ -153,17 +139,10 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
return (
<div>
<Group justify="space-between" align="center" mb="md">
<Title order={3} size="h4" fw={600} style={{ color: 'var(--mantine-color-text)' }}>
{mode === 'create'
? t('automate.creation.title.create', 'Create Automation')
: t('automate.creation.title.configure', 'Configure Automation')
}
</Title>
<ActionIcon variant="subtle" onClick={onBack}>
<ArrowBackIcon />
</ActionIcon>
</Group>
<Text size="sm" mb="md" p="md" style={{borderRadius:'var(--mantine-radius-md)', background: 'var(--color-gray-200)', color: 'var(--mantine-color-text)' }}>
{t("automate.creation.description", "Automations run tools sequentially. To get started, add tools in the order you want them to run.")}
</Text>
<Divider mb="md" />
<Stack gap="md">
{/* Automation Name */}
@ -175,15 +154,9 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
/>
{/* Add Tool Selector */}
<Select
placeholder={t('automate.creation.tools.add', 'Add a tool...')}
data={getAvailableTools()}
searchable
clearable
value={null}
onChange={addTool}
leftSection={<AddIcon />}
size="sm"
<ToolSelector
onSelect={addTool}
excludeTools={['automate']}
/>
{/* Selected Tools */}
@ -194,7 +167,7 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
<Text size="xs" c="dimmed" style={{ minWidth: '1rem', textAlign: 'center' }}>
{index + 1}
</Text>
<div style={{ flex: 1 }}>
<Group justify="space-between" align="center">
<Group gap="xs" align="center">
@ -207,7 +180,7 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
<CloseIcon style={{ fontSize: 14, color: 'orange' }} />
)}
</Group>
<Group gap="xs">
<ActionIcon
variant="subtle"
@ -227,7 +200,7 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
</Group>
</Group>
</div>
{index < selectedTools.length - 1 && (
<Text size="xs" c="dimmed"></Text>
)}
@ -260,4 +233,4 @@ export default function AutomationCreation({ mode, existingAutomation, onBack, o
)}
</div>
);
}
}

View File

@ -0,0 +1,127 @@
import React, { useState, useMemo } from 'react';
import { useTranslation } from 'react-i18next';
import { Menu, Stack, Text, ScrollArea } from '@mantine/core';
import { ToolRegistryEntry } from '../../../data/toolsTaxonomy';
import { useToolSections } from '../../../hooks/useToolSections';
import { renderToolButtons } from '../shared/renderToolButtons';
import ToolSearch from '../toolPicker/ToolSearch';
import { useFlatToolRegistry } from '../../../data/useTranslatedToolRegistry';
interface ToolSelectorProps {
onSelect: (toolKey: string) => void;
excludeTools?: string[];
}
export default function ToolSelector({ onSelect, excludeTools = [] }: ToolSelectorProps) {
const { t } = useTranslation();
const [opened, setOpened] = useState(false);
const [searchTerm, setSearchTerm] = useState('');
const toolRegistry = useFlatToolRegistry();
// Filter out excluded tools (like 'automate' itself)
const baseFilteredTools = useMemo(() => {
return Object.entries(toolRegistry).filter(([key]) => !excludeTools.includes(key));
}, [toolRegistry, excludeTools]);
// Apply search filter
const filteredTools = useMemo(() => {
if (!searchTerm.trim()) {
return baseFilteredTools;
}
const lowercaseSearch = searchTerm.toLowerCase();
return baseFilteredTools.filter(([key, tool]) => {
return (
tool.name.toLowerCase().includes(lowercaseSearch) ||
tool.description?.toLowerCase().includes(lowercaseSearch) ||
key.toLowerCase().includes(lowercaseSearch)
);
});
}, [baseFilteredTools, searchTerm]);
// Create filtered tool registry for ToolSearch
const filteredToolRegistry = useMemo(() => {
const registry: Record<string, ToolRegistryEntry> = {};
baseFilteredTools.forEach(([key, tool]) => {
registry[key] = tool;
});
return registry;
}, [baseFilteredTools]);
// Use the same tool sections logic as the main ToolPicker
const { sections, searchGroups } = useToolSections(filteredTools);
// Determine what to display: search results or organized sections
const isSearching = searchTerm.trim().length > 0;
const displayGroups = useMemo(() => {
if (isSearching) {
return searchGroups || [];
}
if (!sections || sections.length === 0) {
return [];
}
// Find the "all" section which contains all tools without duplicates
const allSection = sections.find(s => (s as any).key === 'all');
return allSection?.subcategories || [];
}, [isSearching, searchGroups, sections]);
const handleToolSelect = (toolKey: string) => {
onSelect(toolKey);
setOpened(false);
setSearchTerm('');
};
const handleSearchFocus = () => {
setOpened(true);
};
const handleSearchChange = (value: string) => {
setSearchTerm(value);
if (!opened) {
setOpened(true);
}
};
return (
<Menu
opened={opened}
onChange={setOpened}
width="300px"
position="bottom-start"
withinPortal
>
<Menu.Target>
<div onClick={handleSearchFocus} style={{ cursor: 'text' }}>
<ToolSearch
value={searchTerm}
onChange={handleSearchChange}
toolRegistry={filteredToolRegistry}
mode="filter"
placeholder={t('automate.creation.tools.add', 'Add a tool...')}
/>
</div>
</Menu.Target>
<Menu.Dropdown p={0}>
<ScrollArea h={350}>
<Stack gap="sm" p="sm">
{displayGroups.length === 0 ? (
<Text size="sm" c="dimmed" ta="center" p="md">
{isSearching
? t('tools.noSearchResults', 'No tools found')
: t('tools.noTools', 'No tools available')
}
</Text>
) : (
displayGroups.map((subcategory) =>
renderToolButtons(subcategory, null, handleToolSelect, !isSearching)
)
)}
</Stack>
</ScrollArea>
</Menu.Dropdown>
</Menu>
);
}

View File

@ -0,0 +1,219 @@
import React, { useState } from 'react';
import { useTranslation } from 'react-i18next';
import {
Button,
Text,
Title,
Stack,
Group,
ActionIcon,
Progress,
Card,
Alert
} from '@mantine/core';
import ArrowBackIcon from '@mui/icons-material/ArrowBack';
import PlayArrowIcon from '@mui/icons-material/PlayArrow';
import CheckIcon from '@mui/icons-material/Check';
import ErrorIcon from '@mui/icons-material/Error';
import { useFileContext } from '../../../contexts/FileContext';
interface ToolSequenceProps {
automation: any;
onBack: () => void;
onComplete: () => void;
}
interface ExecutionStep {
id: string;
operation: string;
name: string;
status: 'pending' | 'running' | 'completed' | 'error';
error?: string;
}
export default function ToolSequence({ automation, onBack, onComplete }: ToolSequenceProps) {
const { t } = useTranslation();
const { activeFiles } = useFileContext();
const [isExecuting, setIsExecuting] = useState(false);
const [executionSteps, setExecutionSteps] = useState<ExecutionStep[]>([]);
const [currentStepIndex, setCurrentStepIndex] = useState(-1);
// Initialize execution steps from automation
React.useEffect(() => {
if (automation?.operations) {
const steps = automation.operations.map((op: any, index: number) => ({
id: `${op.operation}-${index}`,
operation: op.operation,
name: op.operation, // You might want to get the display name from tool registry
status: 'pending' as const
}));
setExecutionSteps(steps);
}
}, [automation]);
const executeAutomation = async () => {
if (!activeFiles || activeFiles.length === 0) {
// Show error - need files to execute automation
return;
}
setIsExecuting(true);
setCurrentStepIndex(0);
try {
for (let i = 0; i < executionSteps.length; i++) {
setCurrentStepIndex(i);
// Update step status to running
setExecutionSteps(prev => prev.map((step, idx) =>
idx === i ? { ...step, status: 'running' } : step
));
// Simulate step execution (replace with actual tool execution)
await new Promise(resolve => setTimeout(resolve, 2000));
// Update step status to completed
setExecutionSteps(prev => prev.map((step, idx) =>
idx === i ? { ...step, status: 'completed' } : step
));
}
setCurrentStepIndex(-1);
setIsExecuting(false);
// All steps completed - show success
} catch (error) {
// Handle error
setExecutionSteps(prev => prev.map((step, idx) =>
idx === currentStepIndex ? { ...step, status: 'error', error: error?.toString() } : step
));
setIsExecuting(false);
}
};
const getStepIcon = (step: ExecutionStep) => {
switch (step.status) {
case 'completed':
return <CheckIcon style={{ fontSize: 16, color: 'green' }} />;
case 'error':
return <ErrorIcon style={{ fontSize: 16, color: 'red' }} />;
case 'running':
return <div style={{ width: 16, height: 16, border: '2px solid #ccc', borderTop: '2px solid #007bff', borderRadius: '50%', animation: 'spin 1s linear infinite' }} />;
default:
return <div style={{ width: 16, height: 16, border: '2px solid #ccc', borderRadius: '50%' }} />;
}
};
const getProgress = () => {
const completedSteps = executionSteps.filter(step => step.status === 'completed').length;
return (completedSteps / executionSteps.length) * 100;
};
const allStepsCompleted = executionSteps.every(step => step.status === 'completed');
const hasErrors = executionSteps.some(step => step.status === 'error');
return (
<div>
<Group justify="space-between" align="center" mb="md">
<Title order={3} size="h4" fw={600} style={{ color: 'var(--mantine-color-text)' }}>
{t('automate.sequence.title', 'Tool Sequence')}
</Title>
<ActionIcon variant="subtle" onClick={onBack}>
<ArrowBackIcon />
</ActionIcon>
</Group>
<Stack gap="md">
{/* Automation Info */}
<Card padding="md" withBorder>
<Text size="sm" fw={500} mb="xs">
{automation?.name || t('automate.sequence.unnamed', 'Unnamed Automation')}
</Text>
<Text size="xs" c="dimmed">
{t('automate.sequence.steps', '{{count}} steps', { count: executionSteps.length })}
</Text>
</Card>
{/* File Selection Warning */}
{(!activeFiles || activeFiles.length === 0) && (
<Alert color="orange" title={t('automate.sequence.noFiles', 'No Files Selected')}>
{t('automate.sequence.noFilesDesc', 'Please select files to process before running the automation.')}
</Alert>
)}
{/* Progress Bar */}
{isExecuting && (
<div>
<Text size="sm" mb="xs">
{t('automate.sequence.progress', 'Progress: {{current}}/{{total}}', {
current: currentStepIndex + 1,
total: executionSteps.length
})}
</Text>
<Progress value={getProgress()} size="lg" />
</div>
)}
{/* Execution Steps */}
<Stack gap="xs">
{executionSteps.map((step, index) => (
<Group key={step.id} gap="sm" align="center">
<Text size="xs" c="dimmed" style={{ minWidth: '1rem', textAlign: 'center' }}>
{index + 1}
</Text>
{getStepIcon(step)}
<div style={{ flex: 1 }}>
<Text size="sm" style={{
color: step.status === 'running' ? 'var(--mantine-color-blue-6)' : 'var(--mantine-color-text)',
fontWeight: step.status === 'running' ? 500 : 400
}}>
{step.name}
</Text>
{step.error && (
<Text size="xs" c="red" mt="xs">
{step.error}
</Text>
)}
</div>
</Group>
))}
</Stack>
{/* Action Buttons */}
<Group justify="space-between" mt="xl">
<Button
leftSection={<PlayArrowIcon />}
onClick={executeAutomation}
disabled={isExecuting || !activeFiles || activeFiles.length === 0}
loading={isExecuting}
>
{isExecuting
? t('automate.sequence.running', 'Running Automation...')
: t('automate.sequence.run', 'Run Automation')
}
</Button>
{(allStepsCompleted || hasErrors) && (
<Button
variant="light"
onClick={onComplete}
>
{t('automate.sequence.finish', 'Finish')}
</Button>
)}
</Group>
</Stack>
<style>
{`
@keyframes spin {
0% { transform: rotate(0deg); }
100% { transform: rotate(360deg); }
}
`}
</style>
</div>
);
}

View File

@ -0,0 +1,29 @@
import React from 'react';
import { Box, Stack } from '@mantine/core';
import ToolButton from '../toolPicker/ToolButton';
import SubcategoryHeader from './SubcategoryHeader';
// Helper function to render tool buttons for a subcategory
export const renderToolButtons = (
subcategory: any,
selectedToolKey: string | null,
onSelect: (id: string) => void,
showSubcategoryHeader: boolean = true
) => (
<Box key={subcategory.subcategory} w="100%">
{showSubcategoryHeader && (
<SubcategoryHeader label={subcategory.subcategory} />
)}
<Stack gap="xs">
{subcategory.tools.map(({ id, tool }: { id: string; tool: any }) => (
<ToolButton
key={id}
id={id}
tool={tool}
isSelected={selectedToolKey === id}
onSelect={onSelect}
/>
))}
</Stack>
</Box>
);

View File

@ -12,6 +12,7 @@ interface ToolSearchProps {
onToolSelect?: (toolId: string) => void;
mode: 'filter' | 'dropdown';
selectedToolKey?: string | null;
placeholder?: string;
}
const ToolSearch = ({
@ -20,7 +21,8 @@ const ToolSearch = ({
toolRegistry,
onToolSelect,
mode = 'filter',
selectedToolKey
selectedToolKey,
placeholder
}: ToolSearchProps) => {
const { t } = useTranslation();
const [dropdownOpen, setDropdownOpen] = useState(false);
@ -61,7 +63,7 @@ const ToolSearch = ({
ref={searchRef}
value={value}
onChange={handleSearchChange}
placeholder={t("toolPicker.searchPlaceholder", "Search tools...")}
placeholder={placeholder || t("toolPicker.searchPlaceholder", "Search tools...")}
icon={<span className="material-symbols-rounded">search</span>}
autoComplete="off"
/>

View File

@ -6,6 +6,7 @@ import { useToolFileSelection } from "../contexts/FileSelectionContext";
import { createToolFlow } from "../components/tools/shared/createToolFlow";
import AutomationSelection from "../components/tools/automate/AutomationSelection";
import AutomationCreation from "../components/tools/automate/AutomationCreation";
import ToolSequence from "../components/tools/automate/ToolSequence";
import { useAutomateOperation } from "../hooks/tools/automate/useAutomateOperation";
import { BaseToolProps } from "../types/tool";
@ -15,7 +16,7 @@ const Automate = ({ onPreviewFile, onComplete, onError }: BaseToolProps) => {
const { setCurrentMode } = useFileContext();
const { selectedFiles } = useToolFileSelection();
const [currentStep, setCurrentStep] = useState<'selection' | 'creation'>('selection');
const [currentStep, setCurrentStep] = useState<'selection' | 'creation' | 'sequence'>('selection');
const [stepData, setStepData] = useState<any>({});
const automateOperation = useAutomateOperation();
@ -49,6 +50,15 @@ const Automate = ({ onPreviewFile, onComplete, onError }: BaseToolProps) => {
mode={stepData.mode}
existingAutomation={stepData.automation}
onBack={() => handleStepChange({ step: 'selection' })}
onComplete={(automation: any) => handleStepChange({ step: 'sequence', automation })}
/>
);
case 'sequence':
return (
<ToolSequence
automation={stepData.automation}
onBack={() => handleStepChange({ step: 'creation', mode: stepData.mode, automation: stepData.automation })}
onComplete={handleComplete}
/>
);
@ -68,7 +78,12 @@ const Automate = ({ onPreviewFile, onComplete, onError }: BaseToolProps) => {
{
title: t('automate.stepTitle', 'Automations'),
isVisible: true,
content: renderCurrentStep()
content: currentStep === 'selection' ? renderCurrentStep() : null
},
{
title: t('automate.sequenceTitle', 'Tool Sequence'),
isVisible: currentStep === 'creation' || currentStep === 'sequence',
content: currentStep === 'creation' || currentStep === 'sequence' ? renderCurrentStep() : null
}
],
review: {