From cff3dc99b9b8a6f997a05a4902ccae1500a35b13 Mon Sep 17 00:00:00 2001 From: Alfred Nutile Date: Tue, 10 Sep 2024 13:04:39 -0400 Subject: [PATCH] Fix mocking --- tests/Feature/ProjectOrchestrateTest.php | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/Feature/ProjectOrchestrateTest.php b/tests/Feature/ProjectOrchestrateTest.php index dfb6c6ab..9450591b 100644 --- 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)