refactor: class-based config is deprecated in Pydantic v2.0, use ConfigDict instead

semantic_router/encoders/base.py

-from typing import Any, List, Optional
+from typing import Any, ClassVar, List, Optional
 
 import numpy as np
-from pydantic import BaseModel, Field, field_validator
+from pydantic import BaseModel, ConfigDict, Field, field_validator
 
 from semantic_router.route import Route
 from semantic_router.schema import SparseEmbedding
@@ -12,8 +12,7 @@ class DenseEncoder(BaseModel):
     score_threshold: Optional[float] = None
     type: str = Field(default="base")
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
     @field_validator("score_threshold")
     def set_score_threshold(cls, v: float | None) -> float | None:
@@ -57,8 +56,7 @@ class SparseEncoder(BaseModel):
     name: str
     type: str = Field(default="base")
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
     def __call__(self, docs: List[str]) -> List[SparseEmbedding]:
         """Sparsely encode a list of documents. Documents can be any type, but the encoder must

semantic_router/index/base.py

@@ -2,10 +2,10 @@ import asyncio
 import json
 import time
 from datetime import datetime
-from typing import Any, Dict, List, Optional, Tuple, Union
+from typing import Any, ClassVar, Dict, List, Optional, Tuple, Union
 
 import numpy as np
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 from semantic_router.route import Route
 from semantic_router.schema import ConfigParameter, SparseEmbedding, Utterance
@@ -515,8 +515,7 @@ class BaseIndex(BaseModel):
         route_info = parse_route_info(metadata=metadata)
         return route_info  # type: ignore
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
 
 def parse_route_info(metadata: List[Dict[str, Any]]) -> List[Tuple]:

semantic_router/index/local.py

-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, ClassVar, Dict, List, Optional, Tuple
 
 import numpy as np
+from pydantic import ConfigDict
 
 from semantic_router.index.base import BaseIndex, IndexConfig
 from semantic_router.linear import similarity_matrix, top_scores
@@ -14,9 +15,8 @@ class LocalIndex(BaseIndex):
     def __init__(self):
         super().__init__()
 
-    class Config:
-        # Stop pydantic from complaining about Optional[np.ndarray]type hints.
-        arbitrary_types_allowed = True
+    # Stop pydantic from complaining about Optional[np.ndarray]type hints.
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
     def add(
         self,

semantic_router/index/postgres.py

 import os
 import uuid
 from enum import Enum
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
+from typing import TYPE_CHECKING, Any, ClassVar, Dict, List, Optional, Tuple, Union
 
 import numpy as np
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 from semantic_router.index.base import BaseIndex, IndexConfig
 from semantic_router.schema import ConfigParameter, Metric, SparseEmbedding
@@ -494,7 +494,5 @@ class PostgresIndex(BaseIndex):
                 return 0
             return count[0]
 
-    class Config:
-        """Configuration for the Pydantic BaseModel."""
-
-        arbitrary_types_allowed = True
+    """Configuration for the Pydantic BaseModel."""
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)

semantic_router/llms/base.py

 import json
-from typing import Any, Dict, List, Optional
+from typing import Any, ClassVar, Dict, List, Optional
 
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 from semantic_router.schema import Message
 from semantic_router.utils.logger import logger
@@ -18,8 +18,7 @@ class BaseLLM(BaseModel):
     temperature: Optional[float] = 0.0
     max_tokens: Optional[int] = None
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
     def __init__(self, name: str, **kwargs):
         """Initialize the BaseLLM.

semantic_router/route.py

 import json
 import re
-from typing import Any, Callable, Dict, List, Optional, Union
+from typing import Any, Callable, ClassVar, Dict, List, Optional, Union
 
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 from semantic_router.llms import BaseLLM
 from semantic_router.schema import Message, RouteChoice
@@ -72,8 +72,7 @@ class Route(BaseModel):
     score_threshold: Optional[float] = None
     metadata: Optional[Dict[str, Any]] = {}
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
     def __call__(self, query: Optional[str] = None) -> RouteChoice:
         """Call the route. If dynamic routes have been provided the query must have been

semantic_router/routers/base.py

@@ -3,11 +3,11 @@ import importlib
 import json
 import os
 import random
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, ClassVar, Dict, List, Optional, Tuple, Union
 
 import numpy as np
 import yaml  # type: ignore
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 from tqdm.auto import tqdm
 from typing_extensions import deprecated
 
@@ -70,8 +70,7 @@ class RouterConfig:
 
     routes: List[Route] = Field(default_factory=list)
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
     def __init__(
         self,
@@ -361,8 +360,7 @@ class BaseRouter(BaseModel):
     aggregation_method: Optional[Callable] = None
     auto_sync: Optional[str] = None
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
     def __init__(
         self,

semantic_router/schema.py

@@ -2,11 +2,11 @@ import json
 from datetime import datetime, timezone
 from difflib import Differ
 from enum import Enum
-from typing import Any, Dict, List, Optional, Tuple, Union
+from typing import Any, ClassVar, Dict, List, Optional, Tuple, Union
 
 import numpy as np
 from aurelio_sdk.schema import SparseEmbedding as BM25SparseEmbedding
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 from semantic_router.utils.logger import logger
 
@@ -484,8 +484,7 @@ class SparseEmbedding(BaseModel):
 
     embedding: np.ndarray
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
     @classmethod
     def from_compact_array(cls, array: np.ndarray):

semantic_router/utils/function_call.py

 import inspect
-from typing import Any, Callable, Dict, List, Optional, Union
+from typing import Any, Callable, ClassVar, Dict, List, Optional, Union
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 from semantic_router.llms import BaseLLM
 from semantic_router.schema import Message, RouteChoice
@@ -23,8 +23,7 @@ class Parameter(BaseModel):
     :type required: bool
     """
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
 
     name: str = Field(description="The name of the parameter")
     description: Optional[str] = Field(