ir_compiler.py

class AgentIRCompiler:
    """
    Converts agent reasoning + context into IR instructions.
    This is a lightweight semantic-to-opcode compiler layer.
    """

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

    # -----------------------------------------
    def compile(self, agent, goal, snapshot, prompt, role):
        """
        Produces IR program as list of ops.
        """

        system_prompt = f"""
You are an IR compiler.

Convert reasoning into structured opcode instructions.

RULES:
- output ONLY JSON list
- each instruction must have:
  - opcode
  - arg1 (optional)
  - arg2 (optional)

GOAL:
{goal}

ROLE:
{role}

AGENT:
{agent.name}

CONTEXT:
{prompt}
"""

        raw = self.llm.call(system_prompt)

        return self._parse(raw)

    # -----------------------------------------
    def _parse(self, raw):
        """
        Robust JSON extraction with fallback.
        """

        import json

        try:
            if isinstance(raw, list):
                return raw

            # attempt direct JSON parse
            return json.loads(raw)

        except Exception:
            # fallback: very weak structured recovery
            return [{
                "opcode": "NOP",
                "arg1": str(raw)[:200],
                "arg2": None
            }]