add tag assign to documents batch
This commit is contained in:
@@ -28,13 +28,13 @@ final class NdjsonHybridRetriever implements RetrieverInterface
|
|||||||
private readonly NdjsonChunkLookup $lookup,
|
private readonly NdjsonChunkLookup $lookup,
|
||||||
private readonly VectorSearchClient $vectorClient,
|
private readonly VectorSearchClient $vectorClient,
|
||||||
private readonly TagRoutingService $tagRouting,
|
private readonly TagRoutingService $tagRouting,
|
||||||
private readonly int $maxChunks = 3,
|
private readonly int $maxChunks = 100,
|
||||||
private readonly int $vectorTopK = 5,
|
private readonly int $vectorTopK = 100,
|
||||||
) {}
|
) {}
|
||||||
|
|
||||||
public function retrieve(string $prompt, int $limit = null): array
|
public function retrieve(string $prompt, int $limit = null): array
|
||||||
{
|
{
|
||||||
$limit ??= $this->maxChunks;
|
$limit = $this->maxChunks;
|
||||||
|
|
||||||
// ---------------------------------------------------------
|
// ---------------------------------------------------------
|
||||||
// 0) Tag-Routing FIRST (soft gate)
|
// 0) Tag-Routing FIRST (soft gate)
|
||||||
|
|||||||
Reference in New Issue
Block a user