Add more functionality to signature description parsing

main.py

@@ -1,5 +1,6 @@
 from claude_dspy import ClaudeCode, ClaudeCodeConfig
 from pydantic import BaseModel
+from modaic import AutoProgram
 import dspy
 
 
@@ -9,7 +10,7 @@ class Output(BaseModel):
 
 class ClaudeCodeSignature(dspy.Signature):
     message: str = dspy.InputField(desc="Request to process")
-    output: list[str] = dspy.OutputField(desc="List of files modified or created")
+    output: Output = dspy.OutputField(desc="List of files modified or created")
 
 
 def main():
@@ -22,24 +23,22 @@ def main():
         signature=ClaudeCodeSignature,
         working_directory=".",
         permission_mode="acceptEdits",
-        allowed_tools=["Read", "Glob", "Write"],
+        allowed_tools=["Read", "Bash", "Write"],
     )
-
-    # use it
-    print("Running ClaudeCode...")
-    result = cc(
-        message="Create a new file called helloworld.txt with the alphabet backwards"
-    )
-
-    print(f"Success: {result.output}")
-    print(result.usage)
-
-    print(f"Session ID: {cc.session_id}")
     cc.push_to_hub(
         "farouk1/claude-code",
         with_code=True,
         commit_message="Add more functionality to signature description parsing",
     )
+    """
+    agent = AutoProgram.from_precompiled("farouk1/claude-code", signature=ClaudeCodeSignature, working_directory=".", permission_mode="acceptEdits", allowed_tools=["Read", "Write", "Bash"])
+    
+    # Test the agent
+    result = agent(message="create a python program that prints 'Hello, World!' and save it to a file in this directory")
+    print(result.output.files)
+    print(result.output)
+    print(result.usage)
+    """
 
 
 if __name__ == "__main__":