fix: check async generator (#41)

microsoft · Jul 22, 2024 · f3e1fd9 · f3e1fd9
1 parent df277ac
commit f3e1fd9
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/worker/orchestration-executor.ts b/src/worker/orchestration-executor.ts
@@ -128,12 +128,8 @@ export class OrchestrationExecutor {
             // [object AsyncGenerator]
             const resultType = result?.toString();
 
-            // Try to extract AsyncGenerator
-            const match = /\[object (\w+)\]/.exec(resultType);
-            const resultTypeName = (match && match[1]) ?? "";
-
-            // Check if the result is a generator
-            if (resultTypeName.indexOf("Generator") > -1) {
+            const isAsyncGenerator = typeof result[Symbol.asyncIterator] === 'function';
+            if (isAsyncGenerator) {
               // Start the orchestrator's generator function
               await ctx.run(result);
             } else {