this should do it but I made a few changes I might convert back

LlmLaraHub · May 8, 2024 · f36ed97 · f36ed97
1 parent a35e327
commit f36ed97
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Modules/LlmDriver/app/Functions/SearchAndSummarize.php b/Modules/LlmDriver/app/Functions/SearchAndSummarize.php
@@ -145,7 +145,7 @@ public function handle(
         $message = $model->getChat()->addInput($response->response, RoleEnum::Assistant);
 
         $this->saveDocumentReference($message, $documentChunkResults);
-        
+
         notify_ui($model->getChat(), 'Complete');
 
         return FunctionResponse::from(

diff --git a/tests/Feature/SearchAndSummarizeChatRepoTest.php b/tests/Feature/SearchAndSummarizeChatRepoTest.php
@@ -24,7 +24,7 @@ public function test_can_search(): void
         $data = 'Foo bar';
         $dto = new \LlmLaraHub\LlmDriver\Responses\CompletionResponse($data);
 
-        LlmDriverFacade::shouldReceive('driver->chat')
+        LlmDriverFacade::shouldReceive('driver->completion')
             ->once()
             ->andReturn($dto);
 

diff --git a/tests/Feature/SearchAndSummarizeTest.php b/tests/Feature/SearchAndSummarizeTest.php
@@ -54,7 +54,7 @@ public function test_gets_user_input()
             ]),
         ]);
 
-        LlmDriverFacade::shouldReceive('driver->chat')
+        LlmDriverFacade::shouldReceive('driver->completion')
             ->once()
             ->andReturn($dto);
-Original file line number
+Diff line change
@@ Expand Up / @@ -54,7 +54,7 @@ public function test_gets_user_input() @@
                 ]),
             ]);
-            LlmDriverFacade::shouldReceive('driver->chat')
+            LlmDriverFacade::shouldReceive('driver->completion')
                 ->once()
                 ->andReturn($dto);
@@ Expand Down @@