<?php

namespace Tests\Fakes;

use App\Services\Llm\LlmClientInterface;

class FakeLlmClient implements LlmClientInterface
{
    /** @var array<string, array<int, float>> */
    public array $embeddings = [];

    /** @var array<int, string> */
    public array $responses = [];

    /** @var array<int, array{prompt:string, options:array}> */
    public array $generatedPrompts = [];

    public function embed(string $text): array
    {
        return $this->embeddings[$text] ?? [0.1, 0.2, 0.3];
    }

    public function generate(string $prompt, array $options = []): string
    {
        $this->generatedPrompts[] = [
            'prompt' => $prompt,
            'options' => $options,
        ];

        return array_shift($this->responses) ?? '{}';
    }
}