fix refresh bug

apache · Dec 27, 2024 · 58eb9f5 · 58eb9f5
1 parent fddc8e0
commit 58eb9f5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/hugegraph-llm/src/hugegraph_llm/demo/rag_demo/app.py b/hugegraph-llm/src/hugegraph_llm/demo/rag_demo/app.py
@@ -120,7 +120,7 @@ def init_rag_ui() -> gr.Interface:
         textbox_array_graph_config = create_configs_block()
 
         with gr.Tab(label="1. Build RAG Index 💡"):
-           textbox_input_text, textbox_input_schema, textbox_info_extract_template = create_vector_graph_block()
+            textbox_input_text, textbox_input_schema, textbox_info_extract_template = create_vector_graph_block()
         with gr.Tab(label="2. (Graph)RAG & User Functions 📖"):
             (
                 textbox_inp,

diff --git a/hugegraph-llm/src/hugegraph_llm/utils/graph_index_utils.py b/hugegraph-llm/src/hugegraph_llm/utils/graph_index_utils.py
@@ -83,7 +83,11 @@ def extract_graph(input_file, input_text, schema, example_prompt) -> str:
         if not context["vertices"] and not context["edges"]:
             log.info("Please check the schema.(The schema may not match the Doc)")
             return json.dumps(
-                {"vertices": context["vertices"], "edges": context["edges"], "warning": "The schema may not match the Doc"},
+                {
+                    "vertices": context["vertices"],
+                    "edges": context["edges"],
+                    "warning": "The schema may not match the Doc"
+                    },
                 ensure_ascii=False,
                 indent=2
             )