From faaa51124ad7967e604ef1c249063fda13a1665e Mon Sep 17 00:00:00 2001 From: Carol Jung Date: Thu, 30 Jan 2025 00:11:39 -0800 Subject: [PATCH 1/2] Fix BeforeValidator import --- src/codegen/sdk/codebase/span.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/codegen/sdk/codebase/span.py b/src/codegen/sdk/codebase/span.py index 785e24587..12c3bb1b6 100644 --- a/src/codegen/sdk/codebase/span.py +++ b/src/codegen/sdk/codebase/span.py @@ -1,6 +1,7 @@ from typing import Annotated, Any -from pydantic import BaseModel, BeforeValidator, ConfigDict, WithJsonSchema +from pydantic import BaseModel, ConfigDict, WithJsonSchema +from pydantic.functional_validators import BeforeValidator from pydantic.json_schema import JsonSchemaValue from pydantic_core.core_schema import ValidationInfo from tree_sitter import Point, Range From a7ca06187aba18b267faa72b7d0fa4c26e0aa117 Mon Sep 17 00:00:00 2001 From: Carol Jung Date: Thu, 30 Jan 2025 11:36:14 -0800 Subject: [PATCH 2/2] Fix ConfigDict + WithJsonSchema import --- 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 12c3bb1b6..d8e7375cb 100644 --- a/src/codegen/sdk/codebase/span.py +++ b/src/codegen/sdk/codebase/span.py @@ -1,8 +1,9 @@ from typing import Annotated, Any -from pydantic import BaseModel, ConfigDict, WithJsonSchema +from pydantic import BaseModel +from pydantic.config import ConfigDict from pydantic.functional_validators import BeforeValidator -from pydantic.json_schema import JsonSchemaValue +from pydantic.json_schema import JsonSchemaValue, WithJsonSchema from pydantic_core.core_schema import ValidationInfo from tree_sitter import Point, Range