Stop supporting Pydantic-V1 JSON-Schema generation (ref #802)

Author: tcompa

fractal_tasks_core/dev/create_manifest.py

@@ -16,7 +16,6 @@
 import logging
 from importlib import import_module
 from pathlib import Path
-from typing import Literal
 from typing import Optional
 
 from fractal_tasks_core.dev.lib_args_schemas import (
@@ -28,26 +27,13 @@
 logging.basicConfig(level=logging.INFO)
 
 
-def check_args_schema_version(args_schema_version: Optional[str]) -> str:
-    if args_schema_version is None:
-        import pydantic
-
-        if pydantic.__version__.startswith("1"):
-            args_schema_version = "pydantic_v1"
-        elif pydantic.__version__.startswith("2"):
-            args_schema_version = "pydantic_v2"
-        else:
-            raise ValueError(f"Unsupported {pydantic.__version__=}")
-    elif args_schema_version not in ["pydantic_v1", "pydantic_v2"]:
-        raise ValueError(f"Unsupported {args_schema_version=}")
-    return args_schema_version
+ARGS_SCHEMA_VERSION = "pydantic_v2"
 
 
 def create_manifest(
     package: str = "fractal_tasks_core",
     manifest_version: str = "2",
     has_args_schemas: bool = True,
-    args_schema_version: Literal["pydantic_v1", "pydantic_v2", None] = None,
     docs_link: Optional[str] = None,
     custom_pydantic_models: Optional[list[tuple[str, str, str]]] = None,
 ):
@@ -85,17 +71,14 @@ def create_manifest(
 
     logging.info("Start generating a new manifest")
 
-    args_schema_version = check_args_schema_version(args_schema_version)
-    logging.warning(f"{args_schema_version=}")
-
     # Prepare an empty manifest
     manifest = dict(
         manifest_version=manifest_version,
         task_list=[],
         has_args_schemas=has_args_schemas,
     )
     if has_args_schemas:
-        manifest["args_schema_version"] = args_schema_version
+        manifest["args_schema_version"] = ARGS_SCHEMA_VERSION
 
     # Prepare a default value of docs_link
     if package == "fractal_tasks_core" and docs_link is None:
@@ -138,7 +121,6 @@ def create_manifest(
                         executable,
                         package=package,
                         custom_pydantic_models=custom_pydantic_models,
-                        args_schema_version=args_schema_version,
                     )
                     logging.info(f"[{executable}] END (new schema)")
                     task_dict[f"args_schema_{kind}"] = schema