feat(generators): add ErnieBotTurbo generator

Author: guanguans

app/Generators/ErnieBotGenerator.php

@@ -47,16 +47,14 @@ public function __construct(array $config)
      */
     public function generate(string $prompt): string
     {
-        $prompt = 'PHP是什么？';
-
         $parameters = [
             'messages' => [
                 ['role' => 'user', 'content' => $prompt],
             ],
-            'user' => Str::uuid()->toString(),
+            'user_id' => Str::uuid()->toString(),
         ] + Arr::get($this->config, 'parameters', []);
 
-        $response = $this->ernie->ernieBot($parameters, $this->buildWriter($messages));
+        [$messages, $response] = $this->getResponse($parameters);
 
         // fake 响应
         return (string) ($messages ?? $this->getCompletionMessages($response));
@@ -76,15 +74,17 @@ protected function getCompletionMessages($response): string
     protected function buildWriter(?string &$messages): \Closure
     {
         return function (string $data) use (&$messages): void {
-            // 流响应完成
-            if (str($data)->startsWith('data: [DONE]')) {
-                return;
-            }
-
             // (正常|错误|流)响应
             $rowResponse = (array) json_decode(Ernie::sanitizeData($data), true);
             $messages .= $text = $this->getCompletionMessages($rowResponse);
             $this->outputStyle->write($text);
         };
     }
+
+    protected function getResponse(array $parameters): array
+    {
+        $response = $this->ernie->ernieBot($parameters, $this->buildWriter($messages));
+
+        return [$messages, $response];
+    }
 }

app/Generators/ErnieBotTurboGenerator.php

@@ -0,0 +1,23 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * This file is part of the guanguans/ai-commit.
+ *
+ * (c) guanguans <ityaozm@gmail.com>
+ *
+ * This source file is subject to the MIT license that is bundled.
+ */
+
+namespace App\Generators;
+
+class ErnieBotTurboGenerator extends ErnieBotGenerator
+{
+    protected function getResponse(array $parameters): array
+    {
+        $response = $this->ernie->ernieBotTurbo($parameters, $this->buildWriter($messages));
+
+        return [$messages, $response];
+    }
+}

app/Support/Ernie.php

@@ -40,8 +40,38 @@ static function (Stringable $data) use ($prefix): Stringable {
     }
 
     /**
-     * @throws RequestException
+     * ```php
+     * [
+     *     'id' => 'chatcmpl-6pqDoRwRGQAlRvJnesR9QMG9rxpyK',
+     *     'object' => 'chat.completion',
+     *     'created' => 1677813488,
+     *     'model' => 'gpt-3.5-turbo-0301',
+     *     'usage' => [
+     *         'prompt_tokens' => 8,
+     *         'completion_tokens' => 16,
+     *         'total_tokens' => 24,
+     *     ],
+     *     'choices' => [
+     *         [
+     *             'message' => [
+     *                 'role' => 'assistant',
+     *                 'content' => 'PHP (Hypertext Preprocessor) is a server-side scripting language used',
+     *             ],
+     *             'finish_reason' => 'length',
+     *             'index' => 0,
+     *         ],
+     *     ],
+     * ];
+     * ```.
+     *
+     * ```stream
+     * data: {'id':'as-rx9g6c5sqp','object':'chat.completion','created':1692253330,'sentence_id':2,'is_end':false,'is_truncated':false,'result':'PHP的语法借鉴吸收C语言、Java和语言的特点，易于一般程序员学习。','need_clear_history':false,'usage':{'prompt_tokens':4,'completion_tokens':35,'total_tokens':87}}
+     *
+     * data: {'id':'as-rx9g6c5sqp','object':'chat.completion','created':1692253331,'sentence_id':3,'is_end':false,'is_truncated':false,'result':'PHP的主要目标是允许网络开发人P也被用于其他很多领域。','need_clear_history':false,'usage':{'prompt_tokens':4,'completion_tokens':35,'total_tokens':122}}
+     * ```
+     *
      * @throws BindingResolutionException
+     * @throws RequestException
      */
     public function ernieBot(array $parameters, ?callable $writer = null): Response
     {
@@ -155,10 +185,7 @@ static function (PendingRequest $pendingRequest) use (&$rowData, $writer): Pendi
                     return $pendingRequest->withOptions([
                         'curl' => [
                             CURLOPT_WRITEFUNCTION => static function ($ch, string $data) use (&$rowData, $writer): int {
-                                if (! str($data)->startsWith('data: [DONE]')) {
-                                    $rowData = $data;
-                                }
-
+                                $rowData = $data;
                                 $writer($data, $ch);
 
                                 return \strlen($data);

config/ai-commit.php

@@ -127,6 +127,20 @@
                 // 'user_id' => Illuminate\Support\Str::uuid()->toString(),
             ],
         ],
+        'ernie_bot_turbo' => [
+            'driver' => 'ernie',
+            'http_options' => [],
+            'api_key' => env('ERNIE_API_KEY', '...'),
+            'secret_key' => env('ERNIE_SECRET_KEY', '...'),
+            'parameters' => [
+                // 'messages' => 'required|array',
+                'temperature' => 0.95,
+                'top_p' => 0.8,
+                'penalty_score' => 1.0,
+                'stream' => true,
+                // 'user_id' => Illuminate\Support\Str::uuid()->toString(),
+            ],
+        ],
     ],
 
     /**