fix stat extractor error (#2208)

fixes an issue with statExtractor expecting an async return value
tokens-studio · Aug 30, 2023 · a141ead · a141ead
1 parent 1c6e83a
commit a141ead
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/app/store/useTokens.tsx b/src/app/store/useTokens.tsx
@@ -365,10 +365,11 @@ export default function useTokens() {
     track('createVariables');
     const createVariableResult = await wrapTransaction({
       name: 'createVariables',
-      statExtractor: (result, transaction) => {
-        result.then((resolve) => {
-          transaction.setMeasurement('variables', resolve.totalVariables, '');
-        });
+      statExtractor: async (result, transaction) => {
+        const data = await result;
+        if (data) {
+          transaction.setMeasurement('variables', data.totalVariables, '');
+        }
       },
     }, async () => await AsyncMessageChannel.ReactInstance.message({
       type: AsyncMessageTypes.CREATE_LOCAL_VARIABLES,