Improve derived MCP companion UI labels

2026-09-03 01:59:23 +00:00 · 2026-03-25 10:44:56 +08:00
parent 6040e56984
commit 7ad035773a
3 changed files with 35 additions and 18 deletions
@@ -1335,11 +1335,17 @@ impl ExtensionManager {
                            Vec::new()
                        };

-                        let display_name = self
-                            .registry
-                            .get_with_kind(&server.name, Some(ExtensionKind::McpServer))
-                            .await
-                            .map(|e| e.display_name);
+                        let display_name =
+                            if crate::tools::mcp::config::is_nearai_companion_server_name(
+                                &server.name,
+                            ) {
+                                Some("NEAR AI Companion".to_string())
+                            } else {
+                                self.registry
+                                    .get_with_kind(&server.name, Some(ExtensionKind::McpServer))
+                                    .await
+                                    .map(|e| e.display_name)
+                            };
                        extensions.push(InstalledExtension {
                            name: server.name.clone(),
                            kind: ExtensionKind::McpServer,