From 57284a25058002911ee571a576740888b6086313 Mon Sep 17 00:00:00 2001 From: Xe Iaso Date: Tue, 27 Aug 2024 17:54:42 -0400 Subject: cmd/mimi: fix tool use Signed-off-by: Xe Iaso --- llm/codeinterpreter/python/python.go | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) (limited to 'llm/codeinterpreter/python/python.go') 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 } -- cgit v1.2.3