Update span to use v2 pydantic

Author: codegen-bot

src/codegen/sdk/codebase/span.py

@@ -1,6 +1,7 @@
 from typing import Annotated, Any
 
-from pydantic import BaseModel, ConfigDict, PlainSerializer, PlainValidator, WithJsonSchema
+from pydantic import BaseModel, ConfigDict, PlainValidator, WithJsonSchema
+from pydantic.json_schema import JsonSchemaValue
 from pydantic_core.core_schema import ValidationInfo
 from tree_sitter import Point, Range
 
@@ -21,43 +22,31 @@ def validate_range(value: Any, info: ValidationInfo) -> Range:
     return value
 
 
-RangeAdapter = Annotated[
-    Range,
-    PlainValidator(validate_range),
-    PlainSerializer(
-        lambda range: {
-            "start_byte": range.start_byte,
-            "end_byte": range.end_byte,
+def range_json_schema() -> JsonSchemaValue:
+    return {
+        "type": "object",
+        "properties": {
+            "start_byte": {"type": "integer"},
+            "end_byte": {"type": "integer"},
             "start_point": {
-                "row": range.start_point.row,
-                "column": range.start_point.column,
+                "type": "object",
+                "properties": {
+                    "row": {"type": "integer"},
+                    "column": {"type": "integer"},
+                },
             },
             "end_point": {
-                "row": range.end_point.row,
-                "column": range.end_point.column,
-            },
-        }
-    ),
-    WithJsonSchema(
-        {
-            "type": "object",
-            "properties": {
-                "start_byte": {"type": "integer"},
-                "end_byte": {"type": "integer"},
-                "start_point": {
-                    "type": "object",
-                    "properties": {
-                        "row": {"type": "integer"},
-                        "column": {"type": "integer"},
-                    },
-                },
-                "end_point": {
-                    "type": "object",
-                    "properties": {"row": {"type": "integer"}, "column": {"type": "integer"}},
-                },
+                "type": "object",
+                "properties": {"row": {"type": "integer"}, "column": {"type": "integer"}},
             },
-        }
-    ),
+        },
+    }
+
+
+RangeAdapter = Annotated[
+    Range,
+    PlainValidator(validate_range),
+    WithJsonSchema(range_json_schema()),
 ]