allow setting min score

Author: RuliLG

src/Client/KnowledgeBaseClient.php

@@ -58,17 +58,19 @@ 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): KnowledgeBaseQueryResponse
+    public function query(string $text, int $k = 10, ?array $entities = null, ?array $where = null, float $minScore = 0.05): KnowledgeBaseQueryResponse
     {
         logger()->debug('[KnowledgeBaseClient] Querying knowledge base', [
             'text' => $text,
             'k' => $k,
             'entities' => $entities,
             'where' => $where,
+            'minScore' => $minScore,
         ]);
         $params = [
             'query' => $text,
             'k' => $k,
+            'min_score' => $minScore,
         ];
 
         if ($entities) {

src/Traits/BelongsToKnowledgeBase.php

@@ -38,7 +38,7 @@ public function knowledgeBasePayload(): array
         return [];
     }
 
-    public function knowledgeBaseKeyAttribute(): ?string 
+    public function knowledgeBaseKeyAttribute(): ?string
     {
         return null;
     }
@@ -66,7 +66,7 @@ public function knowledgeInsertItems(): array
             }
 
             $key = $this->knowledgeBaseKeyAttribute() ? $this->{$this->knowledgeBaseKeyAttribute()} : $this->getKey();
-            
+
             $items[] = new KnowledgeInsertItem(
                 id: $chunk->id,
                 entity: $text->entity,
@@ -91,8 +91,8 @@ public function knowledgeInsertItems(): array
         return $items;
     }
 
-    public static function searchInKnowledgeBase(string $query, int $k = 10, ?array $where = null): KnowledgeBaseQueryResponse
+    public static function searchInKnowledgeBase(string $query, int $k = 10, ?array $where = null, float $minScore = 0.05): KnowledgeBaseQueryResponse
     {
-        return KnowledgeBase::query($query, $k, [class_basename(static::class)], $where);
+        return KnowledgeBase::query($query, $k, [class_basename(static::class)], $where, $minScore);
     }
 }