agent_ir_compiler.py

class AgentIRCompiler:
    def __init__(self, llm):
        self.llm = llm

    def compile(self, agent, goal, memory, input_text, role):
        prompt = f"""
You are an agent compiler.

Convert reasoning into IR instructions only.

GOAL:
{goal}

ROLE:
{role}

INPUT:
{input_text}

MEMORY:
{memory}

RULES:
- Output ONLY IR instructions
- Use: LOAD, ADD, CMP, JZ, JMP, STORE, HALT
- No natural language
- Must form executable program

Example:
LOAD A 10
LOAD B 20
ADD A B C
CMP C 30
JZ success
HALT
"""

        return self.parse_ir(self.llm.call(prompt))

    def parse_ir(self, text):
        program = []

        for line in text.split("\n"):
            line = line.strip()
            if not line:
                continue

            parts = line.split()

            program.append({
                "opcode": parts[0],
                "arg1": parts[1] if len(parts) > 1 else None,
                "arg2": parts[2] if len(parts) > 2 else None,
                "arg3": parts[3] if len(parts) > 3 else None,
            })

        return program