fix: fix some things that should have been caught in last pr

sroussey · Jan 31, 2025 · a533671 · a533671
1 parent 581de0a
commit a533671
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/packages/ai/src/task/TextTranslationTask.ts b/packages/ai/src/task/TextTranslationTask.ts
@@ -66,6 +66,12 @@ export class TextTranslationTask extends JobQueueLlmTask {
   declare defaults: Partial<TextTranslationTaskInput>;
   static readonly type = "TextTranslationTask";
   static readonly category = "Text Model";
+  async validateItem(valueType: string, item: any) {
+    if (valueType == "language") {
+      return typeof item == "string" && item.length == 2;
+    }
+    return super.validateItem(valueType, item);
+  }
 }
 TaskRegistry.registerTask(TextTranslationTask);
 

diff --git a/packages/core/src/task/DebugLogTask.ts b/packages/core/src/task/DebugLogTask.ts
@@ -60,7 +60,15 @@ export class DebugLogTask extends OutputTask {
     this.runOutputData.output = this.runInputData.message;
     return this.runOutputData;
   }
+
+  async validateItem(valueType: string, item: any) {
+    if (valueType == "log_level") {
+      return log_levels.includes(item);
+    }
+    return super.validateItem(valueType, item);
+  }
 }
+
 TaskRegistry.registerTask(DebugLogTask);
 
 export const DebugLog = (input: DebugLogTaskInput) => {

diff --git a/packages/core/src/task/base/ArrayTask.ts b/packages/core/src/task/base/ArrayTask.ts
@@ -260,6 +260,10 @@ export function arrayTaskFactory<
       const { subtasks, ...result } = super.toDependencyJSON();
       return result;
     }
+
+    async validateItem(valueType: string, item: any) {
+      return true; // let children validate
+    }
   }
 
   TaskRegistry.registerTask(ArrayTask);

diff --git a/packages/core/src/task/base/Task.ts b/packages/core/src/task/base/Task.ts
@@ -232,6 +232,8 @@ export abstract class TaskBase {
    */
   async validateItem(valueType: string, item: any) {
     switch (valueType) {
+      case "any":
+        return true;
       case "number":
         return typeof item === "bigint" || typeof item === "number";
       case "text":