allow disabling reranker

Author: RuliLG

src/Client/KnowledgeBaseClient.php

@@ -58,7 +58,14 @@ public function destroy(mixed $id): bool
      * @param  array<string>  $entities The entities to search for.
      * @param  array<string:string>  $where The entities to search for.
      */
-    public function query(string $text, int $k = 10, ?array $entities = null, ?array $where = null, float $minScore = 0.05): KnowledgeBaseQueryResponse
+    public function query(
+        string $text,
+        int $k = 10,
+        ?array $entities = null,
+        ?array $where = null,
+        float $minScore = 0.05,
+        bool $useReranker = false,
+    ): KnowledgeBaseQueryResponse
     {
         logger()->debug('[KnowledgeBaseClient] Querying knowledge base', [
             'text' => $text,
@@ -71,6 +78,7 @@ public function query(string $text, int $k = 10, ?array $entities = null, ?array
             'query' => $text,
             'k' => $k,
             'min_score' => $minScore,
+            'use_rerank' => $useReranker,
         ];
 
         if ($entities) {

src/DTO/KnowledgeBaseQueryResponseItem.php

@@ -20,7 +20,7 @@ public static function from(array $result): static
             id: $result['id'],
             entity: $result['entity'],
             text: $result['text'],
-            rerankingScore: $result['score'],
+            rerankingScore: $result['score'] ?? 0,
             payload: $result['payload'] ?? null,
         );
     }

src/KnowledgeBase.php

@@ -39,11 +39,18 @@ public function destroy(Model $model): bool
         return true;
     }
 
-    public function query(string $text, int $k = 10, ?array $entities = null, ?array $where = null, float $minScore = 0.05): KnowledgeBaseQueryResponse
+    public function query(
+        string $text,
+        int $k = 10,
+        ?array $entities = null,
+        ?array $where = null,
+        float $minScore = 0.05,
+        bool $useReranker = false,
+    ): KnowledgeBaseQueryResponse
     {
         $client = new KnowledgeBaseClient();
 
-        return $client->query($text, $k, $entities, $where, $minScore);
+        return $client->query($text, $k, $entities, $where, $minScore, $useReranker);
     }
 
     /**

src/Traits/BelongsToKnowledgeBase.php

@@ -91,8 +91,8 @@ public function knowledgeInsertItems(): array
         return $items;
     }
 
-    public static function searchInKnowledgeBase(string $query, int $k = 10, ?array $where = null, float $minScore = 0.05): KnowledgeBaseQueryResponse
+    public static function searchInKnowledgeBase(string $query, int $k = 10, ?array $where = null, float $minScore = 0.05, bool $useReranker = false): KnowledgeBaseQueryResponse
     {
-        return KnowledgeBase::query($query, $k, [class_basename(static::class)], $where, $minScore);
+        return KnowledgeBase::query($query, $k, [class_basename(static::class)], $where, $minScore, $useReranker);
     }
 }