farouk1/nanocode
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change signature

Browse Source
This commit is contained in:
Farouk Adeleke
2026-01-23 20:11:43 -08:00
parent 7f0aba9241
commit 378a657595
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nanocode.py
View File

@@ -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: