fix the double toasts

and start adding back chat over completion

Author: alnutile

Modules/LlmDriver/app/Functions/SearchAndSummarize.php

@@ -6,6 +6,8 @@
 use App\Domains\Agents\VerifyPromptOutputDto;
 use App\Domains\Messages\RoleEnum;
 use App\Domains\Prompts\SummarizePrompt;
+use App\Models\Chat;
+use App\Models\Message;
 use App\Models\PromptHistory;
 use Facades\App\Domains\Agents\VerifyResponseAgent;
 use Facades\LlmLaraHub\LlmDriver\DistanceQuery;
@@ -108,10 +110,28 @@ public function handle(
 
         notify_ui($model, 'Building Summary');
 
-        /** @var CompletionResponse $response */
-        $response = LlmDriverFacade::driver(
-            $model->getChatable()->getDriver()
-        )->completion($contentFlattened);
+
+        if(!get_class($model) === Chat::class) {
+            Log::info('[LaraChain] Using the Simple Completion', [
+                'input' => $contentFlattened,
+                'driver' => $model->getChatable()->getDriver(),
+            ]);
+            /** @var CompletionResponse $response */
+            $response = LlmDriverFacade::driver(
+                $model->getChatable()->getDriver()
+            )->completion($contentFlattened);
+        } else {
+            Log::info('[LaraChain] Using the Chat Completion', [
+                'input' => $contentFlattened,
+                'driver' => $model->getChatable()->getDriver(),
+            ]);
+            $messages = $model->getChat()->getChatResponse();
+
+            /** @var CompletionResponse $response */
+            $response = LlmDriverFacade::driver(
+                $model->getChatable()->getDriver()
+            )->chat($messages);
+        }
 
         $this->response = $response->content;
 

app/Domains/Messages/SearchAndSummarizeChatRepo.php

@@ -28,7 +28,10 @@ public function search(Chat $chat,
         string $input,
         ?Filter $filter = null): string
     {
-        Log::info('[LaraChain] Search and Summarize Default Function');
+        Log::info('[LaraChain] Search and Summarize Default Function', [
+            'note' => "Showing input since some system grab the last on the array",
+            'input' => $input
+        ]);
 
         $originalPrompt = $input;
 
@@ -84,14 +87,15 @@ public function search(Chat $chat,
         Log::info('[LaraChain] Getting the Summary', [
             'input' => $contentFlattened,
             'driver' => $chat->chatable->getDriver(),
+            'messages' => count($latestMessagesArray),
         ]);
 
         notify_ui($chat, 'Building Summary');
 
         /** @var CompletionResponse $response */
         $response = LlmDriverFacade::driver(
             $chat->chatable->getDriver()
-        )->completion($contentFlattened);
+        )->chat($latestMessagesArray);
 
         $this->response = $response->content;
 

app/Models/Chat.php

@@ -134,7 +134,7 @@ public function getAiResponse($input)
     public function getChatResponse(int $limit = 5): array
     {
         $latestMessages = $this->messages()
-            ->latest()
+            ->orderBy('id', 'desc')
             ->limit(5)
             ->get();
 

resources/js/Pages/Chat/ChatInputThreaded.vue

@@ -54,6 +54,9 @@ onMounted(() => {
     });
 });
 
+onUnmounted(() => {
+    Echo.leave(`collection.chat.${props.chat.chatable_id}.${props.chat.id}`);
+});
 
 
 const save = () => {

resources/js/Pages/Collection/Chat.vue

@@ -3,7 +3,7 @@ import AppLayout from '@/Layouts/AppLayout.vue';
 import Welcome from '@/Components/Welcome.vue';
 import PrimaryButton from '@/Components/PrimaryButton.vue';
 import SecondaryLink from '@/Components/SecondaryLink.vue';
-import { computed, onMounted, provide, ref } from 'vue';
+import {computed, onMounted, onUnmounted, provide, ref} from 'vue';
 import Nav from '@/Pages/Collection/Components/Nav.vue';
 import CollectionTags from './Components/CollectionTags.vue';
 import { useDropzone } from "vue3-dropzone";
@@ -57,6 +57,11 @@ onMounted(() => {
             });
     });
 });
+
+onUnmounted(() => {
+    Echo.leave(`collection.chat.${props.collection.data.id}.${props.chat.data.id}`);
+});
+
 </script>
 
 <template>

resources/js/Pages/Collection/Components/Documents.vue

@@ -1,5 +1,5 @@
 <script setup>
-import {computed, onMounted, ref} from 'vue';
+import {computed, onMounted, onUnmounted, ref} from 'vue';
 import Tags from '@/Components/Tags.vue';
 import ShowDocument from '@/Pages/Collection/Components/ShowDocument.vue';
 import DocumentReset from '@/Pages/Collection/Components/DocumentReset.vue';
@@ -71,7 +71,9 @@ onMounted(() => {
         });
 });
 
-
+onUnmounted(() => {
+    Echo.leave(`collection.${props.collection.id}`);
+});
 
 </script>
 <template>

tests/Feature/SearchAndSummarizeTest.php

@@ -3,10 +3,12 @@
 namespace Tests\Feature;
 
 use App\Domains\Agents\VerifyPromptOutputDto;
+use App\Domains\Messages\RoleEnum;
 use App\Models\Chat;
 use App\Models\Collection;
 use App\Models\Document;
 use App\Models\DocumentChunk;
+use App\Models\Message;
 use Facades\App\Domains\Agents\VerifyResponseAgent;
 use Facades\LlmLaraHub\LlmDriver\DistanceQuery;
 use LlmLaraHub\LlmDriver\Functions\ParametersDto;
@@ -54,7 +56,7 @@ public function test_gets_user_input()
             ]),
         ]);
 
-        LlmDriverFacade::shouldReceive('driver->completion')
+        LlmDriverFacade::shouldReceive('driver->chat')
             ->once()
             ->andReturn($dto);
 
@@ -75,6 +77,20 @@ public function test_gets_user_input()
             'chatable_id' => $collection->id,
         ]);
 
+        $messageUser = Message::factory()->create([
+            'body' => 'Results Before this one',
+            'role' => RoleEnum::User,
+            'is_chat_ignored' => false,
+            'chat_id'=> $chat->id
+        ]);
+
+        $messageAssistant = Message::factory()->create([
+            'body' => 'Results Before this one',
+            'role' => RoleEnum::Assistant,
+            'is_chat_ignored' => false,
+            'chat_id'=> $chat->id
+        ]);
+
         $document = Document::factory()->create([
             'collection_id' => $collection->id,
         ]);