261 lines
11 KiB
PHP
261 lines
11 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
require dirname(__DIR__) . '/vendor/autoload.php';
|
|
|
|
use app\common\service\DifyChatService;
|
|
|
|
function expectSame($expected, $actual, string $message): void
|
|
{
|
|
if ($expected !== $actual) {
|
|
fwrite(STDERR, "FAIL: {$message}\n");
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
function callPrivate(string $name, array $arguments)
|
|
{
|
|
$method = (new ReflectionClass(DifyChatService::class))->getMethod($name);
|
|
return $method->invoke(null, ...$arguments);
|
|
}
|
|
|
|
$generic = callPrivate('buildRequestSpecs', [
|
|
'https://ai.example.test/v1',
|
|
'model-name',
|
|
['prompt_version' => 'test'],
|
|
'clinical prompt',
|
|
'server-user',
|
|
]);
|
|
expectSame(2, count($generic), 'ambiguous /v1 base should support both protocols');
|
|
expectSame('https://ai.example.test/v1/chat-messages', $generic[0]['url'], 'Dify endpoint');
|
|
expectSame('blocking', $generic[0]['payload']['response_mode'], 'Dify blocking request');
|
|
expectSame('https://ai.example.test/v1/chat/completions', $generic[1]['url'], 'OpenAI endpoint');
|
|
expectSame('model-name', $generic[1]['payload']['model'], 'profile model selection');
|
|
expectSame('clinical prompt', $generic[1]['payload']['messages'][0]['content'], 'OpenAI prompt');
|
|
$serializedSpecs = json_encode($generic, JSON_UNESCAPED_SLASHES) ?: '';
|
|
expectSame(false, str_contains($serializedSpecs, 'api_key'), 'credential field is absent from request bodies');
|
|
expectSame(false, str_contains($serializedSpecs, 'provider'), 'provider override is absent from request bodies');
|
|
expectSame(false, str_contains($serializedSpecs, 'base_url'), 'base URL override is absent from request bodies');
|
|
|
|
// 附件不得让 OpenAI-compatible 回退失效。历史实现遇到报告/录像等非图片附件时只返回
|
|
// Dify 一种协议，网关 404 会直接变成“模型未能处理本次请求”，开处方因此必失败。
|
|
$attachments = [
|
|
['type' => 'document', 'transfer_method' => 'remote_url', 'url' => 'https://cdn.example.test/report.pdf'],
|
|
['type' => 'video', 'transfer_method' => 'remote_url', 'url' => 'https://cdn.example.test/call.mp4'],
|
|
['type' => 'image', 'transfer_method' => 'remote_url', 'url' => 'https://cdn.example.test/tongue.jpg'],
|
|
];
|
|
$withFiles = callPrivate('buildRequestSpecs', [
|
|
'https://ai.example.test/v1',
|
|
'model-name',
|
|
[],
|
|
'clinical prompt',
|
|
'server-user',
|
|
false,
|
|
$attachments,
|
|
]);
|
|
expectSame(2, count($withFiles), 'non-image attachments must keep the OpenAI fallback available');
|
|
expectSame($attachments, $withFiles[0]['payload']['files'], 'Dify still receives every attachment');
|
|
expectSame(
|
|
'clinical prompt',
|
|
$withFiles[0]['payload']['query'],
|
|
'Dify carries attachments in the file channel, not as a manifest'
|
|
);
|
|
$openAiContent = $withFiles[1]['payload']['messages'][0]['content'];
|
|
expectSame(true, is_array($openAiContent), 'OpenAI content becomes multimodal when attachments exist');
|
|
expectSame('text', $openAiContent[0]['type'], 'attachment manifest travels in the text part');
|
|
expectSame(
|
|
true,
|
|
str_contains($openAiContent[0]['text'], 'https://cdn.example.test/report.pdf'),
|
|
'document attachment is listed instead of being silently dropped'
|
|
);
|
|
expectSame(
|
|
true,
|
|
str_contains($openAiContent[0]['text'], 'https://cdn.example.test/call.mp4'),
|
|
'recording attachment is listed instead of being silently dropped'
|
|
);
|
|
expectSame(
|
|
'https://cdn.example.test/tongue.jpg',
|
|
$openAiContent[1]['image_url']['url'],
|
|
'image attachments stay inline for multimodal reading'
|
|
);
|
|
|
|
$openAi = callPrivate('buildRequestSpecs', [
|
|
'https://ai.example.test/v1/chat/completions',
|
|
'model-name',
|
|
[],
|
|
'prompt',
|
|
'server-user',
|
|
]);
|
|
expectSame(1, count($openAi), 'explicit OpenAI endpoint should not probe Dify');
|
|
expectSame('openai', $openAi[0]['protocol'], 'explicit OpenAI protocol');
|
|
|
|
$openAiWithFiles = callPrivate('buildRequestSpecs', [
|
|
'https://ai.example.test/v1/chat/completions',
|
|
'model-name',
|
|
[],
|
|
'prompt',
|
|
'server-user',
|
|
false,
|
|
$attachments,
|
|
]);
|
|
expectSame(1, count($openAiWithFiles), 'explicit OpenAI endpoint stays OpenAI even with attachments');
|
|
expectSame('openai', $openAiWithFiles[0]['protocol'], 'explicit OpenAI protocol with attachments');
|
|
|
|
$difyWithFiles = callPrivate('buildRequestSpecs', [
|
|
'https://ai.example.test/v1/chat-messages',
|
|
'model-name',
|
|
[],
|
|
'prompt',
|
|
'server-user',
|
|
false,
|
|
$attachments,
|
|
]);
|
|
expectSame(1, count($difyWithFiles), 'explicit Dify endpoint stays Dify with attachments');
|
|
expectSame($attachments, $difyWithFiles[0]['payload']['files'], 'explicit Dify keeps the file channel');
|
|
|
|
$dify = callPrivate('buildRequestSpecs', [
|
|
'https://ai.example.test/v1/chat-messages',
|
|
'model-name',
|
|
[],
|
|
'prompt',
|
|
'server-user',
|
|
]);
|
|
expectSame(1, count($dify), 'explicit Dify endpoint should not probe OpenAI');
|
|
expectSame('dify', $dify[0]['protocol'], 'explicit Dify protocol');
|
|
|
|
expectSame('Dify answer', callPrivate('extractContent', [['answer' => ' Dify answer ']]), 'Dify response');
|
|
expectSame(
|
|
'OpenAI answer',
|
|
callPrivate('extractContent', [['choices' => [['message' => ['content' => ' OpenAI answer ']]]]]),
|
|
'OpenAI response'
|
|
);
|
|
expectSame(
|
|
'multipart answer',
|
|
callPrivate('extractContent', [['choices' => [['message' => ['content' => [
|
|
['type' => 'text', 'text' => 'multipart '],
|
|
['type' => 'text', 'text' => 'answer'],
|
|
]]]]]]),
|
|
'OpenAI multipart response'
|
|
);
|
|
|
|
// 附件数量必须按上游应用的 file_upload.number_limits 截断。Dify 超限时返回
|
|
// 400 invalid_param 并整单拒绝，历史实现会把患者的全部舌象/报告一次性送上去，
|
|
// 导致该患者的每一次 AI 请求都固定失败（UPSTREAM_REJECTED）。
|
|
$manyFiles = [];
|
|
for ($i = 0; $i < 5; $i++) {
|
|
$manyFiles[] = ['type' => 'image', 'transfer_method' => 'remote_url', 'url' => "https://cdn.example.test/tongue{$i}.jpg"];
|
|
}
|
|
for ($i = 0; $i < 4; $i++) {
|
|
$manyFiles[] = ['type' => 'document', 'transfer_method' => 'remote_url', 'url' => "https://cdn.example.test/report{$i}.pdf"];
|
|
}
|
|
$capped = callPrivate('normalizeFiles', [$manyFiles, 3]);
|
|
expectSame(3, count($capped['kept']), 'the total attachment count is capped, not each type');
|
|
expectSame(6, count($capped['dropped']), 'attachments past the cap are recorded, not discarded');
|
|
expectSame(
|
|
'https://cdn.example.test/tongue3.jpg',
|
|
$capped['dropped'][0]['url'],
|
|
'the earliest attachments are the ones kept'
|
|
);
|
|
expectSame(
|
|
['kept' => [], 'dropped' => []],
|
|
callPrivate('normalizeFiles', [[['type' => 'image', 'url' => 'ftp://cdn.example.test/x.jpg']], 3]),
|
|
'non-http attachments are still rejected outright'
|
|
);
|
|
|
|
// 被截断的附件必须出现在提示词清单里，否则模型会把“没看到”当成“没有”。
|
|
$cappedSpecs = callPrivate('buildRequestSpecs', [
|
|
'https://ai.example.test/v1/chat-messages',
|
|
'model-name',
|
|
[],
|
|
'clinical prompt',
|
|
'server-user',
|
|
false,
|
|
$capped['kept'],
|
|
$capped['dropped'],
|
|
]);
|
|
$cappedQuery = $cappedSpecs[0]['payload']['query'];
|
|
expectSame(true, str_contains($cappedQuery, '<ATTACHMENTS_NOT_INLINE>'), 'dropped attachments are declared to the model');
|
|
expectSame(true, str_contains($cappedQuery, 'https://cdn.example.test/tongue3.jpg'), 'dropped attachment URL is listed');
|
|
expectSame(false, str_contains($cappedQuery, 'https://cdn.example.test/tongue0.jpg'), 'delivered attachments are not duplicated in the manifest');
|
|
|
|
// 附件被整体拒绝时必须降级为纯文本重试，而不是让整次问诊失败。
|
|
$plan = callPrivate('buildAttemptPlan', [$capped['kept'], $capped['dropped']]);
|
|
expectSame(2, count($plan), 'a request with attachments gets a text-only fallback attempt');
|
|
expectSame([], $plan[1]['files'], 'the fallback attempt sends no attachments');
|
|
expectSame(9, count($plan[1]['omitted']), 'the fallback attempt declares every attachment');
|
|
expectSame(1, count(callPrivate('buildAttemptPlan', [[], []])), 'a request without attachments is attempted once');
|
|
|
|
expectSame(true, callPrivate('shouldRetryWithoutFiles', [400, $capped['kept']]), 'invalid_param retries without attachments');
|
|
expectSame(true, callPrivate('shouldRetryWithoutFiles', [413, $capped['kept']]), 'oversized attachments retry without attachments');
|
|
expectSame(false, callPrivate('shouldRetryWithoutFiles', [400, []]), 'a text-only rejection is not retried');
|
|
expectSame(false, callPrivate('shouldRetryWithoutFiles', [401, $capped['kept']]), 'a credential failure is not retried');
|
|
expectSame(false, callPrivate('shouldRetryWithoutFiles', [500, $capped['kept']]), 'an upstream outage is not retried here');
|
|
|
|
// 协议回退会把最初的“附件被拒”换成另一协议的状态码（Dify 400 -> OpenAI 404），
|
|
// 降级判断必须按每次响应累计，否则去掉附件的重试永远不会发生。
|
|
expectSame(
|
|
true,
|
|
callPrivate('isFileRejection', [['errno' => 0, 'http_code' => 400], $capped['kept']]),
|
|
'an attachment rejection is recognised on the response that carried it'
|
|
);
|
|
expectSame(
|
|
false,
|
|
callPrivate('isFileRejection', [['errno' => 0, 'http_code' => 404], $capped['kept']]),
|
|
'the fallback protocol 404 is not itself an attachment rejection'
|
|
);
|
|
expectSame(
|
|
true,
|
|
callPrivate('isFileRejection', [
|
|
['errno' => 0, 'http_code' => 200, 'upstream_error' => true],
|
|
$capped['kept'],
|
|
]),
|
|
'a 200 stream carrying event:error counts as an attachment rejection'
|
|
);
|
|
expectSame(
|
|
false,
|
|
callPrivate('isFileRejection', [
|
|
['errno' => 0, 'http_code' => 200, 'upstream_error' => true],
|
|
[],
|
|
]),
|
|
'a text-only request never degrades further'
|
|
);
|
|
expectSame(
|
|
false,
|
|
callPrivate('isFileRejection', [['errno' => 28, 'http_code' => 0], $capped['kept']]),
|
|
'a transport failure is not mistaken for an attachment rejection'
|
|
);
|
|
|
|
// Dify 的 inputs 必须是 JSON 对象。PHP 空数组会被编码成 []，上游以
|
|
// invalid_param 拒绝整单——空 inputs 的调用方会 100% 失败。
|
|
$emptyInputs = callPrivate('buildRequestSpecs', [
|
|
'https://ai.example.test/v1/chat-messages', 'model-name', [], 'prompt', 'server-user',
|
|
]);
|
|
expectSame(
|
|
true,
|
|
str_contains((string) json_encode($emptyInputs[0]['payload']), '"inputs":{}'),
|
|
'empty inputs are encoded as a JSON object, never as an array'
|
|
);
|
|
$filledInputs = callPrivate('buildRequestSpecs', [
|
|
'https://ai.example.test/v1/chat-messages', 'model-name', ['prompt_version' => 'v2'], 'prompt', 'server-user',
|
|
]);
|
|
expectSame(
|
|
true,
|
|
str_contains((string) json_encode($filledInputs[0]['payload']), '"inputs":{"prompt_version":"v2"}'),
|
|
'populated inputs keep their keys'
|
|
);
|
|
|
|
expectSame('invalid_param', callPrivate('cleanUpstreamCode', ['invalid_param']), 'enumerable upstream codes are kept for logs');
|
|
expectSame('', callPrivate('cleanUpstreamCode', ["Run failed: 404 for https://cdn.example.test/a.pdf"]), 'upstream prose never reaches the log');
|
|
|
|
expectSame(true, callPrivate('isValidBaseUrl', ['https://ai.example.test/v1']), 'https URL');
|
|
expectSame(true, callPrivate('isValidBaseUrl', ['http://127.0.0.1:8080/v1']), 'internal http URL');
|
|
expectSame(false, callPrivate('isValidBaseUrl', ['file:///tmp/socket']), 'non-http URL');
|
|
expectSame(false, callPrivate('isValidBaseUrl', ['https://user@example.test/v1']), 'userinfo URL');
|
|
expectSame(false, callPrivate('isValidBaseUrl', ['https://ai.example.test/v1?unsafe=query']), 'query URL');
|
|
expectSame(true, callPrivate('isValidTimeout', [90]), 'normal timeout');
|
|
expectSame(false, callPrivate('isValidTimeout', [0]), 'zero timeout');
|
|
expectSame(false, callPrivate('isValidTimeout', [301]), 'excessive timeout');
|
|
|
|
echo "Prescription AI upstream contract: OK\n";
|