From 91616da3b3ea1595b218efc96f7c1f1c7c3d0baf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lu=C3=ADs=20Portela=20Afonso?= <luis.portela@ydata.ai>
Date: Tue, 4 Feb 2025 18:51:44 +0000
Subject: [PATCH] fix: missing migrations for pydantic v2 (#155)

* fix: missing migrations for pydantic v2

* fix(linting): code formatting

---------

Co-authored-by: Azory YData Bot <azory@ydata.ai>
---
 pyproject.toml                                       | 3 ++-
 src/ydata/sdk/common/status.py                       | 8 ++++----
 src/ydata/sdk/datasources/_models/metadata/column.py | 8 ++++----
 3 files changed, 10 insertions(+), 9 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 67896aec..a93e58c9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -43,7 +43,8 @@ dependencies = [
     "prettytable==3.13.*",
     "pydantic>=2.0.0",
     "typeguard==2.13.3",
-    "ydata-datascience"
+    "ydata-datascience",
+    "requests==2.*",
 ]
 
 [project.license]
diff --git a/src/ydata/sdk/common/status.py b/src/ydata/sdk/common/status.py
index 8427adac..5e14dece 100644
--- a/src/ydata/sdk/common/status.py
+++ b/src/ydata/sdk/common/status.py
@@ -1,6 +1,6 @@
 from typing import Generic, Optional, TypeVar
 
-from pydantic import Field
+from pydantic import ConfigDict, Field
 
 from .model import BaseModel
 
@@ -8,7 +8,7 @@
 
 
 class GenericStateErrorStatus(BaseModel, Generic[T]):
+    model_config = ConfigDict(
+        use_enum_values=True
+    )
     state: Optional[T] = Field(None)
-
-    class Config:
-        use_enum_values = True
diff --git a/src/ydata/sdk/datasources/_models/metadata/column.py b/src/ydata/sdk/datasources/_models/metadata/column.py
index 1cc50275..ebf816e9 100644
--- a/src/ydata/sdk/datasources/_models/metadata/column.py
+++ b/src/ydata/sdk/datasources/_models/metadata/column.py
@@ -1,18 +1,18 @@
-from pydantic import Field
+from pydantic import ConfigDict, Field
 
 from ydata.sdk.common.model import BaseModel
 from ydata.sdk.datasources._models.metadata.data_types import DataType, VariableType
 
 
 class Column(BaseModel):
+    model_config = ConfigDict(
+        use_enum_values=True
+    )
 
     name: str
     datatype: DataType = Field(alias='dataType')
     vartype: VariableType = Field(alias='varType')
 
-    class Config:
-        use_enum_values = True
-
     def __repr__(self) -> str:
         return f"Column(name={self.name}, datatype={self.datatype}, vartype={self.vartype})"