this means the mock is working

LlmLaraHub · Mar 26, 2024 · ceac4ba · ceac4ba
1 parent c94b308
commit ceac4ba
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/app/Jobs/VectorlizeDataJob.php b/app/Jobs/VectorlizeDataJob.php
@@ -10,6 +10,7 @@
 use Illuminate\Foundation\Bus\Dispatchable;
 use Illuminate\Queue\InteractsWithQueue;
 use Illuminate\Queue\SerializesModels;
+use App\LlmDriver\Responses\EmbeddingsResponseDto;
 
 class VectorlizeDataJob implements ShouldQueue
 {
@@ -29,10 +30,12 @@ public function __construct(public DocumentChunk $documentChunk)
     public function handle(): void
     {
         $content = $this->documentChunk->content;
-        $results = LlmDriverFacade::embedDara($content);
+
+        /** @var EmbeddingsResponseDto $results */
+        $results = LlmDriverFacade::embedData($content);
 
         $this->documentChunk->update([
-            'embedding' => $results,
+            'embedding' => $results->embedding,
         ]);
     }
 }
diff --git a/tests/Feature/Jobs/VectorlizeDataJobTest.php b/tests/Feature/Jobs/VectorlizeDataJobTest.php
@@ -2,6 +2,8 @@
 
 namespace Tests\Feature\Jobs;
 
+use App\Jobs\VectorlizeDataJob;
+use App\Models\DocumentChunk;
 use Illuminate\Foundation\Testing\RefreshDatabase;
 use Illuminate\Foundation\Testing\WithFaker;
 use Tests\TestCase;
@@ -11,10 +13,15 @@ class VectorlizeDataJobTest extends TestCase
     /**
      * A basic feature test example.
      */
-    public function test_example(): void
+    public function test_gets_data(): void
     {
-        $response = $this->get('/');
+        $documentChunk = DocumentChunk::factory()->create([
+            'embedding' => null
+        ]);
 
-        $response->assertStatus(200);
+        $job = new VectorlizeDataJob($documentChunk);
+        $job->handle();
+
+        $this->assertNotEmpty($documentChunk->embedding);
     }
 }