Update cli.py

This commit is contained in:
ThatLukinhasGuy 2023-11-04 18:16:42 -03:00 committed by GitHub
parent f51fe6322f
commit ef3e297585
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 5 deletions

View File

@ -7,11 +7,9 @@ from g4f import Provider
from g4f.api import Api
from g4f.gui.run import gui_parser, run_gui_args
def run_gui(args):
print("Running GUI...")
def main():
IgnoredProviders = Enum("ignore_providers", {key: key for key in Provider.__all__})
parser = argparse.ArgumentParser(description="Run gpt4free")
@ -26,8 +24,7 @@ def main():
args = parser.parse_args()
if args.mode == "api":
controller=Api(g4f, debug=args.debug)
controller.list_ignored_providers=args.ignored_providers
controller=Api(g4f, debug=args.debug, list_ignored_providers=args.ignored_providers)
controller.run(args.bind, args.num_threads)
elif args.mode == "gui":
run_gui_args(args)
@ -35,6 +32,5 @@ def main():
parser.print_help()
exit(1)
if __name__ == "__main__":
main()