mint/api-v8/app/Console/Commands/MqAiTranslate.php

<?php

namespace App\Console\Commands;

use Illuminate\Console\Command;
use App\Http\Api\Mq;
use Illuminate\Support\Facades\Log;
use Illuminate\Support\Facades\Http;
use Illuminate\Support\Str;
use Illuminate\Http\Client\RequestException;

use App\Http\Controllers\AuthController;
use App\Models\Sentence;
use App\Models\ModelLog;

class MqAiTranslate extends Command
{
    /**
     * The name and signature of the console command.
     * php artisan mq:ai.translate
     * @var string
     */
    protected $signature = 'mq:ai.translate';

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = 'Command description';

    /**
     * Create a new command instance.
     *
     * @return void
     */
    public function __construct()
    {
        parent::__construct();
    }

    /**
     * Execute the console command.
     *
     * @return int
     */
    public function handle()
    {
        if (\App\Tools\Tools::isStop()) {
            return 0;
        }
        $exchange = 'router';
        $queue = 'ai_translate';
        $this->info(" [*] Waiting for {$queue}. To exit press CTRL+C");
        Log::debug("mq:progress start.");
        Mq::worker($exchange, $queue, function ($message) use ($queue) {
            Log::debug('ai translate start', ['message' => $message]);
            //写入 model log
            $modelLog = new ModelLog();
            $modelLog->uid = Str::uuid();

            $param = [
                "model" => $message->model->model,
                "messages" => [
                    ["role" => "system", "content" => $message->model->system_prompt],
                    ["role" => "user", "content" => $message->prompt],
                ],
                'prompt' => $message->prompt,
                "temperature" => 0.7,
                "stream" => false
            ];
            Log::info($queue . ' LLM request' . $message->model->url);
            Log::info($queue . ' model:' . $param['model']);
            Log::debug($queue . ' LLM api request', [
                'url' => $message->model->url,
                'data' => $param
            ]);
            $modelLog->model_id = $message->model->uid;
            $modelLog->request_at = now();
            $modelLog->request_data = json_encode($param, JSON_UNESCAPED_UNICODE);
            try {
                $response = Http::withToken($message->model->key)
                    ->post($message->model->url, $param);

                $response->throw(); // 触发异常（如果请求失败）

                Log::info($queue . ' LLM request successful');
                $modelLog->request_headers = json_encode($response->handlerStats(), JSON_UNESCAPED_UNICODE);
                $modelLog->response_headers = json_encode($response->headers(), JSON_UNESCAPED_UNICODE);
                $modelLog->status = $response->status();
                $modelLog->response_data = json_encode($response->json(), JSON_UNESCAPED_UNICODE);
                /*
                if ($response->failed()) {
                    $modelLog->success = false;
                    $modelLog->save();
                    Log::error($queue . ' http response error', ['data' => $response->json()]);
                    return 1;
                }*/
            } catch (RequestException $e) {
                Log::error($queue . ' LLM request exception: ' . $e->getMessage());
                $failResponse = $e->response;
                $modelLog->request_headers = json_encode($failResponse->handlerStats(), JSON_UNESCAPED_UNICODE);
                $modelLog->response_headers = json_encode($failResponse->headers(), JSON_UNESCAPED_UNICODE);
                $modelLog->status = $failResponse->status();
                $modelLog->response_data = $response->body();
                $modelLog->success = false;
                $modelLog->save();
                return 1;
            }


            $modelLog->save();
            Log::info($queue . ' model log saved');
            $aiData = $response->json();
            Log::debug($queue . ' LLM http response', ['data' => $response->json()]);
            $responseContent = $aiData['choices'][0]['message']['content'];
            if (isset($aiData['choices'][0]['message']['reasoning_content'])) {
                $reasoningContent = $aiData['choices'][0]['message']['reasoning_content'];
            }

            Log::debug($queue . ' LLM response content=' . $responseContent);
            if (empty($reasoningContent)) {
                Log::debug($queue . ' no reasoningContent');
            } else {
                Log::debug($queue . ' reasoning=' . $reasoningContent);
            }

            //获取model token
            Log::debug($queue . ' ai assistant token', ['user' => $message->model->uid]);
            $token = AuthController::getUserToken($message->model->uid);
            Log::debug($queue . ' ai assistant token', ['token' => $token]);

            if ($message->task->info->category === 'translate') {
                //写入句子库
                $url = config('app.url') . '/api/v2/sentence';
                $sentData = [];
                $message->sentence->content = $responseContent;
                $sentData[] = $message->sentence;
                Log::info($queue . " sentence update {$url}");
                $response = Http::withToken($token)->post($url, [
                    'sentences' => $sentData,
                ]);
                if ($response->failed()) {
                    Log::error($queue . ' sentence update failed', [
                        'url' => $url,
                        'data' => $response->json(),
                    ]);
                    return 1;
                } else {
                    Log::info($queue . ' sentence update successful');
                }
            }

            //写入discussion
            #获取句子id
            $sUid = Sentence::where('book_id', $message->sentence->book_id)
                ->where('paragraph', $message->sentence->paragraph)
                ->where('word_start', $message->sentence->word_start)
                ->where('word_end', $message->sentence->word_end)
                ->where('channel_uid', $message->sentence->channel_uid)
                ->value('uid');
            $url = config('app.url') . '/api/v2/discussion';
            $data = [
                'res_id' => $sUid,
                'res_type' => 'sentence',
                'title' => $message->task->info->title,
                'content' => $message->task->info->category,
                'content_type' => 'markdown',
                'type' => 'discussion',
                'notification' => false,
            ];
            $response = Http::withToken($token)->post($url, $data);
            if ($response->failed()) {
                Log::error($queue . ' discussion error', ['data' => $response->json()]);
            } else {
                Log::info($queue . ' discussion topic successful');
                if (isset($response->json()['data']['id'])) {
                    $data['parent'] = $response->json()['data']['id'];
                    unset($data['title']);
                    $topicChildren = [];
                    //提示词
                    $topicChildren[] = $message->prompt;
                    //任务结果
                    $topicChildren[] = $responseContent;
                    //推理过程写入discussion
                    if (isset($reasoningContent) && !empty($reasoningContent)) {
                        $topicChildren[] = $reasoningContent;
                    }
                    foreach ($topicChildren as  $content) {
                        $data['content'] = $content;
                        Log::debug($queue . ' discussion child request', ['url' => $url, 'data' => $data]);
                        $response = Http::withToken($token)->post($url, $data);
                        if ($response->failed()) {
                            Log::error($queue . ' discussion error', ['data' => $response->json()]);
                        } else {
                            Log::info($queue . ' discussion child successful');
                        }
                    }
                } else {
                    Log::error($queue . 'discussion response is null');
                }
            }


            //修改task 完成度
            $taskProgress = $message->task->progress;
            if ($taskProgress->total > 0) {
                $progress = (int)($taskProgress->current * 100 / $taskProgress->total);
            } else {
                $progress = 100;
                Log::error($queue . ' progress total is zero', ['task_id' => $message->task->info->id]);
            }
            $url = config('app.url') . '/api/v2/task/' . $message->task->info->id;
            $data = [
                'progress' => $progress,
            ];
            Log::debug($queue . ' task progress request', ['url' => $url, 'data' => $data]);
            $response = Http::withToken($token)->patch($url, $data);
            if ($response->failed()) {
                Log::error($queue . ' task progress error', ['data' => $response->json()]);
            } else {
                Log::info($queue . ' task progress successful progress=' . $response->json()['data']['progress']);
            }

            //任务完成 修改任务状态为 done
            if ($progress === 100) {
                $url = config('app.url') . '/api/v2/task-status/' . $message->task->info->id;
                $data = [
                    'status' => 'done',
                ];
                Log::debug($queue . ' task status request', ['url' => $url, 'data' => $data]);
                $response = Http::withToken($token)->patch($url, $data);
                //判断状态码
                if ($response->failed()) {
                    Log::error($queue . ' task status error', ['data' => $response->json()]);
                } else {
                    Log::info($queue . ' task status done');
                }
            }
            return 0;
        });
        return 0;
    }
}