diff --git a/hugegraph-llm/src/hugegraph_llm/demo/rag_demo/vector_graph_block.py b/hugegraph-llm/src/hugegraph_llm/demo/rag_demo/vector_graph_block.py
index 7514e76..e1759fc 100644
--- a/hugegraph-llm/src/hugegraph_llm/demo/rag_demo/vector_graph_block.py
+++ b/hugegraph-llm/src/hugegraph_llm/demo/rag_demo/vector_graph_block.py
@@ -122,6 +122,9 @@ def create_vector_graph_block():
     graph_loading_bt.click(import_graph_data, inputs=[out, input_schema], outputs=[out]).then(
         store_prompt,
         inputs=[input_schema, info_extract_template],
+    ).then(fit_vid_index).then(
+        store_prompt,
+        inputs=[input_schema, info_extract_template],
     )
 
     def on_tab_select(input_f, input_t, evt: gr.SelectData):