Fix: Add error handling for command execution in SandboxDockerSession

vndee · Feb 22, 2025 · df887cf · df887cf
1 parent 0636997
commit df887cf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/llm_sandbox/docker.py b/llm_sandbox/docker.py
@@ -198,7 +198,7 @@ def run(self, code: str, libraries: Optional[List] = None) -> ConsoleOutput:
 
             with open(code_file, "w") as f:
                 f.write(code)
-
+                
             self.copy_to_runtime(code_file, code_dest_file)
 
             output = ConsoleOutput(text="", exit_code=0)
@@ -208,6 +208,9 @@ def run(self, code: str, libraries: Optional[List] = None) -> ConsoleOutput:
                     output = self.execute_command(command, workdir="/example")
                 else:
                     output = self.execute_command(command)
+
+                if output.exit_code != 0:
+                    return output
 
             return output