<?php

namespace Database\Factories;

use App\Models\AiModel;
use App\Models\ApiProvider;
use Illuminate\Database\Eloquent\Factories\Factory;

class AiModelFactory extends Factory
{
    protected $model = AiModel::class;

    public function definition(): array
    {
        return [
            'name' => $this->faker->word() . ' Model',
            'model_id' => $this->faker->uuid(),
            'model_type' => $this->faker->randomElement(['text-to-image', 'image-to-image', 'inpainting']),
            'parameters' => json_encode([
                'steps' => 30,
                'cfg_scale' => 7.5,
                'sampler' => 'Euler a',
                'width' => 512,
                'height' => 512
            ]),
            'api_provider_id' => ApiProvider::factory(),
        ];
    }
}