fix failing tests

2023-08-10 13:27:59 -07:00 · 2023-08-10 13:27:59 -07:00 · f6279353ac
commit f6279353ac
parent d2324c9955
2 changed files with 3 additions and 0 deletions
--- a/litellm/tests/test_cache.py
+++ b/litellm/tests/test_cache.py
@ -41,3 +41,4 @@ def test_completion_gpt():
        pytest.fail(f"Error occurred: {e}")


+litellm.cache = False # rest to False for the next test
--- a/litellm/utils.py
+++ b/litellm/utils.py
@ -722,6 +722,8 @@ def message_to_user_question(messages):

 def add_cache(messages, model_response):
  global cache_collection
+  if cache_collection == None:
+    make_collection()
  user_question = message_to_user_question(messages)
  cache_collection.add(
    documents=[user_question],