feat: error handling

app.py

@@ -98,8 +98,7 @@ show_sidebar()
 
 
 col1, col2, col3 = st.columns((1, 4, 1))
-with col2:
-    st.image("https://brianknows.org/brian_logo.png", width=300)
+
 st.write("# Brian Knowledge Base System! 👋")
 
 

pages/manage_knowledge_box.py

@@ -187,7 +187,7 @@ else:
     st.warning(f"{collection_name} KB is empty")
 
 
-tab1, tab2, tab3, tab4, tab5, tab6 = st.tabs(
+tab1, tab2, tab3, tab4, tab5, tab6, tab7 = st.tabs(
     [
         "Remove",
         "Add URL",
@@ -195,20 +195,10 @@ tab1, tab2, tab3, tab4, tab5, tab6 = st.tabs(
         "Add PDF",
         "Add Youtube",
         "Notion and Jina",
+        "Rename",
     ]
 )
 
-# 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
 with tab1:
     # remove a split
@@ -351,24 +341,24 @@ with tab6:
         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()
+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()

retrieve_kb.py

@@ -10,8 +10,11 @@ default_embedding_function = get_embedding_function(openai_key=openai_key)
 
 
 def get_current_knowledge_bases(client):
-    knowledge_boxes = client.list_collections()
-    return knowledge_boxes
+    try:
+        knowledge_boxes = client.list_collections()
+        return knowledge_boxes
+    except:
+        raise Exception("Error retreiving kbs")
 
 
 def get_knowledge_base_information(