AIVideoGenerationService.php 78 KB

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