diff --git a/src/views/AgentChat/index.js b/src/views/AgentChat/index.js
index 635950f1..41834bfc 100644
--- a/src/views/AgentChat/index.js
+++ b/src/views/AgentChat/index.js
@@ -10,17 +10,14 @@ import {
   Input,
   Avatar,
   Chip,
-  Divider,
+  Separator,
   Spinner,
   Tooltip,
-  Badge,
   Checkbox,
   CheckboxGroup,
   Tabs,
-  Tab,
   Kbd,
   Accordion,
-  AccordionItem,
 } from '@heroui/react';
 import { useAuth } from '@contexts/AuthContext';
 import { logger } from '@utils/logger';
@@ -1159,14 +1156,11 @@ const AgentChat = () => {
                   <div className="flex items-center gap-2">
                     <Code className="w-4 h-4" />
                     <span>工具</span>
-                    <Badge
-                      content={selectedTools.length}
-                      color="primary"
-                      size="sm"
-                      classNames={{
-                        badge: 'bg-blue-500',
-                      }}
-                    />
+                    {selectedTools.length > 0 && (
+                      <span className="inline-flex items-center justify-center w-5 h-5 text-xs font-semibold text-white bg-blue-500 rounded-full">
+                        {selectedTools.length}
+                      </span>
+                    )}
                   </div>
                 }
               >