Merge pull request #109 from kdcokenny/main

feat: Allow Pydantic V2+

pyproject.toml

@@ -16,7 +16,7 @@ packages = [{include = "semantic_router"}]
 
 [tool.poetry.dependencies]
 python = "^3.9"
-pydantic = "^1.8.2"
+pydantic = "^2.5.3"
 openai = "^1.3.9"
 cohere = "^4.32"
 numpy = "^1.25.2"

semantic_router/encoders/base.py

@@ -1,6 +1,6 @@
 from typing import List
 
-from pydantic import BaseModel, Field
+from pydantic.v1 import BaseModel, Field
 
 
 class BaseEncoder(BaseModel):

semantic_router/encoders/fastembed.py

@@ -1,7 +1,7 @@
 from typing import Any, List, Optional
 
 import numpy as np
-from pydantic import PrivateAttr
+from pydantic.v1 import PrivateAttr
 
 from semantic_router.encoders import BaseEncoder
 

semantic_router/encoders/huggingface.py

@@ -1,6 +1,6 @@
 from typing import Any, List, Optional
 
-from pydantic import PrivateAttr
+from pydantic.v1 import PrivateAttr
 
 from semantic_router.encoders import BaseEncoder
 

semantic_router/llms/base.py

@@ -1,7 +1,7 @@
 import json
 from typing import Any, List, Optional
 
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 
 from semantic_router.schema import Message
 from semantic_router.utils.logger import logger

semantic_router/route.py

@@ -2,7 +2,7 @@
 import re
 from typing import Any, Callable, Dict, List, Optional, Union
 
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 
 from semantic_router.llms import BaseLLM
 from semantic_router.schema import Message, RouteChoice

semantic_router/schema.py

@@ -1,8 +1,8 @@
 from enum import Enum
 from typing import Dict, List, Literal, Optional
 
-from pydantic import BaseModel
-from pydantic.dataclasses import dataclass
+from pydantic.v1 import BaseModel
+from pydantic.v1.dataclasses import dataclass
 
 from semantic_router.encoders import (
     BaseEncoder,

semantic_router/utils/function_call.py

@@ -1,7 +1,7 @@
 import inspect
 from typing import Any, Callable, Dict, List, Union
 
-from pydantic import BaseModel
+from pydantic.v1 import BaseModel
 
 from semantic_router.llms import BaseLLM
 from semantic_router.schema import Message, RouteChoice

tests/unit/test_schema.py

@@ -1,5 +1,5 @@
 import pytest
-from pydantic import ValidationError
+from pydantic.v1 import ValidationError
 
 from semantic_router.schema import (
     CohereEncoder,