improve interacting with ollama api and models

resources/views/pages/embeddings/⚡embed.blade.php

@@ -1,32 +1,23 @@
 <?php
 
 use App\Actions\Embeddings\RecordEmbeddingUsageEvent;
+use App\Concerns\InteractsWithEmbeddingWorkbench;
 use App\Models\EmbeddingEntry;
 use App\Services\EmbeddingVector;
 use App\Services\EmbeddingWorkbench;
 use Flux\Flux;
-use Illuminate\Support\Facades\Schema;
 use Livewire\Attributes\Title;
 use Livewire\Component;
 
 new #[Title('Embed')] class extends Component
 {
+    use InteractsWithEmbeddingWorkbench;
+
     public string $embeddingInput = '';
 
-    public string $embeddingModel = '';
-
-    /** @var array<int, array{name: string, dimensions: int, default: bool}> */
-    public array $modelOptions = [];
-
     /** @var array<int, array<string, mixed>> */
     public array $embeddingResults = [];
 
-    public string $providerSummary = '';
-
-    public int $storedCount = 0;
-
-    public bool $databaseReady = true;
-
     public function mount(EmbeddingWorkbench $workbench): void
     {
         $this->initializeWorkbench($workbench);
@@ -34,6 +25,10 @@ new #[Title('Embed')] class extends Component
 
     public function generateEmbeddings(EmbeddingWorkbench $workbench, RecordEmbeddingUsageEvent $recordUsageEvent): void
     {
+        if (! $this->ensureWorkbenchAvailable($workbench, 'embeddingInput')) {
+            return;
+        }
+
         $this->validate([
             'embeddingInput' => ['required', 'string', 'max:12000'],
             'embeddingModel' => ['required', 'string'],
@@ -47,12 +42,6 @@ new #[Title('Embed')] class extends Component
             return;
         }
 
-        if (! $this->databaseReady) {
-            $this->addError('embeddingInput', __('The embedding table is not ready.'));
-
-            return;
-        }
-
         try {
             $entries = $workbench->embed($phrases, $this->embeddingModel);
             $tokens = (int) $entries->sum(fn (EmbeddingEntry $entry): int => (int) ($entry->tokens ?? 0));
@@ -78,37 +67,6 @@ new #[Title('Embed')] class extends Component
             $this->addError('embeddingModel', $exception->getMessage());
         }
     }
-
-    protected function initializeWorkbench(EmbeddingWorkbench $workbench): void
-    {
-        $this->modelOptions = $workbench->modelOptions();
-        $this->embeddingModel = $this->embeddingModel ?: $workbench->modelName();
-
-        $this->refreshSummary($workbench);
-    }
-
-    protected function refreshSummary(EmbeddingWorkbench $workbench): void
-    {
-        $provider = $workbench->providerName();
-        $model = $workbench->modelName($this->embeddingModel);
-        $dimensions = $workbench->dimensions($model);
-
-        $this->providerSummary = "{$provider} / {$model} / {$dimensions}d";
-        $this->databaseReady = Schema::hasTable('embedding_entries');
-
-        if (! $this->databaseReady) {
-            $this->storedCount = 0;
-
-            return;
-        }
-
-        $this->storedCount = EmbeddingEntry::query()
-            ->where('provider', $provider)
-            ->where('model', $model)
-            ->where('embedding_dimensions', $dimensions)
-            ->count();
-    }
-
 };
 ?>
 
@@ -124,7 +82,7 @@ new #[Title('Embed')] class extends Component
             </flux:field>
 
             <div>
-                <flux:button type="submit" variant="primary" icon="sparkles" :disabled="! $databaseReady" class="data-loading:opacity-60">
+                <flux:button type="submit" variant="primary" icon="sparkles" :disabled="! $databaseReady || ! $ollamaAvailable" class="data-loading:opacity-60">
                     <span wire:loading.remove wire:target="generateEmbeddings">{{ __('Generate') }}</span>
                     <span wire:loading wire:target="generateEmbeddings">{{ __('Generating') }}</span>
                 </flux:button>