add assistant role (#69)

NetsBlox · May 22, 2024 · 1f1ea54 · 1f1ea54
1 parent 23f27a5
commit 1f1ea54
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/extensions/OpenAIBlocks/index.js b/extensions/OpenAIBlocks/index.js
@@ -89,8 +89,8 @@
             }
             const role = row.contents[0].toLowerCase();
             const content = row.contents[1];
-            if (!['system', 'user'].some((x) => x === role)) {
-                throw Error('speaker must be \'system\' or \'user\'');
+            if (!['system', 'user', 'assistant'].some((x) => x === role)) {
+                throw Error('speaker must be \'system\', \'user\', or \'assistant\'');
             }
             res.push({ role, content });
         }