added --no-stream

This commit is contained in:
Paul Gauthier 2023-06-23 13:29:09 -07:00
parent 7e18fb7f5f
commit d2b9cb9e98
2 changed files with 34 additions and 2 deletions

View file

@ -111,6 +111,13 @@ def main(args=None, input=None, output=None):
dest="pretty",
help="Disable pretty, colorized output",
)
parser.add_argument(
"--no-stream",
action="store_false",
dest="stream",
default=True,
help="Disable streaming responses",
)
parser.add_argument(
"--user-input-color",
default="green",
@ -257,6 +264,7 @@ def main(args=None, input=None, output=None):
map_tokens=args.map_tokens,
verbose=args.verbose,
assistant_output_color=args.assistant_output_color,
stream=args.stream,
)
if args.dirty_commits:
@ -270,6 +278,7 @@ def main(args=None, input=None, output=None):
io.tool_output("Use /help to see in-chat commands.")
if args.command:
io.tool_output()
coder.run(with_message=args.command)
else:
coder.run()