Quellcode durchsuchen

Fix: Apply Metadata Filters Correctly in Full-Text Search Mode for Tencent Cloud Vector Database (#23564)

tags/1.7.2
Qiang Lee vor 2 Monaten
Ursprung
Commit
e9045a8838
Es ist kein Account mit der E-Mail-Adresse des Committers verbunden
1 geänderte Dateien mit 5 neuen und 0 gelöschten Zeilen
  1. 5
    0
      api/core/rag/datasource/vdb/tencent/tencent_vector.py

+ 5
- 0
api/core/rag/datasource/vdb/tencent/tencent_vector.py Datei anzeigen

@@ -246,6 +246,10 @@ class TencentVector(BaseVector):
return self._get_search_res(res, score_threshold)

def search_by_full_text(self, query: str, **kwargs: Any) -> list[Document]:
document_ids_filter = kwargs.get("document_ids_filter")
filter = None
if document_ids_filter:
filter = Filter(Filter.In("metadata.document_id", document_ids_filter))
if not self._enable_hybrid_search:
return []
res = self._client.hybrid_search(
@@ -269,6 +273,7 @@ class TencentVector(BaseVector):
),
retrieve_vector=False,
limit=kwargs.get("top_k", 4),
filter=filter,
)
score_threshold = float(kwargs.get("score_threshold") or 0.0)
return self._get_search_res(res, score_threshold)

Laden…
Abbrechen
Speichern