1

CustomKeyboard/View/KBKeyButton.m

@@ -152,7 +152,16 @@
     // - caseVariant: 0/1/2 => 无变体/小写/大写
     NSString *identifier = self.key.identifier;
     NSInteger variant = (NSInteger)self.key.caseVariant;
-    UIImage *iconImg = [[KBSkinManager shared] iconImageForKeyIdentifier:identifier caseVariant:variant];
+    KBSkinManager *skinManager = [KBSkinManager shared];
+    UIImage *iconImg = [skinManager iconImageForKeyIdentifier:identifier caseVariant:variant];
+
+    if (!iconImg && [identifier isEqualToString:@"ai"]) {
+        NSString *skinId = skinManager.current.skinId ?: @"";
+        BOOL usingDefaultSkin = (skinId.length == 0 || [skinId isEqualToString:@"default"]);
+        if (usingDefaultSkin) {
+            iconImg = [UIImage imageNamed:@"ai_key_icon"];
+        }
+    }
 
     // 设置整块按键背景图（若有）
     self.iconView.image = iconImg;