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()