@bubblelab/bubble-core 0.1.9 → 0.1.10

package/dist/bubbles/tool-bubble/youtube-tool.js

@@ -0,0 +1,352 @@
+import { z } from 'zod';
+import { ToolBubble } from '../../types/tool-bubble-class.js';
+import { CredentialType } from '@bubblelab/shared-schemas';
+import { ApifyBubble } from '../service-bubble/apify/apify.js';
+// Unified YouTube data types (service-agnostic)
+const YouTubeVideoSchema = z.object({
+    title: z.string().nullable().describe('Video title'),
+    id: z.string().nullable().describe('YouTube video ID'),
+    url: z.string().nullable().describe('Video URL'),
+    viewCount: z.number().nullable().describe('Number of views'),
+    likes: z.number().nullable().describe('Number of likes'),
+    date: z.string().nullable().describe('Upload date (ISO format)'),
+    channelName: z.string().nullable().describe('Channel name'),
+    channelUrl: z.string().nullable().describe('Channel URL'),
+    subscribers: z.number().nullable().describe('Number of channel subscribers'),
+    duration: z.string().nullable().describe('Video duration (HH:MM:SS)'),
+    description: z.string().nullable().describe('Video description'),
+    comments: z.number().nullable().describe('Number of comments'),
+    thumbnail: z.string().nullable().describe('Thumbnail URL'),
+});
+const YouTubeTranscriptSegmentSchema = z.object({
+    start: z.string().nullable().describe('Start time in seconds'),
+    duration: z.string().nullable().describe('Duration in seconds'),
+    text: z.string().nullable().describe('Transcript text'),
+});
+// Tool params schema
+const YouTubeToolParamsSchema = z.object({
+    operation: z
+        .enum(['searchVideos', 'getTranscript', 'scrapeChannel'])
+        .describe('Operation: searchVideos for search/URLs, getTranscript for transcripts, scrapeChannel for channel videos. Not all videos will have transcript available.'),
+    // Search operation fields
+    searchQueries: z
+        .array(z.string())
+        .optional()
+        .describe('Search queries for YouTube (for searchVideos). Examples: ["AI tutorials", "react hooks"]'),
+    videoUrls: z
+        .array(z.string().url())
+        .optional()
+        .describe('Direct YouTube URLs - videos, channels, playlists (for searchVideos or getTranscript)'),
+    // Channel scraping
+    channelUrl: z
+        .string()
+        .optional()
+        .describe('YouTube channel URL (for scrapeChannel operation)'),
+    // Transcript operation field
+    videoUrl: z
+        .string()
+        .url()
+        .optional()
+        .describe('Single video URL for transcript extraction (for getTranscript)'),
+    // Common fields
+    maxResults: z
+        .number()
+        .min(0)
+        .max(200)
+        .default(20)
+        .optional()
+        .describe('Max videos to fetch (default: 20)'),
+    includeShorts: z
+        .boolean()
+        .default(false)
+        .optional()
+        .describe('Include YouTube Shorts in results'),
+    credentials: z
+        .record(z.nativeEnum(CredentialType), z.string())
+        .optional()
+        .describe('Required credentials (auto-injected)'),
+});
+// Tool result schema
+const YouTubeToolResultSchema = z.object({
+    operation: z
+        .enum(['searchVideos', 'getTranscript', 'scrapeChannel'])
+        .describe('Operation performed'),
+    // Video data (for searchVideos and scrapeChannel)
+    videos: z
+        .array(YouTubeVideoSchema)
+        .optional()
+        .describe('Array of YouTube videos'),
+    // Transcript data (for getTranscript)
+    transcript: z
+        .array(YouTubeTranscriptSegmentSchema)
+        .optional()
+        .describe('Video transcript with timestamps'),
+    fullTranscriptText: z
+        .string()
+        .optional()
+        .describe('Complete transcript as plain text'),
+    // Common fields
+    totalResults: z.number().describe('Total number of results'),
+    success: z.boolean().describe('Whether operation succeeded'),
+    error: z.string().describe('Error message if failed'),
+});
+/**
+ * Simple, dev-friendly YouTube tool
+ *
+ * Provides easy access to YouTube data through three operations:
+ * - searchVideos: Search YouTube or scrape video/channel/playlist URLs
+ * - getTranscript: Extract video transcripts with timestamps
+ * - scrapeChannel: Get all videos from a channel
+ */
+export class YouTubeTool extends ToolBubble {
+    static bubbleName = 'youtube-tool';
+    static schema = YouTubeToolParamsSchema;
+    static resultSchema = YouTubeToolResultSchema;
+    static shortDescription = 'Search YouTube videos, extract transcripts, and scrape channel content with a simple interface';
+    static longDescription = `
+    Universal YouTube tool for video search, transcript extraction, and channel scraping.
+
+    **OPERATIONS:**
+    1. **searchVideos**: Search YouTube or scrape specific URLs
+       - Search by keywords
+       - Scrape videos, channels, playlists, or hashtags
+       - Get video metadata, views, likes, channel info
+
+    2. **getTranscript**: Extract video transcripts
+       - Get timestamped transcript segments
+       - Full transcript text available
+       - Perfect for content analysis or subtitles
+
+    3. **scrapeChannel**: Get all videos from a channel
+       - Fetch recent videos from any channel
+       - Sort by newest, popular, or oldest
+       - Get comprehensive video data
+
+    **WHEN TO USE THIS TOOL:**
+    - YouTube video search and discovery
+    - Content analysis and research
+    - Transcript extraction for AI/ML
+    - Channel monitoring and tracking
+    - Video metadata collection
+
+    **Simple Interface:**
+    Just specify the operation and provide search terms or URLs.
+    The tool handles service selection, data transformation, and error handling.
+
+    **What you get:**
+    - Clean, structured video data
+    - Timestamped transcripts
+    - Channel and engagement metrics
+    - Ready for analysis or processing
+  `;
+    static alias = 'yt';
+    static type = 'tool';
+    constructor(params = {
+        operation: 'searchVideos',
+        searchQueries: ['AI tutorials'],
+        maxResults: 20,
+    }, context) {
+        super(params, context);
+    }
+    async performAction() {
+        const credentials = this.params?.credentials;
+        if (!credentials || !credentials[CredentialType.APIFY_CRED]) {
+            return this.createErrorResult('YouTube scraping requires authentication. Please configure APIFY_CRED.');
+        }
+        try {
+            const { operation } = this.params;
+            // Validate required fields
+            if (operation === 'searchVideos') {
+                if ((!this.params.searchQueries ||
+                    this.params.searchQueries.length === 0) &&
+                    (!this.params.videoUrls || this.params.videoUrls.length === 0)) {
+                    return this.createErrorResult('searchVideos requires either searchQueries or videoUrls');
+                }
+            }
+            if (operation === 'getTranscript' && !this.params.videoUrl) {
+                return this.createErrorResult('getTranscript requires a videoUrl parameter');
+            }
+            if (operation === 'scrapeChannel' && !this.params.channelUrl) {
+                return this.createErrorResult('scrapeChannel requires a channelUrl parameter');
+            }
+            // Execute operation
+            switch (operation) {
+                case 'searchVideos':
+                    return await this.handleSearchVideos();
+                case 'getTranscript':
+                    return await this.handleGetTranscript();
+                case 'scrapeChannel':
+                    return await this.handleScrapeChannel();
+                default:
+                    throw new Error(`Unsupported operation: ${operation}`);
+            }
+        }
+        catch (error) {
+            return this.createErrorResult(error instanceof Error ? error.message : 'Unknown error occurred');
+        }
+    }
+    /**
+     * Create error result
+     */
+    createErrorResult(errorMessage) {
+        const { operation } = this.params;
+        return {
+            operation: operation || 'searchVideos',
+            videos: operation === 'searchVideos' || operation === 'scrapeChannel'
+                ? []
+                : undefined,
+            transcript: operation === 'getTranscript' ? [] : undefined,
+            fullTranscriptText: operation === 'getTranscript' ? '' : undefined,
+            totalResults: 0,
+            success: false,
+            error: errorMessage,
+        };
+    }
+    /**
+     * Handle searchVideos operation
+     */
+    async handleSearchVideos() {
+        const searchVideo = new ApifyBubble({
+            actorId: 'streamers/youtube-scraper',
+            input: {
+                searchQueries: this.params.searchQueries,
+                startUrls: this.params.videoUrls?.map((url) => ({ url })),
+                maxResults: this.params.maxResults || 20,
+                maxResultsShorts: this.params.includeShorts
+                    ? this.params.maxResults || 20
+                    : 0,
+            },
+            waitForFinish: true,
+            timeout: 180000,
+            credentials: this.params.credentials,
+        }, this.context);
+        const result = await searchVideo.action();
+        if (!result.data.success) {
+            return this.createErrorResult(result.data.error || 'Failed to search YouTube videos');
+        }
+        const items = result.data.items || [];
+        const videos = this.transformVideos(items);
+        return {
+            operation: 'searchVideos',
+            videos,
+            totalResults: videos.length,
+            success: true,
+            error: '',
+        };
+    }
+    /**
+     * Handle getTranscript operation
+     */
+    async handleGetTranscript() {
+        const transcriptScraper = new ApifyBubble({
+            actorId: 'pintostudio/youtube-transcript-scraper',
+            input: {
+                videoUrl: this.params.videoUrl,
+            },
+            waitForFinish: true,
+            timeout: 120000,
+            credentials: this.params.credentials,
+        }, this.context);
+        const result = await transcriptScraper.action();
+        if (!result.data.success) {
+            return this.createErrorResult(result.data.error || 'Failed to extract transcript');
+        }
+        const items = result.data.items || [];
+        const firstItem = items[0];
+        if (!firstItem || typeof firstItem !== 'object') {
+            return this.createErrorResult('No transcript data returned');
+        }
+        const anyItem = firstItem;
+        const transcriptData = anyItem.data || [];
+        const transcript = transcriptData.map((segment) => ({
+            start: segment.start || null,
+            duration: segment.dur || null,
+            text: segment.text || null,
+        }));
+        const fullText = transcript.map((seg) => seg.text || '').join(' ');
+        return {
+            operation: 'getTranscript',
+            transcript,
+            fullTranscriptText: fullText,
+            totalResults: transcript.length,
+            success: true,
+            error: '',
+        };
+    }
+    /**
+     * Handle scrapeChannel operation
+     */
+    async handleScrapeChannel() {
+        const channelScraper = new ApifyBubble({
+            actorId: 'streamers/youtube-scraper',
+            input: {
+                startUrls: [{ url: this.params.channelUrl }],
+                maxResults: this.params.maxResults || 20,
+                sortVideosBy: 'NEWEST',
+            },
+            waitForFinish: true,
+            timeout: 180000,
+            credentials: this.params.credentials,
+        }, this.context);
+        const result = await channelScraper.action();
+        if (!result.data.success) {
+            return this.createErrorResult(result.data.error || 'Failed to scrape channel');
+        }
+        const items = result.data.items || [];
+        const videos = this.transformVideos(items);
+        return {
+            operation: 'scrapeChannel',
+            videos,
+            totalResults: videos.length,
+            success: true,
+            error: '',
+        };
+    }
+    /**
+     * Transform Apify video data to unified format
+     */
+    transformVideos(items) {
+        return items.map((item) => {
+            if (typeof item !== 'object' || item === null) {
+                return this.createEmptyVideo();
+            }
+            const anyItem = item;
+            return {
+                title: anyItem.title || null,
+                id: anyItem.id || null,
+                url: anyItem.url || null,
+                viewCount: anyItem.viewCount || null,
+                likes: anyItem.likes || null,
+                date: anyItem.date || null,
+                channelName: anyItem.channelName || null,
+                channelUrl: anyItem.channelUrl || null,
+                subscribers: anyItem.numberOfSubscribers || null,
+                duration: anyItem.duration || null,
+                description: anyItem.text || null,
+                comments: anyItem.commentsCount || null,
+                thumbnail: anyItem.thumbnailUrl || null,
+            };
+        });
+    }
+    /**
+     * Create empty video object
+     */
+    createEmptyVideo() {
+        return {
+            title: null,
+            id: null,
+            url: null,
+            viewCount: null,
+            likes: null,
+            date: null,
+            channelName: null,
+            channelUrl: null,
+            subscribers: null,
+            duration: null,
+            description: null,
+            comments: null,
+            thumbnail: null,
+        };
+    }
+}
+//# sourceMappingURL=youtube-tool.js.map

package/dist/bubbles/tool-bubble/youtube-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"youtube-tool.js","sourceRoot":"","sources":["../../../src/bubbles/tool-bubble/youtube-tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,kCAAkC,CAAC;AAE9D,OAAO,EAAE,cAAc,EAAmB,MAAM,2BAA2B,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;AAG/D,gDAAgD;AAChD,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IAClC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;IACpD,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;IACtD,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC;IAChD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;IAC5D,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;IACxD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;IAChE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;IAC3D,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;IACzD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,+BAA+B,CAAC;IAC5E,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IACrE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;IAChE,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;IAC9D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC;CAC3D,CAAC,CAAC;AAEH,MAAM,8BAA8B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;IAC9D,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;IAC/D,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;CACxD,CAAC,CAAC;AAEH,qBAAqB;AACrB,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,SAAS,EAAE,CAAC;SACT,IAAI,CAAC,CAAC,cAAc,EAAE,eAAe,EAAE,eAAe,CAAC,CAAC;SACxD,QAAQ,CACP,0JAA0J,CAC3J;IAEH,0BAA0B;IAC1B,aAAa,EAAE,CAAC;SACb,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,QAAQ,EAAE;SACV,QAAQ,CACP,0FAA0F,CAC3F;IAEH,SAAS,EAAE,CAAC;SACT,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC;SACvB,QAAQ,EAAE;SACV,QAAQ,CACP,uFAAuF,CACxF;IAEH,mBAAmB;IACnB,UAAU,EAAE,CAAC;SACV,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,mDAAmD,CAAC;IAEhE,6BAA6B;IAC7B,QAAQ,EAAE,CAAC;SACR,MAAM,EAAE;SACR,GAAG,EAAE;SACL,QAAQ,EAAE;SACV,QAAQ,CAAC,gEAAgE,CAAC;IAE7E,gBAAgB;IAChB,UAAU,EAAE,CAAC;SACV,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,GAAG,CAAC;SACR,OAAO,CAAC,EAAE,CAAC;SACX,QAAQ,EAAE;SACV,QAAQ,CAAC,mCAAmC,CAAC;IAEhD,aAAa,EAAE,CAAC;SACb,OAAO,EAAE;SACT,OAAO,CAAC,KAAK,CAAC;SACd,QAAQ,EAAE;SACV,QAAQ,CAAC,mCAAmC,CAAC;IAEhD,WAAW,EAAE,CAAC;SACX,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC;SAChD,QAAQ,EAAE;SACV,QAAQ,CAAC,sCAAsC,CAAC;CACpD,CAAC,CAAC;AAEH,qBAAqB;AACrB,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,SAAS,EAAE,CAAC;SACT,IAAI,CAAC,CAAC,cAAc,EAAE,eAAe,EAAE,eAAe,CAAC,CAAC;SACxD,QAAQ,CAAC,qBAAqB,CAAC;IAElC,kDAAkD;IAClD,MAAM,EAAE,CAAC;SACN,KAAK,CAAC,kBAAkB,CAAC;SACzB,QAAQ,EAAE;SACV,QAAQ,CAAC,yBAAyB,CAAC;IAEtC,sCAAsC;IACtC,UAAU,EAAE,CAAC;SACV,KAAK,CAAC,8BAA8B,CAAC;SACrC,QAAQ,EAAE;SACV,QAAQ,CAAC,kCAAkC,CAAC;IAE/C,kBAAkB,EAAE,CAAC;SAClB,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,mCAAmC,CAAC;IAEhD,gBAAgB;IAChB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;IAC5D,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;IAC5D,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;CACtD,CAAC,CAAC;AAWH;;;;;;;GAOG;AACH,MAAM,OAAO,WAAY,SAAQ,UAGhC;IACC,MAAM,CAAU,UAAU,GAAe,cAAc,CAAC;IACxD,MAAM,CAAU,MAAM,GAAG,uBAAuB,CAAC;IACjD,MAAM,CAAU,YAAY,GAAG,uBAAuB,CAAC;IACvD,MAAM,CAAU,gBAAgB,GAC9B,gGAAgG,CAAC;IACnG,MAAM,CAAU,eAAe,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmCjC,CAAC;IACF,MAAM,CAAU,KAAK,GAAG,IAAI,CAAC;IAC7B,MAAM,CAAU,IAAI,GAAG,MAAM,CAAC;IAE9B,YACE,SAAiC;QAC/B,SAAS,EAAE,cAAc;QACzB,aAAa,EAAE,CAAC,cAAc,CAAC;QAC/B,UAAU,EAAE,EAAE;KACf,EACD,OAAuB;QAEvB,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC;QAC7C,IAAI,CAAC,WAAW,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;YAC5D,OAAO,IAAI,CAAC,iBAAiB,CAC3B,wEAAwE,CACzE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;YAElC,2BAA2B;YAC3B,IAAI,SAAS,KAAK,cAAc,EAAE,CAAC;gBACjC,IACE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa;oBACzB,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC,CAAC;oBACzC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,EAC9D,CAAC;oBACD,OAAO,IAAI,CAAC,iBAAiB,CAC3B,yDAAyD,CAC1D,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,SAAS,KAAK,eAAe,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAC3D,OAAO,IAAI,CAAC,iBAAiB,CAC3B,6CAA6C,CAC9C,CAAC;YACJ,CAAC;YAED,IAAI,SAAS,KAAK,eAAe,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;gBAC7D,OAAO,IAAI,CAAC,iBAAiB,CAC3B,+CAA+C,CAChD,CAAC;YACJ,CAAC;YAED,oBAAoB;YACpB,QAAQ,SAAS,EAAE,CAAC;gBAClB,KAAK,cAAc;oBACjB,OAAO,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACzC,KAAK,eAAe;oBAClB,OAAO,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC1C,KAAK,eAAe;oBAClB,OAAO,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC1C;oBACE,MAAM,IAAI,KAAK,CAAC,0BAA0B,SAAS,EAAE,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,iBAAiB,CAC3B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,CAClE,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,YAAoB;QAC5C,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAElC,OAAO;YACL,SAAS,EAAE,SAAS,IAAI,cAAc;YACtC,MAAM,EACJ,SAAS,KAAK,cAAc,IAAI,SAAS,KAAK,eAAe;gBAC3D,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,SAAS;YACf,UAAU,EAAE,SAAS,KAAK,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;YAC1D,kBAAkB,EAAE,SAAS,KAAK,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;YAClE,YAAY,EAAE,CAAC;YACf,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,YAAY;SACpB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB;QAC9B,MAAM,WAAW,GAAG,IAAI,WAAW,CACjC;YACE,OAAO,EAAE,2BAA2B;YACpC,KAAK,EAAE;gBACL,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;gBACxC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;gBACzD,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE;gBACxC,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;oBACzC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE;oBAC9B,CAAC,CAAC,CAAC;aACN;YACD,aAAa,EAAE,IAAI;YACnB,OAAO,EAAE,MAAM;YACf,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;SACrC,EACD,IAAI,CAAC,OAAO,CACb,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;QAE1C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,iBAAiB,CAC3B,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,iCAAiC,CACvD,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,OAAO;YACL,SAAS,EAAE,cAAc;YACzB,MAAM;YACN,YAAY,EAAE,MAAM,CAAC,MAAM;YAC3B,OAAO,EAAE,IAAI;YACb,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,mBAAmB;QAC/B,MAAM,iBAAiB,GACrB,IAAI,WAAW,CACb;YACE,OAAO,EAAE,wCAAwC;YACjD,KAAK,EAAE;gBACL,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAS;aAChC;YACD,aAAa,EAAE,IAAI;YACnB,OAAO,EAAE,MAAM;YACf,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;SACrC,EACD,IAAI,CAAC,OAAO,CACb,CAAC;QAEJ,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,CAAC;QAEhD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,iBAAiB,CAC3B,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,8BAA8B,CACpD,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAE3B,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;YAChD,OAAO,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,OAAO,GAAG,SAEf,CAAC;QAEF,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;QAC1C,MAAM,UAAU,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAClD,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,IAAI;YAC5B,QAAQ,EAAE,OAAO,CAAC,GAAG,IAAI,IAAI;YAC7B,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI;SAC3B,CAAC,CAAC,CAAC;QAEJ,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAEnE,OAAO;YACL,SAAS,EAAE,eAAe;YAC1B,UAAU;YACV,kBAAkB,EAAE,QAAQ;YAC5B,YAAY,EAAE,UAAU,CAAC,MAAM;YAC/B,OAAO,EAAE,IAAI;YACb,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,mBAAmB;QAC/B,MAAM,cAAc,GAAG,IAAI,WAAW,CACpC;YACE,OAAO,EAAE,2BAA2B;YACpC,KAAK,EAAE;gBACL,SAAS,EAAE,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,UAAW,EAAE,CAAC;gBAC7C,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE;gBACxC,YAAY,EAAE,QAAQ;aACvB;YACD,aAAa,EAAE,IAAI;YACnB,OAAO,EAAE,MAAM;YACf,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;SACrC,EACD,IAAI,CAAC,OAAO,CACb,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,MAAM,EAAE,CAAC;QAE7C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,iBAAiB,CAC3B,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,0BAA0B,CAChD,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,OAAO;YACL,SAAS,EAAE,eAAe;YAC1B,MAAM;YACN,YAAY,EAAE,MAAM,CAAC,MAAM;YAC3B,OAAO,EAAE,IAAI;YACb,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,eAAe,CACrB,KAAiD;QAEjD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC9C,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACjC,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,CAAC;YAErB,OAAO;gBACL,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,IAAI;gBAC5B,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,IAAI;gBACtB,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,IAAI;gBACxB,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;gBACpC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,IAAI;gBAC5B,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI;gBAC1B,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,IAAI;gBACxC,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,IAAI;gBACtC,WAAW,EAAE,OAAO,CAAC,mBAAmB,IAAI,IAAI;gBAChD,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,IAAI;gBAClC,WAAW,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI;gBACjC,QAAQ,EAAE,OAAO,CAAC,aAAa,IAAI,IAAI;gBACvC,SAAS,EAAE,OAAO,CAAC,YAAY,IAAI,IAAI;aACxC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,gBAAgB;QACtB,OAAO;YACL,KAAK,EAAE,IAAI;YACX,EAAE,EAAE,IAAI;YACR,GAAG,EAAE,IAAI;YACT,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,IAAI;YACV,WAAW,EAAE,IAAI;YACjB,UAAU,EAAE,IAAI;YAChB,WAAW,EAAE,IAAI;YACjB,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,IAAI;YACjB,QAAQ,EAAE,IAAI;YACd,SAAS,EAAE,IAAI;SAChB,CAAC;IACJ,CAAC"}

package/dist/bubbles/workflow-bubble/bubbleflow-generator.workflow.d.ts

@@ -9,7 +9,7 @@
 import { z } from 'zod';
 import { WorkflowBubble } from '../../types/workflow-bubble-class.js';
 import type { BubbleContext } from '../../types/bubble.js';
-import { CredentialType } from '@bubblelab/shared-schemas';
+import { CredentialType, type GenerationResult } from '@bubblelab/shared-schemas';
 import { type StreamingCallback } from '../service-bubble/ai-agent.js';
 /**
  * Parameters schema for the simple BubbleFlow generator
@@ -24,40 +24,11 @@ declare const BubbleFlowGeneratorParamsSchema: z.ZodObject<{
     prompt: string;
     credentials?: Partial<Record<CredentialType, string>> | undefined;
 }>;
-/**
- * Result schema
- */
-declare const BubbleFlowGeneratorResultSchema: z.ZodObject<{
-    generatedCode: z.ZodString;
-    isValid: z.ZodBoolean;
-    success: z.ZodBoolean;
-    error: z.ZodString;
-    toolCalls: z.ZodArray<z.ZodAny, "many">;
-    summary: z.ZodDefault<z.ZodString>;
-    inputsSchema: z.ZodDefault<z.ZodString>;
-}, "strip", z.ZodTypeAny, {
-    success: boolean;
-    error: string;
-    toolCalls: any[];
-    summary: string;
-    generatedCode: string;
-    isValid: boolean;
-    inputsSchema: string;
-}, {
-    success: boolean;
-    error: string;
-    toolCalls: any[];
-    generatedCode: string;
-    isValid: boolean;
-    summary?: string | undefined;
-    inputsSchema?: string | undefined;
-}>;
 type BubbleFlowGeneratorParams = z.output<typeof BubbleFlowGeneratorParamsSchema>;
-type BubbleFlowGeneratorResult = z.output<typeof BubbleFlowGeneratorResultSchema>;
 /**
  * Simple BubbleFlow Generator using AI agent with tools
  */
-export declare class BubbleFlowGeneratorWorkflow extends WorkflowBubble<BubbleFlowGeneratorParams, BubbleFlowGeneratorResult> {
+export declare class BubbleFlowGeneratorWorkflow extends WorkflowBubble<BubbleFlowGeneratorParams, GenerationResult> {
     static readonly type: "workflow";
     static readonly bubbleName = "bubbleflow-generator";
     static readonly schema: z.ZodObject<{
@@ -75,25 +46,65 @@ export declare class BubbleFlowGeneratorWorkflow extends WorkflowBubble<BubbleFl
         isValid: z.ZodBoolean;
         success: z.ZodBoolean;
         error: z.ZodString;
-        toolCalls: z.ZodArray<z.ZodAny, "many">;
+        flowId: z.ZodOptional<z.ZodNumber>;
+        toolCalls: z.ZodArray<z.ZodUnknown, "many">;
         summary: z.ZodDefault<z.ZodString>;
         inputsSchema: z.ZodDefault<z.ZodString>;
+        bubblesUsed: z.ZodDefault<z.ZodArray<z.ZodString, "many">>;
+        tokenUsage: z.ZodOptional<z.ZodObject<{
+            inputTokens: z.ZodNumber;
+            outputTokens: z.ZodNumber;
+            totalTokens: z.ZodNumber;
+            modelName: z.ZodOptional<z.ZodString>;
+        }, "strip", z.ZodTypeAny, {
+            inputTokens: number;
+            outputTokens: number;
+            totalTokens: number;
+            modelName?: string | undefined;
+        }, {
+            inputTokens: number;
+            outputTokens: number;
+            totalTokens: number;
+            modelName?: string | undefined;
+        }>>;
+        bubbleCount: z.ZodOptional<z.ZodNumber>;
+        codeLength: z.ZodOptional<z.ZodNumber>;
     }, "strip", z.ZodTypeAny, {
         success: boolean;
         error: string;
-        toolCalls: any[];
         summary: string;
         generatedCode: string;
         isValid: boolean;
+        toolCalls: unknown[];
         inputsSchema: string;
+        bubblesUsed: string[];
+        tokenUsage?: {
+            inputTokens: number;
+            outputTokens: number;
+            totalTokens: number;
+            modelName?: string | undefined;
+        } | undefined;
+        flowId?: number | undefined;
+        bubbleCount?: number | undefined;
+        codeLength?: number | undefined;
     }, {
         success: boolean;
         error: string;
-        toolCalls: any[];
         generatedCode: string;
         isValid: boolean;
+        toolCalls: unknown[];
+        tokenUsage?: {
+            inputTokens: number;
+            outputTokens: number;
+            totalTokens: number;
+            modelName?: string | undefined;
+        } | undefined;
         summary?: string | undefined;
+        flowId?: number | undefined;
+        bubbleCount?: number | undefined;
+        codeLength?: number | undefined;
         inputsSchema?: string | undefined;
+        bubblesUsed?: string[] | undefined;
     }>;
     static readonly shortDescription = "Generate BubbleFlow code from natural language";
     static readonly longDescription = "\n    Simple BubbleFlow generator that uses AI with validation tools.\n    \n    Just provide a natural language prompt describing what you want your BubbleFlow to do,\n    and it will generate complete TypeScript code with proper validation.\n    \n    Example prompts:\n    - \"Create a flow that queries my database and sends results to Slack\"\n    - \"Build a workflow that processes user data with AI and stores it\"\n    - \"Make a flow that analyzes text and generates a summary\"\n  ";
@@ -104,11 +115,11 @@ export declare class BubbleFlowGeneratorWorkflow extends WorkflowBubble<BubbleFl
     private runSummarizeAgent;
     private createSystemPrompt;
     private createStreamingSystemPrompt;
-    protected performAction(context?: BubbleContext): Promise<BubbleFlowGeneratorResult>;
+    protected performAction(context?: BubbleContext): Promise<GenerationResult>;
     /**
      * Execute the workflow with streaming support for real-time code generation feedback
      */
-    actionWithStreaming(streamingCallback: StreamingCallback, context?: BubbleContext): Promise<BubbleFlowGeneratorResult>;
+    actionWithStreaming(streamingCallback: StreamingCallback, context?: BubbleContext): Promise<GenerationResult>;
 }
 export {};
 //# sourceMappingURL=bubbleflow-generator.workflow.d.ts.map

package/dist/bubbles/workflow-bubble/bubbleflow-generator.workflow.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"bubbleflow-generator.workflow.d.ts","sourceRoot":"","sources":["../../../src/bubbles/workflow-bubble/bubbleflow-generator.workflow.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AACtE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAEL,KAAK,iBAAiB,EACvB,MAAM,+BAA+B,CAAC;AAiCvC;;GAEG;AACH,QAAA,MAAM,+BAA+B;;;;;;;;;EAUnC,CAAC;AAEH;;GAEG;AACH,QAAA,MAAM,+BAA+B;;;;;;;;;;;;;;;;;;;;;;;;EAgBnC,CAAC;AAEH,KAAK,yBAAyB,GAAG,CAAC,CAAC,MAAM,CACvC,OAAO,+BAA+B,CACvC,CAAC;AACF,KAAK,yBAAyB,GAAG,CAAC,CAAC,MAAM,CACvC,OAAO,+BAA+B,CACvC,CAAC;AA0DF;;GAEG;AACH,qBAAa,2BAA4B,SAAQ,cAAc,CAC7D,yBAAyB,EACzB,yBAAyB,CAC1B;IACC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAG,UAAU,CAAU;IAC3C,MAAM,CAAC,QAAQ,CAAC,UAAU,0BAA0B;IACpD,MAAM,CAAC,QAAQ,CAAC,MAAM;;;;;;;;;OAAmC;IACzD,MAAM,CAAC,QAAQ,CAAC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;OAAmC;IAC/D,MAAM,CAAC,QAAQ,CAAC,gBAAgB,oDACmB;IACnD,MAAM,CAAC,QAAQ,CAAC,eAAe,kfAU7B;IACF,MAAM,CAAC,QAAQ,CAAC,KAAK,mBAAmB;IAExC,OAAO,CAAC,aAAa,CAAgB;gBAGnC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,+BAA+B,CAAC,EACvD,OAAO,CAAC,EAAE,aAAa;YAMX,kBAAkB;YAmHlB,iBAAiB;IAkF/B,OAAO,CAAC,kBAAkB;IAmB1B,OAAO,CAAC,2BAA2B;cAmBnB,aAAa,CAC3B,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,yBAAyB,CAAC;IAiPrC;;OAEG;IACU,mBAAmB,CAC9B,iBAAiB,EAAE,iBAAiB,EACpC,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,yBAAyB,CAAC;CAsXtC"}
+{"version":3,"file":"bubbleflow-generator.workflow.d.ts","sourceRoot":"","sources":["../../../src/bubbles/workflow-bubble/bubbleflow-generator.workflow.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AACtE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EACL,cAAc,EAEd,KAAK,gBAAgB,EAItB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAEL,KAAK,iBAAiB,EACvB,MAAM,+BAA+B,CAAC;AA4CvC;;GAEG;AACH,QAAA,MAAM,+BAA+B;;;;;;;;;EAUnC,CAAC;AAEH,KAAK,yBAAyB,GAAG,CAAC,CAAC,MAAM,CACvC,OAAO,+BAA+B,CACvC,CAAC;AA2BF;;GAEG;AACH,qBAAa,2BAA4B,SAAQ,cAAc,CAC7D,yBAAyB,EACzB,gBAAgB,CACjB;IACC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAG,UAAU,CAAU;IAC3C,MAAM,CAAC,QAAQ,CAAC,UAAU,0BAA0B;IACpD,MAAM,CAAC,QAAQ,CAAC,MAAM;;;;;;;;;OAAmC;IACzD,MAAM,CAAC,QAAQ,CAAC,YAAY;;;;;;;;;;;;;;;;;;;qBAoIS,CAAC;;;;;qBAIxB,CAAC;;;;;;;;;;;;;;;;;qBAaH,CAAT;;;;;;;;;;;;;;;qBAsBkB,CAAC;;;;;;;;OA3KgC;IACtD,MAAM,CAAC,QAAQ,CAAC,gBAAgB,oDACmB;IACnD,MAAM,CAAC,QAAQ,CAAC,eAAe,kfAU7B;IACF,MAAM,CAAC,QAAQ,CAAC,KAAK,mBAAmB;IAExC,OAAO,CAAC,aAAa,CAAgB;gBAGnC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,+BAA+B,CAAC,EACvD,OAAO,CAAC,EAAE,aAAa;YAMX,kBAAkB;YA+IlB,iBAAiB;IAkF/B,OAAO,CAAC,kBAAkB;IAqB1B,OAAO,CAAC,2BAA2B;cAmBnB,aAAa,CAC3B,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,gBAAgB,CAAC;IAwR5B;;OAEG;IACU,mBAAmB,CAC9B,iBAAiB,EAAE,iBAAiB,EACpC,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,gBAAgB,CAAC;CA2a7B"}