cmd/mimi: fix tool use

Signed-off-by: Xe Iaso <me@xeiaso.net>

1 files changed, 14 insertions, 6 deletions

diff --git a/llm/codeinterpreter/python/python.go b/llm/codeinterpreter/python/python.go
index 5f2ac6f..c90d9f6 100644
--- a/llm/codeinterpreter/python/python.go
+++ b/llm/codeinterpreter/python/python.go
@@ -32,8 +32,11 @@ func init() {
 }
 
 type Result struct {
-	Stdout string
-	Stderr string
+	UserCode      string
+	Stdout        string
+	Stderr        string
+	PlatformError string
+	Instruction   string
 }
 
 func Run(ctx context.Context, tmpDir, userCode string) (*Result, error) {
@@ -63,8 +66,11 @@ func Run(ctx context.Context, tmpDir, userCode string) (*Result, error) {
 	mod, err := r.InstantiateModule(ctx, code, config)
 	if err != nil {
 		result := &Result{
-			Stdout: fout.String(),
-			Stderr: ferr.String(),
+			UserCode:      userCode,
+			Stdout:        fout.String(),
+			Stderr:        ferr.String(),
+			PlatformError: err.Error(),
+			Instruction:   "Include the user code and any stderr or platform errors in code fences.",
 		}
 		return result, err
 	}
@@ -72,7 +78,9 @@ func Run(ctx context.Context, tmpDir, userCode string) (*Result, error) {
 	defer mod.Close(ctx)
 
 	return &Result{
-		Stdout: fout.String(),
-		Stderr: ferr.String(),
+		UserCode:    userCode,
+		Stdout:      fout.String(),
+		Stderr:      ferr.String(),
+		Instruction: "Include the user code, output, and any stderr or platform errors in code fences.",
 	}, nil
 }