Import re in graph_storage to prevent NameError

@@ -6,6 +6,7 @@
 
 from dataclasses import dataclass, field
 from typing import Dict, Any, List, Optional, Set
+import re
 from datetime import datetime
 import json
 from pathlib import Path