From 32da6c391fa0a42745b1a9e9e24588b9abf5d158 Mon Sep 17 00:00:00 2001
From: KingSkyLi <15566300566@163.com>
Date: Wed, 28 Aug 2024 16:30:31 +0800
Subject: [PATCH] fixed graphvis api;

---
 dbgpt/app/knowledge/service.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/dbgpt/app/knowledge/service.py b/dbgpt/app/knowledge/service.py
index a9faa7a23..b80377d20 100644
--- a/dbgpt/app/knowledge/service.py
+++ b/dbgpt/app/knowledge/service.py
@@ -532,13 +532,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