change signature

This commit is contained in:
2026-01-22 03:33:16 -08:00
parent c086336a0f
commit 367fad475b
2 changed files with 4 additions and 8 deletions

View File

@@ -7,7 +7,6 @@ import json
import tempfile
from modaic import PrecompiledProgram, PrecompiledConfig
import dspy
from dspy.utils.callback import BaseCallback
# --- Modaic ---
@@ -238,7 +237,7 @@ def select_model():
class CodingAssistant(dspy.Signature):
"""You are a concise coding assistant. Help the user with their coding task by using the available tools to read, write, edit files, search the codebase, and run commands."""
"""You are a concise coding assistant with access to sub agents."""
task: str = dspy.InputField(desc="The user's coding task or question")
answer: str = dspy.OutputField(
@@ -299,9 +298,6 @@ class RLMCodingProgram(PrecompiledProgram):
agent.set_lm(self.lm)
self.agent = agent
print(f"Using model: {self.agent.get_lm().model}")
print(f"Using sub-model: {self.agent.sub_lm.model}")
def forward(self, task: str) -> str:
assert task, "Task cannot be empty"
return self.agent(task=task)
@@ -570,5 +566,5 @@ def main():
if __name__ == "__main__":
agent = RLMCodingProgram(RLMCodingConfig())
agent.push_to_hub(MODAIC_REPO_PATH, commit_message="debug", branch="dev")
agent.push_to_hub(MODAIC_REPO_PATH, commit_message="change signature", branch="dev")
#main()