import { openai } from "@ai-sdk/openai";
import { z } from "zod";
export const reviewerTool = {
name: "reviewer",
description:
"Given one or more movie titles, returns reviews and sorts them by rating from highest to lowest.",
parameters: z.object({
movies: z.union([
z.string().describe("A single movie title"),
z.array(z.string()).describe("An array of movie titles"),
]),
}),
execute: async ({ movies }: { movies: string | string[] }) => {
const model = openai("gpt-4o-mini");
const promptText = Array.isArray(movies)
? `Sort the following movies by rating from highest to lowest and provide a short review for each:\n${movies.join(", ")}`
: `Provide a short review and rating for the movie: ${movies}`;
try {
const result = await model.doGenerate({
prompt: [
{ role: "user", content: [{ type: "text", text: promptText }] },
],
temperature: 0.7,
});
const text =
result.content.find((part) => part.type === "text")?.text || "";
return { review: text };
} catch {
return { review: "" };
}
},
};
