存客宝应用接口初始化

2026-01-05 10:25:32 +08:00
parent fa37227954
commit 02797ba50a
414 changed files with 85530 additions and 75 deletions
--- a/application/ai/controller/DouBaoAI.php
+++ b/application/ai/controller/DouBaoAI.php
@@ -0,0 +1,68 @@
+<?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]]);
+        }
+
+    }
+
+
+}