[Schema][Server] Harden sampling tools against the spec (#420)
* harden sampling tools against the spec Report tool-flow violations as -32602 instead of dropping the request, accept resource_link in tool results, gate tools on sampling.tools, and reject empty content. * normalize keyed content arrays to lists array_filter() and friends preserve keys, and a keyed array serializes as a JSON object instead of the content-block array the schema requires.
This commit is contained in:
parent
becce29c41
commit
43815e38a8
1 changed files with 1212 additions and 167 deletions
|
|
@ -11,8 +11,10 @@
|
|||
|
||||
namespace Mcp\Tests\Unit\Schema\Result;
|
||||
|
||||
use Mcp\Exception\InvalidArgumentException;
|
||||
use Mcp\Schema\Content\TextContent;
|
||||
use Mcp\Schema\Content\ToolUseContent;
|
||||
use Mcp\Schema\Enum\Role;
|
||||
use Mcp\Schema\Result\CreateSamplingMessageResult;
|
||||
use PHPUnit\Framework\TestCase;
|
||||
|
||||
|
|
@ -50,4 +52,77 @@ final class CreateSamplingMessageResultTest extends TestCase
|
|||
$this->assertSame('provider-specific', $result->stopReason);
|
||||
$this->assertSame('provider-specific', $result->jsonSerialize()['stopReason']);
|
||||
}
|
||||
|
||||
public function testKnownStopReasonStaysAString(): void
|
||||
{
|
||||
$result = CreateSamplingMessageResult::fromArray([
|
||||
'role' => 'assistant',
|
||||
'content' => ['type' => 'text', 'text' => 'Done'],
|
||||
'model' => 'test-model',
|
||||
'stopReason' => 'endTurn',
|
||||
]);
|
||||
|
||||
$this->assertSame('endTurn', $result->stopReason);
|
||||
}
|
||||
|
||||
public function testSingleContentBlockKeepsItsShape(): void
|
||||
{
|
||||
$result = CreateSamplingMessageResult::fromArray([
|
||||
'role' => 'assistant',
|
||||
'content' => ['type' => 'text', 'text' => 'Done'],
|
||||
'model' => 'test-model',
|
||||
]);
|
||||
|
||||
$this->assertInstanceOf(TextContent::class, $result->content);
|
||||
$this->assertCount(1, $result->getContentBlocks());
|
||||
$this->assertSame('{"type":"text","text":"Done"}', json_encode($result->jsonSerialize()['content']));
|
||||
}
|
||||
|
||||
public function testFilteredContentStillSerializesAsAnArray(): void
|
||||
{
|
||||
$blocks = [new TextContent('thinking'), new ToolUseContent('call-1', 'weather', [])];
|
||||
|
||||
// array_filter() preserves keys, so this list starts at index 1.
|
||||
$toolUses = array_filter($blocks, static fn ($block): bool => $block instanceof ToolUseContent);
|
||||
$result = new CreateSamplingMessageResult(Role::Assistant, $toolUses, 'test-model');
|
||||
|
||||
$this->assertSame(
|
||||
'[{"type":"tool_use","id":"call-1","name":"weather","input":{}}]',
|
||||
json_encode($result->jsonSerialize()['content']),
|
||||
);
|
||||
}
|
||||
|
||||
public function testNonAssistantRoleIsRejected(): void
|
||||
{
|
||||
$this->expectException(InvalidArgumentException::class);
|
||||
$this->expectExceptionMessage('CreateSamplingMessageResult role must be "assistant".');
|
||||
|
||||
CreateSamplingMessageResult::fromArray([
|
||||
'role' => 'user',
|
||||
'content' => ['type' => 'text', 'text' => 'Done'],
|
||||
'model' => 'test-model',
|
||||
]);
|
||||
}
|
||||
|
||||
public function testEmptyContentIsRejected(): void
|
||||
{
|
||||
$this->expectException(InvalidArgumentException::class);
|
||||
|
||||
CreateSamplingMessageResult::fromArray([
|
||||
'role' => 'assistant',
|
||||
'content' => [],
|
||||
'model' => 'test-model',
|
||||
]);
|
||||
}
|
||||
|
||||
public function testToolResultContentIsRejected(): void
|
||||
{
|
||||
$this->expectException(InvalidArgumentException::class);
|
||||
|
||||
CreateSamplingMessageResult::fromArray([
|
||||
'role' => 'assistant',
|
||||
'content' => ['type' => 'tool_result', 'toolUseId' => 'call-1', 'content' => []],
|
||||
'model' => 'test-model',
|
||||
]);
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue