28a2a0c371
- Move AgentsView to consume agents, loading state, and reload logic directly from useAgents - Extend useAgents with filterState/showSystemAgents options and always pass includeEphemeral in fetch filters - Remove duplicate initial fetch/SSE path in AgentsView and rely on hook-managed refresh behavior - Add regression coverage for single initial load and system-agent visibility toggling behavior - Update useAgents hook tests to assert the new includeEphemeral fetch contract
87 lines
2.6 KiB
TypeScript
87 lines
2.6 KiB
TypeScript
import { useState, useEffect, useCallback } from "react";
|
|
import type { Agent, AgentState, AgentCapability, AgentStats } from "../api";
|
|
import { fetchAgents, fetchAgentStats } from "../api";
|
|
import { isEphemeralAgent } from "@fusion/core";
|
|
import { subscribeSse } from "../sse-bus";
|
|
|
|
interface UseAgentsOptions {
|
|
filterState?: AgentState | "all";
|
|
showSystemAgents?: boolean;
|
|
}
|
|
|
|
interface AgentFilter {
|
|
state?: AgentState;
|
|
role?: AgentCapability;
|
|
includeEphemeral?: boolean;
|
|
}
|
|
|
|
export function useAgents(projectId?: string, options?: UseAgentsOptions) {
|
|
const [agents, setAgents] = useState<Agent[]>([]);
|
|
const [stats, setStats] = useState<AgentStats | null>(null);
|
|
const [isLoading, setIsLoading] = useState(false);
|
|
|
|
const loadAgents = useCallback(async (filter?: AgentFilter) => {
|
|
setIsLoading(true);
|
|
try {
|
|
const filterState = options?.filterState;
|
|
const baseFilter = filterState && filterState !== "all" ? { state: filterState } : undefined;
|
|
const includeEphemeral = options?.showSystemAgents ?? false;
|
|
const data = await fetchAgents(
|
|
{
|
|
...baseFilter,
|
|
...filter,
|
|
includeEphemeral: filter?.includeEphemeral ?? includeEphemeral,
|
|
},
|
|
projectId,
|
|
);
|
|
setAgents(data);
|
|
} catch (err) {
|
|
console.error("Failed to load agents:", err);
|
|
} finally {
|
|
setIsLoading(false);
|
|
}
|
|
}, [projectId, options?.filterState, options?.showSystemAgents]);
|
|
|
|
const loadStats = useCallback(async () => {
|
|
try {
|
|
const data = await fetchAgentStats(projectId);
|
|
setStats(data);
|
|
} catch (err) {
|
|
console.error("Failed to load agent stats:", err);
|
|
}
|
|
}, [projectId]);
|
|
|
|
useEffect(() => {
|
|
void loadAgents();
|
|
void loadStats();
|
|
}, [loadAgents, loadStats]);
|
|
|
|
// SSE subscription for agent events
|
|
useEffect(() => {
|
|
const query = projectId ? `?projectId=${encodeURIComponent(projectId)}` : "";
|
|
const refresh = () => {
|
|
void loadAgents();
|
|
void loadStats();
|
|
};
|
|
|
|
return subscribeSse(`/api/events${query}`, {
|
|
events: {
|
|
"agent:created": refresh,
|
|
"agent:updated": refresh,
|
|
"agent:deleted": refresh,
|
|
"agent:stateChanged": refresh,
|
|
},
|
|
});
|
|
}, [projectId, loadAgents, loadStats]);
|
|
|
|
const showSystemAgents = options?.showSystemAgents ?? false;
|
|
const activeAgents = agents.filter((agent) => {
|
|
if (agent.state !== "active" && agent.state !== "running") {
|
|
return false;
|
|
}
|
|
return showSystemAgents || !isEphemeralAgent(agent);
|
|
});
|
|
|
|
return { agents, activeAgents, stats, isLoading, loadAgents, loadStats };
|
|
}
|