Fix Dart APIs

k2-fsa · Sep 20, 2024 · f3fdfc2 · f3fdfc2
1 parent 862211c
commit f3fdfc2
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/flutter/sherpa_onnx/lib/src/keyword_spotter.dart b/flutter/sherpa_onnx/lib/src/keyword_spotter.dart
@@ -19,11 +19,13 @@ class KeywordSpotterConfig {
     this.keywordsScore = 1.0,
     this.keywordsThreshold = 0.25,
     this.keywordsFile = '',
+    this.keywordsBuf = '',
+    this.keywordsBufSize = 0,
   });
 
   @override
   String toString() {
-    return 'KeywordSpotterConfig(feat: $feat, model: $model, maxActivePaths: $maxActivePaths, numTrailingBlanks: $numTrailingBlanks, keywordsScore: $keywordsScore, keywordsThreshold: $keywordsThreshold, keywordsFile: $keywordsFile)';
+    return 'KeywordSpotterConfig(feat: $feat, model: $model, maxActivePaths: $maxActivePaths, numTrailingBlanks: $numTrailingBlanks, keywordsScore: $keywordsScore, keywordsThreshold: $keywordsThreshold, keywordsFile: $keywordsFile, keywordsBuf: $keywordsBuf, keywordsBufSize: $keywordsBufSize)';
   }
 
   final FeatureConfig feat;
@@ -35,6 +37,8 @@ class KeywordSpotterConfig {
   final double keywordsScore;
   final double keywordsThreshold;
   final String keywordsFile;
+  final String keywordsBuf;
+  final int keywordsBufSize;
 }
 
 class KeywordResult {
@@ -89,9 +93,12 @@ class KeywordSpotter {
     c.ref.keywordsScore = config.keywordsScore;
     c.ref.keywordsThreshold = config.keywordsThreshold;
     c.ref.keywordsFile = config.keywordsFile.toNativeUtf8();
+    c.ref.keywordsBuf = config.keywordsBuf.toNativeUtf8();
+    c.ref.keywordsBufSize = config.keywordsBufSize;
 
     final ptr = SherpaOnnxBindings.createKeywordSpotter?.call(c) ?? nullptr;
 
+    calloc.free(c.ref.keywordsBuf);
     calloc.free(c.ref.keywordsFile);
     calloc.free(c.ref.model.bpeVocab);
     calloc.free(c.ref.model.modelingUnit);

diff --git a/flutter/sherpa_onnx/lib/src/sherpa_onnx_bindings.dart b/flutter/sherpa_onnx/lib/src/sherpa_onnx_bindings.dart
@@ -371,6 +371,11 @@ final class SherpaOnnxKeywordSpotterConfig extends Struct {
   external double keywordsThreshold;
 
   external Pointer<Utf8> keywordsFile;
+
+  external Pointer<Utf8> keywordsBuf;
+
+  @Int32()
+  external int keywordsBufSize;
 }
 
 final class SherpaOnnxOfflinePunctuation extends Opaque {}