Add default busy timeout for spellrunner env

Oneirocom · Oct 28, 2023 · 4cf8d45 · 4cf8d45
1 parent 687810c
commit 4cf8d45
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/packages/config/src/config.ts b/packages/config/src/config.ts
@@ -141,3 +141,9 @@ export const MANAGER_WARM_UP_MSEC =
   Number(getVarForEnvironment('MANAGER_WARM_UP_MSEC')) || 5000
 
 export const API_ACCESS_KEY = getVarForEnvironment('API_ACCESS_KEY') || 'apiKey'
+
+export const SPELLRUNNER_BUSY_TIMEOUT_MSEC = getVarForEnvironment(
+  'SPELLRUNNER_BUSY_TIMEOUT_MS'
+)
+  ? Number(getVarForEnvironment('SPELLRUNNER_BUSY_TIMEOUT_MS'))
+  : 120000
diff --git a/packages/core/shared/src/spellManager/SpellRunner.ts b/packages/core/shared/src/spellManager/SpellRunner.ts
@@ -15,6 +15,7 @@ import { extractModuleInputKeys } from './graphHelpers'
 import SpellManager from './SpellManager'
 import { getLogger } from '../logger'
 import { NodeData } from 'rete/types/core/data'
+import { SPELLRUNNER_BUSY_TIMEOUT_MSEC } from '@magickml/config'
 
 export type RunComponentArgs = {
   inputs: MagickSpellInput
@@ -377,7 +378,7 @@ class SpellRunner {
       // A run shouldnt take this long.  This is a hacl but we are replacing all this soon.
       setTimeout(() => {
         this.busy = false
-      }, 120000)
+      }, SPELLRUNNER_BUSY_TIMEOUT_MSEC)
 
       // this running is where the main "work" happens.
       // I do wonder whether we could make this even more elegant by having the node