mirror of
https://github.com/codeany-ai/open-agent-sdk-typescript.git
synced 2026-04-25 07:00:49 +09:00
idoubi
67e120b2ed
Open-source Agent SDK with 30+ built-in tools, MCP integration, multi-turn sessions, subagents, and streaming support. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
102 lines
3.2 KiB
TypeScript
102 lines
3.2 KiB
TypeScript
/**
|
|
* Example 11: Custom Tools with tool() + createSdkMcpServer()
|
|
*
|
|
* Shows the Zod-based tool() helper and in-process MCP server creation.
|
|
* This is the recommended way to add custom tools.
|
|
*
|
|
* Run: npx tsx examples/11-custom-mcp-tools.ts
|
|
*/
|
|
import { z } from 'zod'
|
|
import { query, tool, createSdkMcpServer } from '../src/index.js'
|
|
|
|
// Define tools using Zod schemas for type-safe input validation
|
|
const getTemperature = tool(
|
|
'get_temperature',
|
|
'Get the current temperature at a location',
|
|
{
|
|
city: z.string().describe('City name'),
|
|
unit: z.enum(['celsius', 'fahrenheit']).default('celsius').describe('Temperature unit'),
|
|
},
|
|
async ({ city, unit }) => {
|
|
// Mock weather data
|
|
const temps: Record<string, number> = {
|
|
tokyo: 22, london: 14, paris: 16, 'new york': 18, beijing: 25,
|
|
}
|
|
const tempC = temps[city.toLowerCase()] ?? 20
|
|
const temp = unit === 'fahrenheit' ? tempC * 9 / 5 + 32 : tempC
|
|
const symbol = unit === 'fahrenheit' ? '°F' : '°C'
|
|
|
|
return {
|
|
content: [{ type: 'text' as const, text: `Temperature in ${city}: ${temp}${symbol}` }],
|
|
}
|
|
},
|
|
{ annotations: { readOnlyHint: true } },
|
|
)
|
|
|
|
const convertUnits = tool(
|
|
'convert_units',
|
|
'Convert between measurement units',
|
|
{
|
|
value: z.number().describe('Value to convert'),
|
|
from_unit: z.string().describe('Source unit'),
|
|
to_unit: z.string().describe('Target unit'),
|
|
},
|
|
async ({ value, from_unit, to_unit }) => {
|
|
const conversions: Record<string, Record<string, (v: number) => number>> = {
|
|
km: { miles: (v) => v * 0.621371, m: (v) => v * 1000 },
|
|
miles: { km: (v) => v * 1.60934, m: (v) => v * 1609.34 },
|
|
kg: { lbs: (v) => v * 2.20462, g: (v) => v * 1000 },
|
|
lbs: { kg: (v) => v * 0.453592, g: (v) => v * 453.592 },
|
|
}
|
|
|
|
const fn = conversions[from_unit]?.[to_unit]
|
|
if (!fn) {
|
|
return {
|
|
content: [{ type: 'text' as const, text: `Cannot convert from ${from_unit} to ${to_unit}` }],
|
|
isError: true,
|
|
}
|
|
}
|
|
|
|
const result = fn(value)
|
|
return {
|
|
content: [{ type: 'text' as const, text: `${value} ${from_unit} = ${result.toFixed(2)} ${to_unit}` }],
|
|
}
|
|
},
|
|
)
|
|
|
|
// Bundle tools into an in-process MCP server
|
|
const utilityServer = createSdkMcpServer({
|
|
name: 'utilities',
|
|
version: '1.0.0',
|
|
tools: [getTemperature, convertUnits],
|
|
})
|
|
|
|
async function main() {
|
|
console.log('--- Example 11: Custom MCP Tools (tool + createSdkMcpServer) ---\n')
|
|
|
|
for await (const message of query({
|
|
prompt: 'What is the temperature in Tokyo and Paris? Also convert 10 km to miles. Be brief.',
|
|
options: {
|
|
mcpServers: { utilities: utilityServer as any },
|
|
allowedTools: ['mcp__utilities__*'],
|
|
permissionMode: 'bypassPermissions',
|
|
},
|
|
})) {
|
|
const msg = message as any
|
|
|
|
if (msg.type === 'assistant' && msg.message?.content) {
|
|
for (const block of msg.message.content) {
|
|
if ('text' in block && block.text?.trim()) {
|
|
console.log(block.text)
|
|
} else if ('name' in block) {
|
|
console.log(`[${block.name}] ${JSON.stringify(block.input || {})}`)
|
|
}
|
|
}
|
|
} else if (msg.type === 'result') {
|
|
console.log(`\nDone: ${msg.subtype} (cost: $${msg.total_cost_usd?.toFixed(4) || '0'})`)
|
|
}
|
|
}
|
|
}
|
|
|
|
main().catch(console.error)
|