Hooks

import { Agent, defineTool, shutdown } from "cognitio-agent-sdk"

const agent = new Agent({
  model: "openai/gpt-4.1-mini",
  tools: [defineTool({
    name: "ping",
    description: "Check whether this application is ready.",
    inputJsonSchema: { type: "object", properties: {}, additionalProperties: false },
    execute: () => ({ status: "ready" }),
  })],
  allowedTools: ["sdk_ping"],
  maxTurns: 3,
  hooks: {
    PreToolUse: [{
      matcher: /^sdk_ping$/,
      timeoutMs: 2_000,
      callback(event) {
        console.log("Before tool", event.sessionId, event.toolCallId)
        return { additionalContext: "Report the returned status accurately." }
      },
    }],
    PostToolUse: [{
      matcher: /^sdk_ping$/,
      callback(event) {
        console.log("After tool", event.sessionId, event.toolCallId)
      },
    }],
  },
})

try {
  const result = await agent.run("Call ping once, then tell me the status.", {
    signal: AbortSignal.timeout(60_000),
  })
  if (result.subtype !== "success") throw new Error(`Run ended: ${result.subtype}`)
  console.log(result.text)
} finally {
  await agent.close().finally(() => shutdown())
}