@mux/ai 0.1.1 → 0.1.2

package/dist/chapters.js

@@ -1,255 +0,0 @@
-import Mux from '@mux/mux-node';
-import OpenAI from 'openai';
-import Anthropic from '@anthropic-ai/sdk';
-import { z } from 'zod';
-import { zodTextFormat } from 'openai/helpers/zod';
-const chaptersSchema = z.object({
-    chapters: z.array(z.object({
-        startTime: z.number(),
-        title: z.string()
-    }))
-});
-const DEFAULT_SYSTEM_PROMPT = `Your role is to segment the following captions into chunked chapters, summarising each chapter with a title. 
-
-Analyze the transcript and create logical chapter breaks based on topic changes, major transitions, or distinct sections of content. Each chapter should represent a meaningful segment of the video.
-
-You must respond with valid JSON in exactly this format:
-{
-  "chapters": [
-    {"startTime": 0, "title": "Introduction"},
-    {"startTime": 45.5, "title": "Main Topic Discussion"},
-    {"startTime": 120.0, "title": "Conclusion"}
-  ]
-}
-
-Important rules:
-- startTime must be in seconds (not HH:MM:SS format)
-- Always start with startTime: 0 for the first chapter
-- Create 3-8 chapters depending on content length and natural breaks
-- Chapter titles should be concise and descriptive
-- Do not include any text before or after the JSON
-- The JSON must be valid and parseable`;
-const ANTHROPIC_JSON_PROMPT = `You must respond with valid JSON in exactly this format:
-{
-  "chapters": [
-    {"startTime": 0, "title": "Chapter title here"},
-    {"startTime": 45.5, "title": "Another chapter title"}
-  ]
-}
-
-Do not include any text before or after the JSON. The JSON must be valid and parseable.`;
-/**
- * Converts VTT timestamp (HH:MM:SS.mmm) to seconds
- */
-function vttTimestampToSeconds(timestamp) {
-    const parts = timestamp.split(':');
-    if (parts.length !== 3)
-        return 0;
-    const hours = parseInt(parts[0], 10) || 0;
-    const minutes = parseInt(parts[1], 10) || 0;
-    const seconds = parseFloat(parts[2]) || 0;
-    return hours * 3600 + minutes * 60 + seconds;
-}
-/**
- * Extracts timestamps and text from VTT content for chapter generation
- */
-function extractTimestampsFromVTT(vttContent) {
-    if (!vttContent.trim()) {
-        return '';
-    }
-    const lines = vttContent.split('\n');
-    const segments = [];
-    for (let i = 0; i < lines.length; i++) {
-        const line = lines[i].trim();
-        // Find timestamp lines (contain -->)
-        if (line.includes('-->')) {
-            const startTime = line.split(' --> ')[0].trim();
-            const timeInSeconds = vttTimestampToSeconds(startTime);
-            // Get the subtitle text (next non-empty line)
-            let j = i + 1;
-            while (j < lines.length && !lines[j].trim()) {
-                j++;
-            }
-            if (j < lines.length) {
-                const text = lines[j].trim().replace(/<[^>]*>/g, ''); // Remove formatting tags
-                if (text) {
-                    segments.push({ time: timeInSeconds, text });
-                }
-            }
-        }
-    }
-    // Create a readable transcript with timestamps for the AI
-    return segments
-        .map(segment => `[${Math.floor(segment.time)}s] ${segment.text}`)
-        .join('\n');
-}
-export async function generateChapters(assetId, languageCode, options = {}) {
-    const { provider = 'openai', model, muxTokenId, muxTokenSecret, openaiApiKey, anthropicApiKey, ...config } = options;
-    // Set default models based on provider
-    const defaultModel = provider === 'anthropic' ? 'claude-3-5-haiku-20241022' : 'gpt-4o-mini';
-    const finalModel = model || defaultModel;
-    // Validate required credentials
-    const muxId = muxTokenId || process.env.MUX_TOKEN_ID;
-    const muxSecret = muxTokenSecret || process.env.MUX_TOKEN_SECRET;
-    const openaiKey = openaiApiKey || process.env.OPENAI_API_KEY;
-    const anthropicKey = anthropicApiKey || process.env.ANTHROPIC_API_KEY;
-    if (!muxId || !muxSecret) {
-        throw new Error('Mux credentials are required. Provide muxTokenId and muxTokenSecret in options or set MUX_TOKEN_ID and MUX_TOKEN_SECRET environment variables.');
-    }
-    if (provider === 'openai' && !openaiKey) {
-        throw new Error('OpenAI API key is required for OpenAI provider. Provide openaiApiKey in options or set OPENAI_API_KEY environment variable.');
-    }
-    if (provider === 'anthropic' && !anthropicKey) {
-        throw new Error('Anthropic API key is required for Anthropic provider. Provide anthropicApiKey in options or set ANTHROPIC_API_KEY environment variable.');
-    }
-    // Initialize clients
-    const mux = new Mux({
-        tokenId: muxId,
-        tokenSecret: muxSecret,
-    });
-    let openaiClient;
-    let anthropicClient;
-    if (provider === 'openai') {
-        openaiClient = new OpenAI({
-            apiKey: openaiKey,
-        });
-    }
-    else if (provider === 'anthropic') {
-        anthropicClient = new Anthropic({
-            apiKey: anthropicKey,
-        });
-    }
-    // Fetch asset data from Mux
-    let assetData;
-    try {
-        const asset = await mux.video.assets.retrieve(assetId);
-        assetData = asset;
-    }
-    catch (error) {
-        throw new Error(`Failed to fetch asset from Mux: ${error instanceof Error ? error.message : 'Unknown error'}`);
-    }
-    // Get playback ID
-    const playbackId = assetData.playback_ids?.[0]?.id;
-    if (!playbackId) {
-        throw new Error('No playback ID found for this asset');
-    }
-    // Find caption track in the specified language
-    if (!assetData.tracks) {
-        throw new Error('No tracks found for this asset');
-    }
-    const captionTrack = assetData.tracks.find((track) => track.type === 'text' &&
-        track.status === 'ready' &&
-        track.text_type === 'subtitles' &&
-        track.language_code === languageCode);
-    if (!captionTrack) {
-        throw new Error(`No caption track found for language '${languageCode}'. Available languages: ${assetData.tracks.filter(t => t.type === 'text').map(t => t.language_code).join(', ')}`);
-    }
-    // Fetch the VTT content
-    const transcriptUrl = `https://stream.mux.com/${playbackId}/text/${captionTrack.id}.vtt`;
-    let vttContent;
-    try {
-        const transcriptResponse = await fetch(transcriptUrl);
-        if (!transcriptResponse.ok) {
-            throw new Error(`Failed to fetch VTT: ${transcriptResponse.statusText}`);
-        }
-        vttContent = await transcriptResponse.text();
-    }
-    catch (error) {
-        throw new Error(`Failed to fetch caption track: ${error instanceof Error ? error.message : 'Unknown error'}`);
-    }
-    // Extract timestamped transcript for AI processing
-    const timestampedTranscript = extractTimestampsFromVTT(vttContent);
-    if (!timestampedTranscript) {
-        throw new Error('No usable content found in caption track');
-    }
-    // Generate chapters using AI
-    let chaptersData = null;
-    if (provider === 'openai') {
-        try {
-            const response = await openaiClient.responses.parse({
-                model: finalModel,
-                input: [
-                    {
-                        role: "system",
-                        content: DEFAULT_SYSTEM_PROMPT,
-                    },
-                    {
-                        role: "user",
-                        content: [
-                            {
-                                type: "input_text",
-                                text: timestampedTranscript,
-                            },
-                        ],
-                    },
-                ],
-                text: {
-                    format: zodTextFormat(chaptersSchema, "chapters"),
-                },
-            });
-            chaptersData = response.output_parsed;
-        }
-        catch (error) {
-            throw new Error(`Failed to generate chapters with OpenAI: ${error instanceof Error ? error.message : 'Unknown error'}`);
-        }
-    }
-    else if (provider === 'anthropic') {
-        const anthropicPrompt = `${DEFAULT_SYSTEM_PROMPT}
-
-${ANTHROPIC_JSON_PROMPT}
-
-Transcript:
-${timestampedTranscript}`;
-        try {
-            const response = await anthropicClient.messages.create({
-                model: finalModel,
-                max_tokens: 2000,
-                messages: [
-                    {
-                        role: "user",
-                        content: anthropicPrompt,
-                    },
-                ],
-            });
-            const content = response.content[0];
-            if (content.type === 'text') {
-                const jsonText = content.text.trim();
-                try {
-                    chaptersData = JSON.parse(jsonText);
-                }
-                catch (parseError) {
-                    throw new Error(`Failed to parse JSON response from Anthropic: ${parseError instanceof Error ? parseError.message : 'Unknown error'}`);
-                }
-            }
-            else {
-                throw new Error('Unexpected response type from Anthropic');
-            }
-        }
-        catch (error) {
-            throw new Error(`Failed to generate chapters with Anthropic: ${error instanceof Error ? error.message : 'Unknown error'}`);
-        }
-    }
-    else {
-        throw new Error(`Unsupported provider: ${provider}`);
-    }
-    if (!chaptersData || !chaptersData.chapters) {
-        throw new Error('No chapters generated from AI response');
-    }
-    // Validate and sort chapters
-    const validChapters = chaptersData.chapters
-        .filter(chapter => typeof chapter.startTime === 'number' && typeof chapter.title === 'string')
-        .sort((a, b) => a.startTime - b.startTime);
-    if (validChapters.length === 0) {
-        throw new Error('No valid chapters found in AI response');
-    }
-    // Ensure first chapter starts at 0
-    if (validChapters[0].startTime !== 0) {
-        validChapters[0].startTime = 0;
-    }
-    return {
-        assetId,
-        languageCode,
-        chapters: validChapters,
-    };
-}
-//# sourceMappingURL=chapters.js.map

package/dist/chapters.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"chapters.js","sourceRoot":"","sources":["../src/chapters.ts"],"names":[],"mappings":"AAAA,OAAO,GAAG,MAAM,eAAe,CAAC;AAChC,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAC1C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAqBnD,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9B,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;QACzB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;KAClB,CAAC,CAAC;CACJ,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG;;;;;;;;;;;;;;;;;;;uCAmBS,CAAC;AAExC,MAAM,qBAAqB,GAAG;;;;;;;;wFAQ0D,CAAC;AAEzF;;GAEG;AACH,SAAS,qBAAqB,CAAC,SAAiB;IAC9C,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACnC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAEjC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,OAAO,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAE1C,OAAO,KAAK,GAAG,IAAI,GAAG,OAAO,GAAG,EAAE,GAAG,OAAO,CAAC;AAC/C,CAAC;AAED;;GAEG;AACH,SAAS,wBAAwB,CAAC,UAAkB;IAClD,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC;QACvB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,QAAQ,GAAwC,EAAE,CAAC;IAEzD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE7B,qCAAqC;QACrC,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAChD,MAAM,aAAa,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;YAEvD,8CAA8C;YAC9C,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACd,OAAO,CAAC,GAAG,KAAK,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC5C,CAAC,EAAE,CAAC;YACN,CAAC;YAED,IAAI,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;gBACrB,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC,yBAAyB;gBAC/E,IAAI,IAAI,EAAE,CAAC;oBACT,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC/C,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,0DAA0D;IAC1D,OAAO,QAAQ;SACZ,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;SAChE,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAe,EACf,YAAoB,EACpB,UAA2B,EAAE;IAE7B,MAAM,EACJ,QAAQ,GAAG,QAAQ,EACnB,KAAK,EACL,UAAU,EACV,cAAc,EACd,YAAY,EACZ,eAAe,EACf,GAAG,MAAM,EACV,GAAG,OAAO,CAAC;IAEZ,uCAAuC;IACvC,MAAM,YAAY,GAAG,QAAQ,KAAK,WAAW,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,aAAa,CAAC;IAC5F,MAAM,UAAU,GAAG,KAAK,IAAI,YAAY,CAAC;IAEzC,gCAAgC;IAChC,MAAM,KAAK,GAAG,UAAU,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IACrD,MAAM,SAAS,GAAG,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IACjE,MAAM,SAAS,GAAG,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAC7D,MAAM,YAAY,GAAG,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;IAEtE,IAAI,CAAC,KAAK,IAAI,CAAC,SAAS,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,gJAAgJ,CAAC,CAAC;IACpK,CAAC;IAED,IAAI,QAAQ,KAAK,QAAQ,IAAI,CAAC,SAAS,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CAAC,6HAA6H,CAAC,CAAC;IACjJ,CAAC;IAED,IAAI,QAAQ,KAAK,WAAW,IAAI,CAAC,YAAY,EAAE,CAAC;QAC9C,MAAM,IAAI,KAAK,CAAC,yIAAyI,CAAC,CAAC;IAC7J,CAAC;IAED,qBAAqB;IACrB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC;QAClB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,SAAS;KACvB,CAAC,CAAC;IAEH,IAAI,YAAgC,CAAC;IACrC,IAAI,eAAsC,CAAC;IAE3C,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,YAAY,GAAG,IAAI,MAAM,CAAC;YACxB,MAAM,EAAE,SAAU;SACnB,CAAC,CAAC;IACL,CAAC;SAAM,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QACpC,eAAe,GAAG,IAAI,SAAS,CAAC;YAC9B,MAAM,EAAE,YAAa;SACtB,CAAC,CAAC;IACL,CAAC;IAED,4BAA4B;IAC5B,IAAI,SAAS,CAAC;IACd,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACvD,SAAS,GAAG,KAAK,CAAC;IACpB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;IACjH,CAAC;IAED,kBAAkB;IAClB,MAAM,UAAU,GAAG,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IACnD,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IAED,+CAA+C;IAC/C,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CACnD,KAAK,CAAC,IAAI,KAAK,MAAM;QACrB,KAAK,CAAC,MAAM,KAAK,OAAO;QACxB,KAAK,CAAC,SAAS,KAAK,WAAW;QAC/B,KAAK,CAAC,aAAa,KAAK,YAAY,CACrC,CAAC;IAEF,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,wCAAwC,YAAY,2BAA2B,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACzL,CAAC;IAED,wBAAwB;IACxB,MAAM,aAAa,GAAG,0BAA0B,UAAU,SAAS,YAAY,CAAC,EAAE,MAAM,CAAC;IAEzF,IAAI,UAAkB,CAAC;IACvB,IAAI,CAAC;QACH,MAAM,kBAAkB,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,CAAC;QACtD,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,wBAAwB,kBAAkB,CAAC,UAAU,EAAE,CAAC,CAAC;QAC3E,CAAC;QACD,UAAU,GAAG,MAAM,kBAAkB,CAAC,IAAI,EAAE,CAAC;IAC/C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,kCAAkC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;IAChH,CAAC;IAED,mDAAmD;IACnD,MAAM,qBAAqB,GAAG,wBAAwB,CAAC,UAAU,CAAC,CAAC;IAEnE,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC9D,CAAC;IAED,6BAA6B;IAC7B,IAAI,YAAY,GAAmC,IAAI,CAAC;IAExD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,YAAa,CAAC,SAAS,CAAC,KAAK,CAAC;gBACnD,KAAK,EAAE,UAAU;gBACjB,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,QAAQ;wBACd,OAAO,EAAE,qBAAqB;qBAC/B;oBACD;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,YAAY;gCAClB,IAAI,EAAE,qBAAqB;6BAC5B;yBACF;qBACF;iBACF;gBACD,IAAI,EAAE;oBACJ,MAAM,EAAE,aAAa,CAAC,cAAc,EAAE,UAAU,CAAC;iBAClD;aACF,CAAC,CAAC;YAEH,YAAY,GAAG,QAAQ,CAAC,aAAa,CAAC;QACxC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,4CAA4C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QAC1H,CAAC;IACH,CAAC;SAAM,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QACpC,MAAM,eAAe,GAAG,GAAG,qBAAqB;;EAElD,qBAAqB;;;EAGrB,qBAAqB,EAAE,CAAC;QAEtB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,eAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACtD,KAAK,EAAE,UAAU;gBACjB,UAAU,EAAE,IAAI;gBAChB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,eAAe;qBACzB;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACpC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACrC,IAAI,CAAC;oBACH,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACtC,CAAC;gBAAC,OAAO,UAAU,EAAE,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC,iDAAiD,UAAU,YAAY,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;gBACzI,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,+CAA+C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QAC7H,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,IAAI,CAAC,YAAY,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;QAC5C,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,6BAA6B;IAC7B,MAAM,aAAa,GAAG,YAAY,CAAC,QAAQ;SACxC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ,IAAI,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,CAAC;SAC7F,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IAE7C,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,mCAAmC;IACnC,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,KAAK,CAAC,EAAE,CAAC;QACrC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC;IACjC,CAAC;IAED,OAAO;QACL,OAAO;QACP,YAAY;QACZ,QAAQ,EAAE,aAAa;KACxB,CAAC;AACJ,CAAC"}

package/dist/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,SAAS,CAAC;AACxB,cAAc,cAAc,CAAC;AAC7B,cAAc,iBAAiB,CAAC;AAChC,cAAc,eAAe,CAAC;AAC9B,cAAc,qBAAqB,CAAC;AACpC,cAAc,YAAY,CAAC;AAC3B,cAAc,sBAAsB,CAAC;AAErC,eAAO,MAAM,OAAO,UAAU,CAAC"}

package/dist/moderation.d.ts

@@ -1,39 +0,0 @@
-import { MuxAIOptions } from './types';
-import { ImageDownloadOptions } from './utils/image-download';
-export interface ThumbnailModerationScore {
-    url: string;
-    sexual: number;
-    violence: number;
-    error: boolean;
-}
-export interface ModerationResult {
-    assetId: string;
-    thumbnailScores: ThumbnailModerationScore[];
-    maxScores: {
-        sexual: number;
-        violence: number;
-    };
-    exceedsThreshold: boolean;
-    thresholds: {
-        sexual: number;
-        violence: number;
-    };
-}
-export interface ModerationOptions extends MuxAIOptions {
-    provider?: 'openai' | 'hive';
-    model?: string;
-    thresholds?: {
-        sexual?: number;
-        violence?: number;
-    };
-    thumbnailInterval?: number;
-    thumbnailWidth?: number;
-    maxConcurrent?: number;
-    /** Method for submitting images to AI providers (default: 'url') */
-    imageSubmissionMode?: 'url' | 'base64';
-    /** Options for image download when using base64 submission mode */
-    imageDownloadOptions?: ImageDownloadOptions;
-    hiveApiKey?: string;
-}
-export declare function getModerationScores(assetId: string, options?: ModerationOptions): Promise<ModerationResult>;
-//# sourceMappingURL=moderation.d.ts.map

package/dist/moderation.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"moderation.d.ts","sourceRoot":"","sources":["../src/moderation.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,oBAAoB,EAA0B,MAAM,wBAAwB,CAAC;AAEtF,MAAM,WAAW,wBAAwB;IACvC,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,OAAO,CAAC;CAChB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE,wBAAwB,EAAE,CAAC;IAC5C,SAAS,EAAE;QACT,MAAM,EAAE,MAAM,CAAC;QACf,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,gBAAgB,EAAE,OAAO,CAAC;IAC1B,UAAU,EAAE;QACV,MAAM,EAAE,MAAM,CAAC;QACf,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;CACH;AAED,MAAM,WAAW,iBAAkB,SAAQ,YAAY;IACrD,QAAQ,CAAC,EAAE,QAAQ,GAAG,MAAM,CAAC;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE;QACX,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oEAAoE;IACpE,mBAAmB,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC;IACvC,mEAAmE;IACnE,oBAAoB,CAAC,EAAE,oBAAoB,CAAC;IAC5C,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAwUD,wBAAsB,mBAAmB,CACvC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,iBAAsB,GAC9B,OAAO,CAAC,gBAAgB,CAAC,CAoH3B"}

package/dist/moderation.js

@@ -1,341 +0,0 @@
-import Mux from '@mux/mux-node';
-import OpenAI from 'openai';
-import { downloadImagesAsBase64 } from './utils/image-download';
-const DEFAULT_THRESHOLDS = {
-    sexual: 0.7,
-    violence: 0.8
-};
-// Process promises in batches with maximum concurrency limit
-async function processConcurrently(items, processor, maxConcurrent = 5) {
-    const results = [];
-    for (let i = 0; i < items.length; i += maxConcurrent) {
-        const batch = items.slice(i, i + maxConcurrent);
-        const batchPromises = batch.map(processor);
-        const batchResults = await Promise.all(batchPromises);
-        results.push(...batchResults);
-    }
-    return results;
-}
-// Mapping Hive categories to OpenAI-compatible scores
-const HIVE_SEXUAL_CATEGORIES = [
-    'general_nsfw',
-    'general_suggestive',
-    'yes_sexual_activity',
-    'female_underwear',
-    'male_underwear',
-    'bra',
-    'panties',
-    'sex_toys',
-    'nudity_female',
-    'nudity_male',
-    'cleavage',
-    'swimwear'
-];
-const HIVE_VIOLENCE_CATEGORIES = [
-    'gun_in_hand',
-    'gun_not_in_hand',
-    'animated_gun',
-    'knife_in_hand',
-    'knife_not_in_hand',
-    'culinary_knife_not_in_hand',
-    'culinary_knife_in_hand',
-    'very_bloody',
-    'a_little_bloody',
-    'other_blood',
-    'hanging',
-    'noose',
-    'human_corpse',
-    'animated_corpse',
-    'emaciated_body',
-    'self_harm',
-    'animal_abuse',
-    'fights',
-    'garm_death_injury_or_military_conflict'
-];
-// Generates thumbnail URLs at regular intervals based on video duration
-function getThumbnailUrls(playbackId, duration, options = {}) {
-    const { interval = 10, width = 640 } = options;
-    const timestamps = [];
-    if (duration <= 50) {
-        // Short videos: 5 evenly spaced thumbnails
-        const spacing = duration / 6;
-        for (let i = 1; i <= 5; i++) {
-            timestamps.push(Math.round(i * spacing));
-        }
-    }
-    else {
-        // Longer videos: one thumbnail every interval seconds
-        for (let time = 0; time < duration; time += interval) {
-            timestamps.push(time);
-        }
-    }
-    return timestamps.map((time) => `https://image.mux.com/${playbackId}/thumbnail.png?time=${time}&width=${width}`);
-}
-// Sends thumbnail URLs to OpenAI moderation API with concurrency limiting
-async function requestOpenAIModeration(imageUrls, openaiClient, model, maxConcurrent = 5, submissionMode = 'url', downloadOptions) {
-    // If using base64 mode, download all images first
-    if (submissionMode === 'base64') {
-        try {
-            const downloadResults = await downloadImagesAsBase64(imageUrls, downloadOptions, maxConcurrent);
-            // Process each downloaded image through OpenAI moderation
-            const processor = async (downloadResult) => {
-                try {
-                    const moderation = await openaiClient.moderations.create({
-                        model,
-                        input: [
-                            {
-                                type: "image_url",
-                                image_url: {
-                                    url: downloadResult.base64Data, // Use base64 data URI
-                                },
-                            },
-                        ],
-                    });
-                    const categoryScores = moderation.results[0].category_scores;
-                    return {
-                        url: downloadResult.url, // Return original URL for tracking
-                        sexual: categoryScores.sexual || 0,
-                        violence: categoryScores.violence || 0,
-                        error: false
-                    };
-                }
-                catch (error) {
-                    console.error(`Failed to moderate downloaded image ${downloadResult.url}:`, error);
-                    return {
-                        url: downloadResult.url,
-                        sexual: 0,
-                        violence: 0,
-                        error: true,
-                    };
-                }
-            };
-            return processConcurrently(downloadResults, processor, maxConcurrent);
-        }
-        catch (error) {
-            console.error('Failed to download images for base64 submission:', error);
-            // Return error results for all URLs
-            return imageUrls.map(url => ({
-                url,
-                sexual: 0,
-                violence: 0,
-                error: true,
-            }));
-        }
-    }
-    // Original URL-based submission mode
-    const processor = async (url) => {
-        try {
-            const moderation = await openaiClient.moderations.create({
-                model,
-                input: [
-                    {
-                        type: "image_url",
-                        image_url: {
-                            url: url,
-                        },
-                    },
-                ],
-            });
-            const categoryScores = moderation.results[0].category_scores;
-            return {
-                url,
-                sexual: categoryScores.sexual || 0,
-                violence: categoryScores.violence || 0,
-                error: false
-            };
-        }
-        catch (error) {
-            console.error("Failed to moderate image:", error);
-            return {
-                url,
-                sexual: 0,
-                violence: 0,
-                error: true,
-            };
-        }
-    };
-    return processConcurrently(imageUrls, processor, maxConcurrent);
-}
-// Sends thumbnail URLs to Hive moderation API with concurrency limiting
-async function requestHiveModeration(imageUrls, hiveApiKey, maxConcurrent = 5, submissionMode = 'url', downloadOptions) {
-    // If using base64 mode, download all images first and upload via multipart/form-data
-    if (submissionMode === 'base64') {
-        try {
-            const downloadResults = await downloadImagesAsBase64(imageUrls, downloadOptions, maxConcurrent);
-            // Process each downloaded image through Hive moderation via file upload
-            const processor = async (downloadResult) => {
-                try {
-                    // Create form data with image buffer
-                    const formData = new FormData();
-                    // Create a Blob from the buffer for form data
-                    const imageBlob = new Blob([downloadResult.buffer], {
-                        type: downloadResult.contentType
-                    });
-                    // Get file extension from content type
-                    const extension = downloadResult.contentType.split('/')[1] || 'png';
-                    formData.append('media', imageBlob, `image.${extension}`);
-                    const response = await fetch('https://api.thehive.ai/api/v2/task/sync', {
-                        method: 'POST',
-                        headers: {
-                            'Authorization': `Token ${hiveApiKey}`,
-                            // Don't set Content-Type header - let fetch set it with boundary for multipart
-                        },
-                        body: formData
-                    });
-                    if (!response.ok) {
-                        throw new Error(`Hive API error: ${response.statusText}`);
-                    }
-                    const hiveResult = await response.json();
-                    // Extract scores from Hive response and map to OpenAI format
-                    const classes = hiveResult.status?.[0]?.response?.output?.[0]?.classes || [];
-                    const scoreMap = Object.fromEntries(classes.map((c) => [c.class, c.score]));
-                    const sexualScores = HIVE_SEXUAL_CATEGORIES.map(category => scoreMap[category] || 0);
-                    const violenceScores = HIVE_VIOLENCE_CATEGORIES.map(category => scoreMap[category] || 0);
-                    return {
-                        url: downloadResult.url, // Return original URL for tracking
-                        sexual: Math.max(...sexualScores, 0),
-                        violence: Math.max(...violenceScores, 0),
-                        error: false
-                    };
-                }
-                catch (error) {
-                    console.error(`Failed to moderate uploaded image ${downloadResult.url}:`, error);
-                    return {
-                        url: downloadResult.url,
-                        sexual: 0,
-                        violence: 0,
-                        error: true,
-                    };
-                }
-            };
-            return processConcurrently(downloadResults, processor, maxConcurrent);
-        }
-        catch (error) {
-            console.error('Failed to download images for Hive multipart upload:', error);
-            // Return error results for all URLs
-            return imageUrls.map(url => ({
-                url,
-                sexual: 0,
-                violence: 0,
-                error: true,
-            }));
-        }
-    }
-    // Original URL-based submission mode
-    const processor = async (url) => {
-        try {
-            const response = await fetch('https://api.thehive.ai/api/v2/task/sync', {
-                method: 'POST',
-                headers: {
-                    'Authorization': `Token ${hiveApiKey}`,
-                    'Content-Type': 'application/json'
-                },
-                body: JSON.stringify({ url })
-            });
-            if (!response.ok) {
-                throw new Error(`Hive API error: ${response.statusText}`);
-            }
-            const hiveResult = await response.json();
-            // Extract scores from Hive response and map to OpenAI format
-            // Hive returns scores in status[0].response.output[0].classes as array of {class, score}
-            const classes = hiveResult.status?.[0]?.response?.output?.[0]?.classes || [];
-            const scoreMap = Object.fromEntries(classes.map((c) => [c.class, c.score]));
-            const sexualScores = HIVE_SEXUAL_CATEGORIES.map(category => scoreMap[category] || 0);
-            const violenceScores = HIVE_VIOLENCE_CATEGORIES.map(category => scoreMap[category] || 0);
-            return {
-                url,
-                sexual: Math.max(...sexualScores, 0),
-                violence: Math.max(...violenceScores, 0),
-                error: false
-            };
-        }
-        catch (error) {
-            console.error("Failed to moderate image with Hive:", error);
-            return {
-                url,
-                sexual: 0,
-                violence: 0,
-                error: true,
-            };
-        }
-    };
-    return processConcurrently(imageUrls, processor, maxConcurrent);
-}
-export async function getModerationScores(assetId, options = {}) {
-    const { provider = 'openai', model = 'omni-moderation-latest', thresholds = DEFAULT_THRESHOLDS, thumbnailInterval = 10, thumbnailWidth = 640, maxConcurrent = 5, imageSubmissionMode = 'url', imageDownloadOptions, muxTokenId, muxTokenSecret, openaiApiKey, ...config } = options;
-    if (provider !== 'openai' && provider !== 'hive') {
-        throw new Error('Only OpenAI and Hive providers are currently supported');
-    }
-    // Validate required credentials
-    const muxId = muxTokenId || process.env.MUX_TOKEN_ID;
-    const muxSecret = muxTokenSecret || process.env.MUX_TOKEN_SECRET;
-    const openaiKey = openaiApiKey || process.env.OPENAI_API_KEY;
-    const hiveKey = options.hiveApiKey || process.env.HIVE_API_KEY;
-    if (!muxId || !muxSecret) {
-        throw new Error('Mux credentials are required. Provide muxTokenId and muxTokenSecret in options or set MUX_TOKEN_ID and MUX_TOKEN_SECRET environment variables.');
-    }
-    if (provider === 'openai' && !openaiKey) {
-        throw new Error('OpenAI API key is required for OpenAI provider. Provide openaiApiKey in options or set OPENAI_API_KEY environment variable.');
-    }
-    if (provider === 'hive' && !hiveKey) {
-        throw new Error('Hive API key is required for Hive provider. Provide hiveApiKey in options or set HIVE_API_KEY environment variable.');
-    }
-    // Initialize clients
-    const mux = new Mux({
-        tokenId: muxId,
-        tokenSecret: muxSecret,
-    });
-    let openaiClient;
-    if (provider === 'openai') {
-        openaiClient = new OpenAI({
-            apiKey: openaiKey,
-        });
-    }
-    // Fetch asset data from Mux
-    let assetData;
-    try {
-        const asset = await mux.video.assets.retrieve(assetId);
-        assetData = asset;
-    }
-    catch (error) {
-        throw new Error(`Failed to fetch asset from Mux: ${error instanceof Error ? error.message : 'Unknown error'}`);
-    }
-    // Get playback ID - prefer public playback IDs
-    const publicPlaybackIds = assetData.playback_ids?.filter(pid => pid.policy === 'public') || [];
-    if (publicPlaybackIds.length === 0) {
-        throw new Error('No public playback IDs found for this asset. Moderation requires public playback access.');
-    }
-    const playbackId = publicPlaybackIds[0].id;
-    const duration = assetData.duration || 0;
-    // Generate thumbnail URLs
-    const thumbnailUrls = getThumbnailUrls(playbackId, duration, {
-        interval: thumbnailInterval,
-        width: thumbnailWidth
-    });
-    // Request moderation for all thumbnails
-    let thumbnailScores;
-    if (provider === 'openai') {
-        thumbnailScores = await requestOpenAIModeration(thumbnailUrls, openaiClient, model, maxConcurrent, imageSubmissionMode, imageDownloadOptions);
-    }
-    else if (provider === 'hive') {
-        thumbnailScores = await requestHiveModeration(thumbnailUrls, hiveKey, maxConcurrent, imageSubmissionMode, imageDownloadOptions);
-    }
-    else {
-        throw new Error('Unsupported provider');
-    }
-    // Find highest scores across all thumbnails
-    const maxSexual = Math.max(...thumbnailScores.map(s => s.sexual));
-    const maxViolence = Math.max(...thumbnailScores.map(s => s.violence));
-    const finalThresholds = { ...DEFAULT_THRESHOLDS, ...thresholds };
-    return {
-        assetId,
-        thumbnailScores,
-        maxScores: {
-            sexual: maxSexual,
-            violence: maxViolence
-        },
-        exceedsThreshold: maxSexual > finalThresholds.sexual || maxViolence > finalThresholds.violence,
-        thresholds: finalThresholds
-    };
-}
-//# sourceMappingURL=moderation.js.map