searchSimilar method

  1. @override
Future<List<RetrievalResult>> searchSimilar({
  1. required String query,
  2. int topK = 5,
  3. double threshold = 0.0,
})
override

Search for similar documents.

Implementation

@override
Future<List<RetrievalResult>> searchSimilar({
  required String query,
  int topK = 5,
  double threshold = 0.0,
}) async {
  // Generate embedding for query
  if (initializedEmbeddingModel == null) {
    throw Exception('EmbeddingModel not initialized. Call createEmbeddingModel first.');
  }
  final queryEmbedding = await initializedEmbeddingModel!.generateEmbedding(query);

  // Search similar vectors
  return await _platformService.searchSimilar(
    queryEmbedding: queryEmbedding,
    topK: topK,
    threshold: threshold,
  );
}