fix: add remove chroma cache
Browse files- vector_store.py +2 -0
vector_store.py
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
import streamlit as st
|
|
|
|
| 2 |
from langchain_huggingface import HuggingFaceEmbeddings
|
| 3 |
from langchain_chroma import Chroma
|
| 4 |
from langchain_community.document_loaders import PyPDFLoader
|
|
@@ -22,6 +23,7 @@ def load_vector_store():
|
|
| 22 |
load vector store on every page load
|
| 23 |
"""
|
| 24 |
model = load_embedding_model("sentence-transformers/all-MiniLM-L6-v2")
|
|
|
|
| 25 |
vector_store = Chroma(
|
| 26 |
collection_name="main_store",
|
| 27 |
embedding_function=model,
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
+
import chromadb
|
| 3 |
from langchain_huggingface import HuggingFaceEmbeddings
|
| 4 |
from langchain_chroma import Chroma
|
| 5 |
from langchain_community.document_loaders import PyPDFLoader
|
|
|
|
| 23 |
load vector store on every page load
|
| 24 |
"""
|
| 25 |
model = load_embedding_model("sentence-transformers/all-MiniLM-L6-v2")
|
| 26 |
+
chromadb.api.client.SharedSystemClient.clear_system_cache()
|
| 27 |
vector_store = Chroma(
|
| 28 |
collection_name="main_store",
|
| 29 |
embedding_function=model,
|