mirror of
https://github.com/onestardao/WFGY.git
synced 2026-05-05 15:30:28 +00:00
21 lines
609 B
Python
21 lines
609 B
Python
# wfgy_sdk/cli.py
|
|
|
|
import argparse
|
|
from .initializer import initialize
|
|
from .evaluator import evaluate
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="WFGY CLI")
|
|
parser.add_argument("--init", action="store_true", help="Initialize demo data")
|
|
parser.add_argument("--prompt", type=str, help="Input prompt for evaluation")
|
|
args = parser.parse_args()
|
|
|
|
if args.init:
|
|
initialize()
|
|
return
|
|
|
|
if args.prompt:
|
|
result = evaluate(args.prompt)
|
|
print("Result:", result)
|
|
else:
|
|
print("Please provide a prompt using --prompt")
|