SuperSeller3000/src/Infrastructure/AI/Agent/OllamaVisionAgent.php

<?php

declare(strict_types=1);

namespace App\Infrastructure\AI\Agent;

use App\Infrastructure\AI\OllamaClientInterface;
use App\Infrastructure\AI\PromptTemplateService;

final class OllamaVisionAgent
{
    public function __construct(
        private readonly OllamaClientInterface $ollama,
        private readonly PromptTemplateService $prompts,
        private readonly string $model,
    ) {
    }

    /** @return array{manufacturer: string, modelName: string, modelNumber: string, serial: string} */
    public function analyze(string $imagePath): array
    {
        $prompt = $this->prompts->render('vision_analyze');

        $response = $this->ollama->generateWithImage($this->model, $prompt, $imagePath);

        return [
            'manufacturer' => $this->extractField($response, 'MANUFACTURER'),
            'modelName' => $this->extractField($response, 'MODEL_NAME'),
            'modelNumber' => $this->extractField($response, 'MODEL_NUMBER'),
            'serial' => $this->extractField($response, 'SERIAL'),
        ];
    }

    private function extractField(string $response, string $field): string
    {
        if (preg_match('/^'.$field.':\s*(.+)$/m', $response, $matches)) {
            return trim($matches[1]);
        }

        return '';
    }
}