GoogleSerper.php

<?php

declare(strict_types=1);

namespace Cortex\Tools;

use SensitiveParameter;
use Cortex\LLM\Data\ToolCall;
use Cortex\JsonSchema\SchemaFactory;
use Illuminate\Support\Facades\Http;
use Cortex\JsonSchema\Types\ObjectSchema;

class GoogleSerper extends AbstractTool
{
    public function __construct(
        #[SensitiveParameter]
        protected string $apiKey,
    ) {}

    public function name(): string
    {
        return 'web_search';
    }

    public function description(): string
    {
        return 'Useful for when you need to search the web for information.';
    }

    public function schema(): ObjectSchema
    {
        return SchemaFactory::object()
            ->properties(
                SchemaFactory::string('query')->description('The search query.'),
            );
    }

    /**
     * @param ToolCall|array<int, mixed> $toolCall
     */
    public function invoke(ToolCall|array $toolCall = []): string
    {
        $arguments = $this->getArguments($toolCall);
        $searchQuery = $arguments['query'];

        $response = Http::post('https://google.serper.dev/search', [
            'q' => $searchQuery,
        ])
            ->withHeader('x-api-key', $this->apiKey)
            ->withHeader('Content-Type', 'application/json');

        // @phpstan-ignore method.notFound
        return $response->collect('knowledgeGraph')->toJson();
    }
}