From 73154a05b032ee3c470a9dda76f78cb8af5f95bd Mon Sep 17 00:00:00 2001
From: Han Xiao <han.xiao@jina.ai>
Date: Tue, 17 Jun 2025 21:36:12 -0700
Subject: [PATCH] fix: code gen

---
 src/tools/code-sandbox.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/tools/code-sandbox.ts b/src/tools/code-sandbox.ts
index 12b08d6..3a70657 100644
--- a/src/tools/code-sandbox.ts
+++ b/src/tools/code-sandbox.ts
@@ -31,6 +31,7 @@ ${attempt.error ? `Error: ${attempt.error}
 2. You can access any of these available variables directly:
 ${availableVars}
 3. You don't have access to any third party libraries that need to be installed, so you must write complete, self-contained code.
+4. Must have a return statement.
 </rules>
 
 ${previousAttempts.length > 0 ? `Previous attempts and their errors: