From bee4ca283639aa85a66836dae4a9ff3afa4fc603 Mon Sep 17 00:00:00 2001
From: MrJs133 <jsong010123@gmail.com>
Date: Mon, 30 Dec 2024 22:40:10 +0800
Subject: [PATCH] limit context['vertices'] and context['edges']

---
 .../hugegraph_llm/operators/hugegraph_op/fetch_graph_data.py   | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/hugegraph-llm/src/hugegraph_llm/operators/hugegraph_op/fetch_graph_data.py b/hugegraph-llm/src/hugegraph_llm/operators/hugegraph_op/fetch_graph_data.py
index 6f2064be..a3d10065 100644
--- a/hugegraph-llm/src/hugegraph_llm/operators/hugegraph_op/fetch_graph_data.py
+++ b/hugegraph-llm/src/hugegraph_llm/operators/hugegraph_op/fetch_graph_data.py
@@ -20,7 +20,6 @@
 import gradio as gr
 
 from pyhugegraph.client import PyHugeClient
-from hugegraph_llm.utils.log import log
 
 
 class FetchGraphData:
@@ -31,7 +30,7 @@ def run(self, graph_summary_info: Optional[Dict[str, Any]]) -> Dict[str, Any]:
         limit_vertices = 10000
         limit_edges = 100
         gr.Info(f"Returning a maximum of {limit_vertices} vertices. \n Returning a maximum of {limit_edges} edges.")
-        
+
         if graph_summary_info is None:
             graph_summary_info = {}
         if "num_vertices" not in graph_summary_info: