Sora2API 快速开始

{
  "prompt": "一只可爱的小猫在花园里玩耍，阳光明媚的春天",
  "aspectRatio": "landscape",
  "quality": "hd"
}

class Sora2API {
  constructor(apiKey) {
    this.apiKey = apiKey;
    this.baseUrl = 'https://api.sora2api.ai/api/v1/sora2';
  }
  
  async generateVideo(options) {
    const response = await fetch(`${this.baseUrl}/generate`, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${this.apiKey}`,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify(options)
    });
    
    const result = await response.json();
    if (!response.ok || result.code !== 200) {
      throw new Error(`生成失败: ${result.msg || '未知错误'}`);
    }
    
    return result.data.taskId;
  }
  
  async waitForCompletion(taskId, maxWaitTime = 600000) { // 最长等待10分钟
    const startTime = Date.now();
    
    while (Date.now() - startTime < maxWaitTime) {
      const status = await this.getTaskStatus(taskId);
      
      switch (status.successFlag) {
        case 0:
          console.log('任务生成中，继续等待...');
          break;
          
        case 1:
          console.log('生成成功完成！');
          return status.response;
          
        case 2:
          const taskError = status.errorMessage || '任务创建失败';
          console.error('任务创建失败:', taskError);
          if (status.errorCode) {
            console.error('错误代码:', status.errorCode);
          }
          throw new Error(taskError);
          
        case 3:
          const generateError = status.errorMessage || '任务生成失败';
          console.error('生成失败:', generateError);
          if (status.errorCode) {
            console.error('错误代码:', status.errorCode);
          }
          throw new Error(generateError);
          
        default:
          console.log(`未知状态: ${status.successFlag}`);
          if (status.errorMessage) {
            console.error('错误信息:', status.errorMessage);
          }
          break;
      }
      
      // 等待30秒后再次检查
      await new Promise(resolve => setTimeout(resolve, 30000));
    }
    
    throw new Error('生成超时');
  }
  
  async getTaskStatus(taskId) {
    const response = await fetch(`${this.baseUrl}/record-info?taskId=${taskId}`, {
      method: 'GET',
      headers: {
        'Authorization': `Bearer ${this.apiKey}`
      }
    });
    
    const result = await response.json();
    if (!response.ok || result.code !== 200) {
      throw new Error(`查询状态失败: ${result.msg || '未知错误'}`);
    }
    
    return result.data;
  }
}

// 使用示例
async function main() {
  const api = new Sora2API('YOUR_API_KEY');
  
  try {
    // 文本生成视频
    console.log('开始视频生成...');
    const taskId = await api.generateVideo({
      prompt: '一只可爱的小猫在花园里玩耍，阳光明媚的春天，镜头缓缓推进',
      aspectRatio: 'landscape',
      quality: 'hd',
      watermark: '我的视频'
    });
    
    // 等待完成
    console.log(`任务ID: ${taskId}。等待完成...`);
    const result = await api.waitForCompletion(taskId);
    
    console.log('视频生成成功！');
    console.log('视频URL:', result.imageUrl);
    
  } catch (error) {
    console.error('错误:', error.message);
  }
}

main();