f1b471a219* feat(server): add bidirectional client communication support * fix(server): use correct session ID string conversion method * fix: clean up docblocks for request handler methods in Builder * fix: merge artifacts
92 lines
3.1 KiB
PHP
92 lines
3.1 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the official PHP MCP SDK.
|
|
*
|
|
* A collaboration between Symfony and the PHP Foundation.
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Mcp\Server\Handler\Request;
|
|
|
|
use Mcp\Capability\Completion\ProviderInterface;
|
|
use Mcp\Capability\Registry\ReferenceProviderInterface;
|
|
use Mcp\Schema\JsonRpc\Error;
|
|
use Mcp\Schema\JsonRpc\Request;
|
|
use Mcp\Schema\JsonRpc\Response;
|
|
use Mcp\Schema\Request\CompletionCompleteRequest;
|
|
use Mcp\Schema\Result\CompletionCompleteResult;
|
|
use Mcp\Server\Session\SessionInterface;
|
|
use Psr\Container\ContainerInterface;
|
|
|
|
/**
|
|
* Handles completion/complete requests.
|
|
*
|
|
* @implements RequestHandlerInterface<CompletionCompleteResult>
|
|
*
|
|
* @author Kyrian Obikwelu <koshnawaza@gmail.com>
|
|
*/
|
|
final class CompletionCompleteHandler implements RequestHandlerInterface
|
|
{
|
|
public function __construct(
|
|
private readonly ReferenceProviderInterface $referenceProvider,
|
|
private readonly ?ContainerInterface $container = null,
|
|
) {
|
|
}
|
|
|
|
public function supports(Request $request): bool
|
|
{
|
|
return $request instanceof CompletionCompleteRequest;
|
|
}
|
|
|
|
/**
|
|
* @return Response<CompletionCompleteResult>|Error
|
|
*/
|
|
public function handle(Request $request, SessionInterface $session): Response|Error
|
|
{
|
|
\assert($request instanceof CompletionCompleteRequest);
|
|
|
|
$name = $request->argument['name'] ?? '';
|
|
$value = $request->argument['value'] ?? '';
|
|
|
|
$reference = match ($request->ref->type) {
|
|
'ref/prompt' => $this->referenceProvider->getPrompt($request->ref->name),
|
|
'ref/resource' => $this->referenceProvider->getResourceTemplate($request->ref->uri),
|
|
default => null,
|
|
};
|
|
|
|
if (null === $reference) {
|
|
return new Response($request->getId(), new CompletionCompleteResult([]));
|
|
}
|
|
|
|
$providers = $reference->completionProviders;
|
|
$provider = $providers[$name] ?? null;
|
|
if (null === $provider) {
|
|
return new Response($request->getId(), new CompletionCompleteResult([]));
|
|
}
|
|
|
|
if (\is_string($provider)) {
|
|
if (!class_exists($provider)) {
|
|
return Error::forInternalError('Invalid completion provider', $request->getId());
|
|
}
|
|
$provider = $this->container?->has($provider) ? $this->container->get($provider) : new $provider();
|
|
}
|
|
|
|
if (!$provider instanceof ProviderInterface) {
|
|
return Error::forInternalError('Invalid completion provider type', $request->getId());
|
|
}
|
|
|
|
try {
|
|
$completions = $provider->getCompletions($value);
|
|
$total = \count($completions);
|
|
$hasMore = $total > 100;
|
|
$paged = \array_slice($completions, 0, 100);
|
|
|
|
return new Response($request->getId(), new CompletionCompleteResult($paged, $total, $hasMore));
|
|
} catch (\Throwable) {
|
|
return Error::forInternalError('Error while handling completion request', $request->getId());
|
|
}
|
|
}
|
|
}
|