Fix boolean cli flags

abetlen · abetlen · commit c999325e8e45 · 2023-09-13T22:56:10.000-04:00
diff --git a/llama_cpp/server/__main__.py b/llama_cpp/server/__main__.py
@@ -58,13 +58,31 @@ def contains_list_type(annotation) -> bool:
         description = field.description
         if field.default is not None and description is not None:
             description += f" (default: {field.default})"
-        parser.add_argument(
-            f"--{name}",
-            dest=name,
-            nargs="*" if contains_list_type(field.annotation) else None,
-            type=get_base_type(field.annotation) if field.annotation is not None else str,
-            help=description,
-        )
+        base_type = get_base_type(field.annotation) if field.annotation is not None else str
+        list_type = contains_list_type(field.annotation)
+        if base_type is not bool:
+            parser.add_argument(
+                f"--{name}",
+                dest=name,
+                nargs="*" if list_type else None,
+                type=base_type,
+                help=description,
+            )
+        if base_type is bool:
+            parser.add_argument(
+                f"--{name}",
+                dest=name,
+                action="store_true",
+                help=f"Disable {description}",
+                default=field.default,
+            )
+            parser.add_argument(
+                f"--no-{name}",
+                dest=name,
+                action="store_false",
+                help=f"Disable {description}",
+                default=field.default,
+            )
 
     args = parser.parse_args()
     settings = Settings(**{k: v for k, v in vars(args).items() if v is not None})
