cleanup: use extism.output in example

extism · Oct 16, 2024 · ce41044 · ce41044
1 parent 113da4d
commit ce41044
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/examples/count-vowels.py b/examples/count-vowels.py
@@ -1,5 +1,10 @@
 import extism
 import json
+from dataclasses import dataclass
+
+@dataclass
+class Count(extism.Json):
+    count: int
 
 @extism.plugin_fn
 def count_vowels():
@@ -9,5 +14,5 @@ def count_vowels():
         if ch in ['A', 'a', 'E', 'e', 'I', 'i', 'O', 'o', 'U', 'u']:
             total += 1
     extism.log(extism.LogLevel.Info, "Hello!")
-    extism.output_json({"count": total})
+    extism.output(Count(total))