Route a question to a different AI model

import { anthropic } from "@ai-sdk/anthropic";
import { task } from "@trigger.dev/sdk";
import { generateObject, generateText } from "ai";
import { z } from "zod";

// The router's structured decision. generateObject validates the model
// output against this schema, so there's no manual JSON parsing.
const routingSchema = z.object({
  model: z.enum(["claude-haiku-4-5", "claude-sonnet-4-5"]),
  reason: z.string(),
});

export const routeAndAnswerQuestion = task({
  id: "route-and-answer-question",
  run: async (payload: { question: string }) => {
    // Step 1: Classify the question and pick a model
    const { object: routing } = await generateObject({
      model: anthropic("claude-haiku-4-5"),
      schema: routingSchema,
      system:
        "You are a routing assistant. Pick the model best suited to answer the question:\n" +
        "- claude-haiku-4-5 for simple, common, or straightforward questions\n" +
        "- claude-sonnet-4-5 for complex, unusual, or questions needing deep reasoning",
      prompt: payload.question,
      experimental_telemetry: {
        isEnabled: true,
        functionId: "route-question",
      },
    });

    // Step 2: Answer with the selected model
    const answer = await generateText({
      model: anthropic(routing.model),
      prompt: payload.question,
      experimental_telemetry: {
        isEnabled: true,
        functionId: "answer-question",
      },
    });

    return {
      answer: answer.text,
      selectedModel: routing.model,
      routingReason: routing.reason,
    };
  },
});

{
  "question": "How many planets are there in the solar system?"
}