From fdb81ee6719b57a5a4cc022a7337fa6c391fbbfd Mon Sep 17 00:00:00 2001
From: Farouk Adeleke <farouk@modaic.dev>
Date: Thu, 22 Jan 2026 02:30:55 -0800
Subject: [PATCH] debug

---
 nanocode.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/nanocode.py b/nanocode.py
index fdbbc94..f670616 100644
--- a/nanocode.py
+++ b/nanocode.py
@@ -323,11 +323,13 @@ class RLMCodingProgram(PrecompiledProgram):
             max_iterations=self.config.max_iters,
             verbose=self.config.verbose,
         )
+
         agent.set_lm(self.lm)
-        print(f"Using model: {self.lm.model}")
-        print(f"Using sub-model: {self.sub_lm.model}")
         self.agent = agent
 
+        print(f"Using model: {self.agent.get_lm().model}")
+        print(f"Using sub-model: {self.agent.get_lm().model}")
+
     def forward(self, task: str) -> str:
         assert task, "Task cannot be empty"
         return self.agent(task=task)