from agent_core import fix_loop
from planner import plan

def model_generate(prompt):
    # 여기에는 실제 fine-tuned 모델 inference 연결
    return "def solution(): print('test')"

def agent(prompt):

    reasoning = plan(prompt)

    final_code = fix_loop(model_generate, prompt)

    return {
        "plan": reasoning,
        "code": final_code
    }

if __name__ == "__main__":
    res = agent("LRU 캐시 구현해줘")
    print(res["plan"])
    print(res["code"])