import { z } from "zod";
import { getYouTubeVideoInfo } from "../handlers/get-video-info.handler.js";
import { ToolDefinition } from "../types/tool-definition.js";
const toolName = "get-video-info-for-summary-from-url";
const toolDescription = "Get details or explanation about a YouTube video, get captions or subtitles of Youtube video from a URL";
const toolSchema = {
videoUrl: z.string().describe("The URL or ID of the YouTube video"),
languageCode: z.string().describe("The language code of the video").optional(),
};
const toolHandler = async (args: { videoUrl: string; languageCode?: string }, extra: any) => {
const response = await getYouTubeVideoInfo(args.videoUrl, args.languageCode);
if (response.isError) {
return {
content: [
{
type: "text" as const,
text: `Error getting video info: ${response.error}`,
},
],
};
}
const videoInfo = response.result;
const captions = videoInfo?.subtitles?.map((subtitle) => subtitle.text).join(". ");
if (!captions) {
return {
content: [
{ type: "text" as const, text: "No captions found" },
],
};
}
return {
content: [
{
type: "text" as const,
text: `Title: ${videoInfo?.title}\n\nDescription: ${videoInfo?.description}\n\nCaption: ${captions}`,
},
],
};
};
export const GetVideoInfoTool: ToolDefinition<typeof toolSchema> = {
name: toolName,
description: toolDescription,
schema: toolSchema,
handler: toolHandler,
};
