Add forward compatibility for Pydantic v2

frenck · frenck · commit e5802b192590 · 2023-11-04T22:02:02.000+01:00
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
