| 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"]) | |