Make sure the controller creates meta data

LlmLaraHub · Jul 3, 2024 · f6693ea · f6693ea
1 parent b6ce75b
commit f6693ea
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 1 deletion.
diff --git a/app/Http/Controllers/ChatController.php b/app/Http/Controllers/ChatController.php
@@ -89,7 +89,8 @@ public function chat(Chat $chat)
             $message = $chat->addInput(
                 message: $input,
                 role: RoleEnum::User,
-                show_in_thread: true);
+                show_in_thread: true,
+                meta_data: $meta_data);
 
             $messagesArray = [];
 

diff --git a/app/Models/Message.php b/app/Models/Message.php
@@ -19,6 +19,7 @@ class Message extends Model
         'role',
         'in_out',
         'is_chat_ignored',
+        'meta_data',
     ];
 
     protected $dispatchesEvents = [

diff --git a/tests/Feature/Http/Controllers/ChatControllerTest.php b/tests/Feature/Http/Controllers/ChatControllerTest.php
@@ -3,6 +3,7 @@
 namespace Tests\Feature\Http\Controllers;
 
 use App\Domains\Agents\VerifyPromptOutputDto;
+use App\Domains\Chat\DateRangesEnum;
 use App\Domains\Messages\RoleEnum;
 use App\Models\Chat;
 use App\Models\Collection;
@@ -79,6 +80,35 @@ public function test_will_verify_on_completion(): void
         $this->assertEquals('test', $message->body);
     }
 
+    public function test_adds_meta_data_and_date_range()
+    {
+        $user = User::factory()->create();
+        $collection = Collection::factory()->create();
+        $chat = Chat::factory()->create([
+            'chatable_id' => $collection->id,
+            'chatable_type' => Collection::class,
+            'user_id' => $user->id,
+        ]);
+
+        Orchestrate::shouldReceive('handle')->once()->andReturn('Yo');
+
+        $this->assertDatabaseCount('messages', 0);
+        $this->actingAs($user)->post(route('chats.messages.create', [
+            'chat' => $chat->id,
+        ]),
+            [
+                'system_prompt' => 'Foo',
+                'input' => 'user input',
+                'date_range' => DateRangesEnum::ThisWeek->value,
+            ])->assertOk();
+
+
+        $this->assertDatabaseCount('messages', 1);
+
+        $message = Message::first();
+        $this->assertEquals("this_week", $message->meta_data->date_range);
+    }
+
     public function test_a_function_based_chat()
     {
         $user = User::factory()->create();