chore: switch from pydantic.v1 to pydantic

semantic_router/encoders/aurelio.py

@@ -1,6 +1,6 @@
 import os
 from typing import Any, List, Optional
-from pydantic.v1 import Field
+from pydantic import Field
 
 from aurelio_sdk import AurelioClient, AsyncAurelioClient, EmbeddingResponse
 

semantic_router/encoders/clip.py

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

semantic_router/encoders/cohere.py

@@ -1,7 +1,7 @@
 import os
 from typing import Any, List, Optional
 
-from pydantic.v1 import PrivateAttr
+from pydantic import PrivateAttr
 
 from semantic_router.encoders import DenseEncoder
 from semantic_router.utils.defaults import EncoderDefault

semantic_router/encoders/fastembed.py

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

semantic_router/encoders/huggingface.py

@@ -25,7 +25,7 @@
 import os
 from typing import Any, List, Optional, Dict
 
-from pydantic.v1 import PrivateAttr
+from pydantic import PrivateAttr
 
 from semantic_router.encoders import DenseEncoder
 from semantic_router.utils.logger import logger

semantic_router/encoders/mistral.py

@@ -4,7 +4,7 @@
 from time import sleep
 from typing import Any, List, Optional
 
-from pydantic.v1 import PrivateAttr
+from pydantic import PrivateAttr
 
 from semantic_router.encoders import DenseEncoder
 from semantic_router.utils.defaults import EncoderDefault

semantic_router/encoders/openai.py

@@ -2,7 +2,7 @@
 import os
 from time import sleep
 from typing import Any, List, Optional, Union
-from pydantic.v1 import PrivateAttr
+from pydantic import PrivateAttr
 
 import openai
 from openai import OpenAIError

semantic_router/index/base.py

@@ -2,7 +2,7 @@
 import json
 
 import numpy as np
-from pydantic.v1 import BaseModel
+from pydantic import BaseModel
 
 from semantic_router.schema import ConfigParameter, SparseEmbedding, Utterance
 from semantic_router.route import Route

semantic_router/index/pinecone.py

@@ -8,7 +8,7 @@
 from typing import Any, Dict, List, Optional, Union, Tuple
 
 import numpy as np
-from pydantic.v1 import BaseModel, Field
+from pydantic import BaseModel, Field
 
 from semantic_router.index.base import BaseIndex
 from semantic_router.schema import ConfigParameter, SparseEmbedding

semantic_router/index/qdrant.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict, List, Optional, Tuple, Union
 
 import numpy as np
-from pydantic.v1 import Field
+from pydantic import Field
 
 from semantic_router.index.base import BaseIndex
 from semantic_router.schema import ConfigParameter, Metric, SparseEmbedding, Utterance

semantic_router/llms/base.py

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

semantic_router/llms/cohere.py

@@ -1,7 +1,7 @@
 import os
 from typing import Any, List, Optional
 
-from pydantic.v1 import PrivateAttr
+from pydantic import PrivateAttr
 
 from semantic_router.llms import BaseLLM
 from semantic_router.schema import Message

semantic_router/llms/llamacpp.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 from typing import Any, Optional, List, Dict
 
-from pydantic.v1 import PrivateAttr
+from pydantic import PrivateAttr
 
 from semantic_router.llms.base import BaseLLM
 from semantic_router.schema import Message

semantic_router/llms/mistral.py

@@ -1,7 +1,7 @@
 import os
 from typing import Any, List, Optional
 
-from pydantic.v1 import PrivateAttr
+from pydantic import PrivateAttr
 
 from semantic_router.llms import BaseLLM
 from semantic_router.schema import Message

semantic_router/route.py

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

semantic_router/routers/base.py

@@ -4,7 +4,7 @@
 import random
 import hashlib
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
-from pydantic.v1 import BaseModel, Field
+from pydantic import BaseModel, Field
 
 import numpy as np
 import yaml  # type: ignore

semantic_router/routers/hybrid.py

@@ -1,6 +1,6 @@
 from typing import Dict, List, Optional
 import asyncio
-from pydantic.v1 import Field
+from pydantic import Field
 
 import numpy as np
 

semantic_router/schema.py

@@ -3,7 +3,7 @@
 from enum import Enum
 import numpy as np
 from typing import List, Optional, Union, Any, Dict, Tuple
-from pydantic.v1 import BaseModel, Field
+from pydantic import BaseModel, Field
 from semantic_router.utils.logger import logger
 from aurelio_sdk.schema import BM25Embedding
 

semantic_router/utils/function_call.py

@@ -1,7 +1,7 @@
 import inspect
 from typing import Any, Callable, Dict, List, Optional, Union
 
-from pydantic.v1 import BaseModel
+from pydantic 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.v1 import ValidationError
+from pydantic import ValidationError
 
 from semantic_router.schema import (
     Message,