MqAiTranslate.php 9.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239
  1. <?php
  2. namespace App\Console\Commands;
  3. use Illuminate\Console\Command;
  4. use App\Http\Api\Mq;
  5. use Illuminate\Support\Facades\Log;
  6. use Illuminate\Support\Facades\Http;
  7. use Illuminate\Support\Str;
  8. use Illuminate\Http\Client\RequestException;
  9. use App\Http\Controllers\AuthController;
  10. use App\Models\Sentence;
  11. use App\Models\ModelLog;
  12. class MqAiTranslate extends Command
  13. {
  14. /**
  15. * The name and signature of the console command.
  16. * php artisan mq:ai.translate
  17. * @var string
  18. */
  19. protected $signature = 'mq:ai.translate';
  20. /**
  21. * The console command description.
  22. *
  23. * @var string
  24. */
  25. protected $description = 'Command description';
  26. /**
  27. * Create a new command instance.
  28. *
  29. * @return void
  30. */
  31. public function __construct()
  32. {
  33. parent::__construct();
  34. }
  35. /**
  36. * Execute the console command.
  37. *
  38. * @return int
  39. */
  40. public function handle()
  41. {
  42. if (\App\Tools\Tools::isStop()) {
  43. return 0;
  44. }
  45. $exchange = 'router';
  46. $queue = 'ai_translate';
  47. $this->info(" [*] Waiting for {$queue}. To exit press CTRL+C");
  48. Log::debug("mq:progress start.");
  49. Mq::worker($exchange, $queue, function ($message) use ($queue) {
  50. Log::debug('ai translate start', ['message' => $message]);
  51. //写入 model log
  52. $modelLog = new ModelLog();
  53. $modelLog->uid = Str::uuid();
  54. $param = [
  55. "model" => $message->model->model,
  56. "messages" => [
  57. ["role" => "system", "content" => $message->model->system_prompt],
  58. ["role" => "user", "content" => $message->prompt],
  59. ],
  60. 'prompt' => $message->prompt,
  61. "temperature" => 0.7,
  62. "stream" => false
  63. ];
  64. Log::info($queue . ' ai request' . $message->model->url);
  65. Log::info($queue . ' model:' . $param['model']);
  66. Log::debug($queue . ' ai api request', [
  67. 'url' => $message->model->url,
  68. 'data' => $param
  69. ]);
  70. $modelLog->model_id = $message->model->uid;
  71. $modelLog->request_at = now();
  72. $modelLog->request_data = json_encode($param, JSON_UNESCAPED_UNICODE);
  73. try {
  74. $response = Http::withToken($message->model->key)
  75. ->post($message->model->url, $param);
  76. $response->throw(); // 触发异常(如果请求失败)
  77. $modelLog->request_headers = json_encode($response->handlerStats(), JSON_UNESCAPED_UNICODE);
  78. $modelLog->response_headers = json_encode($response->headers(), JSON_UNESCAPED_UNICODE);
  79. $modelLog->status = $response->status();
  80. $modelLog->response_data = json_encode($response->json(), JSON_UNESCAPED_UNICODE);
  81. /*
  82. if ($response->failed()) {
  83. $modelLog->success = false;
  84. $modelLog->save();
  85. Log::error($queue . ' http response error', ['data' => $response->json()]);
  86. return 1;
  87. }*/
  88. } catch (RequestException $e) {
  89. Log::error('HTTP 请求发生异常: ' . $e->getMessage());
  90. $failResponse = $e->response;
  91. $modelLog->request_headers = json_encode($failResponse->handlerStats(), JSON_UNESCAPED_UNICODE);
  92. $modelLog->response_headers = json_encode($failResponse->headers(), JSON_UNESCAPED_UNICODE);
  93. $modelLog->status = $failResponse->status();
  94. $modelLog->response_data = $response->body();
  95. $modelLog->success = false;
  96. $modelLog->save();
  97. return 1;
  98. }
  99. $modelLog->save();
  100. Log::debug($queue . ' log saved');
  101. $aiData = $response->json();
  102. Log::debug($queue . ' http response', ['data' => $response->json()]);
  103. $responseContent = $aiData['choices'][0]['message']['content'];
  104. if (isset($aiData['choices'][0]['message']['reasoning_content'])) {
  105. $reasoningContent = $aiData['choices'][0]['message']['reasoning_content'];
  106. }
  107. Log::debug($queue . ' ai content=' . $responseContent);
  108. if (empty($reasoningContent)) {
  109. Log::debug($queue . ' no reasoningContent');
  110. } else {
  111. Log::debug($queue . ' reasoning=' . $reasoningContent);
  112. }
  113. //获取model token
  114. Log::debug($queue . ' ai assistant token', ['user' => $message->model->uid]);
  115. $token = AuthController::getUserToken($message->model->uid);
  116. Log::debug($queue . ' ai assistant token', ['token' => $token]);
  117. if ($message->task->info->category === 'translate') {
  118. //写入句子库
  119. $url = config('app.url') . '/api/v2/sentence';
  120. $sentData = [];
  121. $message->sentence->content = $responseContent;
  122. $sentData[] = $message->sentence;
  123. Log::debug($queue . " upload to {$url}");
  124. Log::debug($queue . ' sentence update http request', ['data' => $sentData]);
  125. $response = Http::withToken($token)->post($url, [
  126. 'sentences' => $sentData,
  127. ]);
  128. Log::debug($queue . ' sentence update http response', ['data' => $response->json()]);
  129. if ($response->failed()) {
  130. Log::error($queue . ' upload error', [
  131. 'data' => $response->json(),
  132. 'message' => $message
  133. ]);
  134. return 1;
  135. } else {
  136. Log::info($queue . ' upload successful');
  137. }
  138. }
  139. //写入discussion
  140. #获取句子id
  141. $sUid = Sentence::where('book_id', $message->sentence->book_id)
  142. ->where('paragraph', $message->sentence->paragraph)
  143. ->where('word_start', $message->sentence->word_start)
  144. ->where('word_end', $message->sentence->word_end)
  145. ->where('channel_uid', $message->sentence->channel_uid)
  146. ->value('uid');
  147. $url = config('app.url') . '/api/v2/discussion';
  148. $data = [
  149. 'res_id' => $sUid,
  150. 'res_type' => 'sentence',
  151. 'title' => $message->task->info->title,
  152. 'content' => $message->task->info->category,
  153. 'content_type' => 'markdown',
  154. 'type' => 'discussion',
  155. ];
  156. $response = Http::withToken($token)->post($url, $data);
  157. if ($response->failed()) {
  158. Log::error($queue . ' ai discussion error', ['data' => $response->json()]);
  159. } else {
  160. Log::info($queue . ' ai discussion topic successful');
  161. }
  162. $data['parent'] = $response->json()['data']['id'];
  163. unset($data['title']);
  164. $topicChildren = [];
  165. //提示词
  166. $topicChildren[] = $message->prompt;
  167. //任务结果
  168. $topicChildren[] = $responseContent;
  169. //推理过程写入discussion
  170. if (isset($reasoningContent) && !empty($reasoningContent)) {
  171. $topicChildren[] = $reasoningContent;
  172. }
  173. foreach ($topicChildren as $content) {
  174. $data['content'] = $content;
  175. Log::debug($queue . ' discussion child request', ['url' => $url, 'data' => $data]);
  176. $response = Http::withToken($token)->post($url, $data);
  177. if ($response->failed()) {
  178. Log::error($queue . ' discussion error', ['data' => $response->json()]);
  179. } else {
  180. Log::info($queue . ' discussion child successful');
  181. }
  182. }
  183. //修改task 完成度
  184. $taskProgress = $message->task->progress;
  185. if ($taskProgress->total > 0) {
  186. $progress = (int)($taskProgress->current * 100 / $taskProgress->total);
  187. } else {
  188. $progress = 100;
  189. Log::error($queue . ' progress total is zero', ['task_id' => $message->task->info->id]);
  190. }
  191. $url = config('app.url') . '/api/v2/task/' . $message->task->info->id;
  192. $data = [
  193. 'progress' => $progress,
  194. ];
  195. Log::debug($queue . ' task progress request', ['url' => $url, 'data' => $data]);
  196. $response = Http::withToken($token)->patch($url, $data);
  197. if ($response->failed()) {
  198. Log::error($queue . ' task progress error', ['data' => $response->json()]);
  199. } else {
  200. Log::info($queue . ' task progress successful progress=' . $response->json()['data']['progress']);
  201. }
  202. //任务完成 修改任务状态为 done
  203. if ($progress === 100) {
  204. $url = config('app.url') . '/api/v2/task-status/' . $message->task->info->id;
  205. $data = [
  206. 'status' => 'done',
  207. ];
  208. Log::debug($queue . ' task status request', ['url' => $url, 'data' => $data]);
  209. $response = Http::withToken($token)->patch($url, $data);
  210. //判断状态码
  211. if ($response->failed()) {
  212. Log::error($queue . ' task status error', ['data' => $response->json()]);
  213. } else {
  214. Log::info($queue . ' task status done');
  215. }
  216. }
  217. return 0;
  218. });
  219. return 0;
  220. }
  221. }