<?php

namespace app\ai\controller;

use app\common\util\JwtUtil;
use think\facade\Env;
use think\Controller;

class DouBaoAI extends Controller
{
    protected $apiUrl;
    protected $apiKey;
    protected $headers;

    public function __construct()
    {
        parent::__construct();

        $this->apiUrl = Env::get('doubaoAi.api_url');
        $this->apiKey = Env::get('doubaoAi.api_key');

        // 设置请求头
        $this->headers = [
            'Content-Type: application/json',
            'Authorization: Bearer ' . $this->apiKey
        ];

        if (empty($this->apiKey) || empty($this->apiUrl)) {
            return json_encode(['code' => 500, 'msg' => '参数缺失']);
        }
    }


    public function text($params = [])
    {

        if (empty($params)){
            $content = $this->request->param('content', '');
            $model = $this->request->param('model', 'doubao-seed-1-8-251215');
            if(empty($content)){
                return json_encode(['code' => 500, 'msg' => '提示词缺失']);
            }
            $params = [
                'model' => $model,
                'messages' => [
                    ['role' => 'system', 'content' => '你现在是存客宝的AI助理，你精通中国大陆的法律'],
                    ['role' => 'user', 'content' => $content],
                ],
            ];
        }
        $result = requestCurl($this->apiUrl, $params, 'POST', $this->headers, 'json');
        $result = json_decode($result, true);
        if(isset($result['error'])){
            $error = $result['error'];
            return json_encode(['code' => 500, 'msg' => $error['message']]);
        }else{
            $content = $result['choices'][0]['message']['content'];
            $token = intval($result['usage']['total_tokens']) * 20;

            exit_data($content);
            return json_encode(['code' => 200, 'msg' => '成功','data' => ['token' => $token,'content' => $content]]);
        }

    }


}