customModel length hotfix

Oneirocom · Nov 9, 2023 · 35e9e6a · 35e9e6a
1 parent 26eeb54
commit 35e9e6a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/packages/core/shared/src/nodes/text/GenerateText.ts b/packages/core/shared/src/nodes/text/GenerateText.ts
@@ -180,8 +180,7 @@ export class GenerateText extends MagickComponent<Promise<WorkerReturn>> {
     ]) as CompletionProvider[]
 
     const model = (node.data as { model: string }).model as string
-    const customModel = (node.data as { customModel: string })
-      .customModel as string
+    const customModel = node.data.customModel as string | undefined
 
     // get the provider for the selected model
     const provider = completionProviders.find(provider =>
@@ -195,7 +194,7 @@ export class GenerateText extends MagickComponent<Promise<WorkerReturn>> {
     }
 
     // check if custom model and regex check for word fine-tune in model
-    if (customModel.length > 0) {
+    if (customModel && customModel.length > 0) {
       node.data.customModel = customModel
     }