diff --git a/src/renderer/src/pages/home/Tabs/components/AgentItem.tsx b/src/renderer/src/pages/home/Tabs/components/AgentItem.tsx index ec21ae81de..5f14fde594 100644 --- a/src/renderer/src/pages/home/Tabs/components/AgentItem.tsx +++ b/src/renderer/src/pages/home/Tabs/components/AgentItem.tsx @@ -1,6 +1,7 @@ -import { Avatar, Button, cn } from '@heroui/react' +import { Avatar, Button, Chip, cn } from '@heroui/react' import { DeleteIcon, EditIcon } from '@renderer/components/Icons' import { getAgentAvatar } from '@renderer/config/agent' +import { useSessions } from '@renderer/hooks/agents/useSessions' import AgentSettingsPopup from '@renderer/pages/settings/AgentSettings/AgentSettingsPopup' import { AgentEntity } from '@renderer/types' import { ContextMenu, ContextMenuContent, ContextMenuItem, ContextMenuTrigger } from '@renderer/ui/context-menu' @@ -18,17 +19,16 @@ interface AgentItemProps { const AgentItem: FC = ({ agent, isActive, onDelete, onPress }) => { const { t } = useTranslation() - // const { isOpen, onOpen, onClose } = useDisclosure() - // const { agents } = useAgents() + const { sessions } = useSessions(agent.id) const AgentLabel = useCallback(() => { const displayName = agent.name ?? agent.id const avatar = getAgentAvatar(agent.type) return ( - <> +
{displayName} - +
) }, [agent.id, agent.name, agent.type]) @@ -37,8 +37,17 @@ const AgentItem: FC = ({ agent, isActive, onDelete, onPress }) = - + + {isActive && ( + + {sessions.length} + + )}