October 2026 main drop: mirror from GitHub

c36bd79 verified 30 days ago

2.4 kB

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()

	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()