Add the disable report config

Signed-off-by: SimFG <[email protected]>

examples/data_manager/scalar_store.py

@@ -1,15 +1,15 @@
 import os
+
 import numpy as np
 
-from gptcache.adapter import openai
 from gptcache import cache
+from gptcache.adapter import openai
 from gptcache.manager import get_data_manager, CacheBase, VectorBase
 from gptcache.similarity_evaluation.distance import SearchDistanceEvaluation
 
-
 d = 8
 
-
+# Change the embdding function to your own
 def mock_embeddings(data, **kwargs):
     return np.random.random((d, )).astype('float32')
 

gptcache/adapter/adapter.py

@@ -206,7 +206,7 @@ def post_process():
                 chat_cache.data_manager.add_session(
                     cache_whole_data[2], session.name, pre_embedding_data
                 )
-            if cache_whole_data:
+            if cache_whole_data and not chat_cache.config.disable_report:
                 # user_question / cache_question / cache_question_id / cache_answer / similarity / consume time/ time
                 report_cache_data = cache_whole_data[3]
                 report_search_data = cache_whole_data[2]

gptcache/config.py

@@ -46,6 +46,7 @@ def __init__(
             context_len: Optional[int] = None,
             skip_list: List[str] = None,
             data_check: bool = False,
+            disable_report: bool = False,
     ):
         if similarity_threshold < 0 or similarity_threshold > 1:
             raise CacheError(
@@ -63,3 +64,4 @@ def __init__(
             skip_list = ["system", "assistant"]
         self.skip_list = skip_list
         self.data_check = data_check
+        self.disable_report = disable_report