Add forward compatibility for Pydantic v2 (#440)

frenck · Nov 4, 2023 · 018006c · 018006c
1 parent 3078d8a
commit 018006c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/tailscale/models.py b/src/tailscale/models.py
@@ -3,7 +3,14 @@
 
 from typing import TYPE_CHECKING, Any
 
-from pydantic import BaseModel, Field, validator
+try:
+    from pydantic.v1 import BaseModel, Field, validator
+except ImportError:  # pragma: no cover
+    from pydantic import (  # type: ignore[assignment] # pragma: no cover
+        BaseModel,
+        Field,
+        validator,
+    )
 
 if TYPE_CHECKING:
     from datetime import datetime