File size: 531 Bytes
aef5ed5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | import argparse, sys
from . import Engine
ap = argparse.ArgumentParser()
ap.add_argument("--model", required=True); ap.add_argument("--table", required=True)
ap.add_argument("--archive"); ap.add_argument("--ask"); ap.add_argument("--chat", action="store_true")
a = ap.parse_args()
eng = Engine(a.model, a.table, archive=a.archive)
if a.ask: print(eng.answer(a.ask))
elif a.chat:
while True:
try: q = input("you> ")
except EOFError: break
if not q.strip(): continue
print("shadow>", eng.chat(q))
|