Add unit tests for vector_db.py

[sweep-nightly]

Purpose

This pull request adds unit tests for the vector_db.py module to ensure the correct functionality of the vector database implementation.

Description

The changes in this pull request include:

1. Creating a new file vector_db_test.py in the same directory as vector_db.py to hold the unit tests.
2. Importing the necessary modules and functions from vector_db.py to be tested.
3. Mocking the openai and redis clients to isolate the tests.
4. Adding the following test cases:
   • test_embed_text_array: Tests the embed_text_array function, including caching functionality.
   • test_openai_call_embedding: Tests the openai_call_embedding function.
   • test_caching: Tests the caching functionality of the module.
   • test_multi_get_query_texts_similarity: Tests the multi_get_query_texts_similarity function.
   • test_cosine_similarity: Tests the cosine_similarity function.
   • test_normalize_l2: Tests the normalize_l2 function.

Summary

• Added unit tests for the vector_db.py module in the vector_db_test.py file.
• Implemented test cases for the main functionalities of the vector database, including embedding, caching, and similarity search.
• Mocked the openai and redis clients to isolate the tests.

Suggested changes from Sweep Chat by @kevinlu1248. Continue chatting at http://localhost:3000/c/a8886ff6-9731-4b35-a57b-d40f13514af0.

[vercel]

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name	Status	Preview	Comments	Updated (UTC)
sweep-chat	🔄 Building (Inspect)	Visit Preview	💬 Add feedback	Jun 21, 2024 6:30pm
sweep-chat-demo	✅ Ready (Inspect)	Visit Preview		Jun 21, 2024 6:30pm
sweep-docs	✅ Ready (Inspect)	Visit Preview	💬 Add feedback	Jun 21, 2024 6:30pm
sweep-pulley	🔄 Building (Inspect)	Visit Preview	💬 Add feedback	Jun 21, 2024 6:30pm