fix failing tests

This commit is contained in:
ishaan-jaff 2023-08-10 13:27:59 -07:00
parent 9aefbdd0e7
commit b915c4beb1
2 changed files with 3 additions and 0 deletions

View file

@ -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],