Vercel AI SDK

import { openai } from "@ai-sdk/openai";
import { generateText } from "ai";
import { getRefileTool } from "./refileTool.js";

const { text } = await generateText({
	model: openai("gpt-4o"),
	prompt:
		"Create a research report on the situation in the rain forest. Reply with the URL of the resulting PDF.",
	maxSteps: 3,
	tools: {
		generatePdf: getRefileTool("<token>"),
	},
});

console.log(text);

import { z } from "zod";
import { tool } from "ai";

/**
 * Creates a tool for generating PDFs from markdown text using the Refile API
 *
 * @param {string} refileApiKey - Your Refile API key
 * @param {string} [theme="modern"] - PDF theme (e.g., "modern", "minimal", "academic")
 * @param {string} [pageNumeration="center"] - Page number position ("center", "right", "none")
 * @param {string} [pageSize="A4"] - Page size (e.g., "A4", "Letter", "Legal")
 * @param {string} [refileApiBaseUrl="https://www.refile.co/api/v1"] - Refile API base URL
 * @returns {Object} - AI SDK compatible tool for PDF generation
 */
export function getRefileTool(
	refileApiKey,
	theme = "modern",
	pageNumeration = "center",
	pageSize = "A4",
	refileApiBaseUrl = "https://www.refile.co/api/v1"
) {
	return tool({
		description: "Generate a PDF from a markdown text",
		parameters: z.object({
			markdown: z.string().describe("The markdown to generate a PDF from"),
		}),
		execute: async ({ markdown }) => {
			// Prepare request options
			const options = {
				method: "POST",
				headers: {
					"Content-Type": "application/json",
					Authorization: `Bearer ${refileApiKey}`,
				},
				body: JSON.stringify({
					markdown: markdown,
					theme: theme,
					pageNumeration: pageNumeration,
					pageSize: pageSize,
					output: "url",
				}),
			};

			try {
				// Make the API call
				const response = await fetch(
					`${refileApiBaseUrl}/pdf/markdown`,
					options
				);

				if (!response.ok) {
					const errorData = await response.json().catch(() => null);
					throw new Error(
						`Failed to generate PDF: ${response.status} ${response.statusText}${
							errorData ? ` - ${JSON.stringify(errorData)}` : ""
						}`
					);
				}

				// Parse and return the response
				const data = await response.json();
				return {
					url: data.url,
					success: true,
					message: "PDF generated successfully",
				};
			} catch (error) {
				console.error("Error generating PDF:", error);
				return {
					success: false,
					error: error.message,
					message: "Failed to generate PDF",
				};
			}
		},
	});
}