0.3.4.0

1 files changed, 15 insertions, 12 deletions

diff --git a/g4f/cli.py b/g4f/cli.py
index 408dc914..e3017e31 100644
--- a/g4f/cli.py
+++ b/g4f/cli.py
@@ -11,16 +11,19 @@ def main():
     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", action="store_true", help="Enable verbose logging.")
-    api_parser.add_argument("--model", default=None, help="Default model for chat completion. (incompatible with --debug and --workers)")
+    api_parser.add_argument("--model", default=None, help="Default model for chat completion. (incompatible with --reload and --workers)")
     api_parser.add_argument("--provider", choices=[provider.__name__ for provider in Provider.__providers__ if provider.working],
-                            default=None, help="Default provider for chat completion. (incompatible with --debug and --workers)")
-    api_parser.add_argument("--proxy", default=None, help="Default used proxy.")
+                            default=None, help="Default provider for chat completion. (incompatible with --reload and --workers)")
+    api_parser.add_argument("--image-provider", choices=[provider.__name__ for provider in Provider.__providers__ if provider.working and hasattr(provider, "image_models")],
+                            default=None, help="Default provider for image generation. (incompatible with --reload and --workers)"),
+    api_parser.add_argument("--proxy", default=None, help="Default used proxy. (incompatible with --reload and --workers)")
     api_parser.add_argument("--workers", type=int, default=None, help="Number of workers.")
     api_parser.add_argument("--disable-colors", action="store_true", help="Don't use colors.")
-    api_parser.add_argument("--ignore-cookie-files", action="store_true", help="Don't read .har and cookie files.")
-    api_parser.add_argument("--g4f-api-key", type=str, default=None, help="Sets an authentication key for your API. (incompatible with --debug and --workers)")
+    api_parser.add_argument("--ignore-cookie-files", action="store_true", help="Don't read .har and cookie files. (incompatible with --reload and --workers)")
+    api_parser.add_argument("--g4f-api-key", type=str, default=None, help="Sets an authentication key for your API. (incompatible with --reload and --workers)")
     api_parser.add_argument("--ignored-providers", nargs="+", choices=[provider.__name__ for provider in Provider.__providers__ if provider.working],
-                            default=[], help="List of providers to ignore when processing request. (incompatible with --debug and --workers)")
+                            default=[], help="List of providers to ignore when processing request. (incompatible with --reload and --workers)")
+    api_parser.add_argument("--reload", action="store_true", help="Enable reloading.")
     subparsers.add_parser("gui", parents=[gui_parser()], add_help=False)
 
     args = parser.parse_args()
@@ -39,17 +42,17 @@ def run_api_args(args):
         ignore_cookie_files=args.ignore_cookie_files,
         ignored_providers=args.ignored_providers,
         g4f_api_key=args.g4f_api_key,
-        defaults={
-            "model": args.model,
-            "provider": args.provider,
-            "proxy": args.proxy
-        }
+        provider=args.provider,
+        image_provider=args.image_provider,
+        proxy=args.proxy,
+        model=args.model
     )
     run_api(
         bind=args.bind,
         debug=args.debug,
         workers=args.workers,
-        use_colors=not args.disable_colors
+        use_colors=not args.disable_colors,
+        reload=args.reload
     )
 
 if __name__ == "__main__":