fix: Update ConfigDict + WithJsonSchema import (#223)

Author: caroljung-cg

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