From f84cdd48994d50477c8e5a1c4066065a34f1b9c2 Mon Sep 17 00:00:00 2001 From: Carol Jung Date: Wed, 29 Jan 2025 23:43:15 -0800 Subject: [PATCH] Replace PlainValidator -> BeforeValidator --- src/codegen/sdk/codebase/span.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/codegen/sdk/codebase/span.py b/src/codegen/sdk/codebase/span.py index 799482fa9..785e24587 100644 --- a/src/codegen/sdk/codebase/span.py +++ b/src/codegen/sdk/codebase/span.py @@ -1,6 +1,6 @@ from typing import Annotated, Any -from pydantic import BaseModel, ConfigDict, PlainValidator, WithJsonSchema +from pydantic import BaseModel, BeforeValidator, ConfigDict, WithJsonSchema from pydantic.json_schema import JsonSchemaValue from pydantic_core.core_schema import ValidationInfo from tree_sitter import Point, Range @@ -45,7 +45,7 @@ def range_json_schema() -> JsonSchemaValue: RangeAdapter = Annotated[ Range, - PlainValidator(validate_range), + BeforeValidator(validate_range), WithJsonSchema(range_json_schema()), ] @@ -56,6 +56,7 @@ class Span(BaseModel): model_config = ConfigDict( frozen=True, + arbitrary_types_allowed=True, json_encoders={ Range: lambda r: { "start_byte": r.start_byte,