Add unit tests for vector database implementation

sweepai · Jun 21, 2024 · c1a55ac · c1a55ac
1 parent 15aa182
commit c1a55ac
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sweepai/core/vector_db_test.py b/sweepai/core/vector_db_test.py
@@ -1,5 +1,6 @@
 import pytest
 import numpy as np
+import requests
 from unittest.mock import patch, MagicMock
 from sweepai.core.vector_db import (
     cosine_similarity,