grok.php

<?php

require 'vendor/autoload.php'; // 确保已安装 Guzzle

use GuzzleHttp\Client;

function chat_with_model($content,$token)
{
    $client = new Client([
        'base_uri' => 'https://api.x.ai',
        'stream' => true, // 使用流式响应
        'timeout' => 6000,
    ]);
    $messages = [
        ['role' => 'system', 'content' => $content],
    ];
    $count = 0;
    $response = $client->post('/api/chat', [
        'headers' => [
            'Content-Type' => 'application/json',
            'Authorization' => 'Bearer '.$token
        ],
        'json' => [
            'model' => 'grok-beta',
            'messages' => $messages,
            'stream' => true,
        ],
        'stream' => true,
    ]);
        $buffer = '';
        $body = $response->getBody();

        if (ob_get_level() > 0) {
            ob_flush();
        }
        flush();
        $assistant_response = '';

        while (! $body->eof()) {
            $chunk = $buffer.$body->read(1024); // 每次读取 1024 字节

            if ($chunk) {
                $datas = explode("}\n", $chunk);

                foreach ($datas as $data) {
                    $message_json = json_decode($data.'}', true);

                    if ($message_json) {
                        $message = $message_json['message']['content'] ?? '';
                        echo "data: " . json_encode(['content' => $message]) . "\n\n";
                        $assistant_response .= $message;

                        if (ob_get_level() > 0) {
                            ob_flush();
                        }
                        flush();    // 发送输出到浏览器或终端
                    }
                }

                if ($datas) {
                    $last_match_end = strlen($chunk) - strlen(end($datas));
                    $buffer = substr($chunk, $last_match_end);
                } else {
                    $buffer = $chunk;
                }
            }
        }
        // 将助手响应添加到对话历史
        $messages[] = ['role' => 'assistant', 'content' => $assistant_response];
        $count++;
}
$content = $_GET['content'];
$token = $_GET['token'];
header('Content-Type: text/event-stream');
header('Cache-Control: no-cache');
header('Connection: keep-alive');
chat_with_model($content,$token);