(no commit message)

agent/__init__.py

@@ -1,3 +1,3 @@
-from .modules import SignatureGenerator
+from .modules import signature_generator
 
-__all__ = ["SignatureGenerator"]
+__all__ = ["signature_generator"]

agent/modules.py

@@ -238,3 +238,5 @@ class SignatureGenerator(dspy.Module):
             )
 
         return imports
+
+signature_generator = SignatureGenerator()

main.py

@@ -1,5 +1,5 @@
 from modaic import PrecompiledAgent, PrecompiledConfig
-from agent.modules import SignatureGenerator
+from agent import signature_generator
 import dspy
 
 
@@ -16,7 +16,7 @@ class PromptToSignatureAgent(PrecompiledAgent):
 
     def __init__(self, config: PromptToSignatureConfig, **kwargs):
         super().__init__(config, **kwargs)
-        self.signature_generator = SignatureGenerator()
+        self.signature_generator = signature_generator
         self.signature_refiner = dspy.Refine(
             module=self.signature_generator,
             N=config.max_attempts_to_refine,