Update the cache config template file and example directory

Signed-off-by: SimFG <bang.fu@zilliz.com>
zilliztech · May 23, 2023 · cb3074f · cb3074f
1 parent 1825e90
commit cb3074f
Show file tree

Hide file tree

Showing 18 changed files with 12 additions and 11 deletions.
diff --git a/cache_config_template.yml b/cache_config_template.yml
@@ -21,5 +21,5 @@ pre_function:
 post_function:
     first
 config:
-    threshold: 0.8
+    similarity_threshold: 0.8
     # Set other config here
diff --git a/examples/README.md b/examples/README.md
@@ -592,7 +592,7 @@ pre_function:
 post_function:
     first
 config:
-    threshold: 0.8
+    similarity_threshold: 0.8
     # Set other config here
 ```
 - model_src: The model source.

diff --git a/...les/context_examples/selective_context.py → ...ples/context_process/selective_context.py b/...les/context_examples/selective_context.py → ...ples/context_process/selective_context.py
diff --git a/...context_examples/summarization_context.py → .../context_process/summarization_context.py b/...context_examples/summarization_context.py → .../context_process/summarization_context.py
diff --git a/...es/diffusers_examples/stable_diffusion.py → ...s/integrate/diffusers/stable_diffusion.py b/...es/diffusers_examples/stable_diffusion.py → ...s/integrate/diffusers/stable_diffusion.py
diff --git a/examples/dolly_examples/basic_usage.py → examples/integrate/dolly/basic_usage.py b/examples/dolly_examples/basic_usage.py → examples/integrate/dolly/basic_usage.py
diff --git a/...langchain_examples/langchain_llms_mock.py → ...ntegrate/langchain/langchain_llms_mock.py b/...langchain_examples/langchain_llms_mock.py → ...ntegrate/langchain/langchain_llms_mock.py
diff --git a/...chain_examples/langchain_prompt_openai.py → ...rate/langchain/langchain_prompt_openai.py b/...chain_examples/langchain_prompt_openai.py → ...rate/langchain/langchain_prompt_openai.py
diff --git a/...mples/langchain_similaritycache_openai.py → ...chain/langchain_similaritycache_openai.py b/...mples/langchain_similaritycache_openai.py → ...chain/langchain_similaritycache_openai.py
diff --git a/examples/llama_cpp_examples/basic_usage.py → examples/integrate/llama_cpp/basic_usage.py b/examples/llama_cpp_examples/basic_usage.py → examples/integrate/llama_cpp/basic_usage.py
diff --git a/examples/openai_examples/basic_usage.py → examples/integrate/openai/basic_usage.py b/examples/openai_examples/basic_usage.py → examples/integrate/openai/basic_usage.py
diff --git a/examples/openai_examples/create_image.py → examples/integrate/openai/create_image.py b/examples/openai_examples/create_image.py → examples/integrate/openai/create_image.py
diff --git a/examples/openai_examples/qa.py → examples/integrate/openai/qa.py b/examples/openai_examples/qa.py → examples/integrate/openai/qa.py
diff --git a/examples/openai_examples/readme.py → examples/integrate/openai/readme.py b/examples/openai_examples/readme.py → examples/integrate/openai/readme.py
diff --git a/examples/openai_examples/summarize.py → examples/integrate/openai/summarize.py b/examples/openai_examples/summarize.py → examples/integrate/openai/summarize.py
diff --git a/examples/replicate_examples/vqa.py → examples/integrate/replicate/vqa.py b/examples/replicate_examples/vqa.py → examples/integrate/replicate/vqa.py
@@ -1,12 +1,11 @@
 import time
 
-from gptcache.processor.pre import get_input_image_file_name
 from gptcache import cache
-
-from gptcache.embedding import Timm, Onnx
 from gptcache.adapter import replicate
-from gptcache.similarity_evaluation.np import NumpyNormEvaluation
+from gptcache.embedding import Timm, Onnx
 from gptcache.manager import get_data_manager, CacheBase, VectorBase, ObjectBase
+from gptcache.processor.pre import get_input_image_file_name
+from gptcache.similarity_evaluation.np import NumpyNormEvaluation
 
 timm = Timm('resnet18')
 onnx = Onnx()
@@ -23,7 +22,7 @@
     )
 
 
-image_path = '../../docs/GPTCache.png'
+image_path = '../../../docs/GPTCache.png'
 
 
 # run replicate clinet with gptcache

diff --git a/examples/stability_examples/text_to_image.py → ...ples/integrate/stability/text_to_image.py b/examples/stability_examples/text_to_image.py → ...ples/integrate/stability/text_to_image.py
@@ -1,19 +1,20 @@
-import os
 import io
+import os
 import time
+
 from PIL import Image
 
 from gptcache import cache
-from gptcache.processor.pre import get_prompt
 from gptcache.adapter.stability_sdk import StabilityInference, generation
 from gptcache.embedding import Onnx
 from gptcache.manager.factory import manager_factory
+from gptcache.processor.pre import get_prompt
 from gptcache.similarity_evaluation.distance import SearchDistanceEvaluation
 
 # init gptcache
 onnx = Onnx()
-data_manager = manager_factory('sqlite,faiss,local', 
-                               data_dir='./', 
+data_manager = manager_factory('sqlite,faiss,local',
+                               data_dir='/',
                                vector_params={'dimension': onnx.dimension},
                                object_params={'path': './images'}
                                )

diff --git a/tests/requirements.txt b/tests/requirements.txt
@@ -21,3 +21,4 @@ pexpect
 spacy
 safetensors
 protobuf==3.20.0
+typing_extensions<4.6.0