From 378a6575957aa2fc61dde376058cb44e2d07f458 Mon Sep 17 00:00:00 2001
From: Farouk Adeleke <farouk@modaic.dev>
Date: Fri, 23 Jan 2026 20:11:43 -0800
Subject: [PATCH] change signature

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

diff --git a/nanocode.py b/nanocode.py
index 1bff598..d1f7ab9 100644
--- a/nanocode.py
+++ b/nanocode.py
@@ -199,7 +199,6 @@ class RLMCodingProgram(PrecompiledProgram):
             "grep_files": grep_files,
             "run_bash": run_bash,
         }
-        print(f"{BLUE}CONFIG: {self.config}{RESET}")
 
         self.lm = dspy.LM(
             model=self.config.lm,
@@ -207,14 +206,12 @@ class RLMCodingProgram(PrecompiledProgram):
             max_tokens=self.config.max_tokens,
             track_usage=self.config.track_usage,
         )
-        print(f"{BLUE}NEW LM: {self.lm.model}{RESET}")
         self.sub_lm = dspy.LM(
             model=self.config.sub_lm,
             api_base=self.config.api_base,
             max_tokens=self.config.max_tokens,
             track_usage=self.config.track_usage,
         )
-        print(f"{BLUE}NEW SUB LM: {self.sub_lm.model}{RESET}")
         self.agent = dspy.RLM(
             CodingAssistant,
             sub_lm=self.sub_lm,
@@ -224,6 +221,9 @@ class RLMCodingProgram(PrecompiledProgram):
             verbose=self.config.verbose,
         )
         self.agent.set_lm(self.lm)
+        print(f"{BLUE}CONFIG WAS SET: {self.config}{RESET}")
+        print(f"{BLUE}LM WAS SET: {self.agent.get_lm()}{RESET}")
+        print(f"{BLUE}SUB LM WAS SET: {self.agent.sub_lm}{RESET}")
 
     def forward(self, task: str) -> str:
         if not task: