Sampling compatibility

import {
  MCPClient,
  type OnSamplingCallback,
  type SamplingCreateMessageParams,
} from "@mcp-use/client";

async function completeWithModel(
  params: SamplingCreateMessageParams
): Promise<string> {
  // Send params.messages and the other sampling constraints to your model.
  throw new Error(`Connect a model for ${params.messages.length} messages`);
}

const onSampling: OnSamplingCallback = async (params) => {
  const text = await completeWithModel(params);

  return {
    role: "assistant",
    content: { type: "text", text },
    model: "your-model",
    stopReason: "endTurn",
  };
};

const client = new MCPClient({
  mcpServers: {
    demo: {
      url: "http://localhost:3000/mcp",
      onSampling,
    },
  },
});

try {
  await client.connect("demo");
  // A compatible server flow can now call onSampling.
} finally {
  await client.close();
}