mirror of
https://github.com/tcsenpai/agenticSeek.git
synced 2025-06-03 01:30:11 +00:00
120 lines
3.6 KiB
Python
120 lines
3.6 KiB
Python
import subprocess
|
|
import os, sys
|
|
import tempfile
|
|
import re
|
|
|
|
if __name__ == "__main__": # if running as a script for individual testing
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
|
|
|
|
from sources.tools.tools import Tools
|
|
|
|
class CInterpreter(Tools):
|
|
"""
|
|
This class is a tool to allow agent for C code execution
|
|
"""
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.tag = "c"
|
|
self.name = "C Interpreter"
|
|
self.description = "This tool allows the agent to execute C code."
|
|
|
|
def execute(self, codes: str, safety=False) -> str:
|
|
"""
|
|
Execute C code by compiling and running it.
|
|
"""
|
|
output = ""
|
|
code = '\n'.join(codes) if isinstance(codes, list) else codes
|
|
|
|
if safety and input("Execute code? y/n ") != "y":
|
|
return "Code rejected by user."
|
|
|
|
exec_extension = ".exe" if os.name == "nt" else "" # Windows uses .exe, Linux/Unix does not
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdirname:
|
|
source_file = os.path.join(tmpdirname, "temp.c")
|
|
exec_file = os.path.join(tmpdirname, "temp") + exec_extension
|
|
with open(source_file, 'w') as f:
|
|
f.write(code)
|
|
|
|
try:
|
|
compile_command = ["gcc", source_file, "-o", exec_file]
|
|
compile_result = subprocess.run(
|
|
compile_command,
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=60
|
|
)
|
|
|
|
if compile_result.returncode != 0:
|
|
return f"Compilation failed: {compile_result.stderr}"
|
|
|
|
run_command = [exec_file]
|
|
run_result = subprocess.run(
|
|
run_command,
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=120
|
|
)
|
|
|
|
if run_result.returncode != 0:
|
|
return f"Execution failed: {run_result.stderr}"
|
|
output = run_result.stdout
|
|
|
|
except subprocess.TimeoutExpired as e:
|
|
return f"Execution timed out: {str(e)}"
|
|
except FileNotFoundError:
|
|
return "Error: 'gcc' not found. Ensure a C compiler (e.g., gcc) is installed and in PATH."
|
|
except Exception as e:
|
|
return f"Code execution failed: {str(e)}"
|
|
|
|
return output
|
|
|
|
def interpreter_feedback(self, output: str) -> str:
|
|
"""
|
|
Provide feedback based on the output of the code execution
|
|
"""
|
|
if self.execution_failure_check(output):
|
|
feedback = f"[failure] Error in execution:\n{output}"
|
|
else:
|
|
feedback = "[success] Execution success, code output:\n" + output
|
|
return feedback
|
|
|
|
def execution_failure_check(self, feedback: str) -> bool:
|
|
"""
|
|
Check if the code execution failed.
|
|
"""
|
|
error_patterns = [
|
|
r"error",
|
|
r"failed",
|
|
r"traceback",
|
|
r"invalid",
|
|
r"exception",
|
|
r"syntax",
|
|
r"segmentation fault",
|
|
r"core dumped",
|
|
r"undefined",
|
|
r"cannot"
|
|
]
|
|
combined_pattern = "|".join(error_patterns)
|
|
if re.search(combined_pattern, feedback, re.IGNORECASE):
|
|
return True
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
codes = [
|
|
"""
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
void hello() {
|
|
printf("Hello, World!\\n");
|
|
}
|
|
""",
|
|
"""
|
|
int main() {
|
|
hello();
|
|
return 0;
|
|
}
|
|
"""]
|
|
c = CInterpreter()
|
|
print(c.execute(codes)) |