52 lines
1.6 KiB
TypeScript
52 lines
1.6 KiB
TypeScript
|
import { EOL } from "os"
|
||
|
|
import { basename } from "path"
|
||
|
|
import { Agent } from "../../../agent/agent"
|
||
|
|
import { Provider } from "../../../provider/provider"
|
||
|
|
import { ToolRegistry } from "../../../tool/registry"
|
||
|
|
import { Wildcard } from "../../../util/wildcard"
|
||
|
|
import { bootstrap } from "../../bootstrap"
|
||
|
|
import { cmd } from "../cmd"
|
||
|
|
|
||
|
|
export const AgentCommand = cmd({
|
||
|
|
command: "agent <name>",
|
||
|
|
builder: (yargs) =>
|
||
|
|
yargs.positional("name", {
|
||
|
|
type: "string",
|
||
|
|
demandOption: true,
|
||
|
|
description: "Agent name",
|
||
|
|
}),
|
||
|
|
async handler(args) {
|
||
|
|
await bootstrap(process.cwd(), async () => {
|
||
|
|
const agentName = args.name as string
|
||
|
|
const agent = await Agent.get(agentName)
|
||
|
|
if (!agent) {
|
||
|
|
process.stderr.write(
|
||
|
|
`Agent ${agentName} not found, run '${basename(process.execPath)} agent list' to get an agent list` + EOL,
|
||
|
|
)
|
||
|
|
process.exit(1)
|
||
|
|
}
|
||
|
|
const resolvedTools = await resolveTools(agent)
|
||
|
|
const output = {
|
||
|
|
...agent,
|
||
|
|
tools: resolvedTools,
|
||
|
|
toolOverrides: agent.tools,
|
||
|
|
}
|
||
|
|
process.stdout.write(JSON.stringify(output, null, 2) + EOL)
|
||
|
|
})
|
||
|
|
},
|
||
|
|
})
|
||
|
|
|
||
|
|
async function resolveTools(agent: Agent.Info) {
|
||
|
|
const providerID = agent.model?.providerID ?? (await Provider.defaultModel()).providerID
|
||
|
|
const toolOverrides = {
|
||
|
|
...agent.tools,
|
||
|
|
...(await ToolRegistry.enabled(agent)),
|
||
|
|
}
|
||
|
|
const availableTools = await ToolRegistry.tools(providerID, agent)
|
||
|
|
const resolved: Record<string, boolean> = {}
|
||
|
|
for (const tool of availableTools) {
|
||
|
|
resolved[tool.id] = Wildcard.all(tool.id, toolOverrides) !== false
|
||
|
|
}
|
||
|
|
return resolved
|
||
|
|
}
|