change signature
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user