diff --git a/src/renderer/src/hooks/agents/useAgents.ts b/src/renderer/src/hooks/agents/useAgents.ts index 7dbe22511c..585d9e0c08 100644 --- a/src/renderer/src/hooks/agents/useAgents.ts +++ b/src/renderer/src/hooks/agents/useAgents.ts @@ -40,11 +40,27 @@ export const useAgents = () => { [client, mutate, t] ) + const deleteAgent = useCallback( + async (id: string) => { + try { + await client.deleteAgent(id) + mutate((prev) => ({ + agents: prev?.agents.filter((a) => a.id !== id) ?? [], + total: prev ? prev.total - 1 : 0 + })) + } catch (error) { + window.toast.error(formatErrorMessageWithPrefix(error, t('agent.delete.error.failed'))) + } + }, + [client, mutate, t] + ) + return { agents: data?.agents ?? [], error, isLoading, addAgent, - updateAgent + updateAgent, + deleteAgent } }