Fix deprecation warning with usage of ast.Str (#8837)

pydantic · Feb 16, 2024 · 12ebdfc · 12ebdfc
1 parent 3dd3b42
commit 12ebdfc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pydantic/_internal/_docs_extraction.py b/pydantic/_internal/_docs_extraction.py
@@ -25,8 +25,12 @@ def visit_AnnAssign(self, node: ast.AnnAssign) -> Any:
             self.target = node.target.id
 
     def visit_Expr(self, node: ast.Expr) -> Any:
-        if isinstance(node.value, ast.Str) and self.previous_node_type is ast.AnnAssign:
-            docstring = inspect.cleandoc(node.value.s)
+        if (
+            isinstance(node.value, ast.Constant)
+            and isinstance(node.value.value, str)
+            and self.previous_node_type is ast.AnnAssign
+        ):
+            docstring = inspect.cleandoc(node.value.value)
             if self.target:
                 self.attrs[self.target] = docstring
             self.target = None