Improve cli arguments

This commit is contained in:
Heiner Lohaus 2024-04-20 18:04:16 +02:00
parent 83484c0a56
commit e6863a75e1
2 changed files with 5 additions and 6 deletions

View File

@ -2,7 +2,6 @@ import json
import uuid
import random
import time
import uuid
from typing import Dict, List
from aiohttp import ClientSession, BaseConnector
@ -13,8 +12,8 @@ from .base_provider import AsyncGeneratorProvider
from .helper import format_prompt, get_connector
class Sources():
def __init__(self, list: List[Dict[str, str]]) -> None:
self.list = list
def __init__(self, link_list: List[Dict[str, str]]) -> None:
self.link = link_list
def __str__(self) -> str:
return "\n\n" + ("\n".join([f"[{link['title']}]({link['link']})" for link in self.list]))

View File

@ -15,9 +15,9 @@ def main():
subparsers = parser.add_subparsers(dest="mode", help="Mode to run the g4f in.")
api_parser=subparsers.add_parser("api")
api_parser.add_argument("--bind", default="0.0.0.0:1337", help="The bind string.")
api_parser.add_argument("--debug", type=bool, default=False, help="Enable verbose logging.")
api_parser.add_argument("--debug", action="store_true", help="Enable verbose logging.")
api_parser.add_argument("--workers", type=int, default=None, help="Number of workers.")
api_parser.add_argument("--use_colors", type=bool, default=None, help="Use colors.")
api_parser.add_argument("--disable_colors", action="store_true", help="Don't use colors.")
api_parser.add_argument("--ignored-providers", nargs="+", choices=[provider.name for provider in IgnoredProviders],
default=[], help="List of providers to ignore when processing request.")
subparsers.add_parser("gui", parents=[gui_parser()], add_help=False)
@ -26,7 +26,7 @@ def main():
if args.mode == "api":
import g4f.api
g4f.api.api.set_list_ignored_providers(args.ignored_providers)
g4f.api.run_api(bind=args.bind, debug=args.debug, workers=args.workers, use_colors=args.use_colors)
g4f.api.run_api(bind=args.bind, debug=args.debug, workers=args.workers, use_colors=not args.disable_colors)
elif args.mode == "gui":
run_gui_args(args)
else: