fixed graphvis api;

eosphoros-ai · Aug 28, 2024 · 2f8822c · 2f8822c
1 parent 3473504
commit 2f8822c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dbgpt/app/knowledge/service.py b/dbgpt/app/knowledge/service.py
@@ -646,13 +646,14 @@ def query_graph(self, space_name, limit):
         graph = vector_store_connector.client.query_graph(limit=limit)
         res = {"nodes": [], "edges": []}
         for node in graph.vertices():
-            res["nodes"].append({"vid": node.vid})
+            res["nodes"].append({"vid": node.vid,"communityId":node.get_prop('_community_id'),"props":node.props})
         for edge in graph.edges():
             res["edges"].append(
                 {
                     "src": edge.sid,
                     "dst": edge.tid,
                     "label": edge.name,
+                    "props":edge.props
                 }
             )
         return res