(no commit message)

2025-10-30 13:36:13 -04:00
parent 1385e2e587
commit 11e1ff03f0
3 changed files with 6 additions and 4 deletions
--- a/main.py
+++ b/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,