Fix mocking

LlmLaraHub · Sep 10, 2024 · cff3dc9 · cff3dc9
1 parent 85d078d
commit cff3dc9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/Feature/ProjectOrchestrateTest.php b/tests/Feature/ProjectOrchestrateTest.php
@@ -31,13 +31,13 @@ public function test_tools(): void
         $this->assertDatabaseCount('messages', 0);
         $this->assertDatabaseCount('tasks', 0);
 
-        LlmDriverFacade::shouldReceive('driver->setToolType->chat')
+        LlmDriverFacade::shouldReceive('driver->setSystemPrompt->setToolType->chat')
             ->once()
             ->andReturn(
                 CompletionResponse::from($response)
             );
 
-        LlmDriverFacade::shouldReceive('driver->chat')
+        LlmDriverFacade::shouldReceive('driver->setToolType->setSystemPrompt->chat')
             ->once()
             ->andReturn(
                 CompletionResponse::from($response)