File size: 2,397 Bytes
c36bd79 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | import subprocess
import sys
import os
MAX_RETRIES = 5
def call_llm(prompt):
"""
TODO: Replace this mock function with your actual SDK call
to IBM Watsonx, OpenAI, Anthropic, or local LLM.
"""
print("\n[🤖 LLM is rewriting code...]")
# Return the raw generated code here
return ""
def run_verification_loop():
print("Starting Autonomous Verification Pipeline...")
# Ensure the directory exists
os.makedirs("models", exist_ok=True)
for attempt in range(1, MAX_RETRIES + 1):
print(f"\n▶ Attempt {attempt} of {MAX_RETRIES}...")
# Run standard Go tests
result = subprocess.run(
["go", "test", "./models", "-v"],
capture_output=True,
text=True
)
# If exit code is 0, the code compiled and all tests passed
if result.returncode == 0:
print(result.stdout)
print("✅ SUCCESS: All tests passed. Pipeline complete.")
sys.exit(0)
print("❌ FAILED: Bug detected. Initiating self-correction loop.")
# Combine stdout and stderr to ensure compiler errors are caught
error_output = result.stdout + "\n" + result.stderr
print(f"Captured Error:\n{error_output}")
# Read the broken code to send back to the AI
try:
with open("models/models.go", "r") as f:
broken_code = f.read()
except FileNotFoundError:
broken_code = "// File not found"
correction_prompt = f"""
You are an expert Go Developer fixing a bug in a legacy migration.
Your previous translation failed the test suite or failed to compile.
### CURRENT GO CODE:
{broken_code}
### COMPILER / TEST ERROR:
{error_output}
### TASK:
Rewrite the entire `models.go` file to fix the bug.
Maintain pure int64 math. Do not use float64.
Output ONLY valid Go code. No markdown. No explanations.
"""
fixed_code = call_llm(correction_prompt)
# Write the AI's fix back to disk for the next loop iteration
if fixed_code.strip():
with open("models/models.go", "w") as f:
f.write(fixed_code)
print("\n⚠️ FATAL: Max retries reached. Human intervention required.")
sys.exit(1)
if __name__ == "__main__":
run_verification_loop()
|