createEmbeddingModel method

Implementation

@override
Future<EmbeddingModel> createEmbeddingModel({
  String? modelPath,
  String? tokenizerPath,
  PreferredBackend? preferredBackend,
}) async {
  // Modern API: Use active embedding model if paths not provided
  if (modelPath == null || tokenizerPath == null) {
    final manager = _unifiedManager;
    final activeModel = manager.activeEmbeddingModel;

    // No active embedding model - user must set one first
    if (activeModel == null) {
      throw Exception('No active embedding model set. Use `FlutterGemma.installEmbedder()` or `modelManager.setActiveModel()` to set a model first');
    }

    // Get the actual model file paths through unified system
    final modelFilePaths = await manager.getModelFilePaths(activeModel);
    if (modelFilePaths == null || modelFilePaths.isEmpty) {
      throw Exception('Embedding model file paths not found. Use the `modelManager` to load the model first');
    }

    // Extract model and tokenizer paths from spec
    final activeModelPath = modelFilePaths[PreferencesKeys.embeddingModelFile];
    final activeTokenizerPath = modelFilePaths[PreferencesKeys.embeddingTokenizerFile];

    if (activeModelPath == null || activeTokenizerPath == null) {
      throw Exception('Could not find model or tokenizer path in active embedding model');
    }

    // Check if singleton exists and matches the active model
    if (_initEmbeddingCompleter != null && _initializedEmbeddingModel != null) {
      // For embedding models, we stored paths when creating, need to compare with active model paths
      // Since we don't have stored paths, we need to check if createEmbeddingModel was called with same active model
      // Simpler approach: always close and recreate if active model exists but we're in Modern API path
      // This ensures we always use the correct active model
      debugPrint('ℹ️  Closing existing embedding model to ensure correct active model is used');
      await _initializedEmbeddingModel?.close();
      // onClose callback will reset _initializedEmbeddingModel and _initEmbeddingCompleter
    }

    modelPath = activeModelPath;
    tokenizerPath = activeTokenizerPath;

    debugPrint('Using active embedding model: $modelPath, tokenizer: $tokenizerPath');
  } else {
    // Legacy API with explicit paths - check if singleton exists
    if (_initEmbeddingCompleter case Completer<EmbeddingModel> completer) {
      debugPrint('ℹ️  Reusing existing embedding model instance (Legacy API)');
      return completer.future;
    }
  }

  final completer = _initEmbeddingCompleter = Completer<EmbeddingModel>();

  // Verify the active model is still installed (for Modern API path)
  if (modelPath != null && tokenizerPath != null) {
    final manager = _unifiedManager;
    final activeModel = manager.activeEmbeddingModel;

    if (activeModel != null) {
      final isModelInstalled = await manager.isModelInstalled(activeModel);
      if (!isModelInstalled) {
        completer.completeError(
          Exception('Active embedding model is no longer installed. Use the `modelManager` to load the model first'),
        );
        return completer.future;
      }
    }
  }

  try {
    await _platformService.createEmbeddingModel(
      modelPath: modelPath,
      tokenizerPath: tokenizerPath,
      preferredBackend: preferredBackend,
    );

    final model = _initializedEmbeddingModel = MobileEmbeddingModel(
      onClose: () {
        _initializedEmbeddingModel = null;
        _initEmbeddingCompleter = null;
      },
    );

    completer.complete(model);
    return model;
  } catch (e, st) {
    completer.completeError(e, st);
    Error.throwWithStackTrace(e, st);
  }
}