修改图生图

app/Jobs/ReadDiscordMessageJob.php

@@ -27,8 +27,15 @@ public function handle()
     {
         $data = $this->data;
         $content = $data['message_content'] ?? '';
-        preg_match('/\*\*(.*)\*\*/', $content, $matches);
-        if (isset($matches[1])) {
+        preg_match('/\*\*<(.*)> (.*)\*\*/', $content, $matches);
+        if (blank($matches)) {
+            preg_match('/\*\*(.*)\*\*/', $content, $matches);
+            $content = $matches[1] ?? '';
+        } else {
+            $content = $matches[2] ?? '';
+        }
+
+        if (isset($content) && filled($content)) {
             $key = app(TaskService::class)->getDiscordMessageKey($matches[1]);
             $cachePrompt = Cache::get($key);
             if (!empty($cachePrompt)) {
@@ -38,6 +45,7 @@ public function handle()
                 ];
                 PromptTask::where('id', $cachePrompt['prompt_task_id'])
                     ->update($updateData);
+                Cache::forget($key);
             }
         }
     }

app/Jobs/SendImagineJob.php

@@ -26,13 +26,13 @@ public function __construct(array $data = [])
     public function handle()
     {
         $data = $this->data;
+        $key = app(TaskService::class)->getDiscordMessageKey($data['prompt']);
+        Cache::set($key, ['prompt_task_id' => $data['prompt_id']], (60 * 60 * 24));
         // 处理图转图
         if ($data['type'] == TaskService::TYPE_IMG_TO_IMG) {
             $promptImg = json_decode(($data['prompt_img'] ?? ''));
             $data['prompt'] = ($promptImg[0] ?? '') . ' ' . $data['prompt'];
         }
-        $key = app(TaskService::class)->getDiscordMessageKey($data['prompt']);
-        Cache::set($key, ['prompt_task_id' => $data['prompt_id']], (60 * 60 * 24));
         app(ExternalApiService::class)->imagine($data);
     }
 }