Spaces:
Paused
Paused
File size: 3,137 Bytes
0b9dc2e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | import { useState, useEffect, useCallback } from 'react';
import { workspaceApi } from '@/api';
import type { MCPClient, MCPClientStatus, Skill } from '@/api';
export function useWorkspace(agentId: string | null, sessionId: string | null) {
const [mcps, setMcps] = useState<MCPClientStatus[]>([]);
const [skills, setSkills] = useState<Skill[]>([]);
const [loading, setLoading] = useState(false);
const [skillsLoading, setSkillsLoading] = useState(false);
const [error, setError] = useState<Error | null>(null);
const refetch = useCallback(async () => {
if (!agentId || !sessionId) {
setMcps([]);
return;
}
setLoading(true);
setError(null);
try {
setMcps(await workspaceApi.mcp.list(agentId, sessionId));
} catch (e) {
setError(e as Error);
} finally {
setLoading(false);
}
}, [agentId, sessionId]);
const refetchSkills = useCallback(async () => {
if (!agentId || !sessionId) {
setSkills([]);
return;
}
setSkillsLoading(true);
try {
setSkills(await workspaceApi.skill.list(agentId, sessionId));
} catch (e) {
setError(e as Error);
} finally {
setSkillsLoading(false);
}
}, [agentId, sessionId]);
useEffect(() => {
refetch();
}, [refetch]);
useEffect(() => {
refetchSkills();
}, [refetchSkills]);
const addMcps = useCallback(
async (clients: MCPClient[]) => {
if (!agentId || !sessionId) throw new Error('No agent/session selected');
const existingNames = new Set(mcps.map((m) => m.name));
for (const mcp of clients) {
if (existingNames.has(mcp.name)) {
throw new Error(`MCP server "${mcp.name}" already exists in this workspace.`);
}
}
const batchNames = new Set<string>();
for (const mcp of clients) {
if (batchNames.has(mcp.name)) {
throw new Error(`Duplicate MCP server name "${mcp.name}" in configuration.`);
}
batchNames.add(mcp.name);
}
for (const mcp of clients) {
await workspaceApi.mcp.add(agentId, sessionId, mcp);
}
await refetch();
},
[agentId, sessionId, mcps, refetch],
);
const removeMcp = useCallback(
async (mcpName: string) => {
if (!agentId || !sessionId) throw new Error('No agent/session selected');
await workspaceApi.mcp.remove(mcpName, agentId, sessionId);
await refetch();
},
[agentId, sessionId, refetch],
);
const addSkill = useCallback(
async (skillPath: string) => {
if (!agentId || !sessionId) throw new Error('No agent/session selected');
await workspaceApi.skill.add(agentId, sessionId, { skill_path: skillPath });
await refetchSkills();
},
[agentId, sessionId, refetchSkills],
);
const removeSkill = useCallback(
async (skillName: string) => {
if (!agentId || !sessionId) throw new Error('No agent/session selected');
await workspaceApi.skill.remove(skillName, agentId, sessionId);
await refetchSkills();
},
[agentId, sessionId, refetchSkills],
);
return {
mcps,
loading,
error,
refetch,
addMcps,
removeMcp,
skills,
skillsLoading,
addSkill,
removeSkill,
};
}
|