ai-experiment/core/llm_room.py

6 lines
379 B
Python

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("prompt", nargs="?", default=None, help="Single prompt to answer, then exit. Omit for interactive chat")
parser.add_argument("--system", type=str, default=None, help="Override system prompt from config")
parser.add_argument("--max-tokens", type=int, default=None, help="Override config MAX_TOKENS")