symfony/archived-ai-store
1
0
Fork 0
mirror of https://github.com/symfony/ai-store.git synced 2026-03-23 23:22:17 +01:00
Code Issues Packages Projects Releases Wiki Activity

[Store] Split query string into words for TextQuery and HybridQuery in Retriever

Browse Source
This commit is contained in:
Johannes Wachter
2026-03-17 21:42:47 +01:00
parent 174012d169
commit 6a45db5ef5
2 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/Retriever.php
View File

@@ -113,19 +113,19 @@ final class Retriever implements RetrieverInterface
if (null === $this->vectorizer) {
$this->logger->debug('No vectorizer configured, using TextQuery if supported');
return new TextQuery($query);
return new TextQuery(explode(' ', $query));
}
if (!$this->store->supports(VectorQuery::class)) {
$this->logger->debug('Store does not support vector queries, falling back to TextQuery');
return new TextQuery($query);
return new TextQuery(explode(' ', $query));
}
if ($this->store->supports(HybridQuery::class)) {
$this->logger->debug('Store supports hybrid queries, using HybridQuery with semantic ratio', ['semanticRatio' => $options['semanticRatio'] ?? 0.5]);
return new HybridQuery($this->vectorizer->vectorize($query), $query, $options['semanticRatio'] ?? 0.5);
return new HybridQuery($this->vectorizer->vectorize($query), explode(' ', $query), $options['semanticRatio'] ?? 0.5);
}
$this->logger->debug('Store supports vector queries, using VectorQuery');

14
tests/RetrieverTest.php
View File

@@ -124,7 +124,9 @@ final class RetrieverTest extends TestCase
$store->expects($this->once())
->method('query')
->with(
$this->isInstanceOf(TextQuery::class),
$this->callback(static function ($query) {
return $query instanceof TextQuery && ['test', 'query'] === $query->getTexts();
}),
$this->anything()
)
->willReturn([$document]);
@@ -160,7 +162,9 @@ final class RetrieverTest extends TestCase
$store->expects($this->once())
->method('query')
->with(
$this->isInstanceOf(TextQuery::class),
$this->callback(static function ($query) {
return $query instanceof TextQuery && ['test', 'query'] === $query->getTexts();
}),
$this->anything()
)
->willReturn([$document]);
@@ -196,7 +200,9 @@ final class RetrieverTest extends TestCase
$store->expects($this->once())
->method('query')
->with(
$this->isInstanceOf(HybridQuery::class),
$this->callback(static function ($query) {
return $query instanceof HybridQuery && ['test', 'query'] === $query->getTexts();
}),
$this->anything()
)
->willReturn([$document]);
@@ -403,7 +409,7 @@ final class RetrieverTest extends TestCase
->method('query')
->with(
$this->callback(static function ($query) {
return $query instanceof HybridQuery && 'expanded query terms' === $query->getText();
return $query instanceof HybridQuery && ['expanded', 'query', 'terms'] === $query->getTexts();
}),
$this->anything()
)