AIVideoGenerationService.php 79 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898
  1. <?php
  2. namespace App\Services\AIGeneration;
  3. use App\Libs\Utils;
  4. use App\Models\MpGenerateVideoTask;
  5. use App\Models\AIElement;
  6. use App\Services\VolcEngineService;
  7. use GuzzleHttp\Client;
  8. use Illuminate\Support\Facades\DB;
  9. class AIVideoGenerationService
  10. {
  11. private $volcEngineService;
  12. public function __construct(VolcEngineService $volcEngineService)
  13. {
  14. $this->volcEngineService = $volcEngineService;
  15. }
  16. /**
  17. * 异步创建视频生成任务
  18. *
  19. * @param array $params
  20. * @return MpGenerateVideoTask
  21. */
  22. public function createJimengTask(array $params): MpGenerateVideoTask
  23. {
  24. // 生成唯一的任务ID
  25. $taskId = 'vid_gen_' . time() . '_' . uniqid();
  26. // 创建任务记录
  27. $task = MpGenerateVideoTask::create([
  28. 'task_id' => $taskId,
  29. 'prompt' => $params['prompt'] ?? '',
  30. 'first_frame_url' => $params['first_frame_url'] ?? null,
  31. 'tail_frame_url' => $params['tail_frame_url'] ?? null,
  32. 'seed' => $params['seed'] ?? -1,
  33. 'frames' => $params['video_duration'] ? ($params['video_duration'] * 24 + 1) : 121,
  34. 'video_resolution' => $params['video_resolution'] ?? '720P',
  35. 'status' => MpGenerateVideoTask::STATUS_PENDING,
  36. 'api_type' => 'jimeng',
  37. ]);
  38. // 发送异步请求到即梦AI API创建任务
  39. $this->submitTaskToApi($task);
  40. return $task;
  41. }
  42. /**
  43. * 提交任务到即梦AI API
  44. *
  45. * @param MpGenerateVideoTask $task
  46. * @return void
  47. */
  48. private function submitTaskToApi(MpGenerateVideoTask $task): void
  49. {
  50. try {
  51. // 验证环境变量配置
  52. $accessKey = env('VOLC_AI_AK');
  53. $secretKey = env('VOLC_AI_SK');
  54. if (empty($accessKey) || empty($secretKey)) {
  55. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  56. 'error_message' => '火山引擎访问密钥未配置,请检查环境变量AK和SK'
  57. ]);
  58. return;
  59. }
  60. // 根据任务类型选择对应的req_key
  61. if ($task->first_frame_url && $task->tail_frame_url) {
  62. $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_tail_v30' : 'jimeng_i2v_first_tail_v30_1080';
  63. }else if($task->first_frame_url){
  64. $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_v30' : 'jimeng_i2v_first_v30_1080';
  65. }else{
  66. $req_key = $task->video_resolution == '720P' ? 'jimeng_t2v_v30' : 'jimeng_t2v_v30_1080p';
  67. }
  68. // 构建即梦AI 3.0 视频生成API请求参数
  69. $apiParams = [
  70. 'req_key' => $req_key,
  71. 'prompt' => $task->prompt,
  72. 'seed' => (int)$task->seed,
  73. 'frames' => $task->frames,
  74. ];
  75. // 添加图片URL参数
  76. if ($task->first_frame_url) {
  77. $apiParams['image_urls'] = [
  78. $task->first_frame_url
  79. ];
  80. if ($task->tail_frame_url) $apiParams['image_urls'][] = $task->tail_frame_url;
  81. }
  82. dLog('generate')->info('图生视频参数: ', $apiParams);
  83. logDB('generate', 'info', '即梦AI视频生成任务提交', ['id' => $task->id, 'params' => $apiParams]);
  84. // 调用即梦AI 3.0 视频生成API提交任务
  85. $response = $this->volcEngineService->request(
  86. 'POST',
  87. 'visual.volcengineapi.com', // 即梦AI API域名
  88. '/', // 使用根路径
  89. [], // Query参数
  90. json_encode($apiParams),
  91. $accessKey,
  92. $secretKey,
  93. 'cv', // 服务标识
  94. 'cn-north-1', // 区域
  95. 'CVSync2AsyncSubmitTask',
  96. '2022-08-31',
  97. 'application/json'
  98. );
  99. $responseData = json_decode($response['body'], true);
  100. dLog('generate')->info('即梦AI 3.0 视频生成API提交任务响应: ', $responseData);
  101. logDB('generate', 'info', '即梦AI视频生成API响应', ['id' => $task->id, 'response' => $responseData]);
  102. if ($responseData['code'] !== 10000) {
  103. // API返回错误
  104. logDB('generate', 'error', '即梦AI视频生成任务提交失败', ['id' => $task->id, 'error' => $responseData['message'] ?? 'API Error']);
  105. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  106. 'error_message' => $responseData['message'] ?? 'API Error',
  107. 'extra_params' => $responseData['data'] ?? []
  108. ]);
  109. } else {
  110. // 任务提交成功,更新状态为处理中
  111. $updateData = [];
  112. // 存储API返回的任务ID
  113. if (isset($responseData['data']['task_id'])) {
  114. $updateData['task_id'] = $responseData['data']['task_id'];
  115. $updateData['extra_params'] = $responseData['data'] ?? [];
  116. $updateData['extra_params']['req_key'] = $req_key;
  117. $updateData['extra_params']['request_id'] = isset($responseData['request_id']) ? $responseData['request_id'] : '';
  118. }
  119. $task->updateStatus(MpGenerateVideoTask::STATUS_PROCESSING, $updateData);
  120. }
  121. } catch (\Exception $e) {
  122. // 记录错误
  123. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  124. 'error_message' => 'API请求失败: ' . $e->getMessage()
  125. ]);
  126. Utils::throwError('1001:'.$e->getMessage());
  127. }
  128. }
  129. /**
  130. * 查询任务状态(即梦AI)
  131. *
  132. * @param MpGenerateVideoTask $task
  133. * @return array
  134. */
  135. public function queryJimengTaskStatus(MpGenerateVideoTask $task): array
  136. {
  137. try {
  138. // 获取API任务ID
  139. $apiTaskId = $task->task_id ?? null;
  140. if (!$apiTaskId) {
  141. return [
  142. 'status' => 'failed',
  143. 'error_message' => 'API任务ID不存在',
  144. 'result_url' => null
  145. ];
  146. }
  147. // 根据任务类型选择对应的req_key
  148. if ($task->first_frame_url && $task->tail_frame_url) {
  149. $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_tail_v30' : 'jimeng_i2v_first_tail_v30_1080';
  150. }else if($task->first_frame_url){
  151. $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_v30' : 'jimeng_i2v_first_v30_1080';
  152. }else{
  153. $req_key = $task->video_resolution == '720P' ? 'jimeng_t2v_v30' : 'jimeng_t2v_v30_1080p';
  154. }
  155. // 构建查询参数
  156. $apiParams = [
  157. 'req_key' => $req_key,
  158. 'task_id' => $apiTaskId,
  159. // 'req_json' => '{"return_url":true}' // 返回URL格式
  160. ];
  161. // 调用即梦AI 3.0 视频生成API查询任务状态
  162. $response = $this->volcEngineService->request(
  163. 'POST',
  164. 'visual.volcengineapi.com',
  165. '/', // 使用根路径
  166. [], // Query参数
  167. json_encode($apiParams),
  168. env('VOLC_AI_AK'),
  169. env('VOLC_AI_SK'),
  170. 'cv',
  171. 'cn-north-1',
  172. 'CVSync2AsyncGetResult',
  173. '2022-08-31',
  174. 'application/json'
  175. );
  176. $responseData = json_decode($response['body'], true);
  177. dLog('generate')->info('即梦AI 3.0 视频生成API查询任务状态响应: ', $responseData);
  178. logDB('generate', 'info', '即梦AI视频任务状态查询', ['id' => $task->id, 'response' => $responseData]);
  179. if ($responseData['code'] !== 10000) {
  180. // API返回错误
  181. logDB('generate', 'error', '即梦AI视频任务状态查询失败', ['id' => $task->id, 'error' => $responseData['message'] ?? 'API Error']);
  182. return [
  183. 'status' => 'failed',
  184. 'error_message' => $responseData['message'] ?? 'API Error',
  185. 'result_url' => null,
  186. 'result_json' => $responseData ?? []
  187. ];
  188. }
  189. // 解析API响应
  190. $result = $responseData['data'] ?? [];
  191. $taskStatus = $result['status'] ?? 'failed';
  192. $returnData = [
  193. 'status' => $taskStatus,
  194. 'error_message' => null,
  195. 'result_url' => null,
  196. 'result_json' => $responseData ?? []
  197. ];
  198. // 如果任务成功,获取结果URL
  199. if ($taskStatus === 'done' && isset($result['video_url'])) {
  200. if (isset($result['video_url'])) {
  201. $video_name = 'ai_generation_' . time() . '_' . uniqid();
  202. $video_ext = getVideoExtFromUrl($result['video_url']);
  203. $video_name = $video_name . $video_ext;
  204. // 将视频另存到tos
  205. $url = uploadStreamByTos('video', file_get_contents($result['video_url']), $video_name);
  206. $returnData['result_url'] = $url;
  207. }
  208. // 计算视频时长
  209. $video_duration = 0;
  210. if (isset($result['duration'])) {
  211. $video_duration = (int)$result['duration'];
  212. } elseif (isset($result['frames']) && isset($result['framespersecond'])) {
  213. $frames = (int)$result['frames'];
  214. $fps = (int)$result['framespersecond'];
  215. if ($fps > 0) {
  216. $video_duration = (int)floor($frames / $fps);
  217. }
  218. }
  219. // 如果没有计算出时长,使用任务表中的设定值
  220. if ($video_duration <= 0 && !empty($task->extra_params)) {
  221. $extraParams = is_array($task->extra_params) ? $task->extra_params : json_decode($task->extra_params, true);
  222. if (isset($extraParams['duration'])) {
  223. $video_duration = (int)$extraParams['duration'];
  224. }
  225. }
  226. // 只有当video_duration > 0时才添加到返回数据中
  227. if ($video_duration > 0) {
  228. $returnData['video_duration'] = $video_duration;
  229. $returnData['video_time_point_start'] = 0;
  230. $returnData['video_time_point_end'] = $video_duration;
  231. }
  232. $returnData['status'] = 'success';
  233. } elseif (in_array($taskStatus, ['not_found', 'expired'])) {
  234. $returnData['status'] = 'failed';
  235. $returnData['error_message'] = '任务未找到或已过期';
  236. } elseif ($taskStatus === 'failed') {
  237. $returnData['status'] = 'failed';
  238. $returnData['error_message'] = '任务执行失败';
  239. }
  240. // in_queue 或 generating状态保持原status
  241. return $returnData;
  242. } catch (\Exception $e) {
  243. return [
  244. 'status' => 'failed',
  245. 'error_message' => $e->getMessage(),
  246. 'result_url' => null
  247. ];
  248. }
  249. }
  250. /**
  251. * 更新所有待处理任务的状态
  252. *
  253. * @return void
  254. */
  255. public function updatePendingTasks(): void
  256. {
  257. // 获取所有处理中的任务
  258. $tasks = MpGenerateVideoTask::where('status', MpGenerateVideoTask::STATUS_PROCESSING)
  259. ->get();
  260. foreach ($tasks as $task) {
  261. $api_type = getProp($task, 'api_type');
  262. switch($api_type) {
  263. case 'seedance':
  264. $this->updateSeedanceTask($task);
  265. break;
  266. case 'jimeng':
  267. $this->updateJimengTask($task);
  268. break;
  269. case 'keling':
  270. $this->updateKelingOmniTask($task);
  271. break;
  272. default:
  273. break;
  274. }
  275. }
  276. }
  277. // 更新即梦AI视频任务状态
  278. private function updateJimengTask($task) {
  279. $statusInfo = $this->queryJimengTaskStatus($task);
  280. if (!isset($statusInfo['status'])) return;
  281. if ($statusInfo['status'] === 'success') {
  282. logDB('generate', 'info', '即梦AI视频生成任务成功', ['id' => $task->id, 'result_url' => $statusInfo['result_url']]);
  283. $task->updateStatus(MpGenerateVideoTask::STATUS_SUCCESS, $statusInfo);
  284. // 同步调整分镜视频状态和结果
  285. $segment_id = getProp($task, 'alias_segment_id');
  286. if ($segment_id && isset($statusInfo['result_url'])) {
  287. try {
  288. DB::beginTransaction();
  289. $now = date('Y-m-d H:i:s');
  290. // 构建更新数据
  291. $segmentUpdateData = [
  292. 'video_url' => $statusInfo['result_url'],
  293. 'video_task_status' => '已完成',
  294. 'current_type' => 2,
  295. 'last_frame_url' => $statusInfo['last_frame_url'] ?? '',
  296. 'updated_at' => $now
  297. ];
  298. // 只有当video_duration存在且大于0时才更新
  299. if (isset($statusInfo['video_duration']) && $statusInfo['video_duration'] > 0) {
  300. $segmentUpdateData['video_duration'] = $statusInfo['video_duration'];
  301. $segmentUpdateData['video_time_point_start'] = 0;
  302. $segmentUpdateData['video_time_point_end'] = $statusInfo['video_duration'];
  303. }
  304. // 更新分镜表
  305. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update($segmentUpdateData);
  306. DB::commit();
  307. } catch (\Exception $e) {
  308. DB::rollBack();
  309. dLog('generate')->error('即梦视频任务处理失败', [
  310. 'segment_id' => $segment_id,
  311. 'task_id' => $task->id,
  312. 'error' => $e->getMessage()
  313. ]);
  314. }
  315. }
  316. } elseif ($statusInfo['status'] === 'failed') {
  317. logDB('generate', 'error', '即梦AI视频生成任务失败', ['id' => $task->id, 'error' => $statusInfo['error_message']]);
  318. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, $statusInfo);
  319. // 同步更新分镜表状态为失败
  320. $segment_id = getProp($task, 'alias_segment_id');
  321. if ($segment_id) {
  322. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  323. 'video_task_status' => '失败',
  324. 'updated_at' => date('Y-m-d H:i:s')
  325. ]);
  326. }
  327. }
  328. // 如果仍然是处理中状态,不做任何操作
  329. // 处理:如果任务处理超过12小时,标记为失败
  330. $processingTime = now()->diffInHours($task->created_at);
  331. if ($processingTime > 12) {
  332. logDB('generate', 'warning', '即梦AI视频生成任务超时', ['id' => $task->id, 'processing_hours' => $processingTime]);
  333. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  334. 'error_message' => '任务处理超时(超过12小时)'
  335. ]);
  336. // 同步更新分镜表状态为失败
  337. $segment_id = getProp($task, 'alias_segment_id');
  338. if ($segment_id) {
  339. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  340. 'video_task_status' => '失败',
  341. 'updated_at' => date('Y-m-d H:i:s')
  342. ]);
  343. }
  344. }
  345. }
  346. public function getTaskList($data)
  347. {
  348. $task_id = getProp($data, 'task_id');
  349. $status = getProp($data, 'status');
  350. $query = MpGenerateVideoTask::select('*');
  351. if ($task_id) {
  352. $query->where('id', $task_id);
  353. }
  354. if ($status) {
  355. $query->where('status', $status);
  356. }
  357. return $query->orderBy('created_at', 'desc')->paginate();
  358. }
  359. /**
  360. * 创建 Seedance 1.5 Pro 视频生成任务
  361. *
  362. * @param array $params
  363. * @return MpGenerateVideoTask
  364. */
  365. public function createSeedanceTask(array $params): MpGenerateVideoTask
  366. {
  367. // 生成唯一的任务ID
  368. $taskId = 'seedance_' . time() . '_' . uniqid();
  369. // 创建任务记录,复用现有表结构
  370. $task = MpGenerateVideoTask::create([
  371. 'task_id' => $taskId,
  372. 'alias_segment_id' => $params['alias_segment_id'] ?? '',
  373. 'prompt' => $params['prompt'] ?? '',
  374. 'first_frame_url' => $params['first_frame_url'] ?? '',
  375. 'tail_frame_url' => $params['tail_frame_url'] ?? '',
  376. 'seed' => $params['seed'] ?? -1,
  377. 'frames' => $params['video_duration'] ? ($params['video_duration'] * 24 + 1) : null,
  378. 'video_resolution' => $params['video_resolution'] ?? '720P',
  379. 'status' => MpGenerateVideoTask::STATUS_PENDING,
  380. 'api_type' => 'seedance', // 标识为 Seedance 1.5 Pro API
  381. 'extra_params' => [
  382. 'model' => $params['model'] ?? 'doubao-seedance-1-5-pro-251215',
  383. 'content' => $params['content'],
  384. 'callback_url' => $params['callback_url'] ?? '',
  385. 'return_last_frame' => $params['return_last_frame'] ?? true,
  386. 'service_tier' => $params['service_tier'] ?? 'default',
  387. 'execution_expires_after' => $params['execution_expires_after'] ?? 172800,
  388. 'generate_audio' => $params['generate_audio'] ?? false,
  389. 'draft' => $params['draft'] ?? false,
  390. 'ratio' => $params['ratio'] ?? '9:16',
  391. 'duration' => $params['video_duration'] ?? 5,
  392. 'camera_fixed' => $params['camera_fixed'] ?? false,
  393. 'watermark' => $params['watermark'] ?? false,
  394. ]
  395. ]);
  396. // 提交任务到火山引擎 API
  397. $this->submitSeedanceTaskToApi($task);
  398. return $task;
  399. }
  400. /**
  401. * 提交 Seedance 任务到火山引擎 API
  402. *
  403. * @param MpGenerateVideoTask $task
  404. * @return void
  405. */
  406. private function submitSeedanceTaskToApi(MpGenerateVideoTask $task): void
  407. {
  408. $client = new Client(['verify'=>false,'timeout'=>120]);
  409. try {
  410. // 验证环境变量配置
  411. $apiKey = env('VOLC_AI_API_KEY');
  412. if (empty($apiKey)) {
  413. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  414. 'error_message' => '火山引擎访问密钥未配置,请检查环境变量API Key'
  415. ]);
  416. return;
  417. }
  418. $extraParams = $task->extra_params;
  419. // 构建火山引擎 Seedance API 请求参数
  420. $apiParams = [
  421. 'model' => isset(\App\Consts\BaseConst::MODELS_MAP[$extraParams['model']]) ? \App\Consts\BaseConst::MODELS_MAP[$extraParams['model']] : $extraParams['model'],
  422. 'content' => $extraParams['content'],
  423. ];
  424. // 添加可选参数
  425. if (!empty($extraParams['callback_url'])) {
  426. $apiParams['callback_url'] = $extraParams['callback_url'];
  427. }
  428. if (isset($extraParams['return_last_frame'])) {
  429. $apiParams['return_last_frame'] = $extraParams['return_last_frame'];
  430. }
  431. if (!empty($extraParams['service_tier']) && !in_array($extraParams['model'], ['doubao-seedance-2-0-260128', 'doubao-seedance-2-0-fast-260128'])) {
  432. $apiParams['service_tier'] = $extraParams['service_tier'];
  433. }
  434. if (!empty($extraParams['execution_expires_after'])) {
  435. $apiParams['execution_expires_after'] = $extraParams['execution_expires_after'];
  436. }
  437. if (isset($extraParams['generate_audio'])) {
  438. $apiParams['generate_audio'] = $extraParams['generate_audio'];
  439. }
  440. if (isset($extraParams['draft'])) {
  441. $apiParams['draft'] = $extraParams['draft'];
  442. }
  443. if (!empty($extraParams['video_resolution'])) {
  444. $apiParams['resolution'] = strtolower($task->video_resolution);
  445. }
  446. if (!empty($extraParams['ratio'])) {
  447. $apiParams['ratio'] = $extraParams['ratio'];
  448. }
  449. if (!empty($extraParams['duration'])) {
  450. $apiParams['duration'] = (int)$extraParams['duration'];
  451. }
  452. if (isset($task->frames)) {
  453. // $apiParams['frames'] = $task->frames;
  454. }
  455. if ($task->seed !== -1) {
  456. $apiParams['seed'] = (int)$task->seed;
  457. }
  458. if (isset($extraParams['camera_fixed'])) {
  459. $apiParams['camera_fixed'] = $extraParams['camera_fixed'];
  460. }
  461. if (isset($extraParams['watermark'])) {
  462. $apiParams['watermark'] = $extraParams['watermark'];
  463. }
  464. dLog('generate')->info('Seedance 1.5 Pro 视频生成参数: ', $apiParams);
  465. logDB('generate', 'info', 'Seedance视频生成任务提交', ['id' => $task->id, 'params' => $apiParams]);
  466. // 调用火山引擎 Seedance API
  467. $response = $client->post('https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks', [
  468. 'headers' => [
  469. 'Authorization' => 'Bearer ' . $apiKey,
  470. 'Content-Type' => 'application/json',
  471. ],
  472. 'json' => $apiParams
  473. ]);
  474. $responseData = json_decode($response->getBody(), true);
  475. dLog('generate')->info('Seedance 1.5 Pro API 响应: ', $responseData);
  476. logDB('generate', 'info', 'Seedance视频生成API响应', ['id' => $task->id, 'response' => $responseData]);
  477. if (isset($responseData['error'])) {
  478. // API返回错误
  479. logDB('generate', 'error', 'Seedance视频生成任务提交失败', ['id' => $task->id, 'error' => $responseData['error']['message'] ?? 'API Error']);
  480. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  481. 'error_message' => $responseData['error']['message'] ?? 'API Error',
  482. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  483. ]);
  484. } elseif (isset($responseData['id'])) {
  485. // 任务提交成功,更新状态为处理中
  486. $updateData = [
  487. 'task_id' => $responseData['id'], // 使用 API 返回的任务 ID
  488. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  489. ];
  490. $task->updateStatus(MpGenerateVideoTask::STATUS_PROCESSING, $updateData);
  491. } else {
  492. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  493. 'error_message' => '未知的 API 响应格式',
  494. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  495. ]);
  496. }
  497. } catch (\Exception $e) {
  498. // 记录错误
  499. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  500. 'error_message' => 'API请求失败: ' . $e->getMessage()
  501. ]);
  502. Utils::throwError('1001:'.$e->getMessage());
  503. }
  504. }
  505. // 更新Seedance视频任务状态
  506. private function updateSeedanceTask($task) {
  507. $statusInfo = $this->querySeedanceTaskStatus($task);
  508. if (!isset($statusInfo['status'])) return;
  509. if ($statusInfo['status'] === 'success') {
  510. logDB('generate', 'info', 'Seedance视频生成任务成功', ['id' => $task->id, 'result_url' => $statusInfo['result_url']]);
  511. $task->updateStatus(MpGenerateVideoTask::STATUS_SUCCESS, $statusInfo);
  512. // 同步调整分镜视频状态和结果
  513. $segment_id = getProp($task, 'alias_segment_id');
  514. if ($segment_id && isset($statusInfo['result_url'])) {
  515. try {
  516. DB::beginTransaction();
  517. $now = date('Y-m-d H:i:s');
  518. // 构建更新数据
  519. $segmentUpdateData = [
  520. 'video_url' => $statusInfo['result_url'],
  521. 'video_task_status' => '已完成',
  522. 'current_type' => 2,
  523. 'last_frame_url' => $statusInfo['last_frame_url'] ?? '',
  524. 'updated_at' => $now
  525. ];
  526. // 只有当video_duration存在且大于0时才更新
  527. if (isset($statusInfo['video_duration']) && $statusInfo['video_duration'] > 0) {
  528. $segmentUpdateData['video_duration'] = $statusInfo['video_duration'];
  529. $segmentUpdateData['video_time_point_start'] = 0;
  530. $segmentUpdateData['video_time_point_end'] = $statusInfo['video_duration'];
  531. }
  532. // 更新分镜表
  533. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update($segmentUpdateData);
  534. DB::commit();
  535. } catch (\Exception $e) {
  536. DB::rollBack();
  537. dLog('generate')->error('Seedance视频任务处理失败', [
  538. 'segment_id' => $segment_id,
  539. 'task_id' => $task->id,
  540. 'error' => $e->getMessage()
  541. ]);
  542. }
  543. }
  544. } elseif ($statusInfo['status'] === 'failed') {
  545. logDB('generate', 'error', 'Seedance视频生成任务失败', ['id' => $task->id, 'error' => $statusInfo['error_message']]);
  546. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  547. 'error_message' => $statusInfo['error_message'],
  548. 'result_json' => $statusInfo['result_json'] ?? []
  549. ]);
  550. // 同步更新分镜表状态为失败
  551. $segment_id = getProp($task, 'alias_segment_id');
  552. if ($segment_id) {
  553. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  554. 'video_task_status' => '失败',
  555. 'updated_at' => date('Y-m-d H:i:s')
  556. ]);
  557. }
  558. }
  559. // 如果仍然是处理中状态,不做任何操作
  560. // 处理:如果任务处理超过48小时,标记为失败
  561. $processingTime = now()->diffInHours($task->created_at);
  562. if ($processingTime > 48) {
  563. logDB('generate', 'warning', 'Seedance视频生成任务超时', ['id' => $task->id, 'processing_hours' => $processingTime]);
  564. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  565. 'error_message' => '任务处理超时(超过48小时)'
  566. ]);
  567. // 同步更新分镜表状态为失败
  568. $segment_id = getProp($task, 'alias_segment_id');
  569. if ($segment_id) {
  570. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  571. 'video_task_status' => '失败',
  572. 'updated_at' => date('Y-m-d H:i:s')
  573. ]);
  574. }
  575. }
  576. }
  577. /**
  578. * 查询 Seedance 任务状态
  579. *
  580. * @param MpGenerateVideoTask $task
  581. * @return array
  582. */
  583. public function querySeedanceTaskStatus(MpGenerateVideoTask $task): array
  584. {
  585. $client = new Client(['verify'=>false,'timeout'=>120]);
  586. try {
  587. // 如果任务已完成,直接返回结果
  588. if (in_array($task->status, [MpGenerateVideoTask::STATUS_SUCCESS, MpGenerateVideoTask::STATUS_FAILED])) {
  589. return [
  590. 'success' => true,
  591. 'data' => [
  592. 'id' => $task->id,
  593. 'status' => $this->mapTaskStatus($task->status),
  594. 'result_url' => $task->result_url,
  595. 'error_message' => $task->error_message,
  596. 'created_at' => $task->created_at,
  597. 'started_at' => $task->started_at,
  598. 'completed_at' => $task->completed_at
  599. ]
  600. ];
  601. }
  602. // 调用火山引擎 API 查询任务状态
  603. $apiKey = env('VOLC_AI_API_KEY');
  604. if (empty($apiKey)) {
  605. return [
  606. 'status' => MpGenerateVideoTask::STATUS_FAILED,
  607. 'result_json' => [],
  608. 'error_message' => '火山引擎访问密钥未配置'
  609. ];
  610. }
  611. // 构建查询 URL
  612. $url = 'https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks/' . $task->task_id;
  613. $response = $client->get($url, [
  614. 'headers' => [
  615. 'Authorization' => 'Bearer ' . $apiKey,
  616. 'Content-Type' => 'application/json',
  617. ]
  618. ]);
  619. $responseData = json_decode($response->getBody(), true);
  620. dLog('generate')->info('Seedance 任务状态查询响应: ', $responseData);
  621. logDB('generate', 'info', 'Seedance任务状态查询', ['id' => $task->id, 'response' => $responseData]);
  622. if (isset($responseData['error'])) {
  623. logDB('generate', 'error', 'Seedance任务状态查询失败', ['id' => $task->id, 'error' => $responseData['error']['message'] ?? 'API查询失败']);
  624. return [
  625. 'status' => MpGenerateVideoTask::STATUS_FAILED,
  626. 'result_json' => $responseData,
  627. 'error_message' => $responseData['error']['message'] ?? 'API查询失败'
  628. ];
  629. }
  630. // 解析任务状态
  631. $apiStatus = $responseData['status'] ?? 'unknown';
  632. $taskStatus = $this->mapApiStatusToTaskStatus($apiStatus);
  633. $updateData = ['result_json' => $responseData];
  634. if ($taskStatus === MpGenerateVideoTask::STATUS_SUCCESS) {
  635. $updateData['status'] = MpGenerateVideoTask::STATUS_SUCCESS;
  636. // 任务成功,处理结果
  637. if (isset($responseData['content']['video_url'])) {
  638. $video_url = $responseData['content']['video_url'];
  639. // 下载并保存视频
  640. $video_name = 'ai_generation_' . time() . '_' . uniqid();
  641. $video_ext = getVideoExtFromUrl($video_url);
  642. $video_name = $video_name . $video_ext;
  643. $url = uploadStreamByTos('video', file_get_contents($video_url), $video_name);
  644. $updateData['result_url'] = $url;
  645. }
  646. // 计算视频时长
  647. $video_duration = 0;
  648. if (isset($responseData['duration'])) {
  649. // 如果直接返回duration,使用该值
  650. $video_duration = (int)$responseData['duration'];
  651. } elseif (isset($responseData['frames']) && isset($responseData['framespersecond'])) {
  652. // 如果返回frames和framespersecond,计算时长
  653. $frames = (int)$responseData['frames'];
  654. $fps = (int)$responseData['framespersecond'];
  655. if ($fps > 0) {
  656. $video_duration = (int)floor($frames / $fps);
  657. }
  658. }
  659. // 如果没有计算出时长,使用任务表中的设定值
  660. if ($video_duration <= 0 && !empty($task->extra_params)) {
  661. $extraParams = is_array($task->extra_params) ? $task->extra_params : json_decode($task->extra_params, true);
  662. if (isset($extraParams['duration'])) {
  663. $video_duration = (int)$extraParams['duration'];
  664. }
  665. }
  666. // 只有当video_duration > 0时才添加到更新数据中
  667. if ($video_duration > 0) {
  668. $updateData['video_duration'] = $video_duration;
  669. $updateData['video_time_point_start'] = 0;
  670. $updateData['video_time_point_end'] = $video_duration;
  671. }
  672. // 处理尾帧图片(如果启用了 return_last_frame)
  673. if (isset($responseData['content']['last_frame_url']) && !empty($responseData['content']['last_frame_url'])) {
  674. $pic_url = $responseData['content']['last_frame_url'];
  675. $pic_name = 'ai_generation_' . time() . '_' . uniqid();
  676. $pic_ext = getImgExtFromUrl($pic_url);
  677. $pic_name = $pic_name . $pic_ext;
  678. if ($pic_ext === '.png') {
  679. $comporessed_data = compressRemoteImageUrlToSize($pic_url);
  680. $url = uploadStreamByTos('image', $comporessed_data, $pic_name);
  681. }else {
  682. // 将图片另存到tos
  683. $url = uploadStreamByTos('image', file_get_contents($pic_url), $pic_name);
  684. }
  685. $updateData['last_frame_url'] = $url;
  686. }
  687. return $updateData;
  688. } elseif ($taskStatus === MpGenerateVideoTask::STATUS_FAILED) {
  689. $updateData['status'] = MpGenerateVideoTask::STATUS_FAILED;
  690. $updateData['error_message'] = $responseData['error_message'] ?? '任务执行失败';
  691. return $updateData;
  692. }
  693. // 其他状态保持不变
  694. return [
  695. 'success' => true,
  696. 'data' => [
  697. 'id' => $task->id,
  698. 'status' => $this->mapTaskStatus($task->status),
  699. 'result_url' => $task->result_url,
  700. 'error_message' => $task->error_message,
  701. 'created_at' => $task->created_at,
  702. 'started_at' => $task->started_at,
  703. 'completed_at' => $task->completed_at,
  704. 'api_response' => $responseData
  705. ]
  706. ];
  707. } catch (\Exception $e) {
  708. dLog('generate')->info('Seedance 任务状态查询异常: '.$e->getMessage());
  709. return [
  710. 'success' => false,
  711. 'message' => '查询失败: ' . $e->getMessage()
  712. ];
  713. }
  714. }
  715. /**
  716. * 映射 API 状态到任务状态
  717. *
  718. * @param string $apiStatus
  719. * @return string
  720. */
  721. private function mapApiStatusToTaskStatus(string $apiStatus): string
  722. {
  723. switch ($apiStatus) {
  724. case 'queued':
  725. case 'running':
  726. return MpGenerateVideoTask::STATUS_PROCESSING;
  727. case 'succeeded':
  728. return MpGenerateVideoTask::STATUS_SUCCESS;
  729. case 'failed':
  730. case 'expired':
  731. return MpGenerateVideoTask::STATUS_FAILED;
  732. default:
  733. return MpGenerateVideoTask::STATUS_PROCESSING;
  734. }
  735. }
  736. public function seedanceCallback($data) {
  737. try {
  738. dLog('generate')->info('Seedance 回调数据: ', $data);
  739. logDB('generate', 'info', 'Seedance回调接收', ['task_id' => $data['id'] ?? null, 'status' => $data['status'] ?? null]);
  740. // 从回调数据中获取任务ID
  741. $taskId = $data['id'] ?? null;
  742. if (!$taskId) {
  743. dLog('generate')->error('Seedance 回调缺少任务ID');
  744. logDB('generate', 'error', 'Seedance回调缺少任务ID', ['data' => $data]);
  745. return ['success' => false, 'message' => '缺少任务ID'];
  746. }
  747. // 查找对应的任务记录
  748. $task = MpGenerateVideoTask::where('task_id', $taskId)->first();
  749. if (!$task) {
  750. dLog('generate')->error('Seedance 回调找不到任务: ' . $taskId);
  751. logDB('generate', 'error', 'Seedance回调找不到任务', ['task_id' => $taskId]);
  752. return ['success' => false, 'message' => '任务不存在'];
  753. }
  754. // 解析任务状态
  755. $apiStatus = $data['status'] ?? 'unknown';
  756. $taskStatus = $this->mapApiStatusToTaskStatus($apiStatus);
  757. // 准备更新数据
  758. $updateData = [
  759. 'result_json' => $data,
  760. 'updated_at' => now()
  761. ];
  762. if ($taskStatus === MpGenerateVideoTask::STATUS_SUCCESS) {
  763. $updateData['status'] = MpGenerateVideoTask::STATUS_SUCCESS;
  764. $updateData['completed_at'] = now();
  765. // 处理视频结果
  766. if (isset($data['content']['video_url'])) {
  767. $video_url = $data['content']['video_url'];
  768. try {
  769. // 下载并保存视频
  770. $video_name = 'ai_generation_' . time() . '_' . uniqid();
  771. $video_ext = getVideoExtFromUrl($video_url);
  772. $video_name = $video_name . $video_ext;
  773. $url = uploadStreamByTos('video', file_get_contents($video_url), $video_name);
  774. $updateData['result_url'] = $url;
  775. dLog('generate')->info('Seedance 视频保存成功: ' . $url);
  776. } catch (\Exception $e) {
  777. dLog('generate')->error('Seedance 视频保存失败: ' . $e->getMessage());
  778. // 如果保存失败,直接使用原始URL
  779. $updateData['result_url'] = $video_url;
  780. }
  781. }
  782. // 处理尾帧图片(如果启用了 return_last_frame)
  783. if (isset($data['content']['last_frame_url']) && !empty($data['content']['last_frame_url'])) {
  784. $pic_url = $data['content']['last_frame_url'];
  785. try {
  786. $pic_name = 'ai_generation_' . time() . '_' . uniqid();
  787. $pic_ext = getImgExtFromUrl($pic_url);
  788. $pic_name = $pic_name . $pic_ext;
  789. if ($pic_ext === '.png') {
  790. $compressed_data = compressRemoteImageUrlToSize($pic_url);
  791. $url = uploadStreamByTos('image', $compressed_data, $pic_name);
  792. } else {
  793. $url = uploadStreamByTos('image', file_get_contents($pic_url), $pic_name);
  794. }
  795. $updateData['last_frame_url'] = $url;
  796. dLog('generate')->info('Seedance 尾帧图片保存成功: ' . $url);
  797. } catch (\Exception $e) {
  798. dLog('generate')->error('Seedance 尾帧图片保存失败: ' . $e->getMessage());
  799. // 如果保存失败,直接使用原始URL
  800. $updateData['last_frame_url'] = $pic_url;
  801. }
  802. }
  803. } elseif ($taskStatus === MpGenerateVideoTask::STATUS_FAILED) {
  804. $updateData['status'] = MpGenerateVideoTask::STATUS_FAILED;
  805. $updateData['completed_at'] = now();
  806. $updateData['error_message'] = $data['error_message'] ?? $data['failure_reason'] ?? '任务执行失败';
  807. } else {
  808. // 处理中状态
  809. $updateData['status'] = $taskStatus;
  810. if ($taskStatus === MpGenerateVideoTask::STATUS_PROCESSING && !$task->started_at) {
  811. $updateData['started_at'] = now();
  812. }
  813. }
  814. // 更新任务记录
  815. $task->update($updateData);
  816. // 更新分镜表的视频任务状态
  817. $this->updateSegmentVideoStatus($task, $taskStatus, $updateData);
  818. dLog('generate')->info('Seedance 回调处理完成: ' . $taskId . ', 状态: ' . $taskStatus);
  819. logDB('generate', 'info', 'Seedance回调处理完成', ['task_id' => $taskId, 'status' => $taskStatus]);
  820. return [
  821. 'success' => true,
  822. 'message' => '回调处理成功',
  823. 'task_id' => $taskId,
  824. 'status' => $taskStatus
  825. ];
  826. } catch (\Exception $e) {
  827. dLog('generate')->error('Seedance 回调处理异常: ' . $e->getMessage());
  828. logDB('generate', 'error', 'Seedance回调处理异常', ['error' => $e->getMessage()]);
  829. return [
  830. 'success' => false,
  831. 'message' => '回调处理失败: ' . $e->getMessage()
  832. ];
  833. }
  834. }
  835. /**
  836. * 更新分镜表的视频任务状态
  837. *
  838. * @param MpGenerateVideoTask $task
  839. * @param string $taskStatus
  840. * @param array $updateData
  841. */
  842. private function updateSegmentVideoStatus(MpGenerateVideoTask $task, string $taskStatus, array $updateData) {
  843. try {
  844. // 查找关联的分镜记录
  845. $segment = DB::table('mp_episode_segments')
  846. ->where('video_task_id', $task->id)
  847. ->first();
  848. if (!$segment) {
  849. dLog('generate')->info('未找到关联的分镜记录: ' . $task->id);
  850. return;
  851. }
  852. $segmentUpdateData = ['updated_at' => date('Y-m-d H:i:s')];
  853. if (isset($updateData['last_frame_url']) && $updateData['last_frame_url']) {
  854. $segmentUpdateData['last_frame_url'] = $updateData['last_frame_url'];
  855. }
  856. if ($taskStatus === MpGenerateVideoTask::STATUS_SUCCESS) {
  857. $segmentUpdateData['video_task_status'] = '已完成';
  858. if (isset($updateData['result_url'])) {
  859. $segmentUpdateData['video_url'] = $updateData['result_url'];
  860. }
  861. } elseif ($taskStatus === MpGenerateVideoTask::STATUS_FAILED) {
  862. $segmentUpdateData['video_task_status'] = '失败';
  863. } elseif ($taskStatus === MpGenerateVideoTask::STATUS_PROCESSING) {
  864. $segmentUpdateData['video_task_status'] = '生成中';
  865. }
  866. // 更新分镜表
  867. DB::table('mp_episode_segments')
  868. ->where('video_task_id', $task->id)
  869. ->update($segmentUpdateData);
  870. dLog('generate')->info('分镜视频状态更新成功: ' . $segment->segment_id . ', 状态: ' . ($segmentUpdateData['video_task_status'] ?? '未知'));
  871. } catch (\Exception $e) {
  872. dLog('generate')->error('更新分镜视频状态失败: ' . $e->getMessage());
  873. }
  874. }
  875. /**
  876. * 生成可灵AI的JWT Token
  877. *
  878. * @return string
  879. */
  880. private function generateKelingJwtToken(): string
  881. {
  882. $ak = env('KELING_AK');
  883. $sk = env('KELING_SK');
  884. if (empty($ak) || empty($sk)) {
  885. throw new \Exception('可灵AI访问密钥未配置,请检查环境变量KELING_AK和KELING_SK');
  886. }
  887. // JWT头部
  888. $header = [
  889. 'alg' => 'HS256',
  890. 'typ' => 'JWT'
  891. ];
  892. // JWT载荷
  893. $payload = [
  894. 'iss' => $ak, // 签发者,使用Access Key
  895. 'exp' => time() + 1800, // 过期时间,30分钟后过期
  896. 'nbf' => time() - 5 // 生效时间,5秒前开始生效
  897. ];
  898. // Base64URL编码
  899. $headerEncoded = $this->base64UrlEncode(json_encode($header));
  900. $payloadEncoded = $this->base64UrlEncode(json_encode($payload));
  901. // 创建签名
  902. $signature = hash_hmac('sha256', $headerEncoded . '.' . $payloadEncoded, $sk, true);
  903. $signatureEncoded = $this->base64UrlEncode($signature);
  904. // 返回完整的JWT token
  905. return $headerEncoded . '.' . $payloadEncoded . '.' . $signatureEncoded;
  906. }
  907. /**
  908. * Base64URL编码(JWT标准编码)
  909. *
  910. * @param string $data
  911. * @return string
  912. */
  913. private function base64UrlEncode(string $data): string
  914. {
  915. return rtrim(strtr(base64_encode($data), '+/', '-_'), '=');
  916. }
  917. /**
  918. * 创建可灵AI Omni视频生成任务
  919. *
  920. * @param array $params
  921. * @return MpGenerateVideoTask
  922. */
  923. public function createKelingOmniTask(array $params): MpGenerateVideoTask
  924. {
  925. // 生成唯一的任务ID
  926. $taskId = 'keling_omni_' . time() . '_' . uniqid();
  927. // 创建任务记录,复用现有表结构
  928. $task = MpGenerateVideoTask::create([
  929. 'task_id' => $taskId,
  930. 'prompt' => $params['prompt'] ?? '',
  931. 'first_frame_url' => $params['first_frame_url'] ?? '',
  932. 'tail_frame_url' => $params['tail_frame_url'] ?? '',
  933. 'seed' => $params['seed'] ?? -1,
  934. 'frames' => $params['video_duration'] ? ($params['video_duration'] * 24 + 1) : null,
  935. 'video_resolution' => $params['video_resolution'] ?? '720P',
  936. 'status' => MpGenerateVideoTask::STATUS_PENDING,
  937. 'api_type' => 'keling', // 标识为可灵AI Omni API
  938. 'extra_params' => [
  939. 'api_type' => 'keling',
  940. 'model' => $params['model'] ?? 'kling-video-o1',
  941. 'prompt' => $params['prompt'] ?? '',
  942. 'multi_prompt' => $params['multi_prompt'] ?? [],
  943. 'shot_type' => $params['shot_type'] ?? 'customize',
  944. 'image_list' => $params['image_list'] ?? [],
  945. 'element_list' => $params['element_list'] ?? [],
  946. 'sound' => $params['sound'] ?? 'off',
  947. // 'aspect_ratio' => $params['aspect_ratio'] ?? '16:9',
  948. 'duration' => $params['video_duration'] ?? 5,
  949. 'mode' => $params['mode'] ?? 'std',
  950. 'callback_url' => $params['callback_url'] ?? '',
  951. ]
  952. ]);
  953. // 提交任务到可灵AI Omni API
  954. $this->submitKelingOmniTaskToApi($task);
  955. return $task;
  956. }
  957. /**
  958. * 提交可灵AI Omni任务到API
  959. *
  960. * @param MpGenerateVideoTask $task
  961. * @return void
  962. */
  963. private function submitKelingOmniTaskToApi(MpGenerateVideoTask $task): void
  964. {
  965. $client = new Client(['verify'=>false,'timeout'=>120]);
  966. try {
  967. // 生成JWT token
  968. $jwtToken = $this->generateKelingJwtToken();
  969. $extraParams = $task->extra_params;
  970. // 构建可灵AI Omni API请求参数
  971. $apiParams = [
  972. 'model_name' => $extraParams['model'],
  973. 'prompt' => $extraParams['prompt'],
  974. 'mode' => $extraParams['mode'],
  975. // 'aspect_ratio' => $extraParams['aspect_ratio'],
  976. 'duration' => (int)$extraParams['duration'],
  977. ];
  978. // 添加可选参数
  979. if (!empty($extraParams['multi_prompt'])) {
  980. $apiParams['multi_prompt'] = $extraParams['multi_prompt'];
  981. }
  982. if (!empty($extraParams['shot_type'])) {
  983. $apiParams['shot_type'] = $extraParams['shot_type'];
  984. }
  985. if (!empty($extraParams['image_list'])) {
  986. $apiParams['image_list'] = $extraParams['image_list'];
  987. }
  988. if (!empty($extraParams['element_list'])) {
  989. $apiParams['element_list'] = json_decode($extraParams['element_list'], true);
  990. }
  991. if (isset($extraParams['sound'])) {
  992. $apiParams['sound'] = $extraParams['sound'];
  993. }
  994. // if (!empty($extraParams['callback_url'])) {
  995. // $apiParams['callback_url'] = $extraParams['callback_url'];
  996. // }
  997. dLog('generate')->info('可灵AI Omni视频生成参数: ', $apiParams);
  998. logDB('generate', 'info', '可灵AI Omni视频生成任务提交', ['id' => $task->id, 'params' => $apiParams]);
  999. // 调用可灵AI Omni API
  1000. $response = $client->post('https://api-beijing.klingai.com/v1/videos/omni-video', [
  1001. 'headers' => [
  1002. 'Authorization' => 'Bearer ' . $jwtToken,
  1003. 'Content-Type' => 'application/json',
  1004. ],
  1005. 'json' => $apiParams
  1006. ]);
  1007. $responseData = json_decode($response->getBody(), true);
  1008. dLog('generate')->info('可灵AI Omni API响应: ', $responseData);
  1009. logDB('generate', 'info', '可灵AI Omni API响应', ['id' => $task->id, 'response' => $responseData]);
  1010. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1011. // API返回错误
  1012. $errorMessage = $responseData['error']['message'] ?? $responseData['message'] ?? 'API Error';
  1013. logDB('generate', 'error', '可灵AI Omni任务提交失败', ['id' => $task->id, 'error' => $errorMessage]);
  1014. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1015. 'error_message' => $errorMessage,
  1016. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  1017. ]);
  1018. } elseif (isset($responseData['data']['task_id'])) {
  1019. // 任务提交成功,更新状态为处理中
  1020. $updateData = [
  1021. 'task_id' => $responseData['data']['task_id'], // 使用API返回的任务ID
  1022. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  1023. ];
  1024. $task->updateStatus(MpGenerateVideoTask::STATUS_PROCESSING, $updateData);
  1025. } else {
  1026. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1027. 'error_message' => '未知的API响应格式',
  1028. 'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
  1029. ]);
  1030. }
  1031. } catch (\Exception $e) {
  1032. // 记录错误
  1033. logDB('generate', 'error', '可灵AI Omni任务异常', ['id' => $task->id, 'error' => $e->getMessage()]);
  1034. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1035. 'error_message' => 'API请求失败: ' . $e->getMessage()
  1036. ]);
  1037. dLog('generate')->error('可灵AI Omni API请求异常: ' . $e->getMessage());
  1038. }
  1039. }
  1040. /**
  1041. * 查询可灵AI Omni任务状态
  1042. *
  1043. * @param MpGenerateVideoTask $task
  1044. * @return array
  1045. */
  1046. public function queryKelingOmniTaskStatus(MpGenerateVideoTask $task): array
  1047. {
  1048. $client = new Client(['verify'=>false,'timeout'=>120]);
  1049. try {
  1050. // 如果任务已完成,直接返回结果
  1051. if (in_array($task->status, [MpGenerateVideoTask::STATUS_SUCCESS, MpGenerateVideoTask::STATUS_FAILED])) {
  1052. return [
  1053. 'success' => true,
  1054. 'data' => [
  1055. 'task_id' => $task->task_id,
  1056. 'task_status' => $this->mapTaskStatusToKelingStatus($task->status),
  1057. 'result_url' => $task->result_url,
  1058. 'error_message' => $task->error_message,
  1059. 'created_at' => $task->created_at,
  1060. 'updated_at' => $task->updated_at
  1061. ]
  1062. ];
  1063. }
  1064. // 生成JWT token
  1065. $jwtToken = $this->generateKelingJwtToken();
  1066. // 构建查询URL
  1067. $url = 'https://api-beijing.klingai.com/v1/videos/omni-video/' . $task->task_id;
  1068. $response = $client->get($url, [
  1069. 'headers' => [
  1070. 'Authorization' => 'Bearer ' . $jwtToken,
  1071. 'Content-Type' => 'application/json',
  1072. ]
  1073. ]);
  1074. $responseData = json_decode($response->getBody(), true);
  1075. dLog('generate')->info('可灵AI Omni任务状态查询响应: ', $responseData);
  1076. logDB('generate', 'info', '可灵AI Omni任务状态查询', ['id' => $task->id, 'response' => $responseData]);
  1077. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1078. logDB('generate', 'error', '可灵AI Omni任务状态查询失败', ['id' => $task->id, 'error' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败']);
  1079. return [
  1080. 'status' => 'failed',
  1081. 'error_message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败',
  1082. 'result_json' => $responseData
  1083. ];
  1084. }
  1085. // 解析任务状态
  1086. $taskData = $responseData['data'] ?? [];
  1087. $taskStatus = $taskData['task_status'] ?? 'unknown';
  1088. $updateData = ['result_json' => $responseData];
  1089. if ($taskStatus === 'succeed') {
  1090. $updateData['status'] = 'success';
  1091. // 任务成功,处理结果
  1092. if (isset($taskData['task_result']['videos']) && !empty($taskData['task_result']['videos'])) {
  1093. $video = $taskData['task_result']['videos'][0];
  1094. $video_url = $video['url'];
  1095. // 下载并保存视频
  1096. $video_name = 'ai_generation_' . time() . '_' . uniqid();
  1097. $video_ext = getVideoExtFromUrl($video_url);
  1098. $video_name = $video_name . $video_ext;
  1099. $url = uploadStreamByTos('video', file_get_contents($video_url), $video_name);
  1100. $updateData['result_url'] = $url;
  1101. // 计算视频时长
  1102. $video_duration = 0;
  1103. if (isset($video['duration'])) {
  1104. $video_duration = (int)$video['duration'];
  1105. } elseif (isset($taskData['duration'])) {
  1106. $video_duration = (int)$taskData['duration'];
  1107. } elseif (isset($video['frames']) && isset($video['framespersecond'])) {
  1108. $frames = (int)$video['frames'];
  1109. $fps = (int)$video['framespersecond'];
  1110. if ($fps > 0) {
  1111. $video_duration = (int)floor($frames / $fps);
  1112. }
  1113. }
  1114. // 如果没有计算出时长,使用任务表中的设定值
  1115. if ($video_duration <= 0 && !empty($task->extra_params)) {
  1116. $extraParams = is_array($task->extra_params) ? $task->extra_params : json_decode($task->extra_params, true);
  1117. if (isset($extraParams['duration'])) {
  1118. $video_duration = (int)$extraParams['duration'];
  1119. }
  1120. }
  1121. // 只有当video_duration > 0时才添加到更新数据中
  1122. if ($video_duration > 0) {
  1123. $updateData['video_duration'] = $video_duration;
  1124. $updateData['video_time_point_start'] = 0;
  1125. $updateData['video_time_point_end'] = $video_duration;
  1126. }
  1127. }
  1128. return $updateData;
  1129. } elseif ($taskStatus === 'failed') {
  1130. $updateData = [
  1131. 'status' => 'failed',
  1132. 'error_message' => '任务执行失败',
  1133. 'result_json' => $responseData
  1134. ];
  1135. return $updateData;
  1136. }
  1137. // 其他状态(processing等)保持不变
  1138. return [
  1139. 'success' => true,
  1140. 'data' => [
  1141. 'task_id' => $task->task_id,
  1142. 'task_status' => $taskStatus,
  1143. 'result_url' => $task->result_url,
  1144. 'error_message' => $task->error_message,
  1145. 'created_at' => $task->created_at,
  1146. 'updated_at' => $task->updated_at,
  1147. 'api_response' => $responseData
  1148. ]
  1149. ];
  1150. } catch (\Exception $e) {
  1151. dLog('generate')->error('可灵AI Omni任务状态查询异常: ' . $e->getMessage());
  1152. return [
  1153. 'success' => false,
  1154. 'message' => '查询失败: ' . $e->getMessage()
  1155. ];
  1156. }
  1157. }
  1158. /**
  1159. * 映射任务状态到可灵AI状态格式
  1160. *
  1161. * @param string $taskStatus
  1162. * @return string
  1163. */
  1164. private function mapTaskStatusToKelingStatus(string $taskStatus): string
  1165. {
  1166. switch ($taskStatus) {
  1167. case MpGenerateVideoTask::STATUS_PENDING:
  1168. return 'submitted';
  1169. case MpGenerateVideoTask::STATUS_PROCESSING:
  1170. return 'processing';
  1171. case MpGenerateVideoTask::STATUS_SUCCESS:
  1172. return 'succeed';
  1173. case MpGenerateVideoTask::STATUS_FAILED:
  1174. return 'failed';
  1175. default:
  1176. return 'unknown';
  1177. }
  1178. }
  1179. /**
  1180. * 更新可灵AI Omni视频任务状态
  1181. *
  1182. * @param MpGenerateVideoTask $task
  1183. * @return void
  1184. */
  1185. private function updateKelingOmniTask($task): void
  1186. {
  1187. $statusInfo = $this->queryKelingOmniTaskStatus($task);
  1188. if (!isset($statusInfo['status'])) return;
  1189. if ($statusInfo['status'] === 'success') {
  1190. logDB('generate', 'info', '可灵AI Omni视频生成任务成功', ['id' => $task->id, 'result_url' => $statusInfo['result_url']]);
  1191. $task->updateStatus(MpGenerateVideoTask::STATUS_SUCCESS, $statusInfo);
  1192. // 同步调整分镜视频状态和结果
  1193. $segment_id = getProp($task, 'alias_segment_id');
  1194. if ($segment_id && isset($statusInfo['result_url'])) {
  1195. try {
  1196. DB::beginTransaction();
  1197. $now = date('Y-m-d H:i:s');
  1198. // 构建更新数据
  1199. $segmentUpdateData = [
  1200. 'video_url' => $statusInfo['result_url'],
  1201. 'video_task_status' => '已完成',
  1202. 'current_type' => 2,
  1203. 'last_frame_url' => $statusInfo['last_frame_url'] ?? '',
  1204. 'updated_at' => $now
  1205. ];
  1206. // 只有当video_duration存在且大于0时才更新
  1207. if (isset($statusInfo['video_duration']) && $statusInfo['video_duration'] > 0) {
  1208. $segmentUpdateData['video_duration'] = $statusInfo['video_duration'];
  1209. $segmentUpdateData['video_time_point_start'] = 0;
  1210. $segmentUpdateData['video_time_point_end'] = $statusInfo['video_duration'];
  1211. }
  1212. // 更新分镜表
  1213. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update($segmentUpdateData);
  1214. DB::commit();
  1215. } catch (\Exception $e) {
  1216. DB::rollBack();
  1217. dLog('generate')->error('可灵视频任务处理失败', [
  1218. 'segment_id' => $segment_id,
  1219. 'task_id' => $task->id,
  1220. 'error' => $e->getMessage()
  1221. ]);
  1222. }
  1223. }
  1224. } elseif ($statusInfo['status'] === 'failed') {
  1225. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1226. 'error_message' => $statusInfo['error_message'],
  1227. 'result_json' => $statusInfo['result_json'] ?? []
  1228. ]);
  1229. // 同步更新分镜表状态为失败
  1230. $segment_id = getProp($task, 'alias_segment_id');
  1231. if ($segment_id) {
  1232. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  1233. 'video_task_status' => '失败',
  1234. 'updated_at' => date('Y-m-d H:i:s')
  1235. ]);
  1236. }
  1237. }
  1238. // 如果仍然是处理中状态,不做任何操作
  1239. // 处理:如果任务处理超过24小时,标记为失败
  1240. $processingTime = now()->diffInHours($task->created_at);
  1241. if ($processingTime > 24) {
  1242. logDB('generate', 'warning', '可灵AI Omni视频生成任务超时', ['id' => $task->id, 'processing_hours' => $processingTime]);
  1243. $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
  1244. 'error_message' => '任务处理超时(超过24小时)'
  1245. ]);
  1246. // 同步更新分镜表状态为失败
  1247. $segment_id = getProp($task, 'alias_segment_id');
  1248. if ($segment_id) {
  1249. DB::table('mp_episode_segments')->where('segment_id', $segment_id)->update([
  1250. 'video_task_status' => '失败',
  1251. 'updated_at' => date('Y-m-d H:i:s')
  1252. ]);
  1253. }
  1254. }
  1255. }
  1256. /**
  1257. * 映射任务状态到前端显示状态
  1258. *
  1259. * @param string $taskStatus
  1260. * @return string
  1261. */
  1262. private function mapTaskStatus(string $taskStatus): string
  1263. {
  1264. switch ($taskStatus) {
  1265. case MpGenerateVideoTask::STATUS_PENDING:
  1266. return 'queued';
  1267. case MpGenerateVideoTask::STATUS_PROCESSING:
  1268. return 'running';
  1269. case MpGenerateVideoTask::STATUS_SUCCESS:
  1270. return 'succeeded';
  1271. case MpGenerateVideoTask::STATUS_FAILED:
  1272. return 'failed';
  1273. default:
  1274. return 'unknown';
  1275. }
  1276. }
  1277. // ==================== 主体管理相关方法 ====================
  1278. /**
  1279. * 创建主体
  1280. *
  1281. * @param array $params
  1282. * @return AIElement
  1283. */
  1284. public function createElement(array $params): AIElement
  1285. {
  1286. // 生成唯一的任务ID
  1287. $taskId = 'element_' . time() . '_' . uniqid();
  1288. // 创建主体记录
  1289. $element = AIElement::create([
  1290. 'task_id' => $taskId,
  1291. 'external_task_id' => $params['external_task_id'] ?? null,
  1292. 'element_name' => $params['element_name'],
  1293. 'element_description' => $params['element_description'],
  1294. 'reference_type' => $params['reference_type'],
  1295. 'element_image_list' => $params['element_image_list'] ?? null,
  1296. 'element_video_list' => $params['element_video_list'] ?? null,
  1297. 'element_voice_id' => $params['element_voice_id'] ?? null,
  1298. 'tag_list' => $params['tag_list'] ?? null,
  1299. 'callback_url' => $params['callback_url'] ?? null,
  1300. 'task_status' => AIElement::STATUS_SUBMITTED,
  1301. ]);
  1302. // 提交到可灵AI API
  1303. $this->submitElementToKelingApi($element);
  1304. return $element;
  1305. }
  1306. /**
  1307. * 提交主体创建任务到可灵AI API
  1308. *
  1309. * @param AIElement $element
  1310. * @return void
  1311. */
  1312. private function submitElementToKelingApi(AIElement $element): void
  1313. {
  1314. $client = new Client(['verify' => false, 'timeout' => 120]);
  1315. try {
  1316. // 生成JWT token
  1317. $jwtToken = $this->generateKelingJwtToken();
  1318. // 构建API请求参数
  1319. $apiParams = [
  1320. 'element_name' => $element->element_name,
  1321. 'element_description' => $element->element_description,
  1322. 'reference_type' => $element->reference_type,
  1323. ];
  1324. // 添加图片参考
  1325. if ($element->reference_type === AIElement::REFERENCE_TYPE_IMAGE && $element->element_image_list) {
  1326. $apiParams['element_image_list'] = $element->element_image_list;
  1327. }
  1328. // 添加视频参考
  1329. if ($element->reference_type === AIElement::REFERENCE_TYPE_VIDEO && $element->element_video_list) {
  1330. $apiParams['element_video_list'] = $element->element_video_list;
  1331. }
  1332. // 添加音色ID
  1333. if ($element->element_voice_id) {
  1334. $apiParams['element_voice_id'] = $element->element_voice_id;
  1335. }
  1336. // 添加标签
  1337. if ($element->tag_list) {
  1338. $apiParams['tag_list'] = $element->tag_list;
  1339. }
  1340. // 添加回调URL
  1341. if ($element->callback_url) {
  1342. $apiParams['callback_url'] = $element->callback_url;
  1343. }
  1344. // 添加外部任务ID
  1345. if ($element->external_task_id) {
  1346. $apiParams['external_task_id'] = $element->external_task_id;
  1347. }
  1348. dLog('generate')->info('可灵AI创建主体参数: ', $apiParams);
  1349. // 调用可灵AI创建主体API
  1350. $response = $client->post('https://api-beijing.klingai.com/v1/general/advanced-custom-elements', [
  1351. 'headers' => [
  1352. 'Authorization' => 'Bearer ' . $jwtToken,
  1353. 'Content-Type' => 'application/json',
  1354. ],
  1355. 'json' => $apiParams
  1356. ]);
  1357. $responseData = json_decode($response->getBody(), true);
  1358. dLog('generate')->info('可灵AI创建主体API响应: ', $responseData);
  1359. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1360. // API返回错误
  1361. $errorMessage = $responseData['error']['message'] ?? $responseData['message'] ?? 'API Error';
  1362. $element->update([
  1363. 'task_status' => AIElement::STATUS_FAILED,
  1364. 'task_status_msg' => $errorMessage,
  1365. ]);
  1366. } elseif (isset($responseData['data']['task_id'])) {
  1367. // 任务提交成功,更新状态为处理中
  1368. $element->update([
  1369. 'task_id' => $responseData['data']['task_id'], // 使用API返回的任务ID
  1370. 'task_status' => AIElement::STATUS_PROCESSING,
  1371. 'created_at' => isset($responseData['data']['created_at']) ?
  1372. date('Y-m-d H:i:s', $responseData['data']['created_at'] / 1000) : now(),
  1373. 'updated_at' => isset($responseData['data']['updated_at']) ?
  1374. date('Y-m-d H:i:s', $responseData['data']['updated_at'] / 1000) : now(),
  1375. ]);
  1376. } else {
  1377. $element->update([
  1378. 'task_status' => AIElement::STATUS_FAILED,
  1379. 'task_status_msg' => '未知的API响应格式',
  1380. ]);
  1381. }
  1382. } catch (\Exception $e) {
  1383. // 记录错误
  1384. $element->update([
  1385. 'task_status' => AIElement::STATUS_FAILED,
  1386. 'task_status_msg' => 'API请求失败: ' . $e->getMessage(),
  1387. ]);
  1388. dLog('generate')->error('可灵AI创建主体API请求异常: ' . $e->getMessage());
  1389. }
  1390. }
  1391. /**
  1392. * 查询单个主体状态
  1393. *
  1394. * @param string $taskId
  1395. * @param string|null $externalTaskId
  1396. * @return array
  1397. */
  1398. public function queryElementStatus(string $taskId, string $externalTaskId = null): array
  1399. {
  1400. $client = new Client(['verify' => false, 'timeout' => 120]);
  1401. try {
  1402. // 生成JWT token
  1403. $jwtToken = $this->generateKelingJwtToken();
  1404. // 构建查询URL
  1405. $url = 'https://api-beijing.klingai.com/v1/general/advanced-custom-elements/' . $taskId;
  1406. $queryParams = [];
  1407. if ($externalTaskId) {
  1408. $queryParams['external_task_id'] = $externalTaskId;
  1409. }
  1410. if (!empty($queryParams)) {
  1411. $url .= '?' . http_build_query($queryParams);
  1412. }
  1413. $response = $client->get($url, [
  1414. 'headers' => [
  1415. 'Authorization' => 'Bearer ' . $jwtToken,
  1416. 'Content-Type' => 'application/json',
  1417. ]
  1418. ]);
  1419. $responseData = json_decode($response->getBody(), true);
  1420. dLog('generate')->info('可灵AI查询主体状态响应: ', $responseData);
  1421. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1422. return [
  1423. 'success' => false,
  1424. 'message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败',
  1425. 'data' => null
  1426. ];
  1427. }
  1428. // 更新本地数据库记录
  1429. if (isset($responseData['data'])) {
  1430. $this->updateLocalElementFromApi($responseData['data']);
  1431. }
  1432. return [
  1433. 'success' => true,
  1434. 'data' => $responseData['data']
  1435. ];
  1436. } catch (\Exception $e) {
  1437. dLog('generate')->error('可灵AI查询主体状态异常: ' . $e->getMessage());
  1438. return [
  1439. 'success' => false,
  1440. 'message' => '查询失败: ' . $e->getMessage(),
  1441. 'data' => null
  1442. ];
  1443. }
  1444. }
  1445. /**
  1446. * 根据API响应更新本地主体记录
  1447. *
  1448. * @param array $apiData
  1449. * @return void
  1450. */
  1451. private function updateLocalElementFromApi(array $apiData): void
  1452. {
  1453. $element = AIElement::where('task_id', $apiData['task_id'])->first();
  1454. if (!$element) {
  1455. return;
  1456. }
  1457. $updateData = [
  1458. 'task_status' => $apiData['task_status'],
  1459. 'task_status_msg' => $apiData['task_status_msg'] ?? null,
  1460. 'final_unit_deduction' => $apiData['final_unit_deduction'] ?? null,
  1461. 'updated_at' => isset($apiData['updated_at']) ?
  1462. date('Y-m-d H:i:s', $apiData['updated_at'] / 1000) : now(),
  1463. ];
  1464. // 如果任务成功,更新主体信息
  1465. if ($apiData['task_status'] === AIElement::STATUS_SUCCEED && isset($apiData['task_result']['elements'])) {
  1466. $elementData = $apiData['task_result']['elements'][0];
  1467. $updateData = array_merge($updateData, [
  1468. 'element_id' => $elementData['element_id'],
  1469. 'element_voice_info' => $elementData['element_voice_info'] ?? null,
  1470. 'owned_by' => $elementData['owned_by'] ?? 'custom',
  1471. 'status' => $elementData['status'] ?? AIElement::ELEMENT_STATUS_SUCCEED,
  1472. ]);
  1473. }
  1474. $element->update($updateData);
  1475. }
  1476. /**
  1477. * 查询自定义主体列表
  1478. *
  1479. * @param int $pageNum
  1480. * @param int $pageSize
  1481. * @return array
  1482. */
  1483. public function queryCustomElementsList(int $pageNum = 1, int $pageSize = 30): array
  1484. {
  1485. $client = new Client(['verify' => false, 'timeout' => 120]);
  1486. try {
  1487. // 生成JWT token
  1488. $jwtToken = $this->generateKelingJwtToken();
  1489. // 构建查询URL
  1490. $url = 'https://api-beijing.klingai.com/v1/general/advanced-custom-elements';
  1491. $queryParams = [
  1492. 'pageNum' => $pageNum,
  1493. 'pageSize' => $pageSize,
  1494. ];
  1495. $url .= '?' . http_build_query($queryParams);
  1496. $response = $client->get($url, [
  1497. 'headers' => [
  1498. 'Authorization' => 'Bearer ' . $jwtToken,
  1499. 'Content-Type' => 'application/json',
  1500. ]
  1501. ]);
  1502. $responseData = json_decode($response->getBody(), true);
  1503. dLog('generate')->info('可灵AI查询自定义主体列表响应: ', $responseData);
  1504. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1505. return [
  1506. 'success' => false,
  1507. 'message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败',
  1508. 'data' => []
  1509. ];
  1510. }
  1511. // 更新本地数据库记录
  1512. if (isset($responseData['data']) && is_array($responseData['data'])) {
  1513. foreach ($responseData['data'] as $elementData) {
  1514. $this->updateLocalElementFromApi($elementData);
  1515. }
  1516. }
  1517. return [
  1518. 'success' => true,
  1519. 'data' => $responseData['data'] ?? []
  1520. ];
  1521. } catch (\Exception $e) {
  1522. dLog('generate')->error('可灵AI查询自定义主体列表异常: ' . $e->getMessage());
  1523. return [
  1524. 'success' => false,
  1525. 'message' => '查询失败: ' . $e->getMessage(),
  1526. 'data' => []
  1527. ];
  1528. }
  1529. }
  1530. /**
  1531. * 查询官方主体列表
  1532. *
  1533. * @param int $pageNum
  1534. * @param int $pageSize
  1535. * @return array
  1536. */
  1537. public function queryOfficialElementsList(int $pageNum = 1, int $pageSize = 30): array
  1538. {
  1539. $client = new Client(['verify' => false, 'timeout' => 120]);
  1540. try {
  1541. // 生成JWT token
  1542. $jwtToken = $this->generateKelingJwtToken();
  1543. // 构建查询URL
  1544. $url = 'https://api-beijing.klingai.com/v1/general/advanced-presets-elements';
  1545. $queryParams = [
  1546. 'pageNum' => $pageNum,
  1547. 'pageSize' => $pageSize,
  1548. ];
  1549. $url .= '?' . http_build_query($queryParams);
  1550. $response = $client->get($url, [
  1551. 'headers' => [
  1552. 'Authorization' => 'Bearer ' . $jwtToken,
  1553. 'Content-Type' => 'application/json',
  1554. ]
  1555. ]);
  1556. $responseData = json_decode($response->getBody(), true);
  1557. dLog('generate')->info('可灵AI查询官方主体列表响应: ', $responseData);
  1558. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1559. return [
  1560. 'success' => false,
  1561. 'message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败',
  1562. 'data' => []
  1563. ];
  1564. }
  1565. return [
  1566. 'success' => true,
  1567. 'data' => $responseData['data'] ?? []
  1568. ];
  1569. } catch (\Exception $e) {
  1570. dLog('generate')->error('可灵AI查询官方主体列表异常: ' . $e->getMessage());
  1571. return [
  1572. 'success' => false,
  1573. 'message' => '查询失败: ' . $e->getMessage(),
  1574. 'data' => []
  1575. ];
  1576. }
  1577. }
  1578. /**
  1579. * 删除自定义主体
  1580. *
  1581. * @param string $elementId
  1582. * @return array
  1583. */
  1584. public function deleteElement(string $elementId): array
  1585. {
  1586. $client = new Client(['verify' => false, 'timeout' => 120]);
  1587. try {
  1588. // 生成JWT token
  1589. $jwtToken = $this->generateKelingJwtToken();
  1590. // 构建API请求参数
  1591. $apiParams = [
  1592. 'element_id' => $elementId,
  1593. ];
  1594. dLog('generate')->info('可灵AI删除主体参数: ', $apiParams);
  1595. // 调用可灵AI删除主体API
  1596. $response = $client->post('https://api-beijing.klingai.com/v1/general/delete-elements', [
  1597. 'headers' => [
  1598. 'Authorization' => 'Bearer ' . $jwtToken,
  1599. 'Content-Type' => 'application/json',
  1600. ],
  1601. 'json' => $apiParams
  1602. ]);
  1603. $responseData = json_decode($response->getBody(), true);
  1604. dLog('generate')->info('可灵AI删除主体API响应: ', $responseData);
  1605. if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
  1606. // API返回错误
  1607. return [
  1608. 'success' => false,
  1609. 'message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API Error'
  1610. ];
  1611. } else {
  1612. // 删除成功,更新本地记录状态
  1613. AIElement::where('element_id', $elementId)->update([
  1614. 'status' => AIElement::ELEMENT_STATUS_DELETED,
  1615. 'updated_at' => now(),
  1616. ]);
  1617. return [
  1618. 'success' => true,
  1619. 'message' => '主体删除成功',
  1620. 'data' => $responseData['data'] ?? []
  1621. ];
  1622. }
  1623. } catch (\Exception $e) {
  1624. dLog('generate')->error('可灵AI删除主体API请求异常: ' . $e->getMessage());
  1625. return [
  1626. 'success' => false,
  1627. 'message' => 'API请求失败: ' . $e->getMessage()
  1628. ];
  1629. }
  1630. }
  1631. /**
  1632. * 获取本地主体列表(包含自定义和官方)
  1633. *
  1634. * @param array $filters
  1635. * @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
  1636. */
  1637. public function getLocalElementsList(array $filters = [])
  1638. {
  1639. $query = AIElement::query();
  1640. // 筛选条件
  1641. if (isset($filters['owned_by'])) {
  1642. $query->where('owned_by', $filters['owned_by']);
  1643. }
  1644. if (isset($filters['reference_type'])) {
  1645. $query->where('reference_type', $filters['reference_type']);
  1646. }
  1647. if (isset($filters['status'])) {
  1648. $query->where('status', $filters['status']);
  1649. }
  1650. if (isset($filters['task_status'])) {
  1651. $query->where('task_status', $filters['task_status']);
  1652. }
  1653. // 搜索关键词
  1654. if (isset($filters['keyword']) && !empty($filters['keyword'])) {
  1655. $query->where(function ($q) use ($filters) {
  1656. $q->where('element_name', 'like', '%' . $filters['keyword'] . '%')
  1657. ->orWhere('element_description', 'like', '%' . $filters['keyword'] . '%');
  1658. });
  1659. }
  1660. return $query->orderBy('created_at', 'desc')
  1661. ->paginate($filters['page_size'] ?? 20);
  1662. }
  1663. }