example-usage.ts

/**
 * Example usage of @crewdle/ai-firebase-functions
 * 
 * This example shows how to use the package in your Firebase Functions project
 */

import { onCall } from 'firebase-functions/v2/https';

// Example index.ts for your Firebase Functions project
import {
  runWorkflow,
  streamWorkflow,
  // Import types for better TypeScript support
  type WorkflowMessage,
  type WorkflowFile,
  type WorkflowRequest
} from '@crewdle/ai-firebase-functions';

// Export all functions for deployment
export {
  runWorkflow,
  streamWorkflow
};

// Example: Custom wrapper function for batch processing
export const processBatchWorkflow = onCall({
  region: "northamerica-northeast1",
  timeoutSeconds: 540, // 9 minutes
}, async (request) => {
  const { workflowId, batchMessages } = request.data;

  try {
    const results = [];
    
    // Process multiple message sets with the same workflow
    for (const messageSet of batchMessages) {
      const result = await runWorkflow({
        workflowId,
        messages: messageSet.messages,
        files: messageSet.files || []
      });
      
      results.push({
        id: messageSet.id,
        result: result.choices[0].message.content
      });
    }

    return {
      results,
      totalProcessed: results.length,
      success: true
    };
  } catch (error) {
    console.error('Batch processing failed:', error);
    throw new Error(`Failed to process batch: ${error.message}`);
  }
});

// Example: Streaming workflow with custom handling
export const streamWorkflowWithLogging = onCall({
  region: "northamerica-northeast1",
  timeoutSeconds: 300,
}, async (request, response) => {
  const { workflowId, messages, files, userId } = request.data;

  try {
    console.log(`Starting stream workflow for user: ${userId}`);
    
    const streamResult = await streamWorkflow({
      workflowId,
      messages,
      files: files || []
    });

    console.log(`Stream workflow completed for user: ${userId}`);
    
    return {
      streamOutput: streamResult,
      userId,
      timestamp: new Date().toISOString(),
      success: true
    };
  } catch (error) {
    console.error(`Stream workflow failed for user ${userId}:`, error);
    throw new Error(`Failed to stream workflow: ${error.message}`);
  }
});