49 lines
2.4 KiB
PHP
49 lines
2.4 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace app\common\service {
|
|
class DifyChatService {
|
|
public static array $blocking = [];
|
|
public static bool $emit = false;
|
|
public static int $delayUs = 0;
|
|
public static function streamChat($profile, $inputs, $query, $user, $onDelta, $abort, $files): array {
|
|
if (self::$delayUs) usleep(self::$delayUs);
|
|
if (self::$emit) $onDelta('真实合成模型正文');
|
|
return ['ok' => false, 'error_code' => 'INCOMPLETE_RESPONSE', 'error' => 'fixture'];
|
|
}
|
|
public static function chat($profile, $inputs, $query, $user, $files = [], $options = []): array {
|
|
self::$blocking[] = $options;
|
|
return ['ok' => true, 'content' => '合成模型最终回答'];
|
|
}
|
|
}
|
|
}
|
|
|
|
namespace {
|
|
require dirname(__DIR__) . '/vendor/autoload.php';
|
|
use app\adminapi\logic\tcm\DiagnosisAiLogic;
|
|
use app\adminapi\service\AssistantSseTiming;
|
|
use app\common\service\DifyChatService;
|
|
function retryExpect(bool $ok, string $message): void {
|
|
if (!$ok) throw new \RuntimeException($message);
|
|
}
|
|
$prepared = ['profile' => 'qwen', 'task' => 'summary', 'query' => '合成资料', 'upstream_timeout' => 90];
|
|
DifyChatService::$delayUs = 2000;
|
|
$run = static function (array $request, ?callable $abort = null) {
|
|
return DiagnosisAiLogic::streamPreparedAssistant($request, static fn (string $text): bool => true,
|
|
$abort, new AssistantSseTiming());
|
|
};
|
|
retryExpect($run($prepared) !== null && count(DifyChatService::$blocking) === 1, 'one pre-delta compatibility retry is retained');
|
|
$timeout = DifyChatService::$blocking[0]['timeout'];
|
|
retryExpect($timeout > 0 && $timeout < 90, 'blocking retry only gets the remaining original budget');
|
|
DifyChatService::$blocking = [];
|
|
DifyChatService::$emit = true;
|
|
retryExpect($run($prepared) === null && DifyChatService::$blocking === [], 'no re-submission after real text delivery');
|
|
DifyChatService::$emit = false;
|
|
retryExpect($run($prepared, static fn (): bool => true) === null && DifyChatService::$blocking === [], 'no compatibility request after disconnect');
|
|
DifyChatService::$delayUs = 1100000;
|
|
$prepared['upstream_timeout'] = 1;
|
|
retryExpect($run($prepared) === null && DifyChatService::$blocking === [], 'an exhausted stream cannot start another timeout window');
|
|
echo "Diagnosis AI assistant retry budget: OK\n";
|
|
}
|