(no commit message)

This commit is contained in:
2026-01-02 03:23:55 -08:00
parent a3fc32beef
commit 90a9fd7390
6 changed files with 101 additions and 2 deletions

38
program.py Normal file
View File

@@ -0,0 +1,38 @@
import sys
from typing import Literal
import dspy
from modaic import PrecompiledConfig, PrecompiledProgram
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)
if __name__ == "__main__":
username = sys.argv[1] # ← first arg after script name (username)
program = ExampleProgram(ExampleConfig(output_type="str"), runtime_param="hi")
repo_path = f"{username}/simple_repo"
program.push_to_hub(repo_path, with_code=True)