fix missing mock

LlmLaraHub · Aug 12, 2024 · 0c4ecbd · 0c4ecbd
1 parent 86dd777
commit 0c4ecbd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/Feature/Models/MessageTest.php b/tests/Feature/Models/MessageTest.php
@@ -81,7 +81,7 @@ public function test_run(): void
             'content' => 'test',
         ]);
 
-        LlmDriverFacade::shouldReceive('driver->chat')->once()->andReturn($firstResponse);
+        LlmDriverFacade::shouldReceive('driver->setToolType->chat')->once()->andReturn($firstResponse);
 
         $message = Message::factory()->user()->create([
             'meta_data' => MetaDataDto::from([
@@ -110,7 +110,7 @@ public function test_rerun(): void
             'content' => 'test',
         ]);
 
-        LlmDriverFacade::shouldReceive('driver->chat')->once()->andReturn($firstResponse);
+        LlmDriverFacade::shouldReceive('driver->setToolType->chat')->once()->andReturn($firstResponse);
 
         $message = Message::factory()->user()->create([
             'chat_id' => $chat->id,