(no commit message)
This commit is contained in:
@@ -1,3 +1,3 @@
|
|||||||
from .modules import SignatureGenerator
|
from .modules import signature_generator
|
||||||
|
|
||||||
__all__ = ["SignatureGenerator"]
|
__all__ = ["signature_generator"]
|
||||||
|
|||||||
@@ -238,3 +238,5 @@ class SignatureGenerator(dspy.Module):
|
|||||||
)
|
)
|
||||||
|
|
||||||
return imports
|
return imports
|
||||||
|
|
||||||
|
signature_generator = SignatureGenerator()
|
||||||
4
main.py
4
main.py
@@ -1,5 +1,5 @@
|
|||||||
from modaic import PrecompiledAgent, PrecompiledConfig
|
from modaic import PrecompiledAgent, PrecompiledConfig
|
||||||
from agent.modules import SignatureGenerator
|
from agent import signature_generator
|
||||||
import dspy
|
import dspy
|
||||||
|
|
||||||
|
|
||||||
@@ -16,7 +16,7 @@ class PromptToSignatureAgent(PrecompiledAgent):
|
|||||||
|
|
||||||
def __init__(self, config: PromptToSignatureConfig, **kwargs):
|
def __init__(self, config: PromptToSignatureConfig, **kwargs):
|
||||||
super().__init__(config, **kwargs)
|
super().__init__(config, **kwargs)
|
||||||
self.signature_generator = SignatureGenerator()
|
self.signature_generator = signature_generator
|
||||||
self.signature_refiner = dspy.Refine(
|
self.signature_refiner = dspy.Refine(
|
||||||
module=self.signature_generator,
|
module=self.signature_generator,
|
||||||
N=config.max_attempts_to_refine,
|
N=config.max_attempts_to_refine,
|
||||||
|
|||||||
Reference in New Issue
Block a user