Migrate to laravel ai sdk (#181)

Also * upgrade to livewire 4 * replace rappsoft tables with filament
2026-03-13 15:21:22 -05:00
parent fc6b7a8c52
commit 66889abc72
57 changed files with 4064 additions and 1761 deletions
@@ -0,0 +1,182 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Tests\Feature;
+
+use App\Ai\Agents\ChatWithPortfolioAgent;
+use App\Ai\Agents\ChatWithSuggestedPromptsAgent;
+use App\Models\ChatWithConversation;
+use App\Models\Portfolio;
+use App\Models\User;
+use Illuminate\Foundation\Testing\RefreshDatabase;
+use Illuminate\Support\Facades\DB;
+use Illuminate\Support\Facades\RateLimiter;
+use Illuminate\Support\Str;
+use Livewire\Volt\Volt;
+use Tests\TestCase;
+
+class ChatWithTest extends TestCase
+{
+    use RefreshDatabase;
+
+    public function test_mount_creates_chat_with_conversation_record(): void
+    {
+        ChatWithPortfolioAgent::fake(['Test response']);
+        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);
+
+        $this->actingAs($user = User::factory()->create());
+        $portfolio = Portfolio::factory()->create();
+
+        $this->assertDatabaseCount('agent_conversations', 0);
+
+        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio]);
+
+        $this->assertDatabaseHas('agent_conversations', [
+            'chatable_type' => Portfolio::class,
+            'chatable_id' => $portfolio->id,
+            'user_id' => $user->id,
+        ]);
+    }
+
+    public function test_mount_reuses_existing_chat_with_conversation(): void
+    {
+        ChatWithPortfolioAgent::fake(['Test response']);
+        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);
+
+        $this->actingAs(User::factory()->create());
+        $portfolio = Portfolio::factory()->create();
+
+        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio]);
+        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio]);
+
+        $this->assertDatabaseCount('agent_conversations', 1);
+    }
+
+    public function test_mount_loads_existing_messages(): void
+    {
+        ChatWithPortfolioAgent::fake(['Test response']);
+        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);
+
+        $this->actingAs($user = User::factory()->create());
+        $portfolio = Portfolio::factory()->create();
+
+        $conversation = ChatWithConversation::create([
+            'chatable_type' => Portfolio::class,
+            'chatable_id' => $portfolio->id,
+            'user_id' => $user->id,
+            'title' => 'Chat with investments',
+        ]);
+
+        DB::table('agent_conversation_messages')->insert([
+            'id' => (string) Str::uuid7(),
+            'conversation_id' => $conversation->id,
+            'user_id' => $user->id,
+            'agent' => ChatWithPortfolioAgent::class,
+            'role' => 'user',
+            'content' => 'Previous question',
+            'attachments' => '[]',
+            'tool_calls' => '[]',
+            'tool_results' => '[]',
+            'usage' => '[]',
+            'meta' => '[]',
+            'created_at' => now(),
+            'updated_at' => now(),
+        ]);
+
+        $component = Volt::test('ui.ai-chat-window', ['chatable' => $portfolio]);
+
+        $messages = $component->get('messages');
+        $this->assertCount(1, $messages);
+        $this->assertEquals('user', $messages[0]['role']);
+        $this->assertEquals('Previous question', $messages[0]['content']);
+    }
+
+    public function test_start_completion_adds_user_message_and_sets_streaming(): void
+    {
+        ChatWithPortfolioAgent::fake(['The portfolio looks great!']);
+        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);
+
+        $this->actingAs(User::factory()->create());
+        $portfolio = Portfolio::factory()->create();
+
+        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
+            ->set('prompt', 'How is my portfolio doing?')
+            ->call('startCompletion')
+            ->assertSet('streaming', true)
+            ->assertSee('How is my portfolio doing?');
+    }
+
+    public function test_generate_completion_calls_agent_and_stores_response(): void
+    {
+        ChatWithPortfolioAgent::fake(['The portfolio looks great!']);
+        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);
+
+        $this->actingAs(User::factory()->create());
+        $portfolio = Portfolio::factory()->create();
+
+        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
+            ->set('messages', [['role' => 'user', 'content' => 'How is my portfolio doing?', 'created_at' => now()]])
+            ->call('generateCompletion')
+            ->assertSet('streaming', false);
+
+        ChatWithPortfolioAgent::assertPrompted('How is my portfolio doing?');
+
+        $this->assertDatabaseHas('agent_conversation_messages', [
+            'role' => 'assistant',
+            'content' => 'The portfolio looks great!',
+        ]);
+    }
+
+    public function test_empty_prompt_returns_guidance_without_calling_agent(): void
+    {
+        ChatWithPortfolioAgent::fake(['Test response']);
+        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);
+
+        $this->actingAs(User::factory()->create());
+        $portfolio = Portfolio::factory()->create();
+
+        $component = Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
+            ->set('prompt', '   ')
+            ->call('startCompletion');
+
+        $lastMessage = collect($component->get('messages'))->last()['content'];
+        $this->assertStringContainsString('Feel free to ask me a question!', $lastMessage);
+        ChatWithPortfolioAgent::assertNeverPrompted();
+    }
+
+    public function test_rate_limiting_blocks_excessive_requests(): void
+    {
+        ChatWithPortfolioAgent::fake(['Test response']);
+        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);
+
+        $this->actingAs($user = User::factory()->create());
+        $portfolio = Portfolio::factory()->create();
+
+        for ($i = 0; $i < 20; $i++) {
+            RateLimiter::hit($user->id.'/'.$portfolio->id, 60);
+        }
+
+        $component = Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
+            ->set('prompt', 'Am I rate limited?')
+            ->call('startCompletion');
+
+        $lastMessage = collect($component->get('messages'))->last()['content'];
+        $this->assertStringContainsString("Hang on! You're doing that too much.", $lastMessage);
+        ChatWithPortfolioAgent::assertNeverPrompted();
+    }
+
+    public function test_suggested_prompt_is_used_as_prompt(): void
+    {
+        ChatWithPortfolioAgent::fake(['Here is the best holding!']);
+        ChatWithSuggestedPromptsAgent::fake([['suggested_prompts' => []]]);
+
+        $this->actingAs(User::factory()->create());
+        $portfolio = Portfolio::factory()->create();
+
+        Volt::test('ui.ai-chat-window', ['chatable' => $portfolio])
+            ->call('startCompletion', 'Which holding is most successful in this portfolio?')
+            ->assertSet('streaming', true)
+            ->assertSee('Which holding is most successful in this portfolio?');
+    }
+}