AIVideoGenerationService.php 78 KB

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