import { anthropic } from "@ai-sdk/anthropic"; import { generateText, Output } from "ai"; import { AiModels } from "@/constants/ai-models"; import { CitationFallbackSchema, CitationFallbackSystemPrompt } from "@/lib/ai/prompts/citation-fallback-system.prompt"; interface GetAICitationFallbackProps { originalQuery: string; semanticQuery: string | null; title?: string | null; } export const getAICitationFallback = async ({ originalQuery, semanticQuery, title }: GetAICitationFallbackProps) => { try { const userMessage = title ? `DOCUMENT TITLE (Context): ${title}\n\nOriginal query: ${originalQuery}\n\nSemantic query: ${semanticQuery ?? originalQuery}\n\nPlease provide a citation for the most critical claim in this text. Use the document title as context to understand the domain and academic field.` : `Original query: ${originalQuery}\n\nSemantic query: ${semanticQuery ?? originalQuery}\n\nPlease provide a citation for the most critical claim in this text.`; const { experimental_output } = await generateText({ model: AiModels["anthropic/claude-sonnet-4.5"], system: CitationFallbackSystemPrompt, prompt: userMessage, tools: { web_search: anthropic.tools.webSearch_20250305({ maxUses: 5, }), }, experimental_output: Output.object({ schema: CitationFallbackSchema, }), }); return { error: null, data: experimental_output }; } catch (error) { console.error("Failed to generate AI citation:", error); return { error, data: null, }; } };