From 775d46fecf52a7b0c622c5bab0457b95c57707d8 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Fri, 4 Aug 2023 23:14:43 +0800 Subject: [PATCH] format update in knowledge_base/utils.py --- server/knowledge_base/utils.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/server/knowledge_base/utils.py b/server/knowledge_base/utils.py index 677bb3d..dd0c49f 100644 --- a/server/knowledge_base/utils.py +++ b/server/knowledge_base/utils.py @@ -76,7 +76,7 @@ def load_vector_store( knowledge_base_name: str, embedding_model: str, embedding_device: str, - tick: int, # tick will be changed by upload_doc etc. and make cache refreshed. + tick: int, # tick will be changed by upload_doc etc. and make cache refreshed. ): print(f"loading vector store in '{knowledge_base_name}' with '{embedding_model}' embeddings.") embeddings = load_embeddings(embedding_model, embedding_device) @@ -92,15 +92,18 @@ def lookup_vs( embedding_model: str = EMBEDDING_MODEL, 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) return docs def refresh_vs_cache(kb_name: str): - ''' + """ make vector store cache refreshed when next loading - ''' + """ _VECTOR_STORE_TICKS[kb_name] = _VECTOR_STORE_TICKS.get(kb_name, 0) + 1