Fix generate_code args

koxudaxi · Jul 2, 2024 · 311a138 · 311a138
1 parent 2f41065
commit 311a138
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/fastapi_code_generator/__main__.py b/fastapi_code_generator/__main__.py
@@ -110,8 +110,8 @@ def generate_code(
     output_dir: Path,
     template_dir: Optional[Path],
     model_path: Optional[Path] = None,
-    enum_field_as_literal: Optional[str] = None,
-    custom_visitors: Optional[List[Path]] = [],
+    enum_field_as_literal: Optional[LiteralType] = None,
+    custom_visitors: Optional[List[Path]] = None,
     disable_timestamp: bool = False,
     generate_routers: Optional[bool] = None,
     specify_tags: Optional[str] = None,
@@ -128,7 +128,8 @@ def generate_code(
         template_dir = (
             BUILTIN_MODULAR_TEMPLATE_DIR if generate_routers else BUILTIN_TEMPLATE_DIR
         )
-
+    if not custom_visitors:
+        custom_visitors = []
     data_model_types = get_data_model_types(output_model_type, python_version)
 
     parser = OpenAPIParser(