test: clean error 2

requirements.txt

@@ -9,10 +9,10 @@ pypdf==3.16.4
 python-multipart==0.0.6
 matplotlib==3.8.3
 umap-learn==0.5.5
-ipython
-librosa
-future
-yt-dlp
+ipython==8.24.0
+librosa==0.10.2.post1
+future==1.0.0
+yt-dlp==2024.4.9
 pysqlite3>=0.5.2
 pyspellchecker>=0.8.1
 beautifulsoup4>=4.12.2

retrieve_kb.py

@@ -2,6 +2,7 @@ from fastapi import APIRouter
 from utils import get_chroma_client, get_embedding_function
 import os
 from dotenv import load_dotenv
+import streamlit as st
 
 load_dotenv()
 openai_key = os.getenv("OPENAI_API_KEY")
@@ -14,7 +15,7 @@ def get_current_knowledge_bases(client):
         knowledge_boxes = client.list_collections()
         return knowledge_boxes
     except Exception as e:
-        return f"{str(e)}"
+        st.error(f"{str(e)}")
 
 
 def get_knowledge_base_information(