File size: 482 Bytes
b62c338 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | 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"])
|