32 lines
923 B
Python
32 lines
923 B
Python
from typing import Literal
|
|
|
|
import dspy
|
|
from modaic import PrecompiledConfig, PrecompiledProgram
|
|
|
|
from .mod import function_from_mod # noqa: F401
|
|
|
|
|
|
class Summarize(dspy.Signature):
|
|
question = dspy.InputField()
|
|
context = dspy.InputField()
|
|
answer = dspy.OutputField(desc="Answer to the question, based on the passage")
|
|
|
|
|
|
class ExampleConfig(PrecompiledConfig):
|
|
output_type: Literal["bool", "str"]
|
|
lm: str = "openai/gpt-4o"
|
|
number: int = 1
|
|
|
|
|
|
class ExampleProgram(PrecompiledProgram):
|
|
config: ExampleConfig
|
|
|
|
def __init__(self, config: ExampleConfig, runtime_param: str, **kwargs):
|
|
super().__init__(config, **kwargs)
|
|
self.predictor = dspy.Predict(Summarize)
|
|
self.predictor.lm = dspy.LM(self.config.lm)
|
|
self.runtime_param = runtime_param
|
|
|
|
def forward(self, question: str, context: str) -> str:
|
|
return self.predictor(question=question, context=context)
|