From 55174ae24baa38f0f95f8b517fcb884ed45569a1 Mon Sep 17 00:00:00 2001
From: Heet <heetmankad06@gmail.com>
Date: Wed, 17 Jun 2026 11:54:44 +0530
Subject: [PATCH] Introduce tool renderer registry

---
 packages/web/src/components/tool-call-item.tsx | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/packages/web/src/components/tool-call-item.tsx b/packages/web/src/components/tool-call-item.tsx
index 4a1a8171b..beab5a943 100644
--- a/packages/web/src/components/tool-call-item.tsx
+++ b/packages/web/src/components/tool-call-item.tsx
@@ -23,6 +23,11 @@ interface ToolCallItemProps {
   showTime?: boolean;
 }
 
+//Registry
+const TOOL_RENDERERS: Record<string, React.ComponentType<ToolCallItemProps>> = {
+  "slack-notify": SlackNotifyEvent,
+};
+
 function ToolIcon({ name }: { name: string | null }) {
   if (!name) return null;
 
@@ -51,14 +56,11 @@ function ToolIcon({ name }: { name: string | null }) {
 }
 
 export function ToolCallItem({ event, isExpanded, onToggle, showTime = true }: ToolCallItemProps) {
-  if (event.tool === "slack-notify") {
+  const Renderer = TOOL_RENDERERS[event.tool];
+
+  if (Renderer) {
     return (
-      <SlackNotifyEvent
-        event={event}
-        isExpanded={isExpanded}
-        onToggle={onToggle}
-        showTime={showTime}
-      />
+      <Renderer event={event} isExpanded={isExpanded} onToggle={onToggle} showTime={showTime} />
     );
   }