diff --git a/src/components/nodes/UnifiedDatabaseNode.tsx b/src/components/nodes/UnifiedDatabaseNode.tsx index cbc9a01..3e3002b 100644 --- a/src/components/nodes/UnifiedDatabaseNode.tsx +++ b/src/components/nodes/UnifiedDatabaseNode.tsx @@ -25,7 +25,10 @@ export const UnifiedDatabaseNode: React.FC = ({ // Get operation display info const getOperationInfo = () => { const operation = data.config?.operation as DatabaseOperation; - const operationIcons: Record> = { + const operationIcons: Record< + DatabaseOperation, + React.ComponentType + > = { query: Search, insert: Plus, update: Edit, diff --git a/src/services/processors/ProcessorRegistry.ts b/src/services/processors/ProcessorRegistry.ts index 3ccf825..c494691 100644 --- a/src/services/processors/ProcessorRegistry.ts +++ b/src/services/processors/ProcessorRegistry.ts @@ -144,12 +144,7 @@ export class ProcessorRegistry { { pattern: /llm|ai|gpt|openai/i, types: ["llmTask"] }, { pattern: /db|database|sql/i, - types: [ - "databaseQuery", - "databaseInsert", - "databaseUpdate", - "databaseDelete", - ], + types: ["database"], }, { pattern: /embed|vector/i,