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 uuid
import random import random
import time import time
import uuid
from typing import Dict, List from typing import Dict, List
from aiohttp import ClientSession, BaseConnector from aiohttp import ClientSession, BaseConnector
@ -13,8 +12,8 @@ from .base_provider import AsyncGeneratorProvider
from .helper import format_prompt, get_connector from .helper import format_prompt, get_connector
class Sources(): class Sources():
def __init__(self, list: List[Dict[str, str]]) -> None: def __init__(self, link_list: List[Dict[str, str]]) -> None:
self.list = list self.link = link_list
def __str__(self) -> str: def __str__(self) -> str:
return "\n\n" + ("\n".join([f"[{link['title']}]({link['link']})" for link in self.list])) 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.") subparsers = parser.add_subparsers(dest="mode", help="Mode to run the g4f in.")
api_parser=subparsers.add_parser("api") 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("--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("--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], api_parser.add_argument("--ignored-providers", nargs="+", choices=[provider.name for provider in IgnoredProviders],
default=[], help="List of providers to ignore when processing request.") default=[], help="List of providers to ignore when processing request.")
subparsers.add_parser("gui", parents=[gui_parser()], add_help=False) subparsers.add_parser("gui", parents=[gui_parser()], add_help=False)
@ -26,7 +26,7 @@ def main():
if args.mode == "api": if args.mode == "api":
import g4f.api import g4f.api
g4f.api.api.set_list_ignored_providers(args.ignored_providers) 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": elif args.mode == "gui":
run_gui_args(args) run_gui_args(args)
else: else: