feat: format polardb

Author: lijicode

src/memos/graph_dbs/polardb.py

@@ -1776,28 +1776,50 @@ def export_graph(
 
                 for row in edge_results:
                     source_agtype, target_agtype, edge_agtype = row
-                    
+
                     # Extract and clean source
-                    source_raw = source_agtype.value if hasattr(source_agtype, "value") else str(source_agtype)
-                    if isinstance(source_raw, str) and source_raw.startswith('"') and source_raw.endswith('"'):
+                    source_raw = (
+                        source_agtype.value
+                        if hasattr(source_agtype, "value")
+                        else str(source_agtype)
+                    )
+                    if (
+                        isinstance(source_raw, str)
+                        and source_raw.startswith('"')
+                        and source_raw.endswith('"')
+                    ):
                         source = source_raw[1:-1]
                     else:
                         source = str(source_raw)
-                    
+
                     # Extract and clean target
-                    target_raw = target_agtype.value if hasattr(target_agtype, "value") else str(target_agtype)
-                    if isinstance(target_raw, str) and target_raw.startswith('"') and target_raw.endswith('"'):
+                    target_raw = (
+                        target_agtype.value
+                        if hasattr(target_agtype, "value")
+                        else str(target_agtype)
+                    )
+                    if (
+                        isinstance(target_raw, str)
+                        and target_raw.startswith('"')
+                        and target_raw.endswith('"')
+                    ):
                         target = target_raw[1:-1]
                     else:
                         target = str(target_raw)
-                    
+
                     # Extract and clean edge type
-                    type_raw = edge_agtype.value if hasattr(edge_agtype, "value") else str(edge_agtype)
-                    if isinstance(type_raw, str) and type_raw.startswith('"') and type_raw.endswith('"'):
+                    type_raw = (
+                        edge_agtype.value if hasattr(edge_agtype, "value") else str(edge_agtype)
+                    )
+                    if (
+                        isinstance(type_raw, str)
+                        and type_raw.startswith('"')
+                        and type_raw.endswith('"')
+                    ):
                         edge_type = type_raw[1:-1]
                     else:
                         edge_type = str(type_raw)
-                    
+
                     edges.append(
                         {
                             "source": source,
@@ -2782,21 +2804,33 @@ def get_edges(
                 for row in results:
                     # Extract and clean from_id
                     from_id_raw = row[0].value if hasattr(row[0], "value") else row[0]
-                    if isinstance(from_id_raw, str) and from_id_raw.startswith('"') and from_id_raw.endswith('"'):
+                    if (
+                        isinstance(from_id_raw, str)
+                        and from_id_raw.startswith('"')
+                        and from_id_raw.endswith('"')
+                    ):
                         from_id = from_id_raw[1:-1]
                     else:
                         from_id = str(from_id_raw)
-                    
+
                     # Extract and clean to_id
                     to_id_raw = row[1].value if hasattr(row[1], "value") else row[1]
-                    if isinstance(to_id_raw, str) and to_id_raw.startswith('"') and to_id_raw.endswith('"'):
+                    if (
+                        isinstance(to_id_raw, str)
+                        and to_id_raw.startswith('"')
+                        and to_id_raw.endswith('"')
+                    ):
                         to_id = to_id_raw[1:-1]
                     else:
                         to_id = str(to_id_raw)
-                    
+
                     # Extract and clean edge_type
                     edge_type_raw = row[2].value if hasattr(row[2], "value") else row[2]
-                    if isinstance(edge_type_raw, str) and edge_type_raw.startswith('"') and edge_type_raw.endswith('"'):
+                    if (
+                        isinstance(edge_type_raw, str)
+                        and edge_type_raw.startswith('"')
+                        and edge_type_raw.endswith('"')
+                    ):
                         edge_type = edge_type_raw[1:-1]
                     else:
                         edge_type = str(edge_type_raw)