weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

format update in knowledge_base/utils.py

This commit is contained in:
imClumsyPanda 2023-08-04 23:14:43 +08:00
parent bc3ee83910
commit 775d46fecf
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server/knowledge_base/utils.py
View File

@ -92,15 +92,18 @@ def lookup_vs(
embedding_model: str = EMBEDDING_MODEL, embedding_model: str = EMBEDDING_MODEL,
embedding_device: str = EMBEDDING_DEVICE, embedding_device: str = EMBEDDING_DEVICE,
): ):
search_index = load_vector_store(knowledge_base_name, embedding_model, embedding_device, _VECTOR_STORE_TICKS.get(knowledge_base_name)) search_index = load_vector_store(knowledge_base_name,
embedding_model,
embedding_device,
_VECTOR_STORE_TICKS.get(knowledge_base_name))
docs = search_index.similarity_search(query, k=top_k) docs = search_index.similarity_search(query, k=top_k)
return docs return docs
def refresh_vs_cache(kb_name: str): def refresh_vs_cache(kb_name: str):
''' """
make vector store cache refreshed when next loading make vector store cache refreshed when next loading
''' """
_VECTOR_STORE_TICKS[kb_name] = _VECTOR_STORE_TICKS.get(kb_name, 0) + 1 _VECTOR_STORE_TICKS[kb_name] = _VECTOR_STORE_TICKS.get(kb_name, 0) + 1