auto rebuild

apache · Dec 22, 2024 · 3a75c8f · 3a75c8f
1 parent d2f0b03
commit 3a75c8f
Showing 1 changed file with 3 additions and 0 deletions.
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
@@ -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):