24 lines
607 B
Python
24 lines
607 B
Python
from modaic import PrecompiledAgent, PrecompiledConfig
|
|
|
|
class PromptToSignatureConfig(PrecompiledConfig):
|
|
lm: str = "gpt-4o"
|
|
max_tokens: int = 1024
|
|
|
|
class PromptToSignatureAgent(PrecompiledAgent):
|
|
config: PromptToSignatureConfig
|
|
|
|
def __init__(self, config: PromptToSignatureConfig, **kwargs):
|
|
super().__init__(config, **kwargs)
|
|
|
|
def forward(self, prompt: str) -> str:
|
|
return "hello world"
|
|
|
|
|
|
agent = PromptToSignatureAgent(PromptToSignatureConfig())
|
|
|
|
def main():
|
|
agent.push_to_hub("fadeleke/prompt-to-signature", with_code=True)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|