generateEmbeddings method

  1. @override
Future<List<List<double>>> generateEmbeddings(
  1. List<String> texts, {
  2. TaskType taskType = TaskType.retrievalQuery,
})
override

Generate embedding vectors for multiple texts.

taskType controls the prefix applied before embedding.

Implementation

@override
Future<List<List<double>>> generateEmbeddings(
  List<String> texts, {
  TaskType taskType = TaskType.retrievalQuery,
}) async {
  _assertNotClosed();
  return texts.map((text) {
    final tokens = _prepareTokens(text, taskType: taskType);
    return _interpreter.run(tokens);
  }).toList();
}