Fix V1

Author: guaycuru

pydantic2ts/cli/script.py

@@ -14,7 +14,6 @@
 from types import ModuleType
 from typing import Any, Dict, List, Tuple, Type
 
-from pydantic.json_schema import GenerateJsonSchema, JsonSchemaValue
 from pydantic_core import core_schema
 from uuid import uuid4
 
@@ -33,6 +32,13 @@
     except ImportError:
         GenericModel = None
 
+if V2:
+    try:
+        from pydantic.json_schema import GenerateJsonSchema, JsonSchemaValue
+    except ImportError:
+        GenerateJsonSchema = None
+        JsonSchemaValue = None
+
 logger = logging.getLogger("pydantic2ts")
 
 
@@ -208,15 +214,16 @@ def is_matching_enum(prop_type: Any, schema_title: str, schema_enum: list[str])
     )
 
 
-class CustomGenerateJsonSchema(GenerateJsonSchema):
-    def enum_schema(self, schema: core_schema.EnumSchema) -> JsonSchemaValue:
-        # Call the original method
-        result = super().enum_schema(schema)
+if V2:
+    class CustomGenerateJsonSchema(GenerateJsonSchema):
+        def enum_schema(self, schema: core_schema.EnumSchema) -> JsonSchemaValue:
+            # Call the original method
+            result = super().enum_schema(schema)
 
-        # Add tsEnumNames property
-        result["tsEnumNames"] = [v.name for v in schema["members"]]
+            # Add tsEnumNames property
+            result["tsEnumNames"] = [v.name for v in schema["members"]]
 
-        return result
+            return result
 
 
 def generate_json_schema_v1(