refactor(tools): decompose skill/tools.ts into focused tool creators
This commit is contained in:
@@ -0,0 +1,96 @@
|
||||
import type { Prompt, Resource, Tool } from "@modelcontextprotocol/sdk/types.js"
|
||||
import { sanitizeJsonSchema } from "../../plugin/normalize-tool-arg-schemas"
|
||||
import type {
|
||||
SkillMcpClientInfo,
|
||||
SkillMcpManager,
|
||||
SkillMcpServerContext,
|
||||
} from "../../features/skill-mcp-manager"
|
||||
import type { LoadedSkill } from "../../features/opencode-skill-loader"
|
||||
|
||||
export async function formatMcpCapabilities(
|
||||
skill: LoadedSkill,
|
||||
manager: SkillMcpManager,
|
||||
sessionID: string
|
||||
): Promise<string | null> {
|
||||
if (!skill.mcpConfig || Object.keys(skill.mcpConfig).length === 0) {
|
||||
return null
|
||||
}
|
||||
|
||||
const sections: string[] = ["", "## Available MCP Servers", ""]
|
||||
|
||||
for (const [serverName, config] of Object.entries(skill.mcpConfig)) {
|
||||
const info: SkillMcpClientInfo = {
|
||||
serverName,
|
||||
skillName: skill.name,
|
||||
sessionID,
|
||||
}
|
||||
const context: SkillMcpServerContext = {
|
||||
config,
|
||||
skillName: skill.name,
|
||||
}
|
||||
|
||||
sections.push(`### ${serverName}`, "")
|
||||
|
||||
try {
|
||||
const [tools, resources, prompts] = await Promise.all([
|
||||
manager.listTools(info, context).catch(() => []),
|
||||
manager.listResources(info, context).catch(() => []),
|
||||
manager.listPrompts(info, context).catch(() => []),
|
||||
])
|
||||
|
||||
appendToolSections(sections, tools as Tool[])
|
||||
appendResourceSection(sections, resources as Resource[])
|
||||
appendPromptSection(sections, prompts as Prompt[])
|
||||
|
||||
if (tools.length === 0 && resources.length === 0 && prompts.length === 0) {
|
||||
sections.push("*No capabilities discovered*")
|
||||
}
|
||||
} catch (error) {
|
||||
const errorMessage = error instanceof Error ? error.message : String(error)
|
||||
sections.push(`*Failed to connect: ${errorMessage.split("\n")[0]}*`)
|
||||
}
|
||||
|
||||
sections.push("", `Use \`skill_mcp\` tool with \`mcp_name=\"${serverName}\"\` to invoke.`, "")
|
||||
}
|
||||
|
||||
return sections.join("\n")
|
||||
}
|
||||
|
||||
function appendToolSections(sections: string[], tools: Tool[]): void {
|
||||
if (tools.length === 0) {
|
||||
return
|
||||
}
|
||||
|
||||
sections.push("**Tools:**", "")
|
||||
|
||||
for (const toolDefinition of tools) {
|
||||
sections.push(`#### \`${toolDefinition.name}\``)
|
||||
if (toolDefinition.description) {
|
||||
sections.push(toolDefinition.description)
|
||||
}
|
||||
sections.push(
|
||||
"",
|
||||
"**inputSchema:**",
|
||||
"```json",
|
||||
JSON.stringify(sanitizeJsonSchema(toolDefinition.inputSchema), null, 2),
|
||||
"```",
|
||||
""
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
function appendResourceSection(sections: string[], resources: Resource[]): void {
|
||||
if (resources.length === 0) {
|
||||
return
|
||||
}
|
||||
|
||||
sections.push(`**Resources**: ${resources.map((resource) => resource.uri).join(", ")}`)
|
||||
}
|
||||
|
||||
function appendPromptSection(sections: string[], prompts: Prompt[]): void {
|
||||
if (prompts.length === 0) {
|
||||
return
|
||||
}
|
||||
|
||||
sections.push(`**Prompts**: ${prompts.map((prompt) => prompt.name).join(", ")}`)
|
||||
}
|
||||
Reference in New Issue
Block a user