From f94ea06588053dc1a08e7e6e14e3d17afe2256db Mon Sep 17 00:00:00 2001 From: Prithvi Monangi Date: Wed, 1 Apr 2026 02:32:22 -0700 Subject: [PATCH] fix(configs): migrate CassandraConfig and AzureMySQLConfig to pydantic v2 ConfigDict (#4646) --- mem0/configs/vector_stores/azure_mysql.py | 5 ++--- mem0/configs/vector_stores/cassandra.py | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/mem0/configs/vector_stores/azure_mysql.py b/mem0/configs/vector_stores/azure_mysql.py index e5d468612..861452ac8 100644 --- a/mem0/configs/vector_stores/azure_mysql.py +++ b/mem0/configs/vector_stores/azure_mysql.py @@ -1,6 +1,6 @@ from typing import Any, Dict, Optional -from pydantic import BaseModel, Field, model_validator +from pydantic import BaseModel, ConfigDict, Field, model_validator class AzureMySQLConfig(BaseModel): @@ -80,5 +80,4 @@ class AzureMySQLConfig(BaseModel): return values - class Config: - arbitrary_types_allowed = True + model_config = ConfigDict(arbitrary_types_allowed=True) diff --git a/mem0/configs/vector_stores/cassandra.py b/mem0/configs/vector_stores/cassandra.py index 40e629a81..39dc46dd9 100644 --- a/mem0/configs/vector_stores/cassandra.py +++ b/mem0/configs/vector_stores/cassandra.py @@ -1,6 +1,6 @@ from typing import Any, Dict, List, Optional -from pydantic import BaseModel, Field, model_validator +from pydantic import BaseModel, ConfigDict, Field, model_validator class CassandraConfig(BaseModel): @@ -72,6 +72,5 @@ class CassandraConfig(BaseModel): return values - class Config: - arbitrary_types_allowed = True + model_config = ConfigDict(arbitrary_types_allowed=True)