feat: rename knowledge base

pages/manage_knowledge_box.py

@@ -11,6 +11,7 @@ import wave
 from dotenv import load_dotenv
 import requests
 import uuid
+import time
 
 from retrieve_kb import get_current_knowledge_bases, get_knowledge_base_information
 from generate_kb import add_links_to_knowledge_base
@@ -186,8 +187,16 @@ else:
     st.warning(f"{collection_name} KB is empty")
 
 
-tab1, tab2, tab3, tab4, tab5, tab6 = st.tabs(
-    ["Remove", "Add URL", "Multiple URL", "Add PDF", "Add Youtube", "Notion and Jina"]
+tab1, tab2, tab3, tab4, tab5, tab6, tab7 = st.tabs(
+    [
+        "Remove",
+        "Add URL",
+        "Multiple URL",
+        "Add PDF",
+        "Add Youtube",
+        "Notion and Jina",
+        "Rename",
+    ]
 )
 
 # remove stuff tab
@@ -331,3 +340,25 @@ with tab6:
             st.success("Added")
         except Exception as e:
             st.error(f"{str(e)}")
+
+with tab7:
+
+    # remove a split
+    st.header("Rename collection")
+    new_name = st.text_input("New collection name")
+    collection_info, coll, client = get_knowledge_base_information(
+        client=client,
+        embedding_function=default_embedding_function,
+        kb_name=collection_name,
+    )
+
+    if st.button("rename"):
+        try:
+            coll.modify(
+                name=new_name,
+            )
+        except Exception as e:
+            st.error(f"{str(e)}")
+        st.success("Done")
+        time.sleep(1)
+        st.experimental_rerun()