Skip to content

Commit 6dbf591

Browse files
authored
Revert "Useless quotes (#450)"
This reverts commit e62f72d.
1 parent b8cd27b commit 6dbf591

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

src/memos/graph_dbs/polardb.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -859,9 +859,9 @@ def get_node(
859859

860860
if result:
861861
if include_embedding:
862-
_, properties_json, embedding_json = result
862+
properties_json, embedding_json = result
863863
else:
864-
_, properties_json = result
864+
properties_json = result
865865
embedding_json = None
866866

867867
# Parse properties from JSONB if it's a string
@@ -885,14 +885,12 @@ def get_node(
885885
properties["embedding"] = embedding
886886
except (json.JSONDecodeError, TypeError):
887887
logger.warning(f"Failed to parse embedding for node {id}")
888-
properties.pop("id")
889-
properties.pop("memory")
890-
properties.pop("user_name", None)
888+
891889
return self._parse_node(
892890
{
893891
"id": id,
894-
"memory": properties.get("memory", ""),
895-
**properties,
892+
"memory": json.loads(properties[1]).get("memory", ""),
893+
**json.loads(properties[1]),
896894
}
897895
)
898896
return None
@@ -1292,8 +1290,6 @@ def get_subgraph(
12921290

12931291
user_name = user_name if user_name else self._get_config_value("user_name")
12941292

1295-
if center_id.startswith('"') and center_id.endswith('"'):
1296-
center_id = center_id[1:-1]
12971293
# Use a simplified query to get the subgraph (temporarily only direct neighbors)
12981294
"""
12991295
SELECT * FROM cypher('{self.db_name}_graph', $$

0 commit comments

Comments
 (0)