From 48c9a3a33eff8fedde29c718b78865069d58a138 Mon Sep 17 00:00:00 2001
From: Siddharth Ganesan <siddharthganesan@gmail.com>
Date: Mon, 9 Feb 2026 15:18:35 -0800
Subject: [PATCH] Fix discovery

---
 .../tools/client/tool-display-registry.ts     | 42 +++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/apps/sim/lib/copilot/tools/client/tool-display-registry.ts b/apps/sim/lib/copilot/tools/client/tool-display-registry.ts
index b106ea59f..539414e96 100644
--- a/apps/sim/lib/copilot/tools/client/tool-display-registry.ts
+++ b/apps/sim/lib/copilot/tools/client/tool-display-registry.ts
@@ -306,6 +306,26 @@ const META_custom_tool: ToolMetadata = {
   },
 }
 
+const META_build: ToolMetadata = {
+  displayNames: {
+    [ClientToolCallState.generating]: { text: 'Building', icon: Loader2 },
+    [ClientToolCallState.pending]: { text: 'Building', icon: Loader2 },
+    [ClientToolCallState.executing]: { text: 'Building', icon: Loader2 },
+    [ClientToolCallState.success]: { text: 'Built', icon: Wrench },
+    [ClientToolCallState.error]: { text: 'Failed to build', icon: XCircle },
+    [ClientToolCallState.rejected]: { text: 'Skipped build', icon: XCircle },
+    [ClientToolCallState.aborted]: { text: 'Aborted build', icon: XCircle },
+  },
+  uiConfig: {
+    subagent: {
+      streamingLabel: 'Building',
+      completedLabel: 'Built',
+      shouldCollapse: true,
+      outputArtifacts: [],
+    },
+  },
+}
+
 const META_debug: ToolMetadata = {
   displayNames: {
     [ClientToolCallState.generating]: { text: 'Debugging', icon: Loader2 },
@@ -326,6 +346,26 @@ const META_debug: ToolMetadata = {
   },
 }
 
+const META_discovery: ToolMetadata = {
+  displayNames: {
+    [ClientToolCallState.generating]: { text: 'Discovering', icon: Loader2 },
+    [ClientToolCallState.pending]: { text: 'Discovering', icon: Loader2 },
+    [ClientToolCallState.executing]: { text: 'Discovering', icon: Loader2 },
+    [ClientToolCallState.success]: { text: 'Discovered', icon: Search },
+    [ClientToolCallState.error]: { text: 'Failed to discover', icon: XCircle },
+    [ClientToolCallState.rejected]: { text: 'Skipped discovery', icon: XCircle },
+    [ClientToolCallState.aborted]: { text: 'Aborted discovery', icon: XCircle },
+  },
+  uiConfig: {
+    subagent: {
+      streamingLabel: 'Discovering',
+      completedLabel: 'Discovered',
+      shouldCollapse: true,
+      outputArtifacts: [],
+    },
+  },
+}
+
 const META_deploy: ToolMetadata = {
   displayNames: {
     [ClientToolCallState.generating]: { text: 'Deploying', icon: Loader2 },
@@ -2252,9 +2292,11 @@ const TOOL_METADATA_BY_ID: Record<string, ToolMetadata> = {
   checkoff_todo: META_checkoff_todo,
   crawl_website: META_crawl_website,
   create_workspace_mcp_server: META_create_workspace_mcp_server,
+  build: META_build,
   custom_tool: META_custom_tool,
   debug: META_debug,
   deploy: META_deploy,
+  discovery: META_discovery,
   deploy_api: META_deploy_api,
   deploy_chat: META_deploy_chat,
   deploy_mcp: META_deploy_mcp,