127 lines
4.5 KiB
PHP
127 lines
4.5 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
use App\Jobs\GenerateArtworkEmbeddingJob;
|
|
use App\Models\Artwork;
|
|
use App\Models\ArtworkEmbedding;
|
|
use App\Models\Category;
|
|
use App\Models\ContentType;
|
|
use App\Models\Tag;
|
|
use Illuminate\Foundation\Testing\RefreshDatabase;
|
|
use Illuminate\Support\Facades\Http;
|
|
|
|
uses(RefreshDatabase::class);
|
|
|
|
it('persists a normalized embedding and upserts the artwork to the vector gateway', function () {
|
|
config()->set('recommendations.embedding.enabled', true);
|
|
config()->set('recommendations.embedding.endpoint', '/embed');
|
|
config()->set('recommendations.embedding.min_dim', 2);
|
|
config()->set('vision.clip.base_url', 'https://clip.local');
|
|
config()->set('vision.vector_gateway.enabled', true);
|
|
config()->set('vision.vector_gateway.base_url', 'https://vision.local');
|
|
config()->set('vision.vector_gateway.api_key', 'test-key');
|
|
config()->set('vision.vector_gateway.upsert_file_endpoint', '/vectors/upsert/file');
|
|
config()->set('cdn.files_url', 'https://files.local');
|
|
|
|
Http::fake([
|
|
'https://clip.local/embed' => Http::response([
|
|
'embedding' => [3.0, 4.0],
|
|
], 200),
|
|
'https://files.local/*' => Http::response('fake-image-bytes', 200),
|
|
'https://vision.local/vectors/upsert/file' => Http::response([
|
|
'status' => 'ok',
|
|
], 200),
|
|
]);
|
|
|
|
$contentType = ContentType::query()->create([
|
|
'name' => 'Wallpapers',
|
|
'slug' => 'wallpapers',
|
|
'description' => '',
|
|
]);
|
|
|
|
$category = Category::query()->create([
|
|
'content_type_id' => $contentType->id,
|
|
'parent_id' => null,
|
|
'name' => 'Abstract',
|
|
'slug' => 'abstract',
|
|
'description' => '',
|
|
'image' => null,
|
|
'is_active' => true,
|
|
'sort_order' => 10,
|
|
]);
|
|
|
|
$artwork = Artwork::factory()->create([
|
|
'hash' => 'aabbccddeeff1122',
|
|
'thumb_ext' => 'webp',
|
|
'is_public' => true,
|
|
'is_approved' => true,
|
|
'published_at' => now()->subDay(),
|
|
]);
|
|
$artwork->categories()->attach($category->id);
|
|
$tag = Tag::query()->create(['name' => 'Neon', 'slug' => 'neon']);
|
|
$artwork->tags()->attach($tag->id, ['source' => 'ai', 'confidence' => 0.91]);
|
|
|
|
$job = new GenerateArtworkEmbeddingJob($artwork->id, 'aabbccddeeff1122');
|
|
app()->call([$job, 'handle']);
|
|
|
|
$embedding = ArtworkEmbedding::query()->where('artwork_id', $artwork->id)->first();
|
|
$artwork->refresh();
|
|
|
|
expect($embedding)->not->toBeNull()
|
|
->and($embedding?->dim)->toBe(2)
|
|
->and($embedding?->is_normalized)->toBeTrue()
|
|
->and($artwork->last_vector_indexed_at)->not->toBeNull();
|
|
|
|
$vector = json_decode((string) $embedding?->embedding_json, true, 512, JSON_THROW_ON_ERROR);
|
|
|
|
expect(round((float) $vector[0], 4))->toBe(0.6)
|
|
->and(round((float) $vector[1], 4))->toBe(0.8);
|
|
|
|
Http::assertSent(function (\Illuminate\Http\Client\Request $request): bool {
|
|
return str_contains($request->url(), 'vision.local/vectors/upsert');
|
|
});
|
|
|
|
Http::assertSentCount(3);
|
|
});
|
|
|
|
it('keeps the local embedding when vector upsert fails', function () {
|
|
config()->set('recommendations.embedding.enabled', true);
|
|
config()->set('recommendations.embedding.endpoint', '/embed');
|
|
config()->set('recommendations.embedding.min_dim', 2);
|
|
config()->set('vision.clip.base_url', 'https://clip.local');
|
|
config()->set('vision.vector_gateway.enabled', true);
|
|
config()->set('vision.vector_gateway.base_url', 'https://vision.local');
|
|
config()->set('vision.vector_gateway.api_key', 'test-key');
|
|
config()->set('vision.vector_gateway.upsert_file_endpoint', '/vectors/upsert/file');
|
|
config()->set('cdn.files_url', 'https://files.local');
|
|
|
|
Http::fake([
|
|
'https://clip.local/embed' => Http::response([
|
|
'embedding' => [1.0, 2.0, 2.0],
|
|
], 200),
|
|
'https://files.local/*' => Http::response('fake-image-bytes', 200),
|
|
'https://vision.local/vectors/upsert/file' => Http::response([
|
|
'message' => 'gateway error',
|
|
], 500),
|
|
]);
|
|
|
|
$artwork = Artwork::factory()->create([
|
|
'hash' => '1122334455667788',
|
|
'thumb_ext' => 'webp',
|
|
'is_public' => true,
|
|
'is_approved' => true,
|
|
'published_at' => now()->subDay(),
|
|
]);
|
|
|
|
$job = new GenerateArtworkEmbeddingJob($artwork->id, '1122334455667788');
|
|
app()->call([$job, 'handle']);
|
|
|
|
$artwork->refresh();
|
|
|
|
expect(ArtworkEmbedding::query()->where('artwork_id', $artwork->id)->exists())->toBeTrue()
|
|
->and($artwork->last_vector_indexed_at)->toBeNull();
|
|
|
|
Http::assertSentCount(3);
|
|
});
|