vectordb test removed - api dependency

geemi725 · Feb 28, 2024 · cc2e8d3 · cc2e8d3
1 parent db26fe2
commit cc2e8d3
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/tests/test_xpertai.py b/tests/test_xpertai.py
@@ -34,16 +34,6 @@ def test_load_split_docs():
     docs_split = load_split_docs(doc_path)
     assert isinstance(docs_split[0].page_content, str)
 
-def test_vector_db():
-    from langchain.embeddings import HuggingFaceEmbeddings
-
-    base_dir = os.path.dirname(os.path.realpath("__file__"))
-    lit_file = os.path.join(base_dir, "tests/paper_test.pdf")
-    embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
-    vector_db(lit_file=lit_file, try_meta_data=True, clean=True, embedding=embedding)
-    assert os.path.exists("./data/chroma/") is True
-
-
 def test_explain_shap():
     base_dir = os.path.dirname(os.path.realpath("__file__"))
     df = pd.DataFrame(