Mastra•2mo ago•

2 replies

How to resume a suspended tool call from useChat (Mastra standalone + AI SDK)

AgentTools

Hi,
I have a question about resuming a suspended tool call.
Setup:
Standalone Mastra server (

http://localhost:4111

http://localhost:4111

), Next.js frontend, AI SDK (

useChat

useChat

hook).
All dependencies are on their latest versions.

Current behavior:
When a tool requires approval, I receive a message part with

part.type === "data-tool-call-suspended"

part.type === "data-tool-call-suspended"

,

Question:
How do I resume the tool call after approval using

useChat

useChat

useChat

useChat

exposes

addToolApprovalResponse

addToolApprovalResponse

, but:

* Is this the correct function to call?
* Will it actually resume the suspended tool execution on the Mastra server?

TL;DR
When the user clicks Approve in the tool confirmation UI, which function should be called to resume the suspended tool call? Any way too do this is cool , even if it just a hack

const {
  messages,
  sendMessage,
  status,
  stop,
  regenerate,
  addToolApprovalResponse,
  addToolOutput,
} = useChat({
  transport: new DefaultChatTransport({
    api: "http://localhost:4111/chat/weather-agent",
  }),
});

const handleSubmit = async (message: PromptInputMessage) => {
  sendMessage({ text: message.text }, { body: { model } });
  setText("");
};

const {
  messages,
  sendMessage,
  status,
  stop,
  regenerate,
  addToolApprovalResponse,
  addToolOutput,
} = useChat({
  transport: new DefaultChatTransport({
    api: "http://localhost:4111/chat/weather-agent",
  }),
});

const handleSubmit = async (message: PromptInputMessage) => {
  sendMessage({ text: message.text }, { body: { model } });
  setText("");
};

Mastra weather tool (called by weather agent)

export const weatherTool = createTool({
  id: "get-weather",
  description: "Get current weather for a location",

  inputSchema: z.object({
    location: z.string().describe("City name"),
  }),
  suspendSchema: z.object({ reason: z.string() }),
  resumeSchema: z.object({ approved: z.boolean() }),
  outputSchema: z.object({
  }),
  execute: async (inputData, context) => {
    const { resumeData: { approved } = {}, suspend } = context?.agent ?? {};

    if (!approved) return suspend?.({ reason: "Approval required." });
    return await getWeather(inputData.location);
  },
});

export const weatherTool = createTool({
  id: "get-weather",
  description: "Get current weather for a location",

  inputSchema: z.object({
    location: z.string().describe("City name"),
  }),
  suspendSchema: z.object({ reason: z.string() }),
  resumeSchema: z.object({ approved: z.boolean() }),
  outputSchema: z.object({
  }),
  execute: async (inputData, context) => {
    const { resumeData: { approved } = {}, suspend } = context?.agent ?? {};

    if (!approved) return suspend?.({ reason: "Approval required." });
    return await getWeather(inputData.location);
  },
});

Mastra•2mo ago•

2 replies

ꪑ᥅.ᥴ᥅ꪊ꠸ᦓꫀ !

How to resume a suspended tool call from useChat (Mastra standalone + AI SDK)

AgentTools

Hi,
I have a question about resuming a suspended tool call.
Setup:
Standalone Mastra server (

http://localhost:4111

http://localhost:4111

), Next.js frontend, AI SDK (

useChat

useChat

hook).
All dependencies are on their latest versions.

Current behavior:
When a tool requires approval, I receive a message part with

part.type === "data-tool-call-suspended"

part.type === "data-tool-call-suspended"

,

Question:
How do I resume the tool call after approval using

useChat

useChat

useChat

useChat

exposes

addToolApprovalResponse

addToolApprovalResponse

const {
  messages,
  sendMessage,
  status,
  stop,
  regenerate,
  addToolApprovalResponse,
  addToolOutput,
} = useChat({
  transport: new DefaultChatTransport({
    api: "http://localhost:4111/chat/weather-agent",
  }),
});

const handleSubmit = async (message: PromptInputMessage) => {
  sendMessage({ text: message.text }, { body: { model } });
  setText("");
};

const {
  messages,
  sendMessage,
  status,
  stop,
  regenerate,
  addToolApprovalResponse,
  addToolOutput,
} = useChat({
  transport: new DefaultChatTransport({
    api: "http://localhost:4111/chat/weather-agent",
  }),
});

const handleSubmit = async (message: PromptInputMessage) => {
  sendMessage({ text: message.text }, { body: { model } });
  setText("");
};

Mastra weather tool (called by weather agent)

export const weatherTool = createTool({
  id: "get-weather",
  description: "Get current weather for a location",

  inputSchema: z.object({
    location: z.string().describe("City name"),
  }),
  suspendSchema: z.object({ reason: z.string() }),
  resumeSchema: z.object({ approved: z.boolean() }),
  outputSchema: z.object({
  }),
  execute: async (inputData, context) => {
    const { resumeData: { approved } = {}, suspend } = context?.agent ?? {};

    if (!approved) return suspend?.({ reason: "Approval required." });
    return await getWeather(inputData.location);
  },
});

export const weatherTool = createTool({
  id: "get-weather",
  description: "Get current weather for a location",

  inputSchema: z.object({
    location: z.string().describe("City name"),
  }),
  suspendSchema: z.object({ reason: z.string() }),
  resumeSchema: z.object({ approved: z.boolean() }),
  outputSchema: z.object({
  }),
  execute: async (inputData, context) => {
    const { resumeData: { approved } = {}, suspend } = context?.agent ?? {};

    if (!approved) return suspend?.({ reason: "Approval required." });
    return await getWeather(inputData.location);
  },
});

How to resume a suspended tool call from useChat (Mastra standalone + AI SDK)

Mastra weather tool (called by weather agent)

How to resume a suspended tool call from useChat (Mastra standalone + AI SDK)

Mastra weather tool (called by weather agent)

Similar Threads

Similar Threads

Similar Threads