117 lines
5.1 KiB
PHP
117 lines
5.1 KiB
PHP
<?php
|
|
|
|
use App\Actions\Embeddings\RecordEmbeddingUsageEvent;
|
|
use App\Concerns\InteractsWithEmbeddingWorkbench;
|
|
use App\Services\EmbeddingWorkbench;
|
|
use Livewire\Attributes\Title;
|
|
use Livewire\Component;
|
|
|
|
new #[Title('Embedding search')] class extends Component
|
|
{
|
|
use InteractsWithEmbeddingWorkbench;
|
|
|
|
public string $similarityQuery = '';
|
|
|
|
public int $similarityLimit = 10;
|
|
|
|
public string $minimumSimilarity = '0.30';
|
|
|
|
/** @var array<int, array<string, mixed>> */
|
|
public array $similarityResults = [];
|
|
|
|
public function mount(EmbeddingWorkbench $workbench): void
|
|
{
|
|
$this->initializeWorkbench($workbench);
|
|
}
|
|
|
|
public function searchSimilar(EmbeddingWorkbench $workbench, RecordEmbeddingUsageEvent $recordUsageEvent): void
|
|
{
|
|
if (! $this->ensureWorkbenchAvailable($workbench, 'similarityQuery')) {
|
|
return;
|
|
}
|
|
|
|
$this->validate([
|
|
'similarityQuery' => ['required', 'string', 'max:1000'],
|
|
'similarityLimit' => ['required', 'integer', 'min:1', 'max:50'],
|
|
'minimumSimilarity' => ['required', 'numeric', 'min:0', 'max:1'],
|
|
'embeddingModel' => ['required', 'string'],
|
|
]);
|
|
|
|
try {
|
|
$this->similarityResults = $workbench
|
|
->search($this->similarityQuery, $this->similarityLimit, (float) $this->minimumSimilarity, $this->embeddingModel)
|
|
->map(function (array $result) use ($workbench): array {
|
|
return [
|
|
...$workbench->presentEntry($result['entry']),
|
|
'similarity' => $result['similarity'],
|
|
'distance' => $result['distance'],
|
|
];
|
|
})
|
|
->all();
|
|
|
|
$recordUsageEvent->handle(
|
|
user: auth()->user(),
|
|
workbench: $workbench,
|
|
selectedModel: $this->embeddingModel,
|
|
tool: 'search',
|
|
inputCount: 1,
|
|
resultCount: count($this->similarityResults),
|
|
);
|
|
|
|
$this->refreshSummary($workbench);
|
|
} catch (\Throwable $exception) {
|
|
$this->addError('embeddingModel', $exception->getMessage());
|
|
}
|
|
}
|
|
};
|
|
?>
|
|
|
|
<section class="flex w-full flex-col gap-6">
|
|
@include('pages.embeddings.partials.navigation', ['heading' => __('Similarity search')])
|
|
|
|
<section class="rounded-lg border border-zinc-200 bg-white p-5 dark:border-zinc-700 dark:bg-zinc-900">
|
|
<form wire:submit="searchSimilar" class="flex flex-col gap-4">
|
|
<flux:field>
|
|
<flux:label>{{ __('Query') }}</flux:label>
|
|
<flux:textarea wire:model="similarityQuery" rows="4" placeholder="{{ __('semantic search') }}" />
|
|
<flux:error name="similarityQuery" />
|
|
</flux:field>
|
|
|
|
<div class="grid gap-3 sm:grid-cols-2">
|
|
<flux:input wire:model="similarityLimit" :label="__('Limit')" type="number" min="1" max="50" />
|
|
<flux:input wire:model="minimumSimilarity" :label="__('Minimum similarity')" type="number" min="0" max="1" step="0.05" />
|
|
</div>
|
|
|
|
<div>
|
|
<flux:button type="submit" variant="primary" icon="magnifying-glass" :disabled="! $databaseReady || ! $ollamaAvailable" class="data-loading:opacity-60">
|
|
<span wire:loading.remove wire:target="searchSimilar">{{ __('Search') }}</span>
|
|
<span wire:loading wire:target="searchSimilar">{{ __('Searching') }}</span>
|
|
</flux:button>
|
|
</div>
|
|
|
|
@if ($similarityResults !== [])
|
|
<div class="overflow-x-auto rounded-lg border border-zinc-200 dark:border-zinc-700">
|
|
<table class="min-w-full divide-y divide-zinc-200 text-sm dark:divide-zinc-700">
|
|
<thead class="bg-zinc-50 text-zinc-600 dark:bg-zinc-800 dark:text-zinc-300">
|
|
<tr>
|
|
<th class="px-3 py-2 text-start font-medium">{{ __('Phrase') }}</th>
|
|
<th class="px-3 py-2 text-start font-medium">{{ __('Similarity') }}</th>
|
|
<th class="px-3 py-2 text-start font-medium">{{ __('Distance') }}</th>
|
|
</tr>
|
|
</thead>
|
|
<tbody class="divide-y divide-zinc-200 dark:divide-zinc-700">
|
|
@foreach ($similarityResults as $result)
|
|
<tr wire:key="similarity-result-{{ $result['id'] }}">
|
|
<td class="px-3 py-3 font-medium text-zinc-900 dark:text-zinc-100">{{ $result['source_text'] }}</td>
|
|
<td class="px-3 py-3 text-emerald-700 dark:text-emerald-300">{{ number_format($result['similarity'], 3) }}</td>
|
|
<td class="px-3 py-3 text-zinc-600 dark:text-zinc-300">{{ number_format($result['distance'], 3) }}</td>
|
|
</tr>
|
|
@endforeach
|
|
</tbody>
|
|
</table>
|
|
</div>
|
|
@endif
|
|
</form>
|
|
</section>
|
|
</section>
|