(no commit message)
This commit is contained in:
@@ -1,10 +1,11 @@
|
||||
import dspy
|
||||
from .utils import OPENROUTER_API_BASE, OPENROUTER_API_KEY
|
||||
from .modules import signature_generator
|
||||
from modaic import PrecompiledAgent, PrecompiledConfig
|
||||
|
||||
class PromptToSignatureConfig(PrecompiledConfig):
|
||||
lm: str = "gpt-5-2025-08-07"
|
||||
refine_lm: str = "gpt-5-2025-08-07"
|
||||
lm: str = "openrouter/anthropic/claude-haiku-4.5"
|
||||
refine_lm: str = "openai/gpt-4o"
|
||||
max_tokens: int = 16000
|
||||
temperature: float = 1.0
|
||||
max_attempts_to_refine: int = 5
|
||||
@@ -27,6 +28,8 @@ class PromptToSignatureAgent(PrecompiledAgent):
|
||||
model=config.lm,
|
||||
max_tokens=config.max_tokens,
|
||||
temperature=config.temperature,
|
||||
api_base=OPENROUTER_API_BASE,
|
||||
api_key=OPENROUTER_API_KEY
|
||||
)
|
||||
refine_lm = dspy.LM(
|
||||
model=config.refine_lm,
|
||||
|
||||
Reference in New Issue
Block a user