diff --git a/packages/sample/genaisrc/judge.genai.mts b/packages/sample/genaisrc/judge.genai.mts
index ceaf713ed..60cbf7bb3 100644
--- a/packages/sample/genaisrc/judge.genai.mts
+++ b/packages/sample/genaisrc/judge.genai.mts
@@ -7,8 +7,8 @@
  * @returns
  */
 async function judgeClassify(
-    task: Awaitable<string>,
-    result: Awaitable<string>,
+    task: Awaitable<string | WorkspaceFile>,
+    result: Awaitable<string | WorkspaceFile>,
     categories: Record<string, string>,
     options?: {
         logprobs?: boolean